feat(arabot): add trial mod role command

src/commands/roles/staff/trialMod.ts

@@ -0,0 +1,145 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+/*
+    Animal Rights Advocates Discord Bot
+    Copyright (C) 2023  Anthony Berg
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+*/
+
+import { Args, Command, RegisterBehavior } from '@sapphire/framework';
+import type { Guild, User, Message } from 'discord.js';
+import IDs from '#utils/ids';
+import { roleAddLog, roleRemoveLog } from '#utils/logging/role';
+
+export class TrialModCommand extends Command {
+  public constructor(context: Command.Context, options: Command.Options) {
+    super(context, {
+      ...options,
+      name: 'trialmod',
+      aliases: ['tm'],
+      description: 'Gives/removes the trial mod role',
+      preconditions: ['ModCoordinatorOnly'],
+    });
+  }
+
+  // Registers that this is a slash command
+  public override registerApplicationCommands(registry: Command.Registry) {
+    registry.registerChatInputCommand(
+      (builder) => builder
+        .setName(this.name)
+        .setDescription(this.description)
+        .addUserOption((option) => option.setName('user')
+          .setDescription('User to give/remove trial mod role')
+          .setRequired(true)),
+      {
+        behaviorWhenNotIdentical: RegisterBehavior.Overwrite,
+      },
+    );
+  }
+
+  // Command run
+  public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
+    // Get the arguments
+    const user = interaction.options.getUser('user', true);
+    const mod = interaction.user;
+    const { guild } = interaction;
+
+    // Checks if all the variables are of the right type
+    if (guild === null) {
+      await interaction.reply({
+        content: 'Error fetching guild!',
+        ephemeral: true,
+        fetchReply: true,
+      });
+      return;
+    }
+
+    await interaction.deferReply({ ephemeral: true });
+
+    const info = await this.manageMod(user, mod, guild);
+
+    await interaction.editReply(info.message);
+  }
+
+  public async messageRun(message: Message, args: Args) {
+    // Get arguments
+    let user: User;
+    try {
+      user = await args.pick('user');
+    } catch {
+      await message.react('❌');
+      await message.reply('User was not provided!');
+      return;
+    }
+
+    const mod = message.author;
+
+    if (mod === null) {
+      await message.react('❌');
+      await message.reply('Mod coordinator not found! Try again or contact a developer!');
+      return;
+    }
+
+    const { guild } = message;
+
+    if (guild === null) {
+      await message.react('❌');
+      await message.reply('Guild not found! Try again or contact a developer!');
+      return;
+    }
+
+    const info = await this.manageMod(user, mod, guild);
+
+    await message.reply(info.message);
+    await message.react(info.success ? '✅' : '❌');
+  }
+
+  private async manageMod(user: User, mod: User, guild: Guild) {
+    const info = {
+      message: '',
+      success: false,
+    };
+    const member = guild.members.cache.get(user.id);
+    const moderator = guild.roles.cache.get(IDs.roles.staff.trialModerator);
+
+    // Checks if user's GuildMember was found in cache
+    if (member === undefined) {
+      info.message = 'Error fetching guild member for the user!';
+      return info;
+    }
+
+    if (moderator === undefined) {
+      info.message = 'Error fetching the trial moderator role from cache!';
+      return info;
+    }
+
+    // Checks if the user has Trial Mod and to give them or remove them based on if they have it
+    if (member.roles.cache.has(IDs.roles.staff.trialModerator)) {
+      // Remove the Trial Mod role from the user
+      await member.roles.remove(moderator);
+      await roleRemoveLog(user.id, mod.id, moderator, true);
+      info.message = `Removed the ${moderator.name} role from ${user}`;
+      return info;
+    }
+    // Add Trial Mod role to the user
+    await member.roles.add(moderator);
+    await roleAddLog(user.id, mod.id, moderator, true);
+    info.message = `Gave ${user} the ${moderator.name} role!`;
+
+    await user.send(`You have been given the ${moderator.name} role by ${mod}!`)
+      .catch(() => {});
+    info.success = true;
+    return info;
+  }
+}