Exit transition

1 files changed, 6 insertions, 4 deletions

diff --git a/src/Game/State.hs b/src/Game/State.hs
index 3773268..4f46c01 100644
--- a/src/Game/State.hs
+++ b/src/Game/State.hs
@@ -1,4 +1,4 @@
-module Game.State (State (..), initialState, levelState, maxLives) where
+module Game.State (State (..), initialState, levelState, maxLives, ExitState (..)) where
 
 import Game.Entities.Const (hitDelay)
 import qualified Game.Map as M
@@ -6,6 +6,8 @@ import qualified Game.Map as M
 maxLives :: Int
 maxLives = 4
 
+data ExitState = ExitOff | ExitStarted | ExitDone deriving (Eq)
+
 data State = State
   { batteries :: Int,
     totalBatteries :: Int,
@@ -15,7 +17,7 @@ data State = State
     gameOverDelay :: Int,
     exit :: Bool,
     lastBattery :: (Int, Int),
-    levelCompleted :: Bool,
+    levelCompleted :: ExitState,
     currentLevel :: Int
   }
 
@@ -31,7 +33,7 @@ initialState m =
       exit = False,
       -- doesn't matter where
       lastBattery = (0, 0),
-      levelCompleted = False,
+      levelCompleted = ExitOff,
       currentLevel = 0
     }
 
@@ -45,5 +47,5 @@ levelState s m =
       exit = False,
       -- doesn't matter where
       lastBattery = (0, 0),
-      levelCompleted = False
+      levelCompleted = ExitOff
     }