fix: add cooldowns to delete & edit message commands

Discord-InterChat · Feb 22, 2024 · 8476388 · 8476388
1 parent afd9ecd
commit 8476388
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 6 deletions.
diff --git a/src/commands/context-menu/blacklist.ts b/src/commands/context-menu/blacklist.ts
@@ -194,7 +194,7 @@ export default class Blacklist extends BaseCommand {
       successEmbed.setDescription(
         t(
           { phrase: 'blacklist.user.success', locale: interaction.user.locale },
-          { username: user?.username ?? 'Unknown User', emoji: emojis.tick },
+          { user: user?.username ?? 'Unknown User', emoji: emojis.tick },
         ),
       );
       await blacklistManager.addUserBlacklist(

diff --git a/src/commands/context-menu/deleteMsg.ts b/src/commands/context-menu/deleteMsg.ts
@@ -1,7 +1,6 @@
 import {
   ApplicationCommandType,
-  CacheType,
-  ContextMenuCommandInteraction,
+  MessageContextMenuCommandInteraction,
   RESTPostAPIApplicationCommandsJSONBody,
 } from 'discord.js';
 import BaseCommand from '../BaseCommand.js';
@@ -17,7 +16,12 @@ export default class DeleteMessage extends BaseCommand {
     dm_permission: false,
   };
 
-  async execute(interaction: ContextMenuCommandInteraction<CacheType>) {
+  readonly cooldown = 10_000;
+
+  async execute(interaction: MessageContextMenuCommandInteraction) {
+    const isOnCooldown = await this.handleCooldown(interaction);
+    if (isOnCooldown) return;
+
     await interaction.deferReply({ ephemeral: true });
 
     const messageInDb = await db?.broadcastedMessages.findFirst({

diff --git a/src/commands/context-menu/editMsg.ts b/src/commands/context-menu/editMsg.ts
@@ -14,7 +14,7 @@ import {
 import db from '../../utils/Db.js';
 import BaseCommand from '../BaseCommand.js';
 import { HubSettingsBitField } from '../../utils/BitFields.js';
-import { checkIfStaff, hasVoted, replaceLinks } from '../../utils/Utils.js';
+import { checkIfStaff, replaceLinks, userVotedToday } from '../../utils/Utils.js';
 import { censor } from '../../utils/Profanity.js';
 import { RegisterInteractionHandler } from '../../decorators/Interaction.js';
 import { CustomID } from '../../utils/CustomID.js';
@@ -27,11 +27,15 @@ export default class EditMessage extends BaseCommand {
     name: 'Edit Message',
     dm_permission: false,
   };
+  readonly cooldown = 10_000;
 
   async execute(interaction: MessageContextMenuCommandInteraction) {
+    const isOnCooldown = await this.handleCooldown(interaction);
+    if (isOnCooldown) return;
+
     const target = interaction.targetMessage;
 
-    if (!checkIfStaff(interaction.user.id) && !(await hasVoted(interaction.user.id))) {
+    if (!checkIfStaff(interaction.user.id) && !(await userVotedToday(interaction.user.id))) {
       await interaction.reply({
         content: t(
           { phrase: 'errors.mustVote', locale: interaction.user.locale },

diff --git a/src/utils/Utils.ts b/src/utils/Utils.ts
@@ -84,6 +84,15 @@ export const hasVoted = async (userId: Snowflake): Promise<boolean> => {
   return !!res.voted;
 };
 
+export const userVotedToday = async (userId: Snowflake): Promise<boolean> => {
+  const res = await db.userData.findFirst({
+    where: { userId },
+  });
+
+  if (res?.lastVoted && res.lastVoted > Date.now() - 60 * 60 * 24 * 1000) return true;
+  return false;
+};
+
 export const yesOrNoEmoji = (option: unknown, yesEmoji: string, noEmoji: string) => {
   return option ? yesEmoji : noEmoji;
 };