arabot/src/commands/outreach/outreach.ts

// 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 { Subcommand } from '@sapphire/plugin-subcommands';
import { RegisterBehavior } from '@sapphire/framework';
import type { Snowflake } from 'discord.js';
import { updateUser } from '#utils/database/dbExistingUser';
import {
  addStatUser,
  checkActiveEvent,
  createEvent,
  createStat,
  getCurrentEvent,
  getStatFromLeader,
  getStatFromRole,
  getStatGroups,
  updateStats,
  userInStats,
} from '#utils/database/outreach';
import IDs from '#utils/ids';

export class OutreachCommand extends Subcommand {
  public constructor(context: Subcommand.Context, options: Subcommand.Options) {
    super(context, {
      ...options,
      name: 'outreach',
      description: 'Tools for doing outreach',
      subcommands: [
        {
          name: 'event',
          type: 'group',
          entries: [
            { name: 'create', chatInputRun: 'eventCreate' },
            { name: 'start', chatInputRun: 'eventStart' },
            { name: 'end', chatInputRun: 'eventEnd' },
          ],
        },
        {
          name: 'group',
          type: 'group',
          entries: [
            { name: 'create', chatInputRun: 'groupCreate' },
            { name: 'add', chatInputRun: 'groupAdd' },
            { name: 'update', chatInputRun: 'groupUpdate' },
          ],
        },
      ],
      preconditions: ['ModOnly'],
    });
  }

  // Registers that this is a slash command
  public override registerApplicationCommands(registry: Subcommand.Registry) {
    registry.registerChatInputCommand(
      (builder) => builder
        .setName(this.name)
        .setDescription(this.description)
        .addSubcommandGroup((group) => group.setName('event')
          .setDescription('Commands to do with outreach events')
          .addSubcommand((command) => command.setName('create')
            .setDescription('Start an outreach event')
            .addBooleanOption((option) => option.setName('start')
              .setDescription('Start the event immediately')))
          .addSubcommand((command) => command.setName('start')
            .setDescription('Start an outreach event'))
          .addSubcommand((command) => command.setName('end')
            .setDescription('End an outreach event')))
        .addSubcommandGroup((group) => group.setName('group')
          .setDescription('Commands to do with groups')
          .addSubcommand((command) => command.setName('create')
            .setDescription('Create a group for people doing activism')
            .addUserOption((option) => option.setName('leader')
              .setDescription('This is the person leading the group')
              .setRequired(true)))
          .addSubcommand((command) => command.setName('add')
            .setDescription('Add a person to the group')
            .addUserOption((option) => option.setName('user')
              .setDescription('User to add to the group')
              .setRequired(true))
            .addRoleOption((option) => option.setName('group')
              .setDescription('Group to add the user to')))
          .addSubcommand((command) => command.setName('update')
            .setDescription('Update the statistics for the group')
            .addIntegerOption((option) => option.setName('vegan')
              .setDescription('How many said would go vegan?'))
            .addIntegerOption((option) => option.setName('considered')
              .setDescription('How many seriously considered being vegan?'))
            .addIntegerOption((option) => option.setName('anti-vegan')
              .setDescription('How many people had anti-vegan viewpoints?'))
            .addIntegerOption((option) => option.setName('thanked')
              .setDescription('How many thanked you for the conversation?'))
            .addIntegerOption((option) => option.setName('documentary')
              .setDescription('How many said they would watch a vegan documentary?'))
            .addIntegerOption((option) => option.setName('educated')
              .setDescription('How many got educated on veganism or the animal industry?')))),
      {
        behaviorWhenNotIdentical: RegisterBehavior.Overwrite,
      },
    );
  }

  public async eventCreate(interaction: Subcommand.ChatInputCommandInteraction) {
    // const start = interaction.options.getBoolean('start');
    const modInteraction = interaction.member;
    const { guild } = interaction;

    if (modInteraction === null || guild === null) {
      await interaction.reply({
        content: 'Mod or guild was not found!',
        ephemeral: true,
      });
      return;
    }

    const mod = guild.members.cache.get(modInteraction.user.id);

    if (mod === undefined) {
      await interaction.reply({
        content: 'Mod was not found!',
        ephemeral: true,
      });
      return;
    }

    if (await checkActiveEvent()) {
      await interaction.reply({
        content: 'There is already an active event!',
        ephemeral: true,
      });
      return;
    }

    await updateUser(mod);

    await createEvent(modInteraction.user.id);
  }

  public async groupCreate(interaction: Subcommand.ChatInputCommandInteraction) {
    const leader = interaction.options.getUser('leader', true);
    const { guild } = interaction;

    if (guild === null) {
      await interaction.reply({
        content: 'Guild not found!',
        ephemeral: true,
      });
      return;
    }

    await interaction.deferReply({ ephemeral: true });

    const event = await getCurrentEvent();

    if (event === null) {
      await interaction.editReply({
        content: 'There is no current event!',
      });
      return;
    }

    const statGroups = await getStatGroups(event.id);
    const groupNo = statGroups.length + 1;

    const role = await guild.roles.create({ name: `Outreach Group ${groupNo}` });

    await createStat(event.id, leader.id, role.id);

    const leaderMember = await guild.members.cache.get(leader.id);

    if (leaderMember === undefined) {
      await interaction.editReply({
        content: `Created a group with the leader being ${leader}, however could not give the role.`,
      });
      return;
    }

    await leaderMember.roles.add(role);

    await interaction.editReply({
      content: `Created a group with the leader being ${leader}`,
    });
  }

  public async groupAdd(interaction: Subcommand.ChatInputCommandInteraction) {
    const user = interaction.options.getUser('user', true);
    const group = interaction.options.getRole('group');
    const leader = interaction.user;
    const { guild } = interaction;

    if (guild === null) {
      await interaction.reply({
        content: 'Could not find guild!',
        ephemeral: true,
      });
      return;
    }

    await interaction.deferReply({ ephemeral: true });

    let statId: number;
    let roleId: Snowflake;

    // Find group from role
    if (group !== null) {
      const [stat] = await Promise.all([getStatFromRole(group.id)]);

      if (stat === null) {
        await interaction.editReply({
          content: `Could not find the group for role ${group}`,
        });
        return;
      }

      const leaderMember = guild.members.cache.get(leader.id);

      if (leaderMember === undefined) {
        await interaction.editReply({
          content: 'Could not find your GuildMember in cache!',
        });
        return;
      }

      if (leader.id !== stat.stat.leaderId
          && !leaderMember.roles.cache.has(IDs.roles.staff.outreachCoordinator)) {
        await interaction.editReply({
          content: `You are not the leader for ${group}`,
        });
        return;
      }

      statId = stat.statId;
      roleId = stat.roleId;
    } else {
      // Find group from person who ran the command
      const [stat] = await Promise.all([getStatFromLeader(leader.id)]);

      if (stat === null) {
        await interaction.editReply({
          content: 'You\'re not a group leader!',
        });
        return;
      }

      statId = stat.id;
      roleId = stat.role[0].roleId;
    }

    if (await userInStats(statId, user.id)) {
      await interaction.editReply({
        content: `${user} is already in this group!`,
      });
      return;
    }

    const member = guild.members.cache.get(user.id);

    if (member === undefined) {
      await interaction.editReply({
        content: 'Could not fetch the member!',
      });
      return;
    }

    await updateUser(member);

    await addStatUser(statId, user.id);

    await member.roles.add(roleId);

    await interaction.editReply({
      content: `Added ${user} to the group!`,
    });
  }

  public async groupUpdate(interaction: Subcommand.ChatInputCommandInteraction) {
    const vegan = interaction.options.getInteger('vegan');
    const considered = interaction.options.getInteger('considered');
    const antiVegan = interaction.options.getInteger('anti-vegan');
    const thanked = interaction.options.getInteger('thanked');
    const documentary = interaction.options.getInteger('documentary');
    const educated = interaction.options.getInteger('educated');
    const leader = interaction.user;

    const stats = {
      vegan: vegan !== null ? vegan : 0,
      considered: considered !== null ? considered : 0,
      antiVegan: antiVegan !== null ? antiVegan : 0,
      thanked: thanked !== null ? thanked : 0,
      documentary: documentary !== null ? documentary : 0,
      educated: educated !== null ? educated : 0,
    };

    if (leader === null) {
      await interaction.reply({
        content: 'Could not find your user!',
        ephemeral: true,
      });
      return;
    }

    await interaction.deferReply({ ephemeral: true });

    const stat = await getStatFromLeader(leader.id);

    if (stat === null) {
      await interaction.editReply({
        content: 'You\'re not the leader of a group!',
      });
      return;
    }

    await updateStats(stat.id, stats);

    await interaction.editReply({
      content: 'Updated the database with the stats!',
    });
  }
}