123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480
import {  ApplicationCommandHandler,  ApplicationCommandHandlerCallback,  AutocompleteHandler,  AutocompleteHandlerCallback,  InteractionsClient} from './client.ts'import type { Client } from '../client/mod.ts'import { ApplicationCommandsModule } from './commandModule.ts'import { ApplicationCommandInteraction } from '../structures/applicationCommand.ts'import { GatewayIntents } from '../types/gateway.ts'import { ApplicationCommandType } from '../types/applicationCommand.ts'
/**  Type extension that adds the `_decoratedAppCmd` list. */interface DecoratedAppExt {  _decoratedAppCmd?: ApplicationCommandHandler[]  _decoratedAutocomplete?: AutocompleteHandler[]}
// Maybe a better name for this would be `ApplicationCommandBase` or `ApplicationCommandObject` or something elsetype ApplicationCommandClient =  | Client  | InteractionsClient  | ApplicationCommandsModule
// See abovetype ApplicationCommandClientExt = ApplicationCommandClient & DecoratedAppExt
type CommandValidationCondition = (  i: ApplicationCommandInteraction) => boolean | Promise<boolean>
interface CommandValidation {  condition: CommandValidationCondition  action?: string | ApplicationCommandHandlerCallback}
type ApplicationCommandDecorator = (  client: ApplicationCommandClientExt,  prop: string,  desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>) => void
type AutocompleteDecorator = (  client: ApplicationCommandClientExt,  prop: string,  desc: TypedPropertyDescriptor<AutocompleteHandlerCallback>) => void
/** * Wraps the command handler with a validation function. * @param desc property descriptor * @param validation validation function and action to show or call if validation fails * @returns wrapped function */function wrapConditionApplicationCommandHandler(  desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>,  validation: CommandValidation): ApplicationCommandHandlerCallback {  if (typeof desc.value !== 'function') {    throw new Error('The decorator requires a function')  }  const { condition, action } = validation
  const original = desc.value  return async function (    this: ApplicationCommandClient,    i: ApplicationCommandInteraction  ) {    if (!(await condition(i))) {      // condition not met      if (typeof action === 'string') {        i.reply(action)      } else if (typeof action === 'function') {        action(i)      }      return    } // condition met    return original.call(this, i)  }}
/** * Decorator to add a autocomplete interaction handler. * * Example: * ```ts * class MyClient extends Client { *   // ... * *   @autocomplete("search", "query") *   searchCompletions(i: AutocompleteInteraction) { *     // ... *   } * } * ``` * * @param command Command name of which options' to provide autocompletions for. Can be `*` (all). * @param option Option name to handle autocompletions for. Can be `*` (all). */export function autocomplete(  command: string,  option: string): AutocompleteDecorator {  return function (    client: ApplicationCommandClientExt,    _prop: string,    desc: TypedPropertyDescriptor<AutocompleteHandlerCallback>  ) {    if (client._decoratedAutocomplete === undefined)      client._decoratedAutocomplete = []    if (typeof desc.value !== 'function') {      throw new Error('@autocomplete decorator requires a function')    } else      client._decoratedAutocomplete.push({        cmd: command,        option,        handler: desc.value      })  }}
/** * Decorator to create a Slash Command handler. * * Example: * ```ts * class MyClient extends Client { *   // ... * *   @slash("my-command") *   myCommand(i: ApplicationCommandInteraction) { *     return i.reply("Hello, World!"); *   } * } * ``` * * Note that the name parameter is optional in this decorator, * it can also be inferred from the method name you define and * use decorator on. * * If you want to split these decorators into different * files, you can use these in classes extending * `ApplicationCommandsModule` and then use * `client.interactions.loadModule`. * * For handling sub-commands or grouped sub-commands, look * into docs for `subslash` and `groupslash`. */export function slash(  name?: string,  guild?: string): ApplicationCommandDecorator {  return function (    client: ApplicationCommandClientExt,    prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    if (client._decoratedAppCmd === undefined) client._decoratedAppCmd = []    if (typeof desc.value !== 'function') {      throw new Error('@slash decorator requires a function')    } else      client._decoratedAppCmd.push({        name: name ?? prop,        guild,        handler: desc.value      })  }}
/** * Decorator to create a Sub-Command Command handler for a * Slash Command. * * Example: * ```ts * class MyClient extends Client { *   // ... * *   @subslash("config", "reset") *   configReset(i: ApplicationCommandInteraction) { *     // ... *   } * } * ``` * * Note that only first argument that is `parent` is required, * second can be inferred from the method name you define * and use decorator on. * * For more info on Slash Command handler decorators, look * into docs for `slash` decorator. */export function subslash(  parent: string,  name?: string,  guild?: string): ApplicationCommandDecorator {  return function (    client: ApplicationCommandClientExt,    prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    if (client._decoratedAppCmd === undefined) client._decoratedAppCmd = []    if (typeof desc.value !== 'function') {      throw new Error('@subslash decorator requires a function')    } else      client._decoratedAppCmd.push({        parent,        name: name ?? prop,        guild,        handler: desc.value      })  }}
/** * Decorator to create a Grouped Sub-Command Command handler * for a Slash Command. * * Example: * ```ts * class MyClient extends Client { *   // ... * *   @groupslash("config", "options", "set") *   configOptionsSet(i: ApplicationCommandInteraction) { *     // ... *   } * } * ``` * * Note that only first two arguments i.e. `group` & `parent` are * required, name can be inferred from the method name you define * and use decorator on. * * For more info on Slash Command handler decorators, look * into docs for `slash` decorator. */export function groupslash(  parent: string,  group: string,  name?: string,  guild?: string): ApplicationCommandDecorator {  return function (    client: ApplicationCommandClientExt,    prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    if (client._decoratedAppCmd === undefined) client._decoratedAppCmd = []    if (typeof desc.value !== 'function') {      throw new Error('@groupslash decorator requires a function')    } else      client._decoratedAppCmd.push({        group,        parent,        name: name ?? prop,        guild,        handler: desc.value      })  }}
/** * Decorator to create a Message Context Menu Command handler. * * Example: * ```ts * class MyClient extends Client { *   // ... * *   @messageContextMenu("Content Length") *   contentLength(i: ApplicationCommandInteraction) { *     return i.reply({ *       content: `Length: ${i.targetMessage!.content.length}`, *       ephemeral: true, *     }); *   } * } * ``` * * If you want to split these decorators into different * files, you can use these in classes extending * `ApplicationCommandsModule` and then use * `client.interactions.loadModule`. * * For handling user context menu commands, look into docs for * `userContextMenu` decorator. */export function messageContextMenu(name?: string): ApplicationCommandDecorator {  return function (    client: ApplicationCommandClientExt,    prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    // eslint-disable-next-line @typescript-eslint/no-unnecessary-type-assertion    if (client._decoratedAppCmd === undefined) client._decoratedAppCmd = []    if (typeof desc.value !== 'function') {      throw new Error('@messageContextMenu decorator requires a function')    } else      client._decoratedAppCmd.push({        name: name ?? prop,        type: ApplicationCommandType.MESSAGE,        handler: desc.value      })  }}
/** * Decorator to create a User Context Menu Command handler. * * Example: * ```ts * class MyClient extends Client { *   // ... * *   @userContextMenu("Command Name") *   commandName(i: ApplicationCommandInteraction) { *     // ... *   } * } * ``` * * First argument that is `name` is optional and can be * inferred from method name. * * For handling more docs regarding how context menu command * decorators work, look into `messageContextMenu`'s docs. */export function userContextMenu(name?: string): ApplicationCommandDecorator {  return function (    client: ApplicationCommandClientExt,    prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    // eslint-disable-next-line @typescript-eslint/no-unnecessary-type-assertion    if (client._decoratedAppCmd === undefined) client._decoratedAppCmd = []    if (typeof desc.value !== 'function') {      throw new Error('@userContextMenu decorator requires a function')    } else      client._decoratedAppCmd.push({        name: name ?? prop,        type: ApplicationCommandType.USER,        handler: desc.value      })  }}
/** * The command can only be called from a guild. * @param action message or function called when the condition is not met * @returns wrapped function */export function isInGuild(message: string): ApplicationCommandDecoratorexport function isInGuild(  callback: ApplicationCommandHandlerCallback): ApplicationCommandDecoratorexport function isInGuild(  action: string | ApplicationCommandHandlerCallback): ApplicationCommandDecorator {  return function (    _client: ApplicationCommandClient,    _prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    const validation: CommandValidation = {      condition: (i: ApplicationCommandInteraction) => {        return Boolean(i.guild)      },      action    }    desc.value = wrapConditionApplicationCommandHandler(desc, validation)  }}
/** * The command can only be called if the bot is currently in a voice channel. * `GatewayIntents.GUILD_VOICE_STATES` needs to be set. * @param action message or function called when the condition is not met * @returns wrapped function */export function isBotInVoiceChannel(  message: string): ApplicationCommandDecoratorexport function isBotInVoiceChannel(  callback: ApplicationCommandHandlerCallback): ApplicationCommandDecoratorexport function isBotInVoiceChannel(  action: string | ApplicationCommandHandlerCallback): ApplicationCommandDecorator {  return function (    _client: ApplicationCommandClient,    _prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    const validation: CommandValidation = {      condition: async (i: ApplicationCommandInteraction) => {        // eslint-disable-next-line @typescript-eslint/strict-boolean-expressions        if (!i.client.intents?.includes(GatewayIntents.GUILD_VOICE_STATES)) {          const err =            '@isBotInVoiceChannel: GatewayIntents.GUILD_VOICE_STATES needs to be set.'          console.error(err)          throw new Error(err)        }        return Boolean(await i.guild?.voiceStates.get(i.client.user!.id))      },      action    }    desc.value = wrapConditionApplicationCommandHandler(desc, validation)  }}
/** * The command can only be called if the user is currently in a voice channel. * `GatewayIntents.GUILD_VOICE_STATES` needs to be set. * @param action message or function called when the condition is not met * @returns wrapped function */export function isUserInVoiceChannel(  message: string): ApplicationCommandDecoratorexport function isUserInVoiceChannel(  callback: ApplicationCommandHandlerCallback): ApplicationCommandDecoratorexport function isUserInVoiceChannel(  action: string | ApplicationCommandHandlerCallback): ApplicationCommandDecorator {  return function (    _client: ApplicationCommandClient,    _prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    const validation: CommandValidation = {      condition: async (i: ApplicationCommandInteraction): Promise<boolean> => {        // eslint-disable-next-line @typescript-eslint/strict-boolean-expressions        if (!i.client.intents?.includes(GatewayIntents.GUILD_VOICE_STATES)) {          const err =            '@isUserInVoiceChannel: GatewayIntents.GUILD_VOICE_STATES needs to be set.'          console.error(err)          throw new Error(err)        }        return Boolean(await i.guild?.voiceStates.get(i.user.id))      },      action    }    desc.value = wrapConditionApplicationCommandHandler(desc, validation)  }}
/** * Cusomizable command validation. * @param condition condition that need to succede for the command to be called * @param action message or function called when the condition is not met * @returns wrapped function */export function customValidation(  condition: CommandValidationCondition,  message: string): ApplicationCommandDecoratorexport function customValidation(  condition: CommandValidationCondition,  callback: ApplicationCommandHandlerCallback): ApplicationCommandDecoratorexport function customValidation(  condition: CommandValidationCondition,  action: string | ApplicationCommandHandlerCallback): ApplicationCommandDecorator {  return function (    _client: ApplicationCommandClient,    _prop: string,    desc: TypedPropertyDescriptor<ApplicationCommandHandlerCallback>  ) {    desc.value = wrapConditionApplicationCommandHandler(desc, {      condition,      action    })  }}