Fix equipping bug, select sensible equipment position

src/Dodge/Inventory.hs

@@ -40,9 +40,9 @@ import LensHelp
 
 --import qualified Data.IntSet as IS
 import qualified Data.Map.Strict as M
-import qualified Data.Set as S
+--import qualified Data.Set as S
 import Data.Maybe
-import SDL
+--import SDL
 --import Data.List
 --import System.Random
 
@@ -175,10 +175,11 @@ updateCloseObjects w = w & closeObjects .~ unionBy closeObjEq oldCloseFiltered c
 
 updateRBList :: World -> World
 updateRBList w 
-    | ButtonRight `S.member` _mouseButtons w && w ^? rbOptions . opCurInvPos == Just curinvid 
+    | w ^? rbOptions . opCurInvPos == Just curinvid 
         = w & setEquipAllocation & setEquipActivation
-    | ButtonRight `S.member` _mouseButtons w = case cr ^? crInv . ix (_crInvSel cr) . itUse . eqSite of
-        Just esite -> w & rbOptions .~ EquipOptions (equipSiteToPositions esite) 0 curinvid DoNotMoveEquipment NoChangeActivateEquipment
+    | otherwise = case cr ^? crInv . ix (_crInvSel cr) . itUse . eqSite of
+        Just esite -> w 
+            & rbOptions .~ EquipOptions (equipSiteToPositions esite) (chooseEquipmentPosition cr (equipSiteToPositions esite)) curinvid DoNotMoveEquipment NoChangeActivateEquipment
             & setEquipAllocation & setEquipActivation
         Nothing -> w & rbOptions .~ NoRightButtonOptions
     | otherwise = w & rbOptions .~ NoRightButtonOptions
@@ -186,6 +187,16 @@ updateRBList w
     curinvid = _crInvSel cr
     cr = you w
 
+chooseEquipmentPosition :: Creature -> [EquipPosition] -> Int
+chooseEquipmentPosition cr eps = fromMaybe (chooseFreeSite cr eps) $ do
+    ep <- cr ^? crInvEquipped . ix (_crInvSel cr)
+    findIndex (==ep) eps
+
+chooseFreeSite :: Creature -> [EquipPosition] -> Int
+chooseFreeSite cr = fromMaybe 0 . findIndex hasnoequipment
+  where
+    hasnoequipment ep = isNothing $ cr ^? crEquipment . ix ep
+
 setEquipAllocation :: World -> World
 setEquipAllocation w = case _rbOptions w of
     EquipOptions {_opEquip = es,_opSel=i} ->