From b108b2ed656573da7bd78e7ad5704648716a98fc Mon Sep 17 00:00:00 2001
From: Tibo De Peuter <tibo.depeuter@ugent.be>
Date: Fri, 23 Dec 2022 10:21:56 +0100
Subject: [PATCH] #4 Pick  up items

---
 lib/RPGEngine/Data/Level.hs            |  9 +++++++++
 lib/RPGEngine/Input/ActionSelection.hs | 18 ++++++++++++++----
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/lib/RPGEngine/Data/Level.hs b/lib/RPGEngine/Data/Level.hs
index 7265e25..e3bf9f2 100644
--- a/lib/RPGEngine/Data/Level.hs
+++ b/lib/RPGEngine/Data/Level.hs
@@ -37,6 +37,15 @@ hasAt pos level = match firstItem firstEntity
           firstItem   = find ((== pos) . getICoord) $ items level
           getICoord i = (itemX i, itemY i)
 
+getWithId :: String -> Level -> Maybe (Either Item Entity)
+getWithId id level = match firstItem firstEntity
+    where match :: Maybe Item -> Maybe Entity -> Maybe (Either Item Entity)
+          match (Just a) _ = Just $ Left a
+          match _ (Just a) = Just $ Right a
+          match _ _        = Nothing
+          firstEntity      = find ((== id) . entityId) $ entities level
+          firstItem        = find ((== id) . itemId) $ items level
+
 directionOffsets :: Direction -> (X, Y)
 directionOffsets North  = ( 0,  1)
 directionOffsets East   = ( 1,  0)
diff --git a/lib/RPGEngine/Input/ActionSelection.hs b/lib/RPGEngine/Input/ActionSelection.hs
index 99e6cfe..d454df9 100644
--- a/lib/RPGEngine/Input/ActionSelection.hs
+++ b/lib/RPGEngine/Input/ActionSelection.hs
@@ -4,10 +4,11 @@ module RPGEngine.Input.ActionSelection
 
 import RPGEngine.Input.Core (InputHandler, handleKey, composeInputHandlers, ListSelector (selection))
 
-import RPGEngine.Data (Game (..), State (..), Direction (..), Action (..), ItemId, EntityId)
+import RPGEngine.Data (Game (..), State (..), Direction (..), Action (..), ItemId, EntityId, Level (..), Player (inventory))
 import Graphics.Gloss.Interface.IO.Game (Key(SpecialKey), SpecialKey (KeyUp, KeyDown))
 import Graphics.Gloss.Interface.IO.Interact
     ( SpecialKey(..), KeyState(..) )
+import RPGEngine.Data.Level (getWithId)
 
 ------------------------------ Exported ------------------------------
 
@@ -22,9 +23,10 @@ handleInputActionSelection = composeInputHandlers [
 ----------------------------------------------------------------------
 
 selectAction :: Game -> Game
-selectAction game@Game{ state = ActionSelection list selection continue } = newGame
+selectAction game@Game{ state = ActionSelection list selector continue } = newGame
     where newGame = game{ state = execute selectedAction continue }
-          selectedAction = Leave
+          selectedAction = list !! index
+          index = selection selector
 selectAction g = g
 
 -- TODO Lift this code from LevelSelection
@@ -52,8 +54,16 @@ execute (IncreasePlayerHp iid) s = increasePlayerHp iid s
 execute _ s = s
 
 -- Pick up the item with itemId and put it in the players inventory
+-- Should receive a Playing state
 pickUpItem :: ItemId -> State -> State
-pickUpItem _ s = s -- TODO
+pickUpItem id s@Playing{ level = level, player = player } = newState
+    where (Just (Left pickedUpItem)) = getWithId id level
+          newState     = s{ level = newLevel, player = newPlayer }
+          newLevel     = level{ items = filteredItems }
+          filteredItems = filter (/= pickedUpItem) $ items level
+          newPlayer     = player{ inventory = newInventory }
+          newInventory  = pickedUpItem:inventory player
+pickUpItem _ _ = Error "Something went wrong while picking up an item"
 
 -- Use an item on the player
 useItem :: ItemId -> State -> State -- TODO