From 3b9706ae944bebfdd447aa267c190558d85d6359 Mon Sep 17 00:00:00 2001
From: Ethan Lane <ethan@vylpes.com>
Date: Wed, 10 Jul 2024 17:00:51 +0100
Subject: [PATCH] Update the ImageHelper to check an entity's count

---
 src/database/entities/app/Inventory.ts | 8 ++++++++
 src/helpers/ImageHelper.ts             | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/database/entities/app/Inventory.ts b/src/database/entities/app/Inventory.ts
index 7d659fe..a0aee18 100644
--- a/src/database/entities/app/Inventory.ts
+++ b/src/database/entities/app/Inventory.ts
@@ -47,6 +47,14 @@ export default class Inventory extends AppBaseEntity {
         return single;
     }
 
+    public static async ExistsByCardNumberAndUserId(userId: string, cardNumber: string): Promise<boolean> {
+        const repository = AppDataSource.getRepository(Inventory);
+        
+        const count = await repository.count({ where: { UserId: userId, CardNumber: cardNumber }});
+
+        return count != 0;
+    }
+
     public static async FetchAllByUserId(userId: string): Promise<Inventory[]> {
         const repository = AppDataSource.getRepository(Inventory);
 
diff --git a/src/helpers/ImageHelper.ts b/src/helpers/ImageHelper.ts
index 3fab77b..3a04b92 100644
--- a/src/helpers/ImageHelper.ts
+++ b/src/helpers/ImageHelper.ts
@@ -39,7 +39,7 @@ export default class ImageHelper {
             const imageData = await Jimp.read(filePath);
 
             if (userId != null) {
-                const claimed = await Inventory.FetchOneByCardNumberAndUserId(userId, card.id) != null;
+                const claimed = await Inventory.ExistsByCardNumberAndUserId(userId, card.id);
 
                 if (!claimed) {
                     imageData.greyscale();