Prevent users from being able to claim the latest card unless you were the one to drop it

src/buttonEvents/Claim.ts

@@ -10,6 +10,7 @@ export default class Claim extends ButtonEvent {
 
         const cardNumber = interaction.customId.split(' ')[1];
         const claimId = interaction.customId.split(' ')[2];
+        const droppedBy = interaction.customId.split(' ')[3];
         const userId = interaction.user.id;
 
         const claimed = await eClaim.FetchOneByClaimId(claimId);
@@ -19,8 +20,8 @@ export default class Claim extends ButtonEvent {
             return;
         }
 
-        if (claimId != CoreClient.ClaimId) {
-            await interaction.reply('This card has expired');
+        if (claimId == CoreClient.ClaimId && userId != droppedBy) {
+            await interaction.reply('The latest dropped card can only be claimed by the user who dropped it');
             return;
         }
 
@@ -35,16 +36,10 @@ export default class Claim extends ButtonEvent {
         await inventory.Save(Inventory, inventory);
 
         const claim = new eClaim(claimId);
+        claim.SetInventory(inventory);
+
         await claim.Save(eClaim, claim);
 
-        inventory = await Inventory.FetchOneById(Inventory, inventory.Id, [ "Claims" ]);
-
-        if (inventory) {
-            inventory.AddClaim(claim);
-
-            await inventory.Save(Inventory, inventory);
-        }
-
         await interaction.reply('Card claimed');
     }
 }

src/buttonEvents/Reroll.ts

@@ -45,7 +45,7 @@ export default class Reroll extends ButtonEvent {
 
         row.addComponents(
             new ButtonBuilder()
-                .setCustomId(`claim ${randomCard.CardNumber} ${claimId}`)
+                .setCustomId(`claim ${randomCard.CardNumber} ${claimId} ${interaction.user.id}`)
                 .setLabel("Claim")
                 .setStyle(ButtonStyle.Primary),
             new ButtonBuilder()

src/commands/drop.ts

@@ -51,7 +51,7 @@ export default class Drop extends Command {
 
         row.addComponents(
             new ButtonBuilder()
-                .setCustomId(`claim ${randomCard.CardNumber} ${claimId}`)
+                .setCustomId(`claim ${randomCard.CardNumber} ${claimId} ${interaction.user.id}`)
                 .setLabel("Claim")
                 .setStyle(ButtonStyle.Primary),
             new ButtonBuilder()
@@ -76,7 +76,6 @@ export default class Drop extends Command {
             }
         }
 
-
         CoreClient.ClaimId = claimId;
     }
 }

src/database/entities/app/Claim.ts

@@ -17,6 +17,10 @@ export default class Claim extends AppBaseEntity {
     @ManyToOne(() => Inventory, x => x.Claims)
     Inventory: Inventory;
 
+    public SetInventory(inventory: Inventory) {
+        this.Inventory = inventory;
+    }
+
     public static async FetchOneByClaimId(claimId: string): Promise<Claim | null> {
         const repository = AppDataSource.getRepository(Claim);