feat(arabot): add daily command

src/commands/economy/daily.ts

@@ -0,0 +1,140 @@
+// 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, Message } from 'discord.js';
+import { updateUser } from '#utils/database/dbExistingUser';
+import { daily, getLastDaily } from '#utils/database/economy';
+import { EmbedBuilder } from 'discord.js';
+
+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,
+    });
+    await message.react(info.success ? '✅' : '❌');
+  }
+
+  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;
+    }
+
+    await updateUser(member);
+
+    const [db] = await Promise.all([daily(user.id, amount)]);
+
+    const balance = db.Balance?.balance;
+
+    const embed = new EmbedBuilder()
+      .setColor('#00ff7d')
+      .setAuthor({ name: 'Daily Reward', iconURL: `${user.displayAvatarURL()}` })
+      .addFields(
+        { name: 'Collected', value: `${amount} ARA` },
+      );
+
+    if (balance !== undefined) {
+      embed.setFooter({ text: `New Balance: ${balance}` });
+    }
+
+    info.success = true;
+    info.embeds.push(embed);
+    return info;
+  }
+}

src/utils/database/economy.ts

@@ -0,0 +1,48 @@
+import { container } from '@sapphire/framework';
+import type { Snowflake } from 'discord.js';
+
+export async function daily(userId: Snowflake, amount: number) {
+  const balance = await container.database.user.update({
+    where: {
+      id: userId,
+    },
+    data: {
+      Balance: {
+        upsert: {
+          update: {
+            balance: { increment: amount },
+          },
+          create: {
+            balance: amount,
+          },
+        },
+      },
+      Daily: {
+        create: {
+          amount,
+        },
+      },
+    },
+    include: {
+      Balance: true,
+    },
+  });
+
+  return balance;
+}
+
+export async function getLastDaily(userId: Snowflake) {
+  const time = await container.database.daily.findFirst({
+    where: {
+      userId,
+    },
+    orderBy: {
+      id: 'desc',
+    },
+    select: {
+      time: true,
+    },
+  });
+
+  return time;
+}