Add file

src/Dodge/Data/AmmoType.hs

@@ -0,0 +1,20 @@
+{-# LANGUAGE StrictData #-}
+{-# LANGUAGE TemplateHaskell #-}
+
+module Dodge.Data.AmmoType where
+
+--import Data.Aeson
+--import Data.Aeson.TH
+
+data AmmoType
+    = LauncherAmmo
+    | BulletAmmo
+    | BeltBulletAmmo
+    | ElectricalAmmo
+    | DroneAmmo
+    | GasAmmo
+    | PrintMaterial
+    | ExplosivePutty
+    deriving (Eq, Ord, Show, Read) --Generic, Flat)
+
+--deriveJSON defaultOptions ''AmmoType

src/Dodge/Data/EquipType.hs

@@ -0,0 +1,30 @@
+{-# LANGUAGE LambdaCase #-}
+
+module Dodge.Data.EquipType (equipType) where
+
+import Control.Lens
+import Dodge.Data.Item
+
+equipType :: Item -> Maybe EquipType
+equipType itm = case itm ^. itType of
+    EQUIP eit -> Just $ eitType eit
+    _ -> Nothing
+
+eitType :: EquipItemType -> EquipType
+eitType = \case
+    MAGSHIELD -> GoesOnWrist
+    FLAMESHIELD -> GoesOnWrist
+    FRONTARMOUR -> GoesOnChest
+    WRISTARMOUR -> GoesOnWrist
+    INVISIBILITYEQUIPMENT x -> x
+    BRAINHAT -> GoesOnHead
+    HAT -> GoesOnHead
+    HEADLAMP -> GoesOnHead
+    POWERLEGS -> GoesOnLegs
+    SPEEDLEGS -> GoesOnLegs
+    JUMPLEGS -> GoesOnLegs
+    FUELPACK -> GoesOnBack
+    BULLETBELTPACK -> GoesOnBack
+    BULLETBELTBRACER -> GoesOnWrist
+    BATTERYPACK -> GoesOnBack
+    WRIST_ECG -> GoesOnWrist