feat(arabot): add rank command

src/commands/xp/rank.ts

@@ -0,0 +1,124 @@
+// 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 type { User, Guild, Message } from 'discord.js';
+import { EmbedBuilder } from 'discord.js';
+import { getRank } from '#utils/database/xp';
+
+export class RankCommand extends Command {
+  public constructor(context: Command.Context, options: Command.Options) {
+    super(context, {
+      ...options,
+      name: 'rank',
+      description: 'Gets your current rank on this server',
+    });
+  }
+
+  // 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.rank(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.rank(user, guild);
+
+    await message.reply({
+      content: info.message,
+      embeds: info.embeds,
+    });
+    if (!info.success) {
+      await message.react('❌');
+    }
+  }
+
+  private async rank(user: User, guild: Guild) {
+    const info = {
+      message: '',
+      embeds: [] as EmbedBuilder[],
+      success: false,
+    };
+
+    const member = guild.members.cache.get(user.id);
+
+    if (member === undefined) {
+      info.message = 'Could not find your guild member!';
+      return info;
+    }
+
+    const rank = await getRank(user.id);
+
+    const embed = new EmbedBuilder()
+      .setColor('#00ff7d')
+      .setAuthor({ name: `${member.displayName}'s Rank`, iconURL: `${user.displayAvatarURL()}` })
+      .addFields(
+        { name: 'Rank', value: `${rank.rank}` },
+        { name: 'Level', value: `${rank.level}`, inline: true },
+        { name: 'XP', value: `${rank.xp}`, inline: true },
+      );
+
+    info.success = true;
+    info.embeds.push(embed);
+    return info;
+  }
+}

src/utils/database/xp.ts

@@ -6,7 +6,7 @@ function xpToNextLevel(level: number, xp: number) {
   return 5 * (level * level) + (50 * level) + 100 - xp;
 }
 
-export async function addXp(userId: Snowflake, xp: number) {
+export async function getUser(userId: Snowflake) {
   const user = await container.database.xp.findUnique({
     where: {
       userId,
@@ -16,6 +16,11 @@ export async function addXp(userId: Snowflake, xp: number) {
       level: true,
     },
   });
+  return user;
+}
+
+export async function addXp(userId: Snowflake, xp: number) {
+  const user = await getUser(userId);
 
   let level = 0;
   if (user !== null
@@ -68,3 +73,47 @@ export async function checkCanAddXp(userId: Snowflake) {
 
   return Date.now() - message.lastMessage.getTime() > cooldown;
 }
+
+export async function getRank(userId: Snowflake) {
+  const user = await getUser(userId);
+
+  const info = {
+    rank: 0,
+    level: 0,
+    xp: 0,
+  };
+
+  if (user === null) {
+    return info;
+  }
+
+  info.rank = await container.database.xp.count({
+    where: {
+      xp: {
+        gte: user.xp,
+      },
+    },
+    orderBy: {
+      xp: 'desc',
+    },
+  });
+
+  const xp = await container.database.xp.findUnique({
+    where: {
+      userId,
+    },
+    select: {
+      level: true,
+      xp: true,
+    },
+  });
+
+  if (xp === null) {
+    return info;
+  }
+
+  info.level = xp.level;
+  info.xp = xp.xp;
+
+  return info;
+}