Use play state instead of level complete, that includes game over

2 files changed, 10 insertions, 10 deletions

diff --git a/src/Game/Entities.hs b/src/Game/Entities.hs
index 3cb2510..6ddf589 100644
--- a/src/Game/Entities.hs
+++ b/src/Game/Entities.hs
@@ -71,7 +71,7 @@ addExit es x y = do
 updateAll :: Entities -> GS.State -> IO (Entities, GS.State)
 updateAll es state = do
   -- update the player first (including the reference), unless the level is completed
-  updatedPlayer <- if state.levelCompleted /= GS.ExitOff then pure player else player.update player
+  updatedPlayer <- if state.playState /= GS.InPlay then pure player else player.update player
   void $ writeIORef es.player updatedPlayer
   -- then the other entities
   updated <- (updatedPlayer :) <$> traverse (updateFilter $ state.hitDelay > 0) others
@@ -111,12 +111,12 @@ updateAll es state = do
           processActions s' ents' t
         ActionExitStarted ->
           processActions
-            s {GS.levelCompleted = GS.ExitStarted}
+            s {GS.playState = GS.ExitStarted}
             -- the player is not in the action, changing the type disables collision detection
             ((head ents) {typ = TypeEffect} : tail ents)
             t
-        ActionEntryDone -> processActions s {GS.levelCompleted = GS.ExitOff} ents t
-        ActionExitDone -> processActions s {GS.levelCompleted = GS.ExitDone} ents t
+        ActionEntryDone -> processActions s {GS.playState = GS.InPlay} ents t
+        ActionExitDone -> processActions s {GS.playState = GS.ExitDone} ents t
         ActionAddBlast x y d playerCollision isBlocked -> do
           blast <- mkBlast es.sprites x y d playerCollision isBlocked
           processActions s (ents ++ [blast]) t
@@ -156,7 +156,7 @@ render renderer es state = do
   -- always render player last
   -- won't draw all the frames if the player was hit
   -- or we are exiting the level
-  if testBit state.hitDelay 2 || state.levelCompleted /= GS.ExitOff then pure () else renderOne player
+  if testBit state.hitDelay 2 || state.playState /= GS.InPlay then pure () else renderOne player
   where
     player = head es.entities
     others = tail es.entities
diff --git a/src/Game/State.hs b/src/Game/State.hs
index 083e6d4..df2d4fa 100644
--- a/src/Game/State.hs
+++ b/src/Game/State.hs
@@ -1,4 +1,4 @@
-module Game.State (State (..), initialState, levelState, maxLives, ExitState (..)) where
+module Game.State (State (..), initialState, levelState, maxLives, PlayState (..)) where
 
 import Game.Entities.Const (hitDelay)
 import qualified Game.Map as M
@@ -6,7 +6,7 @@ import qualified Game.Map as M
 maxLives :: Int
 maxLives = 4
 
-data ExitState = ExitOff | ExitEntry | ExitStarted | ExitDone deriving (Eq)
+data PlayState = InPlay | IntoStage | ExitStarted | ExitDone | GameOver deriving (Eq)
 
 data State = State
   { batteries :: Int,
@@ -17,7 +17,7 @@ data State = State
     gameOverDelay :: Int,
     exit :: Bool,
     lastBattery :: (Int, Int),
-    levelCompleted :: ExitState,
+    playState :: PlayState,
     currentLevel :: Int
   }
 
@@ -33,7 +33,7 @@ initialState m =
       exit = False,
       -- doesn't matter where
       lastBattery = (0, 0),
-      levelCompleted = ExitEntry,
+      playState = IntoStage,
       currentLevel = 0
     }
 
@@ -47,5 +47,5 @@ levelState s m =
       exit = False,
       -- doesn't matter where
       lastBattery = (0, 0),
-      levelCompleted = ExitEntry
+      playState = IntoStage
     }