Feature/multi guild support

.gitignore

@@ -1,3 +1,5 @@
-node_modules
-dist
-.env
+node_modules/
+dist/
+.env
+
+.idea/

resources/guildSettings.json

@@ -0,0 +1,40 @@
+{
+  "1325571365079879774": {
+    "channels": {
+      "logging": "1442896999237030031",
+      "commits": "1489996651551522816",
+      "errors": "1491479994994659428"
+    },
+    "roles": {
+      "updates": "1445497401614925914",
+      "qotd": "1445497592573067546",
+      "support": "1445887485853962300",
+      "community": "1416251642185121823"
+    },
+    "trusted": [
+      "1441859003708866601",
+      "768481984242253904",
+      "855798460593733652",
+      "1382022366040686763"
+    ]
+  },
+  "1310540137096282122": {
+    "channels": {
+      "logging": "1492607126638563459",
+      "commits": "1492607196301758525",
+      "errors": "1492607206346981557"
+    },
+    "roles": {
+      "updates": "1492607385112674354",
+      "qotd": "1492607410483761232",
+      "support": "1492607474996346952",
+      "community": "1492607539919978646"
+    },
+    "trusted": [
+      "1441859003708866601",
+      "768481984242253904",
+      "855798460593733652",
+      "1382022366040686763"
+    ]
+  }
+}

src/commands/admin/DevResetCommand.ts

@@ -2,7 +2,7 @@ import { PermissionsBitField } from 'discord.js';
 import { Command, CommandContext, CommandCheckFlags } from '@structures/Command.js';
 import { ExtendedClient } from '@structures/Client.js';
 import { Embeds } from '@utils/Embeds.js';
-import { Constants } from '@utils/Constants.js';
+import * as ds from '@data/DataStore.js';
 
 export default class DevResetCommand extends Command {
   constructor() {
@@ -16,8 +16,17 @@ export default class DevResetCommand extends Command {
 
   public async execute(client: ExtendedClient, context: CommandContext): Promise<void> {
     const author = context.author!;
+    const data = ds.getDataFromContext(context);
 
-    if (!Constants.TRUSTED_USER_IDS.includes(author.id)) {
+    if (!data) {
+      await context.reply({
+        embeds: [Embeds.error('Something went wrong, please try again later.')],
+      });
+
+      return;
+    }
+
+    if (!data.trusted.includes(author.id)) {
       await context.reply({
         embeds: [Embeds.error('You are not authorized to use this command.')],
       });

src/commands/admin/CrashCommand.ts → src/commands/admin/ShutdownCommand.ts

@@ -2,22 +2,29 @@ import { PermissionsBitField } from 'discord.js';
 import { Command, CommandContext, CommandCheckFlags } from '@structures/Command.js';
 import { ExtendedClient } from '@structures/Client.js';
 import { Embeds } from '@utils/Embeds.js';
-import { Constants } from '@utils/Constants.js';
+import * as ds from '@data/DataStore.js';
 
-export default class CrashCommand extends Command {
+export default class ShutdownCommand extends Command {
   constructor() {
     super({
-      name: 'crash',
-      description: 'Crash the bot',
+      name: 'shutdown',
+      description: 'Shut down the bot',
       requiredPermissions: [PermissionsBitField.Flags.Administrator],
       checkFlags: CommandCheckFlags.Author | CommandCheckFlags.Guild,
     });
   }
 
   public async execute(client: ExtendedClient, context: CommandContext): Promise<void> {
     const author = context.author!;
+    const data = ds.getDataFromContext(context);
 
-    if (!Constants.TRUSTED_USER_IDS.includes(author.id)) {
+    if (!data) {
+      return await context.reply({
+        embeds: [Embeds.error('Something went wrong, please try again later.')],
+      });
+    }
+
+    if (!data.trusted.includes(author.id)) {
       await context.reply({
         embeds: [Embeds.error('You are not authorized to use this command.')],
       });
@@ -26,9 +33,10 @@ export default class CrashCommand extends Command {
     }
 
     await context.reply({
-      embeds: [Embeds.error('Crashing...')],
+      embeds: [Embeds.error('Shutting down...')],
     });
 
-    process.exit(0);
+    await client.destroy();
+    process.exit();
   }
 }

src/commands/admin/SyncCommand.ts

@@ -2,7 +2,7 @@ import { PermissionFlagsBits } from 'discord.js';
 import { ExtendedClient } from '@structures/Client.js';
 import { Command, CommandContext, CommandCheckFlags } from '@structures/Command.js';
 import { Embeds } from '@utils/Embeds.js';
-import { Constants } from '@utils/Constants.js';
+import * as ds from '@data/DataStore.js';
 
 export default class SyncCommand extends Command {
   constructor() {
@@ -16,16 +16,23 @@ export default class SyncCommand extends Command {
 
   public async execute(client: ExtendedClient, context: CommandContext): Promise<void> {
     const guild = context.guild!;
+    const data = ds.getDataFromContext(context);
+
+    if (!data) {
+      return await context.reply({
+        embeds: [Embeds.error('Something went wrong, please try again later.')],
+      });
+    }
 
     const members = await guild.members.fetch();
-    const communityRole = guild.roles.cache.get(Constants.ROLES.COMMUNITY);
+    const communityRole = guild.roles.cache.get(data.roles.community);
     const ignoredRoles = new Set(
       [
         guild.id, // @everyone role
         guild.roles.premiumSubscriberRole?.id,
-        guild.roles.cache.get(Constants.ROLES.UPDATES)?.id,
-        guild.roles.cache.get(Constants.ROLES.QOTD_PING)?.id,
-        guild.roles.cache.get(Constants.ROLES.SUPPORT)?.id,
+        guild.roles.cache.get(data.roles.updates)?.id,
+        guild.roles.cache.get(data.roles.qotd)?.id,
+        guild.roles.cache.get(data.roles.support)?.id,
       ].filter((id) => id != null)
     );
 

src/commands/admin/UpdateGeminiKeyCommand.ts

@@ -3,7 +3,7 @@ import { ExtendedClient } from '@structures/Client.js';
 import { Command, CommandContext, CommandCheckFlags } from '@structures/Command.js';
 import { Argument } from '@structures/Argument.js';
 import { Embeds } from '@utils/Embeds.js';
-import { Constants } from '@utils/Constants.js';
+import * as ds from '@data/DataStore.js';
 
 export default class UpdateGeminiKeyCommand extends Command {
   constructor() {
@@ -25,8 +25,17 @@ export default class UpdateGeminiKeyCommand extends Command {
 
   public async execute(client: ExtendedClient, context: CommandContext): Promise<void> {
     const author = context.author!;
+    const data = ds.getDataFromContext(context);
 
-    if (!Constants.TRUSTED_USER_IDS.includes(author.id)) {
+    if (!data) {
+      await context.reply({
+        embeds: [Embeds.error('Something went wrong, please try again later.')],
+      });
+
+      return;
+    }
+
+    if (!data.trusted.includes(author.id)) {
       await context.reply({
         embeds: [Embeds.error('You are not authorized to use this command.')],
       });

src/data/Config.ts

@@ -0,0 +1,2 @@
+export const GUILD_SETTINGS_FILE = 'resources/guildSettings.json';
+export const COBALT_GUILD_ID = '1325571365079879774';

src/data/DataStore.ts

@@ -0,0 +1,82 @@
+type ChannelTypes = 'logging' | 'commits' | 'errors';
+type RoleTypes = 'updates' | 'qotd' | 'support' | 'community';
+import * as fs from 'fs';
+import { GUILD_SETTINGS_FILE } from '@data/Config.js';
+import { CommandContext } from '@structures/Command.js';
+import { validateGuildSettingsFile } from '../utils/JsonValidation.js';
+import { Logger } from '@utils/Logger.js';
+import { IllegalArgumentError } from '../errors/IllegalArgumentError.js';
+
+export class GuildData {
+  readonly guildId: string;
+  channels: Record<ChannelTypes, string>;
+  roles: Record<RoleTypes, string>;
+  trusted: string[];
+
+  constructor(guildId: string) {
+    this.guildId = guildId;
+    this.channels = {
+      logging: '',
+      commits: '',
+      errors: '',
+    };
+    this.roles = {
+      updates: '',
+      qotd: '',
+      support: '',
+      community: '',
+    };
+    this.trusted = [];
+  }
+
+  toJson(): string {
+    return JSON.stringify(
+      {
+        [this.guildId]: {
+          channels: this.channels,
+          roles: this.roles,
+          trusted: this.trusted,
+        },
+      },
+      null,
+      2
+    );
+  }
+}
+
+export let dataStore: GuildData[] = loadFromJson(GUILD_SETTINGS_FILE);
+
+function loadFromJson(jsonFilePath: string): GuildData[] {
+  const fileContent = fs.readFileSync(jsonFilePath, 'utf-8');
+  const data = JSON.parse(fileContent) as unknown;
+  validateGuildSettingsFile(data, jsonFilePath);
+
+  const values: GuildData[] = [];
+  for (const guildId of Object.keys(data)) {
+    const guildObj = data[guildId];
+    const guildData = new GuildData(guildId);
+    guildData.channels = guildObj.channels;
+    guildData.roles = guildObj.roles;
+    guildData.trusted = guildObj.trusted;
+    values.push(guildData);
+  }
+  return values;
+}
+
+export function getDataFromContext(context: CommandContext): GuildData | undefined {
+  if (!context.guild) {
+    return undefined;
+  }
+  return getDataForGuild(context.guild.id);
+}
+
+export function getDataForGuild(guildId: string): GuildData {
+  const data = dataStore.find((guild) => guild.guildId === guildId);
+  if (data === undefined) {
+    throw new IllegalArgumentError(
+      `Requested data for guild: ${guildId}, but this guild is not found in the guild settings file!`
+    );
+  }
+
+  return data;
+}

src/errors/IllegalArgumentError.ts

@@ -0,0 +1,6 @@
+export class IllegalArgumentError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = 'IllegalArgumentError';
+  }
+}

src/errors/RuntimeError.ts

@@ -0,0 +1,6 @@
+export class RuntimeError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = 'RuntimeException';
+  }
+}

src/events/ChatBotHandleEvent.ts

@@ -1,6 +1,8 @@
 import { Event } from '@structures/Event.js';
 import { ExtendedClient } from '@structures/Client.js';
 import { Message, PermissionFlagsBits } from 'discord.js';
+import { isErrorWithMessage, messageOrJsonToMessage } from '@utils/ErrorUtil.js';
+import { Logger } from '@utils/Logger.js';
 
 export default class ChatBotHandleEvent extends Event<'messageCreate'> {
   constructor() {
@@ -32,7 +34,14 @@ export default class ChatBotHandleEvent extends Event<'messageCreate'> {
 
     await message.channel.sendTyping();
 
-    const res = await client.chatBot.generateResponse(content, message.author);
+    let res;
+    try {
+      res = await client.chatBot.generateResponse(content, message.author);
+    } catch (error) {
+      if (isErrorWithMessage(error)) {
+        await Logger.logErrorWithBot(messageOrJsonToMessage(error.message), message.guild);
+      }
+    }
 
     if (!res || res.length === 0) {
       return;

src/events/ClientReadyEvent.ts

@@ -4,7 +4,6 @@ import { Logger } from '@utils/Logger.js';
 import { CommandManager } from '@structures/CommandManager.js';
 import { dirname, join } from 'node:path';
 import { fileURLToPath } from 'node:url';
-import { Constants } from '@utils/Constants.js';
 
 export default class ClientReadyEvent extends Event<'clientReady'> {
   constructor() {
@@ -18,7 +17,7 @@ export default class ClientReadyEvent extends Event<'clientReady'> {
     Logger.success(`Logged in as ${client.user?.tag}`);
 
     const commandsDirectory = join(dirname(fileURLToPath(import.meta.url)), '..', 'commands');
-    const commandManager = new CommandManager(client, Constants.GUILD_ID);
+    const commandManager = new CommandManager(client);
 
     await commandManager.loadCommands(commandsDirectory);
 

src/events/InteractionCreateEvent.ts

@@ -48,7 +48,7 @@ export default class InteractionCreateEvent extends Event {
       }
 
       const context = command.createContext(client, args, interaction);
-      await command.execute(client, context);
+      await command.run(client, context);
     } catch (error) {
       const errorMessage = {
         embeds: [Embeds.error(`${error instanceof Error ? error.message : 'Unknown error'}`)],

src/events/MessageCreateEvent.ts

@@ -33,7 +33,7 @@ export default class MessageCreateEvent extends Event<'messageCreate'> {
       const parsedArgs = await command.parseChatArgs(args, message.guild ?? undefined);
 
       const context = command.createContext(client, parsedArgs, undefined, message);
-      await command.execute(client, context);
+      await command.run(client, context);
     } catch (error) {
       await message.reply({
         embeds: [Embeds.error(`${error instanceof Error ? error.message : 'Unknown error'}`)],