import { channelMentionRegex, convertDelayStringToMS, disableCodeBlocks, disableInlineCode, isSnowflake, isValidSnowflake, resolveMember, resolveUser, resolveUserId, roleMentionRegex, UnknownUser, } from "./utils"; import { GuildChannel, Member, TextChannel, User } from "eris"; import { baseTypeConverters, baseTypeHelpers, CommandContext, TypeConversionError } from "knub"; import { createTypeHelper } from "knub-command-manager"; import { getChannelIdFromMessageId } from "./data/getChannelIdFromMessageId"; import { MessageTarget, resolveMessageTarget } from "./utils/resolveMessageTarget"; import { inputPatternToRegExp } from "./validatorUtils"; export const commandTypes = { ...baseTypeConverters, delay(value) { const result = convertDelayStringToMS(value); if (result == null) { throw new TypeConversionError(`Could not convert ${value} to a delay`); } return result; }, async resolvedUser(value, context: CommandContext) { const result = await resolveUser(context.pluginData.client, value); if (result == null || result instanceof UnknownUser) { throw new TypeConversionError(`User \`${disableCodeBlocks(value)}\` was not found`); } return result; }, async resolvedUserLoose(value, context: CommandContext) { const result = await resolveUser(context.pluginData.client, value); if (result == null) { throw new TypeConversionError(`Invalid user: \`${disableCodeBlocks(value)}\``); } return result; }, async resolvedMember(value, context: CommandContext) { if (!(context.message.channel instanceof GuildChannel)) return null; const result = await resolveMember(context.pluginData.client, context.message.channel.guild, value); if (result == null) { throw new TypeConversionError( `Member \`${disableCodeBlocks(value)}\` was not found or they have left the server`, ); } return result; }, async messageTarget(value: string, context: CommandContext) { value = String(value).trim(); const result = await resolveMessageTarget(context.pluginData, value); if (!result) { throw new TypeConversionError(`Unknown message \`${disableInlineCode(value)}\``); } return result; }, async anyId(value: string, context: CommandContext) { const userId = resolveUserId(context.pluginData.client, value); if (userId) return userId; const channelIdMatch = value.match(channelMentionRegex); if (channelIdMatch) return channelIdMatch[1]; const roleIdMatch = value.match(roleMentionRegex); if (roleIdMatch) return roleIdMatch[1]; if (isValidSnowflake(value)) { return value; } throw new TypeConversionError(`Could not parse ID: \`${disableInlineCode(value)}\``); }, regex(value: string, context: CommandContext): RegExp { try { return inputPatternToRegExp(value); } catch (e) { throw new TypeConversionError(`Could not parse RegExp: \`${disableInlineCode(e.message)}\``); } }, }; export const commandTypeHelpers = { ...baseTypeHelpers, delay: createTypeHelper(commandTypes.delay), resolvedUser: createTypeHelper>(commandTypes.resolvedUser), resolvedUserLoose: createTypeHelper>(commandTypes.resolvedUserLoose), resolvedMember: createTypeHelper>(commandTypes.resolvedMember), messageTarget: createTypeHelper>(commandTypes.messageTarget), anyId: createTypeHelper>(commandTypes.anyId), regex: createTypeHelper(commandTypes.regex), };