ask if a new goalie is a rookie

src/Mtlstats/Control/CreateGoalie.hs

@@ -35,9 +35,10 @@ import Mtlstats.Types
 -- | Handles goalie creation
 createGoalieC :: CreateGoalieState -> Controller
 createGoalieC cgs
-  | null $ cgs^.cgsNumber = getGoalieNumC
+  | null $ cgs^.cgsNumber     = getGoalieNumC
-  | null $ cgs^.cgsName   = getGoalieNameC
+  | null $ cgs^.cgsName       = getGoalieNameC
-  | otherwise             = confirmCreateGoalieC
+  | null $ cgs^.cgsRookieFlag = getRookieFlagC
+  | otherwise                 = confirmCreateGoalieC
 
 getGoalieNumC :: Controller
 getGoalieNumC = promptController goalieNumPrompt
@@ -45,6 +46,16 @@ getGoalieNumC = promptController goalieNumPrompt
 getGoalieNameC :: Controller
 getGoalieNameC = promptController goalieNamePrompt
 
+getRookieFlagC :: Controller
+getRookieFlagC = Controller
+  { drawController = const $ do
+    C.drawString "Is this goalie a rookie? (Y/N)"
+    return C.CursorInvisible
+  , handleController = \e -> do
+    modify $ progMode.createGoalieStateL.cgsRookieFlag .~ ynHandler e
+    return True
+  }
+
 confirmCreateGoalieC :: Controller
 confirmCreateGoalieC = Controller
   { drawController = \s -> do