discordbot-ng/src/index.ts

import { ButtonInteraction, Client, GatewayIntentBits, MessageFlags, Partials } from 'discord.js';
import buttons from './buttons';
import { config } from './config';
import events from './events';
import { Database } from './shared/Database';
import { openRedisClient } from './shared/RedisClient';
import { Handler } from './types';
import { initIfNecessary, loadHandlersFrom, refreshCommands } from './utils';
import { initializeWebserver } from './webserver';
import { Encrypter } from './shared/Encrypter';
import tasks from './tasks';


let ready = false;

console.log('Starting...');

const client = new Client({
  intents: [
    GatewayIntentBits.Guilds,
    GatewayIntentBits.GuildMessages,
    GatewayIntentBits.GuildMembers,
    GatewayIntentBits.GuildModeration,
    GatewayIntentBits.GuildVoiceStates,
    GatewayIntentBits.MessageContent
  ],
  partials: [Partials.Message, Partials.GuildMember, Partials.User, Partials.Channel],

  rest: {
    version: '10'
  },

  allowedMentions: {
    parse: [],
    repliedUser: false
  },
});

const commands: Handler[] = [];
const contextMenus: Handler[] = [];
const modals: Handler[] = [];
const menus: Handler[] = [];

// 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.',
        flags: [MessageFlags.Ephemeral],
      });

    return;
  }

  if (interaction.isChatInputCommand()) {
    // Handle chat
    for (const command of commands) {
      if (interaction.commandName == command.name) {
        command.handler(client, interaction);
        return;
      }
    }
  } else if (interaction.isContextMenuCommand()) {
    // Handle context menu commands.
    for (const command of contextMenus) {
      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 (button.id != id) continue;

      Promise.resolve((button.execute as (context: Client, interaction: ButtonInteraction, ...args: unknown[]) => Promise<void>)(client, interaction, ...interaction.customId.split('_').slice(1))
        .catch(e => console.error('An error occurred while handling a button interaction: ', e)));
      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('_').slice(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('_').slice(1));
        return;
      }
    }
  }
});

client.on('clientReady', async () => {
  console.log(`Logged in as ${client.user!.tag}!`);

  await loadHandlersFrom('commands', commands);
  await loadHandlersFrom('context-menus', contextMenus);
  await loadHandlersFrom('modals', modals);
  await loadHandlersFrom('menus', menus);

  await refreshCommands(client);

  await initIfNecessary(client, commands);
  await initIfNecessary(client, modals);
  await initIfNecessary(client, menus);

  Encrypter.initialize(config.DATABASE_SECRET!);
  await Database.open('./data/discord-main.db');
  await openRedisClient(config.REDIS_URL!, client);

  await initializeWebserver(client);

  tasks.forEach((task) => {
    const handler = () => {
      Promise.resolve(task.handle(client))
        .catch(e => console.error(`Error in '${task.id}' task:`, e));
    };

    if (task.firstRun) handler();
    setInterval(handler, task.interval);
  });

  events.forEach(event =>
    client.on(event.event, (...args) =>
      Promise.resolve((event.execute as (context: Client, ...args: unknown[]) => Promise<void>)(client, ...args))
        .catch(e => console.error(`Error in ${event.event} handler:`, e))
    )
  );

  ready = true;
  console.log('Ready');
});

client.on('error', console.error);
process.on('uncaughtException', console.error);

client.login(config.DISCORD_TOKEN);