StyAndCo/zappyzep
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add !invite utility command

Browse source
This commit is contained in:
Dragory 2020-08-05 18:52:15 +03:00
parent ebdeabdc32
commit 024078ccdc
No known key found for this signature in database
GPG key ID: 5F387BA66DF8AAC1
7 changed files with 187 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

31
backend/src/utils.ts
View file

@ -1,6 +1,7 @@
import {
AnyInvite,
Attachment,
BaseInvite,
ChannelInvite,
Client,
Embed,
@ -15,6 +16,8 @@ import {
Message,
MessageContent,
PossiblyUncachedMessage,
RESTChannelInvite,
RESTPrivateInvite,
TextableChannel,
TextChannel,
User,
@ -450,6 +453,14 @@ export function getUrlsInString(str: string, onlyUnique = false): MatchedURL[] {
}, []);
}
export function parseInviteCodeInput(str: string): string {
if (str.match(/^[a-z0-9]{6,}$/i)) {
return str;
}
return getInviteCodesInString(str)[0];
}
export function getInviteCodesInString(str: string): string[] {
const inviteCodeRegex = /(?:discord.gg|discordapp.com\/invite|discord.com\/invite)\/([a-z0-9]+)/gi;
return Array.from(str.matchAll(inviteCodeRegex)).map(m => m[1]);
@ -1064,13 +1075,15 @@ export async function resolveRoleId(bot: Client, guildId: string, value: string)
const inviteCache = new SimpleCache<Promise<ChannelInvite>>(10 * MINUTES, 200);
export async function resolveInvite(client: Client, code: string): Promise<ChannelInvite | null> {
if (inviteCache.has(code)) {
return inviteCache.get(code);
export async function resolveInvite(client: Client, code: string, withCounts?: boolean): Promise<AnyInvite | null> {
const key = `${code}:${withCounts ? 1 : 0}`;
if (inviteCache.has(key)) {
return inviteCache.get(key);
}
const promise = client.getInvite(code).catch(() => null);
inviteCache.set(code, promise);
const promise = client.getInvite(code, withCounts).catch(() => null);
inviteCache.set(key, promise);
return promise;
}
@ -1229,6 +1242,14 @@ export function isGuildInvite(invite: AnyInvite): invite is GuildInvite {
return (invite as GuildInvite).guild != null;
}
export function isRESTGuildInvite(invite: BaseInvite): invite is RESTChannelInvite {
return (invite as any).guild != null;
}
export function isRESTGroupDMInvite(invite: BaseInvite): invite is RESTPrivateInvite {
return (invite as any).guild == null && (invite as any).channel != null;
}
export function asyncMap<T, R>(arr: T[], fn: (item: T) => Promise<R>): Promise<R[]> {
return Promise.all(arr.map((item, index) => fn(item)));
}