feat: autocompleter (#14)

* feat: autocompleter

* fix: don't format dist

* fix: copypasted namings

Author: Chicken

.prettierignore

@@ -0,0 +1 @@
+dist/

src/Autocompleter.ts

@@ -0,0 +1,11 @@
+import { AutocompleteInteraction } from "discord.js";
+
+export abstract class Autocompleter {
+	public readonly command: string;
+
+	protected constructor(command: string) {
+		this.command = command;
+	}
+
+	abstract execute(interaction: AutocompleteInteraction): Promise<void>;
+}

src/autocompleters/AutocompleterSample.ts.disable

@@ -0,0 +1,19 @@
+import { Autocompleter } from "../Autocompleter";
+import { AutocompleteInteraction } from "discord.js";
+
+class CountryAutocompleter extends Autocompleter {
+	constructor() {
+		super("country"); // command which this autocompleter is for
+	}
+
+	async execute(interaction: AutocompleteInteraction): Promise<void> {
+		// your options
+		const country = ["Finland", "Sweden", "Norway"];
+		// for example return all options which start with the user input
+		await interaction.respond(
+			country.filter((c) => c.startsWith(interaction.options.getFocused() as string)).map((c) => ({ name: c, value: c })),
+		);
+	}
+}
+
+export default new CountryAutocompleter();

src/ctx.ts

@@ -1,11 +1,13 @@
 import { ButtonHandler } from "./ButtonHandler";
 import { Collection } from "discord.js";
 import { Command } from "./Command";
+import { Autocompleter } from "Autocompleter";
 
 class Ctx {
 	public readonly defaultGuild;
 	public readonly commands: Collection<string, Command> = new Collection<string, Command>();
 	public readonly buttons: Collection<string, ButtonHandler> = new Collection<string, ButtonHandler>();
+	public readonly autocompleters: Collection<string, Autocompleter> = new Collection<string, Autocompleter>();
 	public readonly logLevel: string;
 
 	// Databases are gonna go here
@@ -18,9 +20,14 @@ class Ctx {
 		this.logLevel = LOG_LEVEL || "error";
 	}
 
-	update(commands: Collection<string, Command>, buttons: Collection<string, ButtonHandler>) {
+	update(
+		commands: Collection<string, Command>,
+		buttons: Collection<string, ButtonHandler>,
+		autocompleters: Collection<string, Autocompleter>
+	) {
 		commands.forEach((v, k) => this.commands.set(k, v));
 		buttons.forEach((v, k) => this.buttons.set(k, v));
+		autocompleters.forEach((v, k) => this.autocompleters.set(k, v));
 	}
 }
 

src/events/interactionCreate.ts

@@ -1,4 +1,4 @@
-import { ButtonInteraction, CommandInteraction, Interaction } from "discord.js";
+import { AutocompleteInteraction, ButtonInteraction, CommandInteraction, Interaction } from "discord.js";
 import { ctx } from "../ctx";
 import { logger } from "../logger";
 
@@ -8,6 +8,8 @@ export default async function interactionCreate(interaction: Interaction) {
 			await handleCommandInteractions(interaction);
 		} else if (interaction.isButton()) {
 			await handleButtonInteractions(interaction);
+		} else if (interaction.isAutocomplete()) {
+			await handleAutocompleteInteraction(interaction);
 		}
 	} catch (err: unknown) {
 		if (err instanceof Error) {
@@ -41,3 +43,13 @@ async function handleButtonInteractions(interaction: ButtonInteraction) {
 		await interaction.reply("An error occurred. Please contact a developer");
 	}
 }
+
+async function handleAutocompleteInteraction(interaction: AutocompleteInteraction) {
+	const autocompleter = ctx.autocompleters.get(interaction.commandName);
+	if (autocompleter) {
+		await autocompleter.execute(interaction);
+	} else {
+		logger.error(`error resolving autocompleter for ${interaction.commandName}`);
+		await interaction.respond([]);
+	}
+}

src/loader.ts

@@ -1,5 +1,6 @@
 import { Client, Collection } from "discord.js";
 import { Command } from "./Command";
+import { Autocompleter } from "./Autocompleter";
 import * as fs from "fs";
 import { ButtonHandler } from "./ButtonHandler";
 import { logger } from "./logger";
@@ -46,3 +47,18 @@ export async function loadEvents(client: Client) {
 			})
 	);
 }
+
+export async function loadAutocompleters(): Promise<Collection<string, Autocompleter>> {
+	logger.debug("loading autocompleters...");
+	const loadedAutocompleters = new Collection<string, Autocompleter>();
+	await Promise.all(
+		fs
+			.readdirSync("./dist/autocompleters")
+			.filter((f) => f.endsWith(".js"))
+			.map(async (filename) => {
+				const autocompleter = (await import(`./autocompleters/${filename}`)).default as Autocompleter;
+				loadedAutocompleters.set(autocompleter.command, autocompleter);
+			})
+	);
+	return loadedAutocompleters;
+}

src/start.ts

@@ -1,5 +1,5 @@
 import "dotenv/config";
-import { loadButtonHandlers, loadCommands, loadEvents } from "./loader";
+import { loadAutocompleters, loadButtonHandlers, loadCommands, loadEvents } from "./loader";
 import { Client } from "discord.js";
 import { ctx } from "./ctx";
 import { logger } from "./logger";
@@ -19,7 +19,7 @@ async function start(): Promise<Client> {
 	const client = new Client({ intents: [] });
 
 	logger.debug("Loading context...");
-	ctx.update(await loadCommands(), await loadButtonHandlers());
+	ctx.update(await loadCommands(), await loadButtonHandlers(), await loadAutocompleters());
 	logger.debug("Loading events...");
 	await loadEvents(client);
 	logger.debug("Attempting login");