taking damage

1 files changed, 12 insertions, 0 deletions

diff --git a/src/Main.hs b/src/Main.hs
index d07cad5..e239bec 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -20,6 +20,9 @@ import qualified Data.CaseInsensitive as CI
 import Data.Map.Strict (Map)
 import qualified Data.Map.Strict as Map
 
+import Data.Set (Set)
+import qualified Data.Set as Set
+
 import Data.Bimap (Bimap)
 import qualified Data.Bimap as Bimap
 
@@ -84,7 +87,10 @@ main = do
                                     , cmd "spend" spendSeq "Spend some of the current focus´ AP"
                                     , cmd "delay" delay "Spend AP until the current focus´ sequence is no higher than the next highest"
                                     , cmd "note" addNote "Add a note to the current focus"
+                                    , cmd "hit" takeHit "Damage the focused entity"
+                                    , cmd "fatigue" takeFatigue "Inflict fatigue damage upon the focused entity"
                                     ]
+                  , wordBreakChars = wordBreakChars initialShellDescription \\ [',', '*']
                   }
   void $ runShell description haskelineBackend (def :: GameState)
 
@@ -244,3 +250,9 @@ delay = withFocus $ \focusId -> () <$ runMaybeT (delay' focusId)
 
 addNote :: String -> Sh GameState ()
 addNote note = withFocus $ \focusId -> gEntities . ix focusId . eNotes %= (note :)
+
+takeHit :: Int -> Completable (Set Hitzone) -> Sh GameState ()
+takeHit dmg = withArg $ \zones -> withFocus $ \focusId -> forM_ zones $ \zone -> gEntities . ix focusId . eStats . sDamage . ix zone += dmg
+
+takeFatigue :: Int -> Sh GameState ()
+takeFatigue dmg = withFocus $ \focusId -> gEntities . ix focusId . eStats . sFatigue += dmg