From 74809ced2ee97513dce2ac42a054e4f6a16cee7f Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jlamothe1980@gmail.com>
Date: Sat, 24 Aug 2019 20:14:49 -0400
Subject: [PATCH] prompt for away team score

---
 src/Mtlstats/Events.hs | 3 +++
 src/Mtlstats/Prompt.hs | 7 ++++++-
 src/Mtlstats/UI.hs     | 1 +
 3 files changed, 10 insertions(+), 1 deletion(-)

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