Revisited the game over sequence

1 files changed, 5 insertions, 1 deletions

diff --git a/src/Game/Entities.hs b/src/Game/Entities.hs
index 71f9791..a168e83 100644
--- a/src/Game/Entities.hs
+++ b/src/Game/Entities.hs
@@ -51,8 +51,12 @@ updateAll es = do
   updated <- (updatedPlayer :) <$> traverse (updateFilter playerWasHit) others
   -- collect new entities
   new <- traverse (processSpawn es.sprites) (concatMap (\e -> e.spawns) updated)
+  -- is the player dead?
+  updated' <- do
+    s <- readIORef es.state
+    pure $ if s.lives == 0 && updatedPlayer.dir /= Dying then (head updated) {dir = Dying, gravity = gravityUp} : tail updated else updated
   -- clear spawns (new entities), filter out destroyed entities, and add the new ones
-  pure es {entities = map (\e -> e {spawns = []}) (filter (\e -> not e.destroy) updated) ++ new}
+  pure es {entities = map (\e -> e {spawns = []}) (filter (\e -> not e.destroy) updated') ++ new}
   where
     player = head es.entities
     others = tail es.entities