Merge pull request #7 from withuno/smithki/discord-bot-health-check

Add health checks to discord bot service

Author: smithki

.github/workflows/bot-refresh.yml

@@ -5,11 +5,52 @@
 import fs from 'fs';
 import path from 'path';
 
-import { Client, GatewayIntentBits } from 'discord.js';
+import { Client, GatewayIntentBits, REST, RESTPostAPIChatInputApplicationCommandsJSONBody, Routes } from 'discord.js';
+import fastify from 'fastify';
 
-import { DiscordEvent } from './interfaces';
+import { DiscordEvent, SlashCommand } from './interfaces';
 
-async function main() {
+/**
+ * Start a `fastify` server to provide the app platform with health-checks.
+ */
+async function startServer() {
+  const server = fastify();
+  server.get('/', () => '[@withuno/locust] Locust Test Bot');
+  await server.listen({
+    port: Number(process.env.PORT ?? 8080),
+  });
+}
+
+/**
+ * Update the Discord Bot with the latest slash commands spec.
+ */
+async function loadSlashCommands() {
+  const commandsPath = path.resolve(__dirname, './commands');
+  const commandFiles = await fs.promises.readdir(commandsPath);
+  const commands: SlashCommand[] = commandFiles
+    .map((filepath) => {
+      return require(path.join(commandsPath, filepath))?.default;
+    })
+    .filter(Boolean);
+
+  const commandData: Array<RESTPostAPIChatInputApplicationCommandsJSONBody> = commands.map((cmd) => {
+    return cmd.slash.toJSON();
+  });
+
+  console.log(`◌ Refreshing ${commands.length} application command(s)...`);
+
+  const rest = new REST({ version: '10' }).setToken(process.env.DISCORD_TOKEN!);
+  const data: any = await rest.put(Routes.applicationCommands(process.env.DISCORD_CLIENT_ID!), {
+    body: commandData,
+  });
+
+  console.log(`◉ Refreshed ${data.length} application command(s)`);
+}
+
+/**
+ * Bootstrap a Discord client & start listening for events.
+ */
+async function startDiscordClient() {
   const client = new Client({ intents: [GatewayIntentBits.Guilds] });
 
   const eventsPath = path.resolve(__dirname, './events');
@@ -31,6 +72,12 @@ async function main() {
   await client.login(process.env.DISCORD_TOKEN);
 }
 
+async function main() {
+  await loadSlashCommands();
+  await startDiscordClient();
+  await startServer();
+}
+
 main().catch((err) => {
   console.error('----- FATAL ERROR -----');
   console.error(err);