Initial commit.

src/index.ts

@@ -0,0 +1,155 @@
+import 'source-map-support/register';
+import { Client as DiscordClient, GatewayIntentBits, Guild, Partials } from 'discord.js';
+import { config } from './config';
+import { Handler } from './types';
+import { initIfNecessary, loadHandlersFrom, openRedisClient, refreshCommands } from './utils';
+import { initializeDiscordJoiner } from './discord-joiner';
+import { Database } from './shared/Database';
+import { handleAuditLogCreate, handleBulkMessageDelete, handleGuildCreate, handleMemberJoin, handleMessageCreate, handleMessageDelete, handleMessageUpdate, handleThreadCreate, handleVoiceStateUpdate } from './events';
+import { ticketCooldownMap } from './shared/ticket-cooldown';
+
+let ready = false;
+
+console.log('Starting...');
+
+const client = new DiscordClient({
+  intents: [
+    GatewayIntentBits.Guilds,
+    GatewayIntentBits.GuildMessages,
+    GatewayIntentBits.GuildMembers,
+    GatewayIntentBits.GuildModeration,
+    GatewayIntentBits.GuildVoiceStates,
+    GatewayIntentBits.MessageContent
+  ],
+  partials: [Partials.Message, Partials.GuildMember, Partials.User, Partials.Channel],
+  rest: { timeout: 30000 },
+  allowedMentions: {
+    parse: [],
+    repliedUser: false
+  }
+});
+
+const commands: Handler[] = [];
+const buttons: Handler[] = [];
+const modals: Handler[] = [];
+const menus: Handler[] = [];
+
+loadHandlersFrom('commands', commands);
+loadHandlersFrom('buttons', buttons);
+loadHandlersFrom('modals', modals);
+loadHandlersFrom('menus', menus);
+
+// Due to their reliance on each other, these two events (interactionCreate, and ready) have to stay here.
+// Alternatively, they can move to another single file. Or use static classes.
+client.on('interactionCreate', async (interaction) => {
+  if (!ready) {
+    if (
+      interaction.isChatInputCommand()
+      || interaction.isContextMenuCommand()
+      || interaction.isButton()
+      || interaction.isModalSubmit()
+      || interaction.isAnySelectMenu()
+    )
+      interaction.reply({
+        content: 'Bot is still starting up. Please wait a few seconds.',
+        ephemeral: true,
+      });
+
+    return;
+  }
+
+  if (interaction.isChatInputCommand() || interaction.isContextMenuCommand()) {
+    // Handle chat and context menu commands.
+    for (const command of commands) {
+      if (interaction.commandName == command.name) {
+        command.handler(client, interaction);
+        return;
+      }
+    }
+  } else if (interaction.isAutocomplete()) {
+    // Handle autocomplete requests.
+    for (const command of commands) {
+      if (interaction.commandName == command.name) {
+        if (command.autoComplete) {
+          command
+            .autoComplete(client, interaction)
+            .catch(e => console.error(e));
+        }
+        return;
+      }
+    }
+  } else if (interaction.isButton()) {
+    // Handle button presses.
+    const id = interaction.customId.split('_')[0];
+
+    for (const button of buttons) {
+      if (id == button.name) {
+        button.handler(client, interaction, interaction.customId.split('_')[1]);
+        return;
+      }
+    }
+  } else if (interaction.isModalSubmit()) {
+    // Handle modal submissions.
+    const id = interaction.customId.split('_')[0];
+
+    for (const modal of modals) {
+      if (id == modal.name) {
+        modal.handler(client, interaction, interaction.customId.split('_')[1]);
+        return;
+      }
+    }
+  } else if (interaction.isAnySelectMenu()) {
+    // Handle menu selections.
+    const id = interaction.customId.split('_')[0];
+
+    for (const menu of menus) {
+      if (id == menu.name) {
+        menu.handler(client, interaction, interaction.customId.split('_')[1]);
+        return;
+      }
+    }
+  }
+});
+
+client.on('ready', async () => {
+  console.log(`Logged in as ${client.user!.tag}!`);
+
+  await refreshCommands(client);
+
+  await initIfNecessary(client, commands);
+  await initIfNecessary(client, buttons);
+  await initIfNecessary(client, modals);
+  await initIfNecessary(client, menus);
+
+  await Database.open('./data/discord-main.db');
+  await openRedisClient(config.REDIS_URL!, client);
+
+  await initializeDiscordJoiner();
+
+  // Prune ticket cooldowns that are expired every day
+  setInterval(() => {
+    const keys = Array.from(ticketCooldownMap.keys());
+    for (const key of keys) {
+      if (Date.now() >= ticketCooldownMap.get(key)!)
+        ticketCooldownMap.delete(key);
+    }
+  }, 8.64e+7);
+
+  ready = true;
+
+  console.log('Ready');
+});
+
+client.on('guildAuditLogEntryCreate', handleAuditLogCreate);
+client.on('guildCreate', handleGuildCreate);
+client.on('guildMemberAdd', handleMemberJoin);
+client.on('messageCreate', handleMessageCreate);
+client.on('messageDelete', handleMessageDelete);
+client.on('messageDeleteBulk', handleBulkMessageDelete);
+client.on('messageUpdate', handleMessageUpdate);
+client.on('threadCreate', handleThreadCreate);
+client.on('voiceStateUpdate', handleVoiceStateUpdate);
+
+client.on('error', console.error);
+
+client.login(config.DISCORD_TOKEN);