Allow adding firearms in grid

8kto · Apr 28, 2024 · 52d51fc · 52d51fc
1 parent c4cfe23
commit 52d51fc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/components/EquipmentList/helpers.ts b/src/components/EquipmentList/helpers.ts
@@ -4,6 +4,7 @@ import { trivialSort } from '@/components/DataGrid/helpers'
 import type { SortConfig } from '@/components/DataGrid/types'
 import {
   isArmorItem,
+  isFirearmEquipmentItem,
   isMeleeWeaponItem,
   isMiscEquipmentItem,
   isMissileItem,
@@ -26,6 +27,7 @@ import CurrencyConverter from '@/shared/services/CurrencyConverter'
 import {
   addArmor,
   addEquipmentItem,
+  addFirearmWeapon,
   addMeleeWeapon,
   addMissileWeapon,
   InventoryState,
@@ -145,6 +147,8 @@ export const handleAddEquipmentItemClick = <T extends EquipmentItem>(
       addMissileWeapon(clone)
     } else if (isMiscEquipmentItem(clone)) {
       addEquipmentItem(clone)
+    } else if (isFirearmEquipmentItem(clone)) {
+      addFirearmWeapon(clone)
     } else {
       throw new Error('Unknown item')
     }

diff --git a/src/components/Inventory/ItemDetails/helpers.ts b/src/components/Inventory/ItemDetails/helpers.ts
@@ -2,6 +2,7 @@ import type { ArmorItem } from '@/domain/armor'
 import type { EquipmentItem } from '@/domain/equipment'
 import type { InventoryItem } from '@/domain/inventory'
 import type {
+  FirearmWeaponItem,
   MeleeWeaponItem,
   MissileWeaponItem,
   WeaponItem,
@@ -40,3 +41,9 @@ export const isMiscEquipmentItem = (
 ): item is EquipmentItem => {
   return item.categoryKey === 'miscEquipment'
 }
+
+export const isFirearmEquipmentItem = (
+  item: EquipmentItem,
+): item is FirearmWeaponItem => {
+  return item.categoryKey === 'firearmWeapons'
+}