diff --git a/src/Mtlstats/Events.hs b/src/Mtlstats/Events.hs index 602a8c9..52ad1cc 100644 --- a/src/Mtlstats/Events.hs +++ b/src/Mtlstats/Events.hs @@ -48,4 +48,7 @@ handleEvent e = gets (view progMode) >>= \case | null $ gs ^. homeScore -> do promptHandler homeScorePrompt e return True + | null $ gs ^. awayScore -> do + promptHandler awayScorePrompt e + return True | otherwise -> undefined diff --git a/src/Mtlstats/Prompt.hs b/src/Mtlstats/Prompt.hs index a70eed9..5c69ec4 100644 --- a/src/Mtlstats/Prompt.hs +++ b/src/Mtlstats/Prompt.hs @@ -25,7 +25,8 @@ module Mtlstats.Prompt ( promptHandler, numPrompt, -- * Individual prompts - homeScorePrompt + homeScorePrompt, + awayScorePrompt ) where import Control.Monad (when) @@ -79,3 +80,7 @@ numPrompt pStr act = Prompt homeScorePrompt :: Prompt homeScorePrompt = numPrompt "Home score: " $ modify . (progMode . homeScoreL ?~) + +awayScorePrompt :: Prompt +awayScorePrompt = numPrompt "Away score: " $ + modify . (progMode . awayScoreL ?~) diff --git a/src/Mtlstats/UI.hs b/src/Mtlstats/UI.hs index bf9de7c..42d296f 100644 --- a/src/Mtlstats/UI.hs +++ b/src/Mtlstats/UI.hs @@ -42,6 +42,7 @@ draw s = do NewGame gs | null $ gs ^. gameType -> drawMenu gameTypeMenu | null $ gs ^. homeScore -> drawPrompt homeScorePrompt s + | null $ gs ^. awayScore -> drawPrompt awayScorePrompt s | otherwise -> undefined C.render void $ C.setCursorMode cm