arabot/src/commands/economy/daily.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 { Command, RegisterBehavior } from '@sapphire/framework';
import { Time } from '@sapphire/time-utilities';
import type {
  User,
  Guild,
  GuildMember,
  Message,
} from 'discord.js';
import { updateUser } from '#utils/database/dbExistingUser';
import { daily, getLastDaily } from '#utils/database/economy';
import { EmbedBuilder } from 'discord.js';
import IDs from '#utils/ids';

export class DailyCommand extends Command {
  public constructor(context: Command.Context, options: Command.Options) {
    super(context, {
      ...options,
      name: 'daily',
      description: 'Get given an amount of money once a day',
    });
  }

  // Registers that this is a slash command
  public override registerApplicationCommands(registry: Command.Registry) {
    registry.registerChatInputCommand(
      (builder) => builder
        .setName(this.name)
        .setDescription(this.description),
      {
        behaviorWhenNotIdentical: RegisterBehavior.Overwrite,
      },
    );
  }

  // Command run
  public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
    const { user, guild } = interaction;

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

    await interaction.deferReply();

    const info = await this.runDaily(user, guild);

    await interaction.editReply({
      content: info.message,
      embeds: info.embeds,
    });
  }

  public async messageRun(message: Message) {
    const user = message.member?.user;
    const { guild } = message;

    if (user === undefined) {
      await message.react('❌');
      await message.reply('Could not find your user!');
      return;
    }

    if (guild === null) {
      await message.react('❌');
      await message.reply('Could not find the guild!');
      return;
    }

    const info = await this.runDaily(user, guild);

    await message.reply({
      content: info.message,
      embeds: info.embeds,
    });

    if (!info.success) {
      await message.react('❌');
    }
  }

  private async runDaily(user: User, guild: Guild) {
    const amount = 10;
    const time = Time.Hour * 18;
    const info = {
      message: '',
      embeds: [] as EmbedBuilder[],
      success: false,
    };

    const lastDaily = await getLastDaily(user.id);

    if (lastDaily !== null
      && (new Date().getTime() - lastDaily.time.getTime()) < time) {
      info.message = 'You have already claimed your daily, come back later to claim it!';
      return info;
    }

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

    if (member === undefined) {
      info.message = 'Could not find your guild member!';
      return info;
    }

    // Give bonus for the user
    const bonus = await this.giveBonus(member);

    await updateUser(member);

    const [db] = await Promise.all([daily(user.id, amount + bonus)]);

    const balance = db.Balance?.balance;

    const embed = new EmbedBuilder()
      .setColor('#00ff7d')
      .setAuthor({ name: 'Daily Reward', iconURL: `${user.displayAvatarURL()}` })
      .addFields({ name: 'Collected', value: `${amount} ARA`, inline: bonus > 0 });

    if (bonus > 0) {
      embed.addFields(
        { name: 'Bonus', value: `${bonus} ARA`, inline: true },
        { name: '\u200B', value: 'Thank you for contributing to ARA! :D' },
      );
    }

    if (balance !== undefined) {
      embed.setFooter({ text: `New Balance: ${balance}` });
    }

    info.success = true;
    info.embeds.push(embed);
    return info;
  }

  private async giveBonus(member: GuildMember) {
    let bonus = 0;

    const amount = [
      { role: member.roles.premiumSubscriberRole?.id, amount: 5 },
      { role: IDs.roles.staff.coordinator, amount: 2 },
      { role: IDs.roles.staff.moderator, amount: 2 },
      { role: IDs.roles.staff.trialModerator, amount: 2 },
      { role: IDs.roles.staff.restricted, amount: 1 },
      { role: IDs.roles.staff.verifier, amount: 2 },
      { role: IDs.roles.staff.trialVerifier, amount: 2 },
      { role: IDs.roles.staff.developer, amount: 2 },
      { role: IDs.roles.staff.mentor, amount: 2 },
      { role: IDs.roles.stageHost, amount: 1 },
    ];

    member.roles.cache.forEach((role) => {
      amount.forEach((check) => {
        if (role.id === check.role) {
          bonus += check.amount;
        }
      });
    });

    return bonus;
  }
}