feat: add autocomplete command

Author: davidabou

src/harmonix.ts

@@ -29,7 +29,8 @@ import {
   registerCommands,
   registerButtons,
   registerModals,
-  registerSelectMenus
+  registerSelectMenus,
+  registerAutocomplete
 } from './register'
 import {
   resolveButton,
@@ -130,6 +131,7 @@ export const createHarmonix = async (
   registerButtons(harmonix)
   registerModals(harmonix)
   registerSelectMenus(harmonix)
+  registerAutocomplete(harmonix)
 
   return harmonix
 }

src/register.ts

@@ -2,7 +2,7 @@ import { ClientEvents, Events } from 'discord.js'
 import consola from 'consola'
 import { colors } from 'consola/utils'
 import { resolveOption } from './utils'
-import { ctx } from './harmonix'
+import { createError, ctx } from './harmonix'
 import type { Harmonix, ParsedInputs, ParsedOptions } from './types'
 
 export const registerEvents = (harmonix: Harmonix) => {
@@ -143,3 +143,17 @@ export const registerSelectMenus = (harmonix: Harmonix) => {
     slm.callback(interaction)
   })
 }
+
+export const registerAutocomplete = (harmonix: Harmonix) => {
+  harmonix.client?.on(Events.InteractionCreate, async (interaction) => {
+    if (!interaction.isAutocomplete()) return
+    const cmd = harmonix.commands.get(interaction.commandName)
+
+    if (!cmd || !cmd.config.autocomplete) return
+    try {
+      await cmd.config.autocomplete(interaction)
+    } catch (error: any) {
+      createError(error.message)
+    }
+  })
+}

src/types/commands.ts

@@ -9,7 +9,8 @@ import type {
   APIInteractionDataResolvedChannel,
   GuildMember,
   APIInteractionDataResolvedGuildMember,
-  Attachment
+  Attachment,
+  AutocompleteInteraction
 } from 'discord.js'
 
 type OptionType =
@@ -28,7 +29,9 @@ type _OptionDef<T extends OptionType> = {
   description?: string
   required?: boolean
   metadata?: Record<string, any>
-}
+} & (T extends 'String' | 'Integer' | 'Number'
+  ? { autocomplete?: boolean }
+  : {})
 
 type StringOptionDef = _OptionDef<'String'>
 type IntegerOptionDef = _OptionDef<'Integer'>
@@ -170,6 +173,7 @@ export interface CommandConfig<T extends OptionsDef = OptionsDef> {
   userPermissions?: PermissionsString[]
   preconditions?: string[]
   dm?: boolean
+  autocomplete?: (interaction: AutocompleteInteraction) => Promise<void> | void
 }
 
 export type CommandExecute<T extends OptionsDef = OptionsDef> = (

src/utils.ts

@@ -37,6 +37,7 @@ export const slashToJSON = (cmd: HarmonixCommand<OptionsDef>) => {
               .setName(name)
               .setDescription(arg.description ?? 'No description provided')
               .setRequired(arg.required ?? true)
+              .setAutocomplete(arg.autocomplete ?? false)
 
             if (arg.metadata?.minLength) {
               opt.setMinLength(arg.metadata.minLength)
@@ -60,6 +61,7 @@ export const slashToJSON = (cmd: HarmonixCommand<OptionsDef>) => {
               .setName(name)
               .setDescription(arg.description ?? 'No description provided')
               .setRequired(arg.required ?? true)
+              .setAutocomplete(arg.autocomplete ?? false)
 
             if (arg.metadata?.minValue) {
               opt.setMinValue(arg.metadata.minValue)
@@ -123,6 +125,7 @@ export const slashToJSON = (cmd: HarmonixCommand<OptionsDef>) => {
               .setName(name)
               .setDescription(arg.description ?? 'No description provided')
               .setRequired(arg.required ?? true)
+              .setAutocomplete(arg.autocomplete ?? false)
 
             if (arg.metadata?.minValue) {
               opt.setMinValue(arg.metadata.minValue)