combat rounds

1 files changed, 22 insertions, 1 deletions

diff --git a/src/Main.hs b/src/Main.hs
index a38b514..e93d725 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -31,10 +31,12 @@ import Data.List
 import Data.List
 import Data.Maybe
 import Data.Bool
+import Data.Monoid (All(..))
 
 import Data.Function
 
 import Control.Monad.State.Strict
+import Control.Monad.Reader
 import Control.Monad.Trans.Maybe
 import Control.Monad.List
 
@@ -157,9 +159,28 @@ stateMaintenance = do
     isUnconscious <- evalF =<< MaybeT (preuse $ lStats . sUnconscious)
     guard $ isDead || isUnconscious
     when isDead . lift . shellPutStrLn $ name ++ " died"
-    when isUnconscious . lift . shellPutStrLn $ name ++ " is unconscious"
+    when (isUnconscious && not isDead) . lift . shellPutStrLn $ name ++ " is unconscious"
     gFocus' . eSeqVal .= Nothing
     gFocus .= Nothing
+  void $ do
+    round <- use gRound
+    let
+      finished sVal = fromMaybe False (previews (seqVal . _Just) (<= 0) sVal) || view (seqRound . _Wrapped) sVal /= round
+    allFinished <- getAll . foldMapOf (gEntities . each . eStats . sSequence . _Just) (All . finished) <$> get
+    when allFinished $ do
+      let
+        advanceRound' :: EntityIdentifier -> Entity -> Sh GameState Entity
+        advanceRound' ident entity = fmap (fromMaybe entity . (\(m, s) -> s <$ m)) . (runStateT ?? entity) . runMaybeT $ do
+          cRound <- MaybeT . preuse $ eStats . sSequence . _Just . seqRound . _Wrapped
+          guard $ cRound < 0
+          cVal <- MaybeT . preuse $ eStats . sSequence . _Just . seqVal . _Just
+          name <- lift . lift $ toName ident
+          nVal' <- MaybeT . preuse $ eStats . sSeqVal
+          nVal <- MaybeT . focusState eStats $ evalFormula' name nVal'
+          eStats . sSequence . _Just . seqVal . _Just += nVal
+          eStats . sSequence . _Just . seqRound . _Wrapped += 1
+      gRounds -= 1
+      gEntities <~ (imapM advanceRound' =<< use gEntities)
 
 -- Query state
 listFactions, listEntities :: Sh GameState ()