diff --git a/database/0.1.5/1694609771821-CreateClaim/Up/01-CreateClaim.sql b/database/0.1.5/1694609771821-CreateClaim/Up/01-CreateClaim.sql new file mode 100644 index 0000000..bfd8057 --- /dev/null +++ b/database/0.1.5/1694609771821-CreateClaim/Up/01-CreateClaim.sql @@ -0,0 +1,8 @@ +CREATE TABLE `claim` ( + `Id` varchar(255) NOT NULL, + `WhenCreated` datetime NOT NULL, + `WhenUpdated` datetime NOT NULL, + `ClaimId` varchar(255) NOT NULL, + `InventoryId` varchar(255) NOT NULL, + PRIMARY KEY (`Id`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci; \ No newline at end of file diff --git a/database/0.1.5/1694609771821-CreateClaim/Up/02-MoveToClaim.sql b/database/0.1.5/1694609771821-CreateClaim/Up/02-MoveToClaim.sql new file mode 100644 index 0000000..b73eb32 --- /dev/null +++ b/database/0.1.5/1694609771821-CreateClaim/Up/02-MoveToClaim.sql @@ -0,0 +1,10 @@ +INSERT INTO claim ( + Id, + ClaimId, + InventoryId +) +SELECT + UUID(), + ClaimId, + Id +FROM inventory; \ No newline at end of file diff --git a/database/0.1.5/1694609771821-CreateClaim/Up/03-AlterInventory.sql b/database/0.1.5/1694609771821-CreateClaim/Up/03-AlterInventory.sql new file mode 100644 index 0000000..7005ca3 --- /dev/null +++ b/database/0.1.5/1694609771821-CreateClaim/Up/03-AlterInventory.sql @@ -0,0 +1,2 @@ +ALTER TABLE inventory +DROP ClaimId; \ No newline at end of file diff --git a/src/buttonEvents/Claim.ts b/src/buttonEvents/Claim.ts index 5412db8..6f9ae58 100644 --- a/src/buttonEvents/Claim.ts +++ b/src/buttonEvents/Claim.ts @@ -2,6 +2,7 @@ import { ButtonInteraction } from "discord.js"; import { ButtonEvent } from "../type/buttonEvent"; import Inventory from "../database/entities/app/Inventory"; import { CoreClient } from "../client/client"; +import { default as eClaim } from "../database/entities/app/Claim"; export default class Claim extends ButtonEvent { public override async execute(interaction: ButtonInteraction) { @@ -9,30 +10,36 @@ 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 Inventory.FetchOneByClaimId(claimId); + const claimed = await eClaim.FetchOneByClaimId(claimId); if (claimed) { await interaction.reply('This card has already been claimed'); 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; } let inventory = await Inventory.FetchOneByCardNumberAndUserId(userId, cardNumber); if (!inventory) { - inventory = new Inventory(userId, cardNumber, 1, claimId); + inventory = new Inventory(userId, cardNumber, 1); } else { inventory.SetQuantity(inventory.Quantity + 1); } await inventory.Save(Inventory, inventory); + const claim = new eClaim(claimId); + claim.SetInventory(inventory); + + await claim.Save(eClaim, claim); + await interaction.reply('Card claimed'); } } \ No newline at end of file diff --git a/src/buttonEvents/Reroll.ts b/src/buttonEvents/Reroll.ts index c1a54ea..bc91c90 100644 --- a/src/buttonEvents/Reroll.ts +++ b/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() diff --git a/src/commands/drop.ts b/src/commands/drop.ts index 5770903..7503bf7 100644 --- a/src/commands/drop.ts +++ b/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; } } \ No newline at end of file diff --git a/src/database/entities/app/Claim.ts b/src/database/entities/app/Claim.ts new file mode 100644 index 0000000..b4cde5e --- /dev/null +++ b/src/database/entities/app/Claim.ts @@ -0,0 +1,31 @@ +import { Column, Entity, ManyToOne } from "typeorm"; +import AppBaseEntity from "../../../contracts/AppBaseEntity"; +import Inventory from "./Inventory"; +import AppDataSource from "../../dataSources/appDataSource"; + +@Entity() +export default class Claim extends AppBaseEntity { + constructor(claimId: string) { + super(); + + this.ClaimId = claimId; + } + + @Column() + ClaimId: string; + + @ManyToOne(() => Inventory, x => x.Claims) + Inventory: Inventory; + + public SetInventory(inventory: Inventory) { + this.Inventory = inventory; + } + + public static async FetchOneByClaimId(claimId: string): Promise { + const repository = AppDataSource.getRepository(Claim); + + const single = await repository.findOne({ where: { ClaimId: claimId }}); + + return single; + } +} \ No newline at end of file diff --git a/src/database/entities/app/Inventory.ts b/src/database/entities/app/Inventory.ts index c6319a9..adec824 100644 --- a/src/database/entities/app/Inventory.ts +++ b/src/database/entities/app/Inventory.ts @@ -1,16 +1,16 @@ -import { Column, Entity } from "typeorm"; +import { Column, Entity, OneToMany } from "typeorm"; import AppBaseEntity from "../../../contracts/AppBaseEntity"; import AppDataSource from "../../dataSources/appDataSource"; +import Claim from "./Claim"; @Entity() export default class Inventory extends AppBaseEntity { - constructor(userId: string, cardNumber: string, quantity: number, claimId: string) { + constructor(userId: string, cardNumber: string, quantity: number) { super(); this.UserId = userId; this.CardNumber = cardNumber; this.Quantity = quantity; - this.ClaimId = claimId; } @Column() @@ -22,13 +22,17 @@ export default class Inventory extends AppBaseEntity { @Column() Quantity: number; - @Column() - ClaimId: string; + @OneToMany(() => Claim, x => x.Inventory) + Claims: Claim[]; public SetQuantity(quantity: number) { this.Quantity = quantity; } + public AddClaim(claim: Claim) { + this.Claims.push(claim); + } + public static async FetchOneByCardNumberAndUserId(userId: string, cardNumber: string): Promise { const repository = AppDataSource.getRepository(Inventory); @@ -36,12 +40,4 @@ export default class Inventory extends AppBaseEntity { return single; } - - public static async FetchOneByClaimId(claimId: string): Promise { - const repository = AppDataSource.getRepository(Inventory); - - const single = await repository.findOne({ where: { ClaimId: claimId }}); - - return single; - } } \ No newline at end of file diff --git a/src/database/migrations/app/0.1.5/1694609771821-CreateClaim.ts b/src/database/migrations/app/0.1.5/1694609771821-CreateClaim.ts new file mode 100644 index 0000000..22fe74c --- /dev/null +++ b/src/database/migrations/app/0.1.5/1694609771821-CreateClaim.ts @@ -0,0 +1,17 @@ +import { MigrationInterface, QueryRunner } from "typeorm" +import MigrationHelper from "../../../../helpers/MigrationHelper" + +export class CreateClaim1694609771821 implements MigrationInterface { + + public async up(queryRunner: QueryRunner): Promise { + MigrationHelper.Up('1694609771821-CreateClaim', '0.1.5', [ + '01-CreateClaim', + '02-MoveToClaim', + '03-AlterInventory', + ], queryRunner); + } + + public async down(queryRunner: QueryRunner): Promise { + } + +}