don't ask which goalie to assign the game to when there's only one

2020-02-14 22:56:35 -05:00
parent 2607fc5ce8
commit 119032ca80
3 changed files with 44 additions and 16 deletions
--- a/src/Mtlstats/Actions/NewGame/GoalieInput.hs
+++ b/src/Mtlstats/Actions/NewGame/GoalieInput.hs
@@ -36,8 +36,12 @@ import Mtlstats.Util

 -- | Attempts to finish game goalie entry
 finishGoalieEntry :: ProgState -> ProgState
-finishGoalieEntry s = s & progMode.gameStateL.gameGoaliesRecorded
-  .~ not (null $ s^.progMode.gameStateL.gameGoalieStats)
+finishGoalieEntry s = case M.toList $ s^.progMode.gameStateL.gameGoalieStats of
+  []         -> s
+  [(gid, _)] -> setGameGoalie gid s'
+  _          -> s'
+  where
+    s' = s & progMode.gameStateL.gameGoaliesRecorded .~ True

 -- | Records the goalie's game stats
 recordGoalieStats :: ProgState -> ProgState