Allow for pump attachment to flamethrower

src/Dodge/Item/Ammo.hs

@@ -11,6 +11,7 @@ module Dodge.Item.Ammo (
     megaTinMag,
     megaBattery,
     capacitor,
+    compressor,
 ) where
 
 import Control.Lens
@@ -70,6 +71,13 @@ capacitor =
         & itUse .~ UseNothing
         & itConsumables ?~ 0
 
+compressor :: Item
+compressor =
+    defHeldItem
+        & itType .~ ATTACH COMPRESSOR
+        & itUse .~ UseNothing
+        & itConsumables ?~ 0
+
 chemFuelPouch :: Item
 chemFuelPouch =
     defHeldItem

src/Dodge/Item/Display.hs

@@ -103,6 +103,7 @@ showAttachItem = \case
     GIMBAL -> "GIMBAL"
     GYROSCOPE -> "GYROSCOPE"
     CAPACITOR -> "CAPACITOR"
+    COMPRESSOR -> "COMPRESSOR"
 
 showEquipItem :: EquipItemType -> String
 showEquipItem eit = case eit of

src/Dodge/Item/Grammar.hs

@@ -36,16 +36,19 @@ tryAttachItems = leftRightCombine leftIsParentCombine rightIsParentCombine
 
 itemAboveAttachables :: CItem -> [ItemSF]
 itemAboveAttachables (itm,sf) = case (itm ^. itType, sf) of
-    (_, HeldPlatformSF) ->
-         [WeaponTargetingSF, WeaponScopeSF,TorchSF]
-            <> getAutoSpringLinks itm
-            <> extraWeaponLinks itm
+    (HELD FLAMETHROWER,_) -> PumpSF:helddefaults
+    (_, HeldPlatformSF) -> helddefaults
     (DETECTOR {}, _) -> [ARHUDSF,TriggerSF,MapperSF]
     --(CRAFT TRANSFORMER, _) ->  [LaserWeaponSF]
     (LASER, _) ->  [TransformerSF]
     (MAPPER, _) -> [ARHUDSF]
     (_, GadgetPlatformSF) -> [TriggerSF, WeaponTargetingSF, WeaponScopeSF]
     _ -> []
+  where
+    helddefaults = 
+         [WeaponTargetingSF, WeaponScopeSF,TorchSF]
+            <> getAutoSpringLinks itm
+            <> extraWeaponLinks itm
 
 itemBelowAttachables :: CItem -> [ItemSF]
 itemBelowAttachables (itm,sf) = case (itm ^. itType, sf) of
@@ -138,6 +141,7 @@ itemToFunction itm = case itm ^. itType of
     CLICKER{} -> GadgetPlatformSF
     ATTACH CAPACITOR -> CapacitorSF
     CRAFT TRANSFORMER -> TransformerSF
+    CRAFT PUMP -> PumpSF
     _ -> NoSF
 
 treeToPotentialFunction :: DTree CItem -> S.Set ItemSF

src/Dodge/Item/InventoryColor.hs

@@ -42,7 +42,9 @@ sfInvColor = \case
     PulseLaserSF -> white
     CapacitorSF -> yellow
     PulseBallSF -> white
+    PlasmaSF -> white
     TransformerSF -> yellow
+    PumpSF -> yellow
     TorchSF{} -> green
 
 --ammoTypeColor :: AmmoType -> Color