From 6406f6d763777feefdb4101a11430711dd3a6968 Mon Sep 17 00:00:00 2001
From: Sainan <sainan@calamity.gg>
Date: Fri, 7 Jun 2024 16:43:06 +0200
Subject: [PATCH] improve: type safety for WeaponTypeInternal

---
 src/services/itemDataService.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/services/itemDataService.ts b/src/services/itemDataService.ts
index 6fb3cbac..207be787 100644
--- a/src/services/itemDataService.ts
+++ b/src/services/itemDataService.ts
@@ -18,7 +18,7 @@ export const warframes: MinWarframe[] = Array.from(new Items({ category: ["Warfr
         return next;
     });
 
-export type WeaponTypeInternal = "LongGuns" | "Pistols" | "Melee";
+export type WeaponTypeInternal = "LongGuns" | "Pistols" | "Melee" | "SpaceMelee" | "SpaceGuns" | "SentinelWeapons" | "OperatorAmps" | "SpecialItems";
 
 export const items: MinItem[] = Array.from(new Items({ category: ["All"] }) as MinimalItem[]).map(item => {
     const next = { ...item };
@@ -38,7 +38,7 @@ export const getWeaponType = (weaponName: string): WeaponTypeInternal => {
         throw new Error(`${weaponName} doesn't quack like a weapon`);
     }
 
-    const weaponType = weaponInfo.productCategory as WeaponTypeInternal;
+    const weaponType = weaponInfo.productCategory;
 
     if (!weaponType) {
         logger.error(`unknown weapon category for item ${weaponName}`);