From 3681df871f21dd68191153751256515728360753 Mon Sep 17 00:00:00 2001
From: Ethan Lane <ethan@vylpes.com>
Date: Sun, 6 Oct 2024 09:27:46 +0100
Subject: [PATCH] Create keep button handler

---
 src/buttonEvents/Multidrop.ts  | 42 +++++++++++++++++++++++++++++++---
 src/constants/ErrorMessages.ts |  1 +
 2 files changed, 40 insertions(+), 3 deletions(-)

diff --git a/src/buttonEvents/Multidrop.ts b/src/buttonEvents/Multidrop.ts
index 88bd240..dce691a 100644
--- a/src/buttonEvents/Multidrop.ts
+++ b/src/buttonEvents/Multidrop.ts
@@ -1,12 +1,13 @@
-import { ButtonInteraction, EmbedBuilder } from "discord.js";
+import { AttachmentBuilder, ButtonInteraction, EmbedBuilder } from "discord.js";
 import { ButtonEvent } from "../type/buttonEvent";
 import AppLogger from "../client/appLogger";
-import { CoreClient } from "../client/client";
 import CardDropHelperMetadata from "../helpers/CardDropHelperMetadata";
 import Inventory from "../database/entities/app/Inventory";
 import EmbedColours from "../constants/EmbedColours";
 import { readFileSync } from "fs";
 import path from "path";
+import ErrorMessages from "../constants/ErrorMessages";
+import User from "../database/entities/app/User";
 
 export default class Multidrop extends ButtonEvent {
     public override async execute(interaction: ButtonInteraction) {
@@ -45,6 +46,14 @@ export default class Multidrop extends ButtonEvent {
             return;
         }
 
+        const user = await User.FetchOneById(User, interaction.user.id);
+
+        if (!user) {
+            AppLogger.LogWarn("Button/Multidrop/Keep", ErrorMessages.UnableToFetchUser);
+            await interaction.reply(ErrorMessages.UnableToFetchUser);
+            return;
+        }
+
         // Claim
         let inventory = await Inventory.FetchOneByCardNumberAndUserId(interaction.user.id, cardNumber);
 
@@ -67,12 +76,39 @@ export default class Multidrop extends ButtonEvent {
         }
 
         // Drop next card
+        const randomCard = CardDropHelperMetadata.GetRandomCard();
         cardsRemaining -= 1;
 
+        if (!randomCard) {
+            AppLogger.LogWarn("Button/Multidrop/Keep", ErrorMessages.UnableToFetchCard);
+            await interaction.reply(ErrorMessages.UnableToFetchCard);
+            return;
+        }
+
         await interaction.deferUpdate();
 
         try {
-            const image = readFileSync(path.join(process.env.DATA_DIR!, "cards", ))
+            const image = readFileSync(path.join(process.env.DATA_DIR!, "cards", ));
+            const imageFileName = randomCard.card.path.split("/").pop()!;
+
+            const attachment = new AttachmentBuilder(image, { name: imageFileName });
+
+            const inventory = await Inventory.FetchOneByCardNumberAndUserId(interaction.user.id, randomCard.card.id);
+            const quantityClaimed = inventory ? inventory.Quantity : 0;
+
+            const embed = CardDropHelperMetadata.GenerateMultidropEmbed(randomCard, quantityClaimed, imageFileName, cardsRemaining, undefined, user.Currency);
+
+            const row = CardDropHelperMetadata.GenerateMultidropButtons(randomCard, cardsRemaining, interaction.user.id, cardsRemaining < 0);
+
+            await interaction.update({
+                embeds: [ embed ],
+                files: [ attachment ],
+                components: [ row ],
+            });
+        } catch (e) {
+            AppLogger.LogError("Button/Multidrop/Keep", `Error sending next drop for card ${randomCard.card.id}: ${e}`);
+
+            await interaction.reply(`Unable to send next drop. Please try again, and report this if it keeps happening. (${randomCard.card.id})`);
         }
     }
 }
\ No newline at end of file
diff --git a/src/constants/ErrorMessages.ts b/src/constants/ErrorMessages.ts
index a7f4ee6..c8749ec 100644
--- a/src/constants/ErrorMessages.ts
+++ b/src/constants/ErrorMessages.ts
@@ -2,6 +2,7 @@ export default class ErrorMessages {
     public static readonly BotSyncing = "Bot is currently syncing, please wait until its done.";
     public static readonly SafeMode = "Safe Mode has been activated, please resync to continue.";
     public static readonly UnableToFetchCard = "Unable to fetch card, please try again.";
+    public static readonly UnableToFetchUser = "Unable to fetch user, please try again.";
 
     public static readonly NotEnoughCurrency = (need: number, have: number) => `Not enough currency! You need ${need} currency, you have ${have}!`;
 }
\ No newline at end of file