From 59d344b6786733182879ba732193b8d6d2fcfd56 Mon Sep 17 00:00:00 2001
From: Kiana Sheibani <kiana.a.sheibani@gmail.com>
Date: Sat, 28 Jan 2023 21:50:09 -0500
Subject: [PATCH] Add pausing

---
 src/Game/Engine.hs | 23 ++++++++++++++---------
 src/Game/Utils.hs  | 10 ++++++++++
 2 files changed, 24 insertions(+), 9 deletions(-)

diff --git a/src/Game/Engine.hs b/src/Game/Engine.hs
index 28c57b8..7ab92eb 100644
--- a/src/Game/Engine.hs
+++ b/src/Game/Engine.hs
@@ -8,6 +8,7 @@ import Control.Monad (guard)
 import Control.Monad.Trans.MSF.Maybe
 import Control.Monad.Trans.MSF.Reader
 import Control.Monad.Trans.MSF.Writer
+import Data.Bits (xor)
 import Data.Functor (($>))
 import Data.Maybe
 import Data.MonadicStreamFunction
@@ -67,8 +68,8 @@ tick = feedback initialState $ proc (dir, state) -> do
   newstate <- maybeExit -< tickState dir state
   returnA -< (newstate, newstate)
 
-mainSF :: Monad m => MSF (MaybeT (DrawerT m)) () ()
-mainSF = proc () -> do
+gameSF :: Monad m => MSF (MaybeT (DrawerT m)) () GameState
+gameSF = proc () -> do
   -- A "tick" is each frame that the snake advances
   n <- count -< ()
   let isTick = n `mod` 8 == 1
@@ -77,12 +78,16 @@ mainSF = proc () -> do
   dirs <- liftTransS handleEvents -< ()
   dir <- fifoGate -< (dirs, isTick)
 
-  state' <-
-    if isTick
-      then fmap Just tick -< dir
-      else returnA -< Nothing
-  -- undefined is safe here because the first frame is guaranteed to be a tick
-  state <- hold undefined -< state'
+  -- only run `tick` whenever there's a tick
+  pauseMSF undefined tick -< (dir, isTick)
+
+mainSF :: Monad m => MSF (DrawerT m) () ()
+mainSF = proc () -> do
+  keys <- getKeys -< ()
+  let esc = SpecialKey KeyEsc `elem` keys
+
+  paused <- accumulateWith xor True -< esc
+  state <- pauseMSF undefined (loopMaybe gameSF) -< ((), paused)
 
   -- Display the current state
-  liftTransS displayState -< state
+  displayState -< state
diff --git a/src/Game/Utils.hs b/src/Game/Utils.hs
index c6aa297..3ff02f3 100644
--- a/src/Game/Utils.hs
+++ b/src/Game/Utils.hs
@@ -1,3 +1,5 @@
+{-# LANGUAGE Arrows #-}
+
 module Game.Utils where
 
 import Control.Monad.Trans.MSF.Maybe
@@ -75,3 +77,11 @@ fifoGate =
 
 loopMaybe :: Monad m => MSF (MaybeT m) a b -> MSF m a b
 loopMaybe msf = msf `catchMaybe` loopMaybe msf
+
+pauseMSF :: Monad m => b -> MSF m a b -> MSF m (a, Bool) b
+pauseMSF def msf = proc (x, b) -> do
+  y <-
+    if b
+      then fmap Just msf -< x
+      else returnA -< Nothing
+  hold def -< y