version 0.11.0

Master menu

ChangeLog.md

@@ -1,5 +1,22 @@
 # Changelog for mtlstats
 
+## 0.11.0
+- Added active flag to players/goalies
+- Clear rookie flag on new (regular) season
+- Save a copy of the database on new season
+- Implemented game standings editing
+- Added title screen
+- Changed sytling of menus
+
+## 0.10.0
+- Don't show player number zero in reports
+- Fixed player/goalie name capitalisation on edit
+- Return to correct edit menus after editing player stats
+- Enabled batch editing of player/goalie YTD/lifetime stats
+- Bugfix: allow user to edit goalie shutouts
+- Added line numbers to lifetime player/goalie reports
+- Implemented rookie flag
+
 ## 0.9.0
 - Bugfix: Display lifetime stats in report, not YTD
 - Force expected capitalization on player/goalie names
@@ -24,24 +41,20 @@
 - Reset game standings on new season
 
 ## 0.5.0
-
 - Fixed player creation bug
 - Prompt for goalie informaiton on game data entry
 - Implemented player editing
 
 ## v0.4.0
-
 - Record penalty minutes
 - Calculate total game statistics
 - Generate year-to-date statistics report
 
 ## v0.3.0
-
 - Record goals and assists
 - Track goals for and goals against
 
 ## v0.2.0
-
 - Overtime losses don't count in the loss column
 - Confirm game data with user before updating stats
 - Implemented player creation

app/Main.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

package.yaml

@@ -1,5 +1,5 @@
 name:                mtlstats
-version:             0.9.0
+version:             0.11.0
 github:              "mtlstats/mtlstats"
 license:             GPL-3
 author:              "Jonathan Lamothe"

src/Mtlstats.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Actions.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -24,6 +24,7 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 module Mtlstats.Actions
   ( startNewSeason
   , resetYtd
+  , clearRookies
   , resetStandings
   , startNewGame
   , addChar
@@ -32,7 +33,9 @@ module Mtlstats.Actions
   , createGoalie
   , edit
   , editPlayer
+  , editSelectedPlayer
   , editGoalie
+  , editSelectedGoalie
   , addPlayer
   , addGoalie
   , resetCreatePlayerState
@@ -40,17 +43,29 @@ module Mtlstats.Actions
   , backHome
   , scrollUp
   , scrollDown
+  , saveDatabase
   ) where
 
-import Control.Monad.Trans.State (modify)
+import Control.Monad.IO.Class (liftIO)
+import Control.Monad.Trans.State (gets, modify)
+import Data.Aeson (encodeFile)
 import Data.Maybe (fromMaybe)
 import Lens.Micro ((^.), (&), (.~), (%~))
+import System.EasyFile
+  ( createDirectoryIfMissing
+  , getAppUserDataDirectory
+  , (</>)
+  )
 
+import Mtlstats.Config
 import Mtlstats.Types
+import Mtlstats.Util
 
 -- | Starts a new season
 startNewSeason :: ProgState -> ProgState
-startNewSeason = (progMode .~ NewSeason) . (database . dbGames .~ 0)
+startNewSeason
+  = (progMode .~ NewSeason False)
+  . (database.dbGames .~ 0)
 
 -- | Resets all players year-to-date stats
 resetYtd :: ProgState -> ProgState
@@ -58,6 +73,12 @@ resetYtd
   = (database . dbPlayers %~ map (pYtd .~ newPlayerStats))
   . (database . dbGoalies %~ map (gYtd .~ newGoalieStats))
 
+-- | Clears the rookie flag from all players/goalies
+clearRookies :: ProgState -> ProgState
+clearRookies = database
+  %~ (dbPlayers %~ map (pRookie .~ False))
+  .  (dbGoalies %~ map (gRookie .~ False))
+
 -- | Resets game standings
 resetStandings :: ProgState -> ProgState
 resetStandings = database
@@ -106,10 +127,36 @@ edit = progMode .~ EditMenu
 editPlayer :: ProgState -> ProgState
 editPlayer = progMode .~ EditPlayer newEditPlayerState
 
+-- | Edits the selected 'Player'
+editSelectedPlayer
+  :: (Player -> Player)
+  -- ^ The modification to be made to the 'Player'
+  -> ProgState
+  -> ProgState
+editSelectedPlayer f s = fromMaybe s $ do
+  n <- s^.progMode.editPlayerStateL.epsSelectedPlayer
+  let
+    players  = s^.database.dbPlayers
+    players' = modifyNth n f players
+  Just $ s & database.dbPlayers .~ players'
+
 -- | Starts the 'Goalie' editing process
 editGoalie :: ProgState -> ProgState
 editGoalie = progMode .~ EditGoalie newEditGoalieState
 
+-- | Edits the selected 'Goalie'
+editSelectedGoalie
+  :: (Goalie -> Goalie)
+  -- ^ The modification to be made to the 'Goalie'
+  -> ProgState
+  -> ProgState
+editSelectedGoalie f s = fromMaybe s $ do
+  n <- s^.progMode.editGoalieStateL.egsSelectedGoalie
+  let
+    goalies  = s^.database.dbGoalies
+    goalies' = modifyNth n f goalies
+  Just $ s & database.dbGoalies .~ goalies'
+
 -- | Adds the entered player to the roster
 addPlayer :: ProgState -> ProgState
 addPlayer s = fromMaybe s $ do
@@ -160,3 +207,13 @@ scrollUp = scrollOffset %~ max 0 . pred
 -- | Scrolls the display down
 scrollDown :: ProgState -> ProgState
 scrollDown = scrollOffset %~ succ
+
+-- | Saves the database
+saveDatabase :: String -> Action ()
+saveDatabase fn = do
+  db <- gets (^.database)
+  liftIO $ do
+    dir <- getAppUserDataDirectory appName
+    let dbFile = dir </> fn
+    createDirectoryIfMissing True dir
+    encodeFile dbFile db

src/Mtlstats/Actions/EditGoalie.hs

@@ -1,164 +0,0 @@
-{- |
-
-mtlstats
-Copyright (C) 2019 Rhéal Lamothe
-<rheal.lamothe@gmail.com>
-
-This program is free software: you can redistribute it and/or modify
-it under the terms of the GNU General Public License as published by
-the Free Software Foundation, either version 3 of the License, or (at
-your option) any later version.
-
-This program is distributed in the hope that it will be useful, but
-WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-General Public License for more details.
-
-You should have received a copy of the GNU General Public License
-along with this program.  If not, see <https://www.gnu.org/licenses/>.
-
--}
-
-module Mtlstats.Actions.EditGoalie
-  ( editGoalieNumber
-  , editGoalieName
-  , editGoalieYtdGames
-  , editGoalieYtdMins
-  , editGoalieYtdGoals
-  , editGoalieYtdWins
-  , editGoalieYtdLosses
-  , editGoalieYtdTies
-  , editGoalieLtGames
-  , editGoalieLtMins
-  , editGoalieLtGoals
-  , editGoalieLtWins
-  , editGoalieLtLosses
-  , editGoalieLtTies
-  ) where
-
-import Control.Monad (void)
-import Data.Maybe (fromMaybe)
-import Lens.Micro ((^.), (&), (.~), (%~))
-
-import Mtlstats.Types
-import Mtlstats.Util
-
--- | Edits a goalie's number
-editGoalieNumber
-  :: Int
-  -- ^ New goalie number
-  -> ProgState
-  -> ProgState
-editGoalieNumber num = editGoalie (gNumber .~ num) EGMenu
-
--- | Edits a goalie's name
-editGoalieName
-  :: String
-  -- ^ The new name
-  -> ProgState
-  -> ProgState
-editGoalieName name = editGoalie (gName .~ name) EGMenu
-
--- | Edits a goalie's YTD games
-editGoalieYtdGames
-  :: Int
-  -- ^ The number of games played
-  -> ProgState
-  -> ProgState
-editGoalieYtdGames games = editGoalie (gYtd.gsGames .~ games) EGYtd
-
--- | Edits a goalie's YTD minutes
-editGoalieYtdMins
-  :: Int
-  -- ^ The number of minutes played
-  -> ProgState
-  -> ProgState
-editGoalieYtdMins mins = editGoalie (gYtd.gsMinsPlayed .~ mins) EGYtd
-
--- | Edits a goalie's YTD goals allowed
-editGoalieYtdGoals
-  :: Int
-  -- ^ The number of goals
-  -> ProgState
-  -> ProgState
-editGoalieYtdGoals goals = editGoalie (gYtd.gsGoalsAllowed .~ goals) EGYtd
-
--- | Edits a goalie's YTD wins
-editGoalieYtdWins
-  :: Int
-  -- ^ The number of wins
-  -> ProgState
-  -> ProgState
-editGoalieYtdWins wins = editGoalie (gYtd.gsWins .~ wins) EGYtd
-
--- | Edits a goalie's YTD losses
-editGoalieYtdLosses
-  :: Int
-  -- ^ The number of losses
-  -> ProgState
-  -> ProgState
-editGoalieYtdLosses losses = editGoalie (gYtd.gsLosses .~ losses) EGYtd
-
--- | Edits a goalie's YTD ties
-editGoalieYtdTies
-  :: Int
-  -- ^ The number of ties
-  -> ProgState
-  -> ProgState
-editGoalieYtdTies ties = editGoalie (gYtd.gsTies .~ ties) EGYtd
-
--- | Edits a goalie's lifetime games played
-editGoalieLtGames
-  :: Int
-  -- ^ The number of games
-  -> ProgState
-  -> ProgState
-editGoalieLtGames games = editGoalie (gLifetime.gsGames .~ games) EGLifetime
-
--- | Edits a goalie's lifetime minutes played
-editGoalieLtMins
-  :: Int
-  -- ^ The number of minutes
-  -> ProgState
-  -> ProgState
-editGoalieLtMins mins = editGoalie (gLifetime.gsMinsPlayed .~ mins) EGLifetime
-
--- | Edits a goalie's lifetime goals allowed
-editGoalieLtGoals
-  :: Int
-  -- ^ The number of goals
-  -> ProgState
-  -> ProgState
-editGoalieLtGoals goals = editGoalie (gLifetime.gsGoalsAllowed .~ goals) EGLifetime
-
--- | Edits a goalie's lifetime wins
-editGoalieLtWins
-  :: Int
-  -- ^ The number of wins
-  -> ProgState
-  -> ProgState
-editGoalieLtWins wins = editGoalie (gLifetime.gsWins .~ wins) EGLifetime
-
--- | Edits a goalie's lifetime losses
-editGoalieLtLosses
-  :: Int
-  -- ^ The number of losses
-  -> ProgState
-  -> ProgState
-editGoalieLtLosses losses = editGoalie (gLifetime.gsLosses .~ losses) EGLifetime
-
--- | Edits a goalie's lifetime ties
-editGoalieLtTies
-  :: Int
-  -- ^ The number of ties
-  -> ProgState
-  -> ProgState
-editGoalieLtTies ties = editGoalie (gLifetime.gsTies .~ ties) EGLifetime
-
-editGoalie :: (Goalie -> Goalie) -> EditGoalieMode -> ProgState -> ProgState
-editGoalie f mode s = fromMaybe s $ do
-  gid <- s^.progMode.editGoalieStateL.egsSelectedGoalie
-  void $ nth gid $ s^.database.dbGoalies
-  Just $ s
-    & database.dbGoalies %~ modifyNth gid f
-    & progMode.editGoalieStateL.egsMode .~ mode

src/Mtlstats/Actions/EditStandings.hs

@@ -0,0 +1,70 @@
+{- |
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+module Mtlstats.Actions.EditStandings
+  ( editStandings
+  , editHomeStandings
+  , editAwayStandings
+  , editWins
+  , editLosses
+  , editOvertime
+  , editGoalsFor
+  , editGoalsAgainst
+  ) where
+
+import Lens.Micro ((.~))
+
+import Mtlstats.Types
+
+-- | Enters edit standings mode
+editStandings :: ProgState -> ProgState
+editStandings = progMode .~ EditStandings ESMMenu
+
+-- | Edits the home standings
+editHomeStandings :: ProgState -> ProgState
+editHomeStandings = progMode .~ EditStandings (ESMHome ESMSubMenu)
+
+-- | Edits the road standings
+editAwayStandings :: ProgState -> ProgState
+editAwayStandings = progMode .~ EditStandings (ESMAway ESMSubMenu)
+
+-- | Changes to edit wins mode
+editWins :: ProgState -> ProgState
+editWins = doEdit ESMEditWins
+
+-- | Changes to edit losses mode
+editLosses :: ProgState -> ProgState
+editLosses = doEdit ESMEditLosses
+
+-- | Changes to edit overtime mode
+editOvertime :: ProgState -> ProgState
+editOvertime = doEdit ESMEditOvertime
+
+-- | Changes to edit goals for mode
+editGoalsFor :: ProgState -> ProgState
+editGoalsFor = doEdit ESMEditGoalsFor
+
+-- | Changes to edit goals against mode
+editGoalsAgainst :: ProgState -> ProgState
+editGoalsAgainst = doEdit ESMEditGoalsAgainst
+
+doEdit :: ESMSubMode -> ProgState -> ProgState
+doEdit = (progMode.editStandingsModeL.esmSubModeL .~)

src/Mtlstats/Actions/NewGame.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Actions/NewGame/GoalieInput.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Config.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Control.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -29,8 +29,10 @@ import Lens.Micro.Extras (view)
 import qualified UI.NCurses as C
 
 import Mtlstats.Actions
+import Mtlstats.Control.TitleScreen
 import Mtlstats.Control.EditGoalie
 import Mtlstats.Control.EditPlayer
+import Mtlstats.Control.EditStandings
 import Mtlstats.Control.NewGame
 import Mtlstats.Handlers
 import Mtlstats.Menu
@@ -41,10 +43,11 @@ import Mtlstats.Types
 -- run
 dispatch :: ProgState -> Controller
 dispatch s = case s^.progMode of
-  MainMenu   -> mainMenuC
-  NewSeason  -> newSeasonC
-  NewGame gs -> newGameC gs
-  EditMenu   -> editMenuC
+  TitleScreen    -> titleScreenC
+  MainMenu       -> mainMenuC
+  NewSeason flag -> newSeasonC flag
+  NewGame gs     -> newGameC gs
+  EditMenu       -> editMenuC
   CreatePlayer cps
     | null $ cps^.cpsNumber   -> getPlayerNumC
     | null $ cps^.cpsName     -> getPlayerNameC
@@ -54,8 +57,9 @@ dispatch s = case s^.progMode of
     | null $ cgs^.cgsNumber -> getGoalieNumC
     | null $ cgs^.cgsName   -> getGoalieNameC
     | otherwise             -> confirmCreateGoalieC
-  EditPlayer eps -> editPlayerC eps
-  EditGoalie egs -> editGoalieC egs
+  EditPlayer eps      -> editPlayerC eps
+  EditGoalie egs      -> editGoalieC egs
+  (EditStandings esm) -> editStandingsC esm
 
 mainMenuC :: Controller
 mainMenuC = Controller
@@ -63,13 +67,9 @@ mainMenuC = Controller
   , handleController = menuHandler mainMenu
   }
 
-newSeasonC :: Controller
-newSeasonC = Controller
-  { drawController   = const $ drawMenu newSeasonMenu
-  , handleController = \e -> do
-    menuHandler newSeasonMenu e
-    return True
-  }
+newSeasonC :: Bool -> Controller
+newSeasonC False = promptController newSeasonPrompt
+newSeasonC True  = menuController newSeasonMenu
 
 editMenuC :: Controller
 editMenuC = menuController editMenu

src/Mtlstats/Control/EditGoalie.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -46,23 +46,25 @@ selectC = promptController goalieToEditPrompt
 
 editC :: EditGoalieMode -> Controller
 editC = \case
-  EGMenu      -> menuC
-  EGNumber    -> numberC
-  EGName      -> nameC
-  EGYtd       -> ytdMenuC
-  EGLifetime  -> lifetimeMenuC
-  EGYtdGames  -> ytdGamesC
-  EGYtdMins   -> ytdMinsC
-  EGYtdGoals  -> ytdGoalsC
-  EGYtdWins   -> ytdWinsC
-  EGYtdLosses -> ytdLossesC
-  EGYtdTies   -> ytdTiesC
-  EGLtGames   -> ltGamesC
-  EGLtMins    -> ltMinsC
-  EGLtGoals   -> ltGoalsC
-  EGLtWins    -> ltWinsC
-  EGLtLosses  -> ltLossesC
-  EGLtTies    -> ltTiesC
+  EGMenu          -> menuC
+  EGNumber        -> numberC
+  EGName          -> nameC
+  EGYtd           -> ytdMenuC
+  EGLifetime      -> lifetimeMenuC
+  EGYtdGames b    -> ytdGamesC b
+  EGYtdMins b     -> ytdMinsC b
+  EGYtdGoals b    -> ytdGoalsC b
+  EGYtdShutouts b -> ytdShutoutsC b
+  EGYtdWins b     -> ytdWinsC b
+  EGYtdLosses b   -> ytdLossesC b
+  EGYtdTies       -> ytdTiesC
+  EGLtGames b     -> ltGamesC b
+  EGLtMins b      -> ltMinsC b
+  EGLtGoals b     -> ltGoalsC b
+  EGLtShutouts b  -> ltShutoutsC b
+  EGLtWins b      -> ltWinsC b
+  EGLtLosses b    -> ltLossesC b
+  EGLtTies        -> ltTiesC
 
 menuC :: Controller
 menuC = menuControllerWith header editGoalieMenu
@@ -79,38 +81,44 @@ ytdMenuC = menuControllerWith header editGoalieYtdMenu
 lifetimeMenuC :: Controller
 lifetimeMenuC = menuControllerWith header editGoalieLtMenu
 
-ytdGamesC :: Controller
-ytdGamesC = promptController editGoalieYtdGamesPrompt
+ytdGamesC :: Bool -> Controller
+ytdGamesC = promptController . editGoalieYtdGamesPrompt
 
-ytdMinsC :: Controller
-ytdMinsC = promptController editGoalieYtdMinsPrompt
+ytdMinsC :: Bool -> Controller
+ytdMinsC = promptController . editGoalieYtdMinsPrompt
 
-ytdGoalsC :: Controller
-ytdGoalsC = promptController editGoalieYtdGoalsPrompt
+ytdGoalsC :: Bool -> Controller
+ytdGoalsC = promptController . editGoalieYtdGoalsPrompt
 
-ytdWinsC :: Controller
-ytdWinsC = promptController editGoalieYtdWinsPrompt
+ytdShutoutsC :: Bool -> Controller
+ytdShutoutsC = promptController . editGoalieYtdShutoutsPrompt
 
-ytdLossesC :: Controller
-ytdLossesC = promptController editGoalieYtdLossesPrompt
+ytdWinsC :: Bool -> Controller
+ytdWinsC = promptController . editGoalieYtdWinsPrompt
+
+ytdLossesC :: Bool -> Controller
+ytdLossesC = promptController . editGoalieYtdLossesPrompt
 
 ytdTiesC :: Controller
 ytdTiesC = promptController editGoalieYtdTiesPrompt
 
-ltGamesC :: Controller
-ltGamesC = promptController editGoalieLtGamesPrompt
+ltGamesC :: Bool -> Controller
+ltGamesC = promptController . editGoalieLtGamesPrompt
 
-ltMinsC :: Controller
-ltMinsC = promptController editGoalieLtMinsPrompt
+ltMinsC :: Bool -> Controller
+ltMinsC = promptController . editGoalieLtMinsPrompt
 
-ltGoalsC :: Controller
-ltGoalsC = promptController editGoalieLtGoalsPrompt
+ltGoalsC :: Bool -> Controller
+ltGoalsC = promptController . editGoalieLtGoalsPrompt
 
-ltWinsC :: Controller
-ltWinsC = promptController editGoalieLtWinsPrompt
+ltShutoutsC :: Bool -> Controller
+ltShutoutsC = promptController . editGoalieLtShutoutsPrompt
 
-ltLossesC :: Controller
-ltLossesC = promptController editGoalieLtLossesPrompt
+ltWinsC :: Bool -> Controller
+ltWinsC = promptController . editGoalieLtWinsPrompt
+
+ltLossesC :: Bool -> Controller
+ltLossesC = promptController . editGoalieLtLossesPrompt
 
 ltTiesC :: Controller
 ltTiesC = promptController editGoalieLtTiesPrompt

src/Mtlstats/Control/EditPlayer.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -38,18 +38,18 @@ editPlayerC :: EditPlayerState -> Controller
 editPlayerC eps
   | null $ eps^.epsSelectedPlayer = selectPlayerC
   | otherwise = case eps^.epsMode of
-    EPMenu       -> menuC
-    EPNumber     -> numberC
-    EPName       -> nameC
-    EPPosition   -> positionC
-    EPYtd        -> ytdC
-    EPLifetime   -> lifetimeC
-    EPYtdGoals   -> ytdGoalsC
-    EPYtdAssists -> ytdAssistsC
-    EPYtdPMin    -> ytdPMinC
-    EPLtGoals    -> ltGoalsC
-    EPLtAssists  -> ltAssistsC
-    EPLtPMin     -> ltPMinC
+    EPMenu         -> menuC
+    EPNumber       -> numberC
+    EPName         -> nameC
+    EPPosition     -> positionC
+    EPYtd          -> ytdC
+    EPLifetime     -> lifetimeC
+    EPYtdGoals   b -> ytdGoalsC   b
+    EPYtdAssists b -> ytdAssistsC b
+    EPYtdPMin      -> ytdPMinC
+    EPLtGoals    b -> ltGoalsC    b
+    EPLtAssists  b -> ltAssistsC  b
+    EPLtPMin       -> ltPMinC
 
 selectPlayerC :: Controller
 selectPlayerC = promptController playerToEditPrompt
@@ -72,20 +72,20 @@ ytdC = menuControllerWith header editPlayerYtdMenu
 lifetimeC :: Controller
 lifetimeC = menuControllerWith header editPlayerLtMenu
 
-ytdGoalsC :: Controller
-ytdGoalsC = promptController editPlayerYtdGoalsPrompt
+ytdGoalsC :: Bool -> Controller
+ytdGoalsC = promptController . editPlayerYtdGoalsPrompt
 
-ytdAssistsC :: Controller
-ytdAssistsC = promptController editPlayerYtdAssistsPrompt
+ytdAssistsC :: Bool -> Controller
+ytdAssistsC = promptController . editPlayerYtdAssistsPrompt
 
 ytdPMinC :: Controller
 ytdPMinC = promptController editPlayerYtdPMinPrompt
 
-ltGoalsC :: Controller
-ltGoalsC = promptController editPlayerLtGoalsPrompt
+ltGoalsC :: Bool -> Controller
+ltGoalsC = promptController . editPlayerLtGoalsPrompt
 
-ltAssistsC :: Controller
-ltAssistsC = promptController editPlayerLtAssistsPrompt
+ltAssistsC :: Bool -> Controller
+ltAssistsC = promptController . editPlayerLtAssistsPrompt
 
 ltPMinC :: Controller
 ltPMinC = promptController editPlayerLtPMinPrompt

src/Mtlstats/Control/EditStandings.hs

@@ -0,0 +1,87 @@
+{- |
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+{-# LANGUAGE LambdaCase #-}
+
+module Mtlstats.Control.EditStandings (editStandingsC) where
+
+import Lens.Micro ((^.))
+import qualified UI.NCurses as C
+
+import Mtlstats.Format
+import Mtlstats.Menu
+import Mtlstats.Menu.EditStandings
+import Mtlstats.Prompt
+import Mtlstats.Prompt.EditStandings
+import Mtlstats.Types
+import Mtlstats.Types.Menu
+
+-- | Controller for the edit standings menu
+editStandingsC :: EditStandingsMode -> Controller
+editStandingsC = \case
+  ESMMenu   -> menuControllerWith header editStandingsMenu
+  ESMHome m -> editHomeStandingsC m
+  ESMAway m -> editAwayStandingsC m
+
+editHomeStandingsC :: ESMSubMode -> Controller
+editHomeStandingsC = \case
+  ESMSubMenu          -> menuC editHomeStandingsMenu
+  ESMEditWins         -> promptC editHomeWinsPrompt
+  ESMEditLosses       -> promptC editHomeLossesPrompt
+  ESMEditOvertime     -> promptC editHomeOvertimePrompt
+  ESMEditGoalsFor     -> promptC editHomeGoalsForPrompt
+  ESMEditGoalsAgainst -> promptC editHomeGoalsAgainstPrompt
+
+editAwayStandingsC :: ESMSubMode -> Controller
+editAwayStandingsC = \case
+  ESMSubMenu          -> menuC editAwayStandingsMenu
+  ESMEditWins         -> promptC editAwayWinsPrompt
+  ESMEditLosses       -> promptC editAwayLossesPrompt
+  ESMEditOvertime     -> promptC editAwayOvertimePrompt
+  ESMEditGoalsFor     -> promptC editAwayGoalsForPrompt
+  ESMEditGoalsAgainst -> promptC editAwayGoalsAgainstPrompt
+
+menuC :: Menu () -> Controller
+menuC = menuControllerWith header
+
+promptC :: Prompt -> Controller
+promptC = promptControllerWith header
+
+header :: ProgState -> C.Update ()
+header = do
+  db <- (^.database)
+  let
+    home  = db^.dbHomeGameStats
+    away  = db^.dbAwayGameStats
+    table = numTable ["   W", "   L", "  OT", "  GF", "  GA"]
+      [ ( "HOME", valsFor home )
+      , ( "ROAD", valsFor away )
+      ]
+  return $ C.drawString $ unlines $ table ++ [""]
+
+valsFor :: GameStats -> [Int]
+valsFor gs =
+  [ gs^.gmsWins
+  , gs^.gmsLosses
+  , gs^.gmsOvertime
+  , gs^.gmsGoalsFor
+  , gs^.gmsGoalsAgainst
+  ]

src/Mtlstats/Control/NewGame.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Control/NewGame/GoalieInput.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Control/TitleScreen.hs

@@ -0,0 +1,142 @@
+{- |
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+{-# LANGUAGE LambdaCase #-}
+
+module Mtlstats.Control.TitleScreen (titleScreenC) where
+
+import Control.Monad.Trans.State (modify)
+import Data.Char (chr)
+import qualified UI.NCurses as C
+
+import Mtlstats.Actions
+import Mtlstats.Format
+import Mtlstats.Types
+
+titleScreenC :: Controller
+titleScreenC = Controller
+  { drawController = const $ do
+    (_, cols) <- C.windowSize
+    C.drawString $ unlines $ map (centre $ fromIntegral $ pred cols)
+      $  [ ""
+         , "MONTREAL CANADIENS STATISTICS"
+         ]
+      ++ titleText
+      ++ [ ""
+         , "Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe"
+         , "<rheal.lamothe@gmail.com>"
+         , ""
+         , "Press any key to continue..."
+         ]
+    return C.CursorInvisible
+  , handleController = \case
+    C.EventCharacter _  -> modify backHome >> return True
+    C.EventSpecialKey _ -> modify backHome >> return True
+    _                   -> return True
+  }
+
+titleText :: [String]
+titleText = box $ map (map blockify) $ foldl joinBlocks (repeat "")
+  [chM, chT, chL, chS, chT, chA, chT, chS]
+
+box :: [String] -> [String]
+box strs
+  =  [[tl] ++ replicate width horiz ++ [tr]]
+  ++ map (\str -> [vert] ++ str ++ [vert]) strs
+  ++ [[bl] ++ replicate width horiz ++ [br]]
+  where
+    width = length $ head strs
+    tl    = chr 0x2554
+    tr    = chr 0x2557
+    bl    = chr 0x255a
+    br    = chr 0x255d
+    horiz = chr 0x2550
+    vert  = chr 0x2551
+
+blockify :: Char -> Char
+blockify = \case
+  '#' -> chr 0x2588
+  '>' -> chr 0x2590
+  '<' -> chr 0x258c
+  ch  -> ch
+
+joinBlocks :: [String] -> [String] -> [String]
+joinBlocks = zipWith (++)
+
+chM :: [String]
+chM =
+  [ "##<   >##"
+  , ">##   ##<"
+  , ">##< >##<"
+  , ">### ###<"
+  , ">#######<"
+  , ">#<###>#<"
+  , ">#<>#<>#<"
+  , "##<   >##"
+  ]
+
+chT :: [String]
+chT =
+  [ ">########<"
+  , ">## ## ##<"
+  , ">#< ## >#<"
+  , "    ##    "
+  , "    ##    "
+  , "    ##    "
+  , "    ##    "
+  , "   >##<   "
+  ]
+
+chL :: [String]
+chL =
+  [ "###    "
+  , ">#<    "
+  , ">#<    "
+  , ">#<    "
+  , ">#<    "
+  , ">#<  ##"
+  , ">#< >##"
+  , "#######"
+  ]
+
+chS :: [String]
+chS =
+  [ " #####< "
+  , ">#<  ## "
+  , "##      "
+  , " #####< "
+  , "     >#<"
+  , "      ##"
+  , ">#<  >#<"
+  , " ###### "
+  ]
+
+chA :: [String]
+chA =
+  [ "  >##<  "
+  , "   ##   "
+  , "  >##<  "
+  , "  ####  "
+  , " >#<>#< "
+  , " ###### "
+  , ">#<  >#<"
+  , "###  ###"
+  ]

src/Mtlstats/Format.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Handlers.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Helpers/Goalie.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -19,7 +19,7 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 -}
 
-module Mtlstats.Helpers.Goalie (goalieDetails) where
+module Mtlstats.Helpers.Goalie (goalieDetails, goalieName) where
 
 import Lens.Micro ((^.))
 
@@ -31,7 +31,7 @@ goalieDetails :: Goalie -> String
 goalieDetails g = let
   header = unlines $ labelTable
     [ ( "Number", show $ g^.gNumber )
-    , ( "Name",   g^.gName          )
+    , ( "Name",   goalieName g      )
     ]
 
   body = unlines $ numTable ["YTD", "Lifetime"] $ map
@@ -39,9 +39,24 @@ goalieDetails g = let
     [ ( "Games played",  gsGames        )
     , ( "Mins played",   gsMinsPlayed   )
     , ( "Goals allowed", gsGoalsAllowed )
+    , ( "Shutouts",      gsShutouts     )
     , ( "Wins",          gsWins         )
     , ( "Losses",        gsLosses       )
     , ( "Ties",          gsTies         )
     ]
 
   in header ++ "\n" ++ body
+
+-- | Returns the goalie name, modified if they are a rookie
+goalieName :: Goalie -> String
+goalieName g = let
+
+  prefix = if g^.gActive
+    then ""
+    else "*"
+
+  suffix = if g^.gRookie
+    then "*"
+    else ""
+
+  in prefix ++ g^.gName ++ suffix

src/Mtlstats/Helpers/Player.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -19,7 +19,7 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 -}
 
-module Mtlstats.Helpers.Player (playerDetails) where
+module Mtlstats.Helpers.Player (playerDetails, playerName) where
 
 import Lens.Micro ((^.))
 
@@ -32,7 +32,7 @@ playerDetails p = unlines $ top ++ [""] ++ table
   where
     top = labelTable
       [ ( "Number",   show $ p^.pNumber )
-      , ( "Name",     p^.pName          )
+      , ( "Name",     playerName p      )
       , ( "Position", p^.pPosition      )
       ]
 
@@ -43,3 +43,18 @@ playerDetails p = unlines $ top ++ [""] ++ table
       , ( "Assists",      psAssists )
       , ( "Penalty mins", psPMin    )
       ]
+
+-- | Presents a modified version of the player's name indicating
+-- whether or not they're a rookie
+playerName :: Player -> String
+playerName p = let
+
+  prefix = if p^.pActive
+    then ""
+    else "*"
+
+  suffix = if p^.pRookie
+    then "*"
+    else ""
+
+  in prefix ++ p^.pName ++ suffix

src/Mtlstats/Menu.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -35,24 +35,18 @@ module Mtlstats.Menu (
   editMenu
 ) where
 
-import Control.Monad.IO.Class (liftIO)
 import Control.Monad.Trans.State (gets, modify)
-import Data.Aeson (encodeFile)
 import Data.Char (toUpper)
 import qualified Data.Map as M
 import Data.Maybe (mapMaybe)
 import Lens.Micro ((^.), (?~))
-import Lens.Micro.Extras (view)
-import System.EasyFile
-  ( createDirectoryIfMissing
-  , getAppUserDataDirectory
-  , (</>)
-  )
 import qualified UI.NCurses as C
 
 import Mtlstats.Actions
 import qualified Mtlstats.Actions.NewGame.GoalieInput as GI
+import Mtlstats.Actions.EditStandings
 import Mtlstats.Config
+import Mtlstats.Format
 import Mtlstats.Types
 import Mtlstats.Types.Menu
 import Mtlstats.Util
@@ -96,7 +90,11 @@ menuStateController menuFunc = Controller
 -- | The draw function for a 'Menu'
 drawMenu :: Menu a -> C.Update C.CursorMode
 drawMenu m = do
-  C.drawString $ show m
+  (_, cols) <- C.windowSize
+  let
+    width    = fromIntegral $ pred cols
+    menuText = map (centre width) $ lines $ show m
+  C.drawString $ unlines menuText
   return C.CursorInvisible
 
 -- | The event handler for a 'Menu'
@@ -109,61 +107,56 @@ menuHandler m _ = return $ m^.menuDefault
 
 -- | The main menu
 mainMenu :: Menu Bool
-mainMenu = Menu "*** MAIN MENU ***" True
-  [ MenuItem '1' "New Season" $
+mainMenu = Menu "MASTER MENU" True
+  [ MenuItem 'A' "NEW SEASON" $
     modify startNewSeason >> return True
-  , MenuItem '2' "New Game" $
+  , MenuItem 'B' "NEW GAME" $
     modify startNewGame >> return True
-  , MenuItem '3' "Edit" $
+  , MenuItem 'C' "EDIT MENU" $
     modify edit >> return True
-  , MenuItem 'X' "Exit" $ do
-    db <- gets $ view database
-    liftIO $ do
-      dir <- getAppUserDataDirectory appName
-      let dbFile = dir </> dbFname
-      createDirectoryIfMissing True dir
-      encodeFile dbFile db
-    return False
+  , MenuItem 'E' "EXIT" $
+    saveDatabase dbFname >> return False
   ]
 
 -- | The new season menu
 newSeasonMenu :: Menu ()
-newSeasonMenu = Menu "*** SEASON TYPE ***" ()
-  [ MenuItem 'R' "Regular Season" $ modify
+newSeasonMenu = Menu "SEASON TYPE" ()
+  [ MenuItem 'R' "REGULAR SEASON" $ modify
     $ resetYtd
+    . clearRookies
     . resetStandings
     . startNewGame
-  , MenuItem 'P' "Playoffs" $ modify
+  , MenuItem 'P' "PLAYOFFS" $ modify
     $ resetStandings
     . startNewGame
   ]
 
 -- | Requests the month in which the game took place
 gameMonthMenu :: Menu ()
-gameMonthMenu = Menu "Month:" () $ map
+gameMonthMenu = Menu "MONTH:" () $ map
   (\(ch, name, val) ->
     MenuItem ch name $
     modify $ progMode.gameStateL.gameMonth ?~ val)
-  [ ( 'A', "January",   1  )
-  , ( 'B', "February",  2  )
-  , ( 'C', "March",     3  )
-  , ( 'D', "April",     4  )
-  , ( 'E', "May",       5  )
-  , ( 'F', "June",      6  )
-  , ( 'G', "July",      7  )
-  , ( 'H', "August",    8  )
-  , ( 'I', "September", 9  )
-  , ( 'J', "October",   10 )
-  , ( 'K', "November",  11 )
-  , ( 'L', "December",  12 )
+  [ ( 'A', "JANUARY",   1  )
+  , ( 'B', "FEBRUARY",  2  )
+  , ( 'C', "MARCH",     3  )
+  , ( 'D', "APRIL",     4  )
+  , ( 'E', "MAY",       5  )
+  , ( 'F', "JUNE",      6  )
+  , ( 'G', "JULY",      7  )
+  , ( 'H', "AUGUST",    8  )
+  , ( 'I', "SEPTEMBER", 9  )
+  , ( 'J', "OCTOBER",   10 )
+  , ( 'K', "NOVEMBER",  11 )
+  , ( 'L', "DECEMBER",  12 )
   ]
 
 -- | The game type menu (home/away)
 gameTypeMenu :: Menu ()
-gameTypeMenu = Menu "Game type:" ()
-  [ MenuItem '1' "Home Game" $
+gameTypeMenu = Menu "GAME TYPE:" ()
+  [ MenuItem 'H' "HOME GAME" $
     modify $ progMode.gameStateL.gameType ?~ HomeGame
-  , MenuItem '2' "Away Game" $
+  , MenuItem 'A' "AWAY GAME" $
     modify $ progMode.gameStateL.gameType ?~ AwayGame
   ]
 
@@ -184,15 +177,17 @@ gameGoalieMenu s = let
 
 -- | The edit menu
 editMenu :: Menu ()
-editMenu = Menu "*** EDIT ***" ()
-  [ MenuItem '1' "Create Player" $
+editMenu = Menu "EDIT MENU" ()
+  [ MenuItem 'A' "CREATE PLAYER" $
     modify createPlayer
-  , MenuItem '2' "Create Goalie" $
+  , MenuItem 'B' "CREATE GOALIE" $
     modify createGoalie
-  , MenuItem '3' "Edit Player" $
+  , MenuItem 'C' "EDIT PLAYER" $
     modify editPlayer
-  , MenuItem '4' "Edit Goalie" $
+  , MenuItem 'D' "EDIT GOALIE" $
     modify editGoalie
-  , MenuItem 'R' "Return to Main Menu" $
+  , MenuItem 'E' "EDIT STANDINGS" $
+    modify editStandings
+  , MenuItem 'R' "RETURN TO MAIN MENU" $
     modify backHome
   ]

src/Mtlstats/Menu/EditGoalie.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -26,7 +26,7 @@ module Mtlstats.Menu.EditGoalie
   ) where
 
 import Control.Monad.Trans.State (modify)
-import Lens.Micro ((.~))
+import Lens.Micro ((.~), (%~))
 
 import Mtlstats.Actions
 import Mtlstats.Types
@@ -34,44 +34,53 @@ import Mtlstats.Types.Menu
 
 -- | The 'Goalie' edit menu
 editGoalieMenu :: Menu ()
-editGoalieMenu = Menu "*** EDIT GOALTENDER ***" () $ map
-  (\(ch, label, mode) -> MenuItem ch label $
-    modify $ case mode of
-      Nothing -> edit
-      Just m  -> progMode.editGoalieStateL.egsMode .~ m)
+editGoalieMenu = Menu "EDIT GOALTENDER" () $ map
+  (\(ch, label, action) -> MenuItem ch label $ modify action)
+
   --  key, label,                 value
-  [ ( '1', "Edit number",         Just EGNumber   )
-  , ( '2', "Edit name",           Just EGName     )
-  , ( '3', "Edit YTD stats",      Just EGYtd      )
-  , ( '4', "Edit Lifetime stats", Just EGLifetime )
-  , ( 'R', "Return to Edit Menu", Nothing         )
+  [ ( 'A', "EDIT NUMBER",         set EGNumber   )
+  , ( 'B', "EDIT NAME",           set EGName     )
+  , ( 'C', "TOGGLE ROOKIE FLAG",  toggleRookie   )
+  , ( 'D', "TOGGLE ACTIVE FLAG",  toggleActive   )
+  , ( 'E', "EDIT YTD STATS",      set EGYtd      )
+  , ( 'F', "EDIT LIFETIME STATS", set EGLifetime )
+  , ( 'R', "RETURN TO EDIT MENU", edit           )
   ]
 
+  where
+    set mode     = progMode.editGoalieStateL.egsMode .~ mode
+    toggleRookie = editSelectedGoalie (gRookie %~ not)
+    toggleActive = editSelectedGoalie (gActive %~ not)
+
 -- | The 'Goalie' YTD edit menu
 editGoalieYtdMenu :: Menu ()
-editGoalieYtdMenu = editMenu "*** EDIT GOALTENDER YEAR-TO-DATE ***"
+editGoalieYtdMenu = editMenu "EDIT GOALTENDER YEAR-TO-DATE"
   --  key, label,                 value
-  [ ( '1', "Edit YTD games",      EGYtdGames  )
-  , ( '2', "Edit YTD minutes",    EGYtdMins   )
-  , ( '3', "Edit YTD goals",      EGYtdGoals  )
-  , ( '4', "Edit YTD wins",       EGYtdWins   )
-  , ( '5', "Edit YTD losses",     EGYtdLosses )
-  , ( '6', "Edit YTD ties",       EGYtdTies   )
-  , ( 'R', "Return to edit menu", EGMenu      )
+  [ ( 'A', "EDIT ALL YTD STATS",  EGYtdGames    True  )
+  , ( 'B', "EDIT YTD GAMES",      EGYtdGames    False )
+  , ( 'C', "EDIT YTD MINUTES",    EGYtdMins     False )
+  , ( 'D', "EDIT YTD GOALS",      EGYtdGoals    False )
+  , ( 'E', "EDIT YTD SHUTOUTS",   EGYtdShutouts False )
+  , ( 'F', "EDIT YTD WINS",       EGYtdWins     False )
+  , ( 'G', "EDIT YTD LOSSES",     EGYtdLosses   False )
+  , ( 'H', "EDIT YTD TIES",       EGYtdTies           )
+  , ( 'R', "RETURN TO EDIT MENU", EGMenu              )
   ]
 
 -- | The 'Goalie' lifetime edit menu
 editGoalieLtMenu :: Menu ()
 editGoalieLtMenu = editMenu
-  "*** EDIT GOALTENDER LIFETIME ***"
+  "EDIT GOALTENDER LIFETIME"
   --  key, label,                   value
-  [ ( '1', "Edit lifetime games",   EGLtGames  )
-  , ( '2', "Edit lifetime minutes", EGLtMins   )
-  , ( '3', "Edit lifetime goals",   EGLtGoals  )
-  , ( '4', "Edit lifetime wins",    EGLtWins   )
-  , ( '5', "Edit lifetime losses",  EGLtLosses )
-  , ( '6', "Edit lifetime ties",    EGLtTies   )
-  , ( 'R', "Return to edit menu",   EGMenu     )
+  [ ( 'A', "EDIT ALL LIFETIME STATS", EGLtGames    True  )
+  , ( 'B', "EDIT LIFETIME GAMES",     EGLtGames    False )
+  , ( 'C', "EDIT LIFETIME MINUTES",   EGLtMins     False )
+  , ( 'D', "EDIT LIFETIME GOALS",     EGLtGoals    False )
+  , ( 'E', "EDIT LIFETIME SHUTOUTS",  EGLtShutouts False )
+  , ( 'F', "EDIT LIFETIME WINS",      EGLtWins     False )
+  , ( 'G', "EDIT LIFETIME LOSSES",    EGLtLosses   False )
+  , ( 'H', "EDIT LIFETIME TIES",      EGLtTies           )
+  , ( 'R', "RETURN TO EDIT MENU",     EGMenu             )
   ]
 
 editMenu :: String -> [(Char, String, EditGoalieMode)] -> Menu ()

src/Mtlstats/Menu/EditPlayer.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -26,7 +26,7 @@ module Mtlstats.Menu.EditPlayer
   ) where
 
 import Control.Monad.Trans.State (modify)
-import Lens.Micro ((.~))
+import Lens.Micro ((.~), (%~))
 
 import Mtlstats.Actions
 import Mtlstats.Types
@@ -34,40 +34,47 @@ import Mtlstats.Types.Menu
 
 -- | The 'Player' edit menu
 editPlayerMenu :: Menu ()
-editPlayerMenu = Menu "*** EDIT PLAYER ***" () $ map
-  (\(ch, label, mode) -> MenuItem ch label $
-    modify $ case mode of
-      Nothing -> edit
-      Just m  -> progMode.editPlayerStateL.epsMode .~ m)
+editPlayerMenu = Menu "EDIT PLAYER" () $ map
+  (\(ch, label, action) -> MenuItem ch label $ modify action)
+
   --  key, label,                 value
-  [ ( '1', "Edit number",         Just EPNumber   )
-  , ( '2', "Edit name",           Just EPName     )
-  , ( '3', "Edit position",       Just EPPosition )
-  , ( '4', "Edit YTD stats",      Just EPYtd      )
-  , ( '5', "Edit lifetime stats", Just EPLifetime )
-  , ( 'R', "Return to Edit Menu", Nothing         )
+  [ ( 'A', "EDIT NUMBER",         set EPNumber   )
+  , ( 'B', "EDIT NAME",           set EPName     )
+  , ( 'C', "EDIT POSITION",       set EPPosition )
+  , ( 'D', "TOGGLE ROOKIE FLAG",  toggleRookie   )
+  , ( 'E', "TOGGLE ACTIVE FLAG",  toggleActive   )
+  , ( 'F', "EDIT YTD STATS",      set EPYtd      )
+  , ( 'G', "EDIT LIFETIME STATS", set EPLifetime )
+  , ( 'R', "RETURN TO EDIT MENU", edit           )
   ]
 
+  where
+    set mode     = progMode.editPlayerStateL.epsMode .~ mode
+    toggleRookie = editSelectedPlayer $ pRookie %~ not
+    toggleActive = editSelectedPlayer $ pActive %~ not
+
 -- | The 'Player' YTD stats edit menu
 editPlayerYtdMenu :: Menu ()
 editPlayerYtdMenu = editMenu
-  "*** EDIT PLAYER YEAR-TO-DATE ***"
+  "EDIT PLAYER YEAR-TO-DATE"
   --  key, label,                        value
-  [ ( '1', "Edit YTD goals",             EPYtdGoals   )
-  , ( '2', "Edit YTD assists",           EPYtdAssists )
-  , ( '3', "Edit YTD penalty mins",      EPYtdPMin    )
-  , ( 'R', "Return to player edit menu", EPMenu       )
+  [ ( 'A', "EDIT ALL YTD STATS",         EPYtdGoals   True  )
+  , ( 'B', "EDIT YTD GOALS",             EPYtdGoals   False )
+  , ( 'C', "EDIT YTD ASSISTS",           EPYtdAssists False )
+  , ( 'D', "EDIT YTD PENALTY MINS",      EPYtdPMin          )
+  , ( 'R', "RETURN TO PLAYER EDIT MENU", EPMenu             )
   ]
 
 -- | The 'Player' lifetime stats edit menu
 editPlayerLtMenu :: Menu ()
 editPlayerLtMenu = editMenu
-  "*** EDIT PLAYER LIFETIME ***"
+  "EDIT PLAYER LIFETIME"
   --  key, label,                        value
-  [ ( '1', "Edit lifetime goals",        EPLtGoals   )
-  , ( '2', "Edit lifetime assits",       EPLtAssists )
-  , ( '3', "Edit lifetime penalty mins", EPLtPMin    )
-  , ( 'R', "Return to edit player menu", EPMenu      )
+  [ ( 'A', "EDIT ALL LIFETIME STATS",    EPLtGoals   True  )
+  , ( 'B', "EDIT LIFETIME GOALS",        EPLtGoals   False )
+  , ( 'C', "EDIT LIFETIME ASSITS",       EPLtAssists False )
+  , ( 'D', "EDIT LIFETIME PENALTY MINS", EPLtPMin          )
+  , ( 'R', "RETURN TO EDIT PLAYER MENU", EPMenu            )
   ]
 
 editMenu :: String -> [(Char, String, EditPlayerMode)] -> Menu ()

src/Mtlstats/Menu/EditStandings.hs

@@ -0,0 +1,64 @@
+{- |
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+module Mtlstats.Menu.EditStandings
+  ( editStandingsMenu
+  , editHomeStandingsMenu
+  , editAwayStandingsMenu
+  ) where
+
+import Control.Monad.Trans.State (modify)
+
+import Mtlstats.Actions
+import Mtlstats.Actions.EditStandings
+import Mtlstats.Types.Menu
+
+editStandingsMenu :: Menu ()
+editStandingsMenu = Menu "EDIT STANDINGS" ()
+  [ MenuItem 'A' "EDIT HOME STANDINGS" $
+    modify editHomeStandings
+  , MenuItem 'B' "EDIT ROAD STANDINGS" $
+    modify editAwayStandings
+  , MenuItem 'R' "RETURN TO MAIN MENU" $
+    modify backHome
+  ]
+
+editHomeStandingsMenu :: Menu ()
+editHomeStandingsMenu = subMenu "HOME"
+
+editAwayStandingsMenu :: Menu ()
+editAwayStandingsMenu = subMenu "ROAD"
+
+subMenu :: String -> Menu ()
+subMenu str = Menu (str ++ " STANDINGS") ()
+  [ MenuItem 'W' "EDIT WINS" $
+    modify editWins
+  , MenuItem 'L' "EDIT LOSSES" $
+    modify editLosses
+  , MenuItem 'O' "EDIT OVERTIME GAMES" $
+    modify editOvertime
+  , MenuItem 'F' "EDIT GOALS FOR" $
+    modify editGoalsFor
+  , MenuItem 'A' "EDIT GOALS AGAINST" $
+    modify editGoalsAgainst
+  , MenuItem 'R' "RETURN TO EDIT STANDINGS MENU" $
+    modify editStandings
+  ]

src/Mtlstats/Prompt.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -31,8 +31,10 @@ module Mtlstats.Prompt (
   ucStrPrompt,
   namePrompt,
   numPrompt,
+  numPromptWithFallback,
   selectPrompt,
   -- * Individual prompts
+  newSeasonPrompt,
   playerNumPrompt,
   playerNamePrompt,
   playerPosPrompt,
@@ -46,8 +48,7 @@ module Mtlstats.Prompt (
 import Control.Monad (when)
 import Control.Monad.Extra (whenJust)
 import Control.Monad.Trans.State (gets, modify)
-import Data.Char (isDigit, toUpper)
-import Data.Foldable (forM_)
+import Data.Char (isAlphaNum, isDigit, toUpper)
 import Lens.Micro ((^.), (&), (.~), (?~), (%~))
 import Lens.Micro.Extras (view)
 import Text.Read (readMaybe)
@@ -145,15 +146,44 @@ numPrompt
   -> (Int -> Action ())
   -- ^ The callback function for the result
   -> Prompt
-numPrompt pStr act = Prompt
+numPrompt pStr = numPromptWithFallback pStr $ return ()
+
+-- | Builds a numeric prompt with a fallback action
+numPromptWithFallback
+  :: String
+  -- ^ The prompt string
+  -> Action ()
+  -- ^ The action to call on invalid (or blank) input
+  -> (Int -> Action ())
+  -- ^ The callback function for the result
+  -> Prompt
+numPromptWithFallback pStr fallback act = Prompt
   { promptDrawer      = drawSimplePrompt pStr
   , promptProcessChar = \ch str -> if isDigit ch
     then str ++ [ch]
     else str
-  , promptAction      = \inStr -> forM_ (readMaybe inStr) act
+  , promptAction      = \inStr -> case readMaybe inStr of
+      Nothing -> fallback
+      Just n  -> act n
   , promptSpecialKey  = const $ return ()
   }
 
+-- | Prompts the user for a filename to save a backup of the database
+-- to
+newSeasonPrompt :: Prompt
+newSeasonPrompt = prompt
+  { promptProcessChar = \ch str -> if isAlphaNum ch
+    then str ++ [toUpper ch]
+    else str
+  }
+  where
+    prompt = strPrompt "Filename to save database: " $ \fn ->
+      if null fn
+      then modify backHome
+      else do
+        saveDatabase $ fn ++ ".json"
+        modify $ progMode .~ NewSeason True
+
 -- | Builds a selection prompt
 selectPrompt :: SelectParams a -> Prompt
 selectPrompt params = Prompt

src/Mtlstats/Prompt/EditGoalie.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -26,12 +26,14 @@ module Mtlstats.Prompt.EditGoalie
   , editGoalieYtdGamesPrompt
   , editGoalieYtdMinsPrompt
   , editGoalieYtdGoalsPrompt
+  , editGoalieYtdShutoutsPrompt
   , editGoalieYtdWinsPrompt
   , editGoalieYtdLossesPrompt
   , editGoalieYtdTiesPrompt
   , editGoalieLtGamesPrompt
   , editGoalieLtMinsPrompt
   , editGoalieLtGoalsPrompt
+  , editGoalieLtShutoutsPrompt
   , editGoalieLtWinsPrompt
   , editGoalieLtLossesPrompt
   , editGoalieLtTiesPrompt
@@ -40,7 +42,7 @@ module Mtlstats.Prompt.EditGoalie
 import Control.Monad.Trans.State (modify)
 import Lens.Micro ((.~))
 
-import Mtlstats.Actions.EditGoalie
+import Mtlstats.Actions
 import Mtlstats.Prompt
 import Mtlstats.Types
 
@@ -51,70 +53,171 @@ goalieToEditPrompt = selectGoaliePrompt "Goalie to edit: " $
 
 -- | Prompt to edit a goalie's number
 editGoalieNumberPrompt :: Prompt
-editGoalieNumberPrompt = numPrompt "Goalie number: " $
-  modify . editGoalieNumber
+editGoalieNumberPrompt = editNum "Goalie number: " EGMenu
+  (gNumber .~)
 
 -- | Prompt to edit a goalie's name
 editGoalieNamePrompt :: Prompt
-editGoalieNamePrompt = strPrompt "Goalie name: " $
-  modify . editGoalieName
+editGoalieNamePrompt = namePrompt "Goalie name: " $ \name ->
+  if null name
+  then goto EGMenu
+  else doEdit EGMenu $ gName .~ name
 
 -- | Prompt to edit a goalie's YTD games played
-editGoalieYtdGamesPrompt :: Prompt
-editGoalieYtdGamesPrompt = numPrompt "Year-to-date games played: " $
-  modify . editGoalieYtdGames
+editGoalieYtdGamesPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieYtdGamesPrompt batchMode =
+  editNum "Year-to-date games played: " mode
+  (gYtd.gsGames .~)
+  where
+    mode = if batchMode then EGYtdMins True else EGYtd
 
 -- | Prompt to edit a goalie's YTD minutes played
-editGoalieYtdMinsPrompt :: Prompt
-editGoalieYtdMinsPrompt = numPrompt "Year-to-date minutes played: " $
-  modify . editGoalieYtdMins
+editGoalieYtdMinsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieYtdMinsPrompt batchMode =
+  editNum "Year-to-date minutes played: " mode
+  (gYtd.gsMinsPlayed .~)
+  where
+    mode = if batchMode then EGYtdGoals True else EGYtd
 
 -- | Prompt to edit a goalie's YTD goales allowed
-editGoalieYtdGoalsPrompt :: Prompt
-editGoalieYtdGoalsPrompt = numPrompt "Year-to-date goals allowed: " $
-  modify . editGoalieYtdGoals
+editGoalieYtdGoalsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieYtdGoalsPrompt batchMode =
+  editNum "Year-to-date goals allowed: " mode
+  (gYtd.gsGoalsAllowed .~)
+  where
+    mode = if batchMode then EGYtdShutouts True else EGYtd
+
+-- | Prompt to edit a goalie's YTD shutouts
+editGoalieYtdShutoutsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieYtdShutoutsPrompt batchMode =
+  editNum "Year-to-date shutouts: " mode
+  (gYtd.gsShutouts .~)
+  where
+    mode = if batchMode then EGYtdWins True else EGYtd
 
 -- | Prompt to edit a goalie's YTD wins
-editGoalieYtdWinsPrompt :: Prompt
-editGoalieYtdWinsPrompt = numPrompt "Year-to-date wins: " $
-  modify . editGoalieYtdWins
+editGoalieYtdWinsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieYtdWinsPrompt batchMode =
+  editNum "Year-to-date wins: " mode
+  (gYtd.gsWins .~)
+  where
+    mode = if batchMode then EGYtdLosses True else EGYtd
 
 -- | Prompt to edit a goalie's YTD losses
-editGoalieYtdLossesPrompt :: Prompt
-editGoalieYtdLossesPrompt = numPrompt "Year-to-date losses: " $
-  modify . editGoalieYtdLosses
+editGoalieYtdLossesPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieYtdLossesPrompt batchMode =
+  editNum "Year-to-date losses: " mode
+  (gYtd.gsLosses .~)
+  where
+    mode = if batchMode then EGYtdTies else EGYtd
 
 -- | Prompt to edit a goalie's YTD ties
 editGoalieYtdTiesPrompt :: Prompt
-editGoalieYtdTiesPrompt = numPrompt "Year-to-date ties: " $
-  modify . editGoalieYtdTies
+editGoalieYtdTiesPrompt = editNum "Year-to-date ties: " EGYtd
+  (gYtd.gsTies .~)
 
 -- | Prompt to edit a goalie's lifetime games played
-editGoalieLtGamesPrompt :: Prompt
-editGoalieLtGamesPrompt = numPrompt "Lifetime games played: " $
-  modify . editGoalieLtGames
+editGoalieLtGamesPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieLtGamesPrompt batchMode =
+  editNum "Lifetime games played: " mode
+  (gLifetime.gsGames .~)
+  where
+    mode = if batchMode then EGLtMins True else EGLifetime
 
 -- | Prompt to edit a goalie's lifetime minutes played
-editGoalieLtMinsPrompt :: Prompt
-editGoalieLtMinsPrompt = numPrompt "Lifetime minutes played: " $
-  modify . editGoalieLtMins
+editGoalieLtMinsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieLtMinsPrompt batchMode =
+  editNum "Lifetime minutes played: " mode
+  (gLifetime.gsMinsPlayed .~)
+  where
+    mode = if batchMode then EGLtGoals True else EGLifetime
 
 -- | Prompt to edit a goalie's lifetime goals allowed
-editGoalieLtGoalsPrompt :: Prompt
-editGoalieLtGoalsPrompt = numPrompt "Lifetime goals allowed: " $
-  modify . editGoalieLtGoals
+editGoalieLtGoalsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieLtGoalsPrompt batchMode =
+  editNum "Lifetime goals allowed: " mode
+  (gLifetime.gsGoalsAllowed .~)
+  where
+    mode = if batchMode then EGLtShutouts True else EGLifetime
+
+-- | Prompt to edit a goalie's lifetime shutouts
+editGoalieLtShutoutsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieLtShutoutsPrompt batchMode =
+  editNum "Lifetime shutouts: " mode
+  (gLifetime.gsShutouts .~)
+  where
+    mode = if batchMode then EGLtWins True else EGLifetime
 
 -- | Prompt to edit a goalie's lifetime wins
-editGoalieLtWinsPrompt :: Prompt
-editGoalieLtWinsPrompt = numPrompt "Lifetime wins: " $
-  modify . editGoalieLtWins
+editGoalieLtWinsPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieLtWinsPrompt batchMode =
+  editNum "Lifetime wins: " mode
+  (gLifetime.gsWins .~)
+  where
+    mode = if batchMode then EGLtLosses True else EGLifetime
 
 -- | Prompt to edit a goalie's lifetime losses
-editGoalieLtLossesPrompt :: Prompt
-editGoalieLtLossesPrompt = numPrompt "Lifetime losses: " $
-  modify . editGoalieLtLosses
+editGoalieLtLossesPrompt
+  :: Bool
+  -- ^ Indicates whether or not we're in batch mode
+  -> Prompt
+editGoalieLtLossesPrompt batchMode =
+  editNum "Lifetime losses: " mode
+  (gLifetime.gsLosses .~)
+  where
+    mode = if batchMode then EGLtTies else EGLifetime
 
 -- | Prompt to edit a goalie's lifetime ties
 editGoalieLtTiesPrompt :: Prompt
-editGoalieLtTiesPrompt = numPrompt "Lifetime ties: " $
-  modify . editGoalieLtTies
+editGoalieLtTiesPrompt = editNum "Lifetime ties: " EGLifetime
+  (gLifetime.gsTies .~)
+
+editNum
+  :: String
+  -> EditGoalieMode
+  -> (Int -> Goalie -> Goalie)
+  -> Prompt
+editNum pStr mode f = numPromptWithFallback pStr
+  (goto mode)
+  (doEdit mode . f)
+
+doEdit :: EditGoalieMode -> (Goalie -> Goalie) -> Action ()
+doEdit mode f = do
+  modify $ editSelectedGoalie f
+  goto mode
+
+goto :: EditGoalieMode -> Action ()
+goto = modify . (progMode.editGoalieStateL.egsMode .~)

src/Mtlstats/Prompt/EditPlayer.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -31,62 +31,95 @@ module Mtlstats.Prompt.EditPlayer
   , editPlayerLtPMinPrompt
   ) where
 
-import Control.Monad.Extra (whenJustM)
-import Control.Monad.Trans.State (gets, modify)
-import Lens.Micro ((^.), (.~), (%~))
+import Control.Monad.Trans.State (modify)
+import Lens.Micro ((.~))
 
+import Mtlstats.Actions
 import Mtlstats.Prompt
 import Mtlstats.Types
-import Mtlstats.Util
 
 -- | Prompt to edit a player's number
 editPlayerNumPrompt :: Prompt
-editPlayerNumPrompt = numPrompt "Player number: " $
-  editPlayer . (pNumber .~)
+editPlayerNumPrompt = editNum "Player number: " EPMenu
+  (pNumber .~)
 
 -- | Prompt to edit a player's name
 editPlayerNamePrompt :: Prompt
-editPlayerNamePrompt = strPrompt "Player name: " $
-  editPlayer . (pName .~)
+editPlayerNamePrompt = namePrompt "Player name: " $ \name ->
+  if null name
+  then goto EPMenu
+  else doEdit EPMenu $ pName .~ name
 
 -- | Prompt to edit a player's position
 editPlayerPosPrompt :: Prompt
-editPlayerPosPrompt = ucStrPrompt "Player position: " $
-  editPlayer . (pPosition .~)
+editPlayerPosPrompt = ucStrPrompt "Player position: " $ \pos ->
+  if null pos
+  then goto EPMenu
+  else doEdit EPMenu $ pPosition .~ pos
 
 -- | Prompt to edit a player's year-to-date goals
-editPlayerYtdGoalsPrompt :: Prompt
-editPlayerYtdGoalsPrompt = numPrompt "Year-to-date goals: " $
-  editPlayer . (pYtd.psGoals .~)
+editPlayerYtdGoalsPrompt
+  :: Bool
+  -- ^ Indicates wheter or not we're editing in batch mode
+  -> Prompt
+editPlayerYtdGoalsPrompt batchMode = editNum "Year-to-date goals: " mode
+  (pYtd.psGoals .~)
+  where
+    mode = if batchMode then EPYtdAssists True else EPYtd
 
 -- | Prompt to edit a player's year-to-date assists
-editPlayerYtdAssistsPrompt :: Prompt
-editPlayerYtdAssistsPrompt = numPrompt "Year-to-date assists: " $
-  editPlayer . (pYtd.psAssists .~)
+editPlayerYtdAssistsPrompt
+  :: Bool
+  -- ^ Indicates wheter or not we're editing in batch mode
+  -> Prompt
+editPlayerYtdAssistsPrompt batchMode = editNum "Year-to-date assists: " mode
+  (pYtd.psAssists .~)
+  where
+    mode = if batchMode then EPYtdPMin else EPYtd
 
 -- | Prompt to edit a player's year-to-date penalty minutes
 editPlayerYtdPMinPrompt :: Prompt
-editPlayerYtdPMinPrompt = numPrompt "Year-to-date penalty minutes: " $
-  editPlayer . (pYtd.psPMin .~)
+editPlayerYtdPMinPrompt = editNum "Year-to-date penalty minutes: " EPYtd
+  (pYtd.psPMin .~)
 
 -- | Prompt to edit a player's lifetime goals
-editPlayerLtGoalsPrompt :: Prompt
-editPlayerLtGoalsPrompt = numPrompt "Lifetime goals: " $
-  editPlayer . (pLifetime.psGoals .~)
+editPlayerLtGoalsPrompt
+  :: Bool
+  -- ^ Indicates wheter or not we're editing in batch mode
+  -> Prompt
+editPlayerLtGoalsPrompt batchMode = editNum "Lifetime goals: " mode
+  (pLifetime.psGoals .~)
+  where
+    mode = if batchMode then EPLtAssists True else EPLifetime
 
 -- | Prompt to edit a player's lifetime assists
-editPlayerLtAssistsPrompt :: Prompt
-editPlayerLtAssistsPrompt = numPrompt "Lifetime assists: " $
-  editPlayer . (pLifetime.psAssists .~)
+editPlayerLtAssistsPrompt
+  :: Bool
+  -- ^ Indicates wheter or not we're editing in batch mode
+  -> Prompt
+editPlayerLtAssistsPrompt batchMode = editNum "Lifetime assists: " mode
+  (pLifetime.psAssists .~)
+  where
+    mode = if batchMode then EPLtPMin else EPLifetime
 
 -- | Prompt to edit a player's lifetime penalty minutes
 editPlayerLtPMinPrompt :: Prompt
-editPlayerLtPMinPrompt = numPrompt "Lifetime penalty minutes: " $
-  editPlayer . (pLifetime.psPMin .~)
+editPlayerLtPMinPrompt = editNum "Lifetime penalty minutes: " EPLifetime
+  (pLifetime.psPMin .~)
 
-editPlayer :: (Player -> Player) -> Action ()
-editPlayer f =
-  whenJustM (gets (^.progMode.editPlayerStateL.epsSelectedPlayer)) $ \pid ->
-    modify
-      $ (database.dbPlayers %~ modifyNth pid f)
-      . (progMode.editPlayerStateL.epsMode .~ EPMenu)
+editNum
+  :: String
+  -> EditPlayerMode
+  -> (Int -> Player -> Player)
+  -> Prompt
+editNum pStr mode f = numPromptWithFallback pStr
+  (goto mode)
+  (doEdit mode . f)
+
+doEdit :: EditPlayerMode -> (Player -> Player) -> Action ()
+doEdit mode f = do
+  modify $ editSelectedPlayer f
+  goto mode
+
+goto :: EditPlayerMode -> Action ()
+goto = modify . (progMode.editPlayerStateL.epsMode .~)

src/Mtlstats/Prompt/EditStandings.hs

@@ -0,0 +1,89 @@
+{- |
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+module Mtlstats.Prompt.EditStandings
+  ( editHomeWinsPrompt
+  , editHomeLossesPrompt
+  , editHomeOvertimePrompt
+  , editHomeGoalsForPrompt
+  , editHomeGoalsAgainstPrompt
+  , editAwayWinsPrompt
+  , editAwayLossesPrompt
+  , editAwayOvertimePrompt
+  , editAwayGoalsForPrompt
+  , editAwayGoalsAgainstPrompt
+  ) where
+
+import Control.Monad.Trans.State (modify)
+import Lens.Micro ((.~), (%~))
+
+import Mtlstats.Prompt
+import Mtlstats.Types
+
+editHomeWinsPrompt :: Prompt
+editHomeWinsPrompt =
+  mkPrompt "Home wins: " (dbHomeGameStats.gmsWins .~)
+
+editHomeLossesPrompt :: Prompt
+editHomeLossesPrompt =
+  mkPrompt "Home losses: " (dbHomeGameStats.gmsLosses .~)
+
+editHomeOvertimePrompt :: Prompt
+editHomeOvertimePrompt =
+  mkPrompt "Home overtime games: " (dbHomeGameStats.gmsOvertime .~)
+
+editHomeGoalsForPrompt :: Prompt
+editHomeGoalsForPrompt =
+  mkPrompt "Home goals for: " (dbHomeGameStats.gmsGoalsFor .~)
+
+editHomeGoalsAgainstPrompt :: Prompt
+editHomeGoalsAgainstPrompt =
+  mkPrompt "Home goals against: " (dbHomeGameStats.gmsGoalsAgainst .~)
+
+editAwayWinsPrompt :: Prompt
+editAwayWinsPrompt =
+  mkPrompt "Road wins: " (dbAwayGameStats.gmsWins .~)
+
+editAwayLossesPrompt :: Prompt
+editAwayLossesPrompt =
+  mkPrompt "Road losses: " (dbAwayGameStats.gmsLosses .~)
+
+editAwayOvertimePrompt :: Prompt
+editAwayOvertimePrompt =
+  mkPrompt "Road overtime games: " (dbAwayGameStats.gmsOvertime .~)
+
+editAwayGoalsForPrompt :: Prompt
+editAwayGoalsForPrompt =
+  mkPrompt "Road goals for: " (dbAwayGameStats.gmsGoalsFor .~)
+
+editAwayGoalsAgainstPrompt :: Prompt
+editAwayGoalsAgainstPrompt =
+  mkPrompt "Road goals against: " (dbAwayGameStats.gmsGoalsAgainst .~)
+
+mkPrompt :: String -> (Int -> Database -> Database) -> Prompt
+mkPrompt pStr f = numPromptWithFallback pStr
+  (modify subMenu)
+  (\n -> modify
+    $ (database %~ f n)
+    . subMenu)
+
+subMenu :: ProgState -> ProgState
+subMenu = progMode.editStandingsModeL.esmSubModeL .~ ESMSubMenu

src/Mtlstats/Prompt/NewGame.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Prompt/NewGame/GoalieInput.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

src/Mtlstats/Report.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -29,6 +29,8 @@ import Lens.Micro ((^.))
 
 import Mtlstats.Config
 import Mtlstats.Format
+import Mtlstats.Helpers.Goalie
+import Mtlstats.Helpers.Player
 import Mtlstats.Types
 import Mtlstats.Util
 
@@ -129,7 +131,7 @@ gameStatsReport width s = let
 
   criteria (_, ps) = psPoints ps > 0
 
-  in filteredPlayerReport width "GAME" criteria True playerStats
+  in filteredPlayerReport width "GAME" criteria True False playerStats
   ++ [""]
   ++ gameGoalieReport width goalieStats
 
@@ -146,9 +148,9 @@ yearToDateStatsReport width s = let
     $ filter goalieIsActive
     $ db^.dbGoalies
 
-  in playerReport width "YEAR TO DATE" True playerStats
+  in playerReport width "YEAR TO DATE" True False playerStats
   ++ [""]
-  ++ goalieReport width True goalieStats
+  ++ goalieReport width True False goalieStats
 
 lifetimeStatsReport :: Int -> ProgState -> [String]
 lifetimeStatsReport width s = let
@@ -161,9 +163,9 @@ lifetimeStatsReport width s = let
   goalieStats = map (\g -> (g, g^.gLifetime))
     $ db^.dbGoalies
 
-  in playerReport width "LIFETIME" False playerStats
+  in playerReport width "LIFETIME" False True playerStats
   ++ [""]
-  ++ goalieReport width False goalieStats
+  ++ goalieReport width False True goalieStats
 
 gameDate :: GameState -> String
 gameDate gs = fromMaybe "" $ do
@@ -176,6 +178,7 @@ playerReport
   :: Int
   -> String
   -> Bool
+  -> Bool
   -> [(Player, PlayerStats)]
   -> [String]
 playerReport width label =
@@ -186,11 +189,13 @@ filteredPlayerReport
   -> String
   -> ((Player, PlayerStats) -> Bool)
   -> Bool
+  -> Bool
   -> [(Player, PlayerStats)]
   -> [String]
-filteredPlayerReport width label criteria showTotals ps = let
-  tStats = foldl addPlayerStats newPlayerStats $ map snd ps
-  fps    = filter criteria ps
+filteredPlayerReport width label criteria showTotals lineNumbers ps = let
+  tStats    = foldl addPlayerStats newPlayerStats $ map snd ps
+  criteria' = (&&) <$> criteria <*> \(p, _) -> p^.pNumber /= 0
+  fps       = filter criteria' ps
 
   rHeader =
     [ centre width (label ++ " STATISTICS")
@@ -216,7 +221,7 @@ filteredPlayerReport width label criteria showTotals ps = let
   body = map
     (\(p, stats) ->
       [ CellText $ show (p^.pNumber) ++ " "
-      , CellText $ p^.pName
+      , CellText $ playerName p
       ] ++ statsCells stats)
     fps
 
@@ -231,8 +236,13 @@ filteredPlayerReport width label criteria showTotals ps = let
     then label ++ " TOTALS"
     else ""
 
+  lnOverlay = if lineNumbers
+    then "" : [right 2 $ show x | x <- [(1 :: Int)..]]
+    else repeat ""
+
   table = overlayLast olayText
-    $ map (centre width)
+    $ map (\(ln, line) -> overlay ln $ centre width line)
+    $ zip lnOverlay
     $ complexTable ([right, left] ++ repeat right)
     $ tHeader : body ++ if showTotals
       then [separator, totals]
@@ -243,9 +253,10 @@ filteredPlayerReport width label criteria showTotals ps = let
 goalieReport
   :: Int
   -> Bool
+  -> Bool
   -> [(Goalie, GoalieStats)]
   -> [String]
-goalieReport width showTotals goalieData = let
+goalieReport width showTotals lineNumbers goalieData = let
   olayText = if showTotals
     then "GOALTENDING TOTALS"
     else ""
@@ -274,7 +285,7 @@ goalieReport width showTotals goalieData = let
   body = map
     (\(goalie, stats) ->
       [ CellText $ show (goalie^.gNumber) ++ " "
-      , CellText $ goalie^.gName
+      , CellText $ goalieName goalie
       ] ++ rowCells stats)
     goalieData
 
@@ -284,7 +295,12 @@ goalieReport width showTotals goalieData = let
 
   summary = replicate 2 (CellText  "") ++ rowCells tData
 
-  in map (centre width)
+  lnOverlay = if lineNumbers
+    then "" : [right 2 $ show x | x <- [(1 :: Int)..]]
+    else repeat ""
+
+  in map (\(ln, line) -> overlay ln $ centre width line)
+    $ zip lnOverlay
     $ overlayLast olayText
     $ complexTable ([right, left] ++ repeat right)
     $ header : body ++ if showTotals
@@ -304,7 +320,7 @@ gameGoalieReport width goalieData = let
   body = map
     (\(goalie, stats) ->
       [ CellText $ show (goalie^.gNumber) ++ " "
-      , CellText $ goalie^.gName
+      , CellText $ goalieName goalie
       , CellText $ show $ stats^.gsMinsPlayed
       , CellText $ show $ stats^.gsGoalsAllowed
       , CellText $ showFloating $ gsAverage stats

src/Mtlstats/Types.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -35,6 +35,8 @@ module Mtlstats.Types (
   EditPlayerMode (..),
   EditGoalieState (..),
   EditGoalieMode (..),
+  EditStandingsMode (..),
+  ESMSubMode (..),
   Database (..),
   Player (..),
   PlayerStats (..),
@@ -56,6 +58,9 @@ module Mtlstats.Types (
   createGoalieStateL,
   editPlayerStateL,
   editGoalieStateL,
+  editStandingsModeL,
+  -- ** EditStandingsMode Lenses
+  esmSubModeL,
   -- ** GameState Lenses
   gameYear,
   gameMonth,
@@ -106,6 +111,8 @@ module Mtlstats.Types (
   pNumber,
   pName,
   pPosition,
+  pRookie,
+  pActive,
   pYtd,
   pLifetime,
   -- ** PlayerStats Lenses
@@ -115,6 +122,8 @@ module Mtlstats.Types (
   -- ** Goalie Lenses
   gNumber,
   gName,
+  gRookie,
+  gActive,
   gYtd,
   gLifetime,
   -- ** GoalieStats Lenses
@@ -227,24 +236,28 @@ data ProgState = ProgState
 
 -- | The program mode
 data ProgMode
-  = MainMenu
-  | NewSeason
+  = TitleScreen
+  | MainMenu
+  | NewSeason Bool
   | NewGame GameState
   | EditMenu
   | CreatePlayer CreatePlayerState
   | CreateGoalie CreateGoalieState
   | EditPlayer EditPlayerState
   | EditGoalie EditGoalieState
+  | EditStandings EditStandingsMode
 
 instance Show ProgMode where
-  show MainMenu         = "MainMenu"
-  show NewSeason        = "NewSeason"
-  show (NewGame _)      = "NewGame"
-  show EditMenu         = "EditMenu"
-  show (CreatePlayer _) = "CreatePlayer"
-  show (CreateGoalie _) = "CreateGoalie"
-  show (EditPlayer _)   = "EditPlayer"
-  show (EditGoalie _)   = "EditGoalie"
+  show TitleScreen       = "TitleScreen"
+  show MainMenu          = "MainMenu"
+  show (NewSeason _)     = "NewSeason"
+  show (NewGame _)       = "NewGame"
+  show EditMenu          = "EditMenu"
+  show (CreatePlayer _)  = "CreatePlayer"
+  show (CreateGoalie _)  = "CreateGoalie"
+  show (EditPlayer _)    = "EditPlayer"
+  show (EditGoalie _)    = "EditGoalie"
+  show (EditStandings _) = "EditStandings"
 
 -- | The game state
 data GameState = GameState
@@ -347,11 +360,11 @@ data EditPlayerMode
   | EPPosition
   | EPYtd
   | EPLifetime
-  | EPYtdGoals
-  | EPYtdAssists
+  | EPYtdGoals Bool
+  | EPYtdAssists Bool
   | EPYtdPMin
-  | EPLtGoals
-  | EPLtAssists
+  | EPLtGoals Bool
+  | EPLtAssists Bool
   | EPLtPMin
   deriving (Eq, Show)
 
@@ -369,20 +382,39 @@ data EditGoalieMode
   | EGName
   | EGYtd
   | EGLifetime
-  | EGYtdGames
-  | EGYtdMins
-  | EGYtdGoals
-  | EGYtdWins
-  | EGYtdLosses
+  | EGYtdGames    Bool
+  | EGYtdMins     Bool
+  | EGYtdGoals    Bool
+  | EGYtdShutouts Bool
+  | EGYtdWins     Bool
+  | EGYtdLosses   Bool
   | EGYtdTies
-  | EGLtGames
-  | EGLtMins
-  | EGLtGoals
-  | EGLtWins
-  | EGLtLosses
+  | EGLtGames     Bool
+  | EGLtMins      Bool
+  | EGLtGoals     Bool
+  | EGLtShutouts  Bool
+  | EGLtWins      Bool
+  | EGLtLosses    Bool
   | EGLtTies
   deriving (Eq, Show)
 
+-- | Represents the standings edit mode
+data EditStandingsMode
+  = ESMMenu
+  | ESMHome ESMSubMode
+  | ESMAway ESMSubMode
+  deriving (Eq, Show)
+
+-- | Represents the standings edit sub-mode
+data ESMSubMode
+  = ESMSubMenu
+  | ESMEditWins
+  | ESMEditLosses
+  | ESMEditOvertime
+  | ESMEditGoalsFor
+  | ESMEditGoalsAgainst
+  deriving (Eq, Show)
+
 -- | Represents the database
 data Database = Database
   { _dbPlayers       :: [Player]
@@ -397,29 +429,6 @@ data Database = Database
   -- ^ Statistics for away games
   } deriving (Eq, Show)
 
-instance FromJSON Database where
-  parseJSON = withObject "Database" $ \v -> Database
-    <$> v .: "players"
-    <*> v .: "goalies"
-    <*> v .: "games"
-    <*> v .: "home_game_stats"
-    <*> v .: "away_game_stats"
-
-instance ToJSON Database where
-  toJSON (Database players goalies games hgs ags) = object
-    [ "players"         .= players
-    , "goalies"         .= goalies
-    , "games"           .= games
-    , "home_game_stats" .= hgs
-    , "away_game_stats" .= ags
-    ]
-  toEncoding (Database players goalies games hgs ags) = pairs $
-    "players"         .= players <>
-    "goalies"         .= goalies <>
-    "games"           .= games   <>
-    "home_game_stats" .= hgs     <>
-    "away_game_stats" .= ags
-
 -- | Represents a (non-goalie) player
 data Player = Player
   { _pNumber   :: Int
@@ -428,35 +437,16 @@ data Player = Player
   -- ^ The player's name
   , _pPosition :: String
   -- ^ The player's position
+  , _pRookie   :: Bool
+  -- ^ Indicates that the player is a rookie
+  , _pActive   :: Bool
+  -- ^ Indicates that the player is active
   , _pYtd      :: PlayerStats
   -- ^ The Player's year-to-date stats
   , _pLifetime :: PlayerStats
   -- ^ The player's lifetime stats
   } deriving (Eq, Show)
 
-instance FromJSON Player where
-  parseJSON = withObject "Player" $ \v -> Player
-    <$> v .: "number"
-    <*> v .: "name"
-    <*> v .: "position"
-    <*> v .: "ytd"
-    <*> v .: "lifetime"
-
-instance ToJSON Player where
-  toJSON (Player num name pos ytd lt) = object
-    [ "number"   .= num
-    , "name"     .= name
-    , "position" .= pos
-    , "ytd"      .= ytd
-    , "lifetime" .= lt
-    ]
-  toEncoding (Player num name pos ytd lt) = pairs $
-    "number"   .= num  <>
-    "name"     .= name <>
-    "position" .= pos  <>
-    "ytd"      .= ytd  <>
-    "lifetime" .= lt
-
 -- | Represents a (non-goalie) player's stats
 data PlayerStats = PlayerStats
   { _psGoals   :: Int
@@ -467,55 +457,22 @@ data PlayerStats = PlayerStats
   -- ^ The number of penalty minutes
   } deriving (Eq, Show)
 
-instance FromJSON PlayerStats where
-  parseJSON = withObject "PlayerStats" $ \v -> PlayerStats
-    <$> v .: "goals"
-    <*> v .: "assists"
-    <*> v .: "penalty_mins"
-
-instance ToJSON PlayerStats where
-  toJSON (PlayerStats g a pm) = object
-    [ "goals"        .= g
-    , "assists"      .= a
-    , "penalty_mins" .= pm
-    ]
-  toEncoding (PlayerStats g a pm) = pairs $
-    "goals"        .= g  <>
-    "assists"      .= a  <>
-    "penalty_mins" .= pm
-
 -- | Represents a goalie
 data Goalie = Goalie
   { _gNumber   :: Int
   -- ^ The goalie's number
   , _gName     :: String
   -- ^ The goalie's name
+  , _gRookie   :: Bool
+  -- ^ Indicates that the goalie is a rookie
+  , _gActive   :: Bool
+  -- ^ Indicates that the goalie is active
   , _gYtd      :: GoalieStats
   -- ^ The goalie's year-to-date stats
   , _gLifetime :: GoalieStats
   -- ^ The goalie's lifetime stats
   } deriving (Eq, Show)
 
-instance FromJSON Goalie where
-  parseJSON = withObject "Goalie" $ \v -> Goalie
-    <$> v .: "number"
-    <*> v .: "name"
-    <*> v .: "ytd"
-    <*> v .: "lifetime"
-
-instance ToJSON Goalie where
-  toJSON (Goalie num name ytd lt) = object
-    [ "number"   .= num
-    , "name"     .= name
-    , "ytd"      .= ytd
-    , "lifetime" .= lt
-    ]
-  toEncoding (Goalie num name ytd lt) = pairs $
-    "number"   .= num  <>
-    "name"     .= name <>
-    "ytd"      .= ytd  <>
-    "lifetime" .= lt
-
 -- | Represents a goalie's stats
 data GoalieStats = GoalieStats
   { _gsGames        :: Int
@@ -534,35 +491,6 @@ data GoalieStats = GoalieStats
   -- ^ The number of ties
   } deriving (Eq, Show)
 
-instance FromJSON GoalieStats where
-  parseJSON = withObject "GoalieStats" $ \v -> GoalieStats
-    <$> v .:? "games"         .!= 0
-    <*> v .:? "mins_played"   .!= 0
-    <*> v .:? "goals_allowed" .!= 0
-    <*> v .:? "shutouts"      .!= 0
-    <*> v .:? "wins"          .!= 0
-    <*> v .:? "losses"        .!= 0
-    <*> v .:? "ties"          .!= 0
-
-instance ToJSON GoalieStats where
-  toJSON (GoalieStats g m a s w l t) = object
-    [ "games"         .= g
-    , "mins_played"   .= m
-    , "goals_allowed" .= a
-    , "shutouts"      .= s
-    , "wins"          .= w
-    , "losses"        .= l
-    , "ties"          .= t
-    ]
-  toEncoding (GoalieStats g m a s w l t) = pairs $
-      "games"         .= g <>
-      "mins_played"   .= m <>
-      "goals_allowed" .= a <>
-      "shutouts"      .= s <>
-      "wins"          .= w <>
-      "losses"        .= l <>
-      "ties"          .= t
-
 -- | Game statistics
 data GameStats = GameStats
   { _gmsWins         :: Int
@@ -577,29 +505,6 @@ data GameStats = GameStats
   -- ^ Goals against the team
   } deriving (Eq, Show)
 
-instance FromJSON GameStats where
-  parseJSON = withObject "GameStats" $ \v -> GameStats
-    <$> v .: "wins"
-    <*> v .: "losses"
-    <*> v .: "overtime"
-    <*> v .: "goals_for"
-    <*> v .: "goals_against"
-
-instance ToJSON GameStats where
-  toJSON (GameStats w l ot gf ga) = object
-    [ "wins"          .= w
-    , "losses"        .= l
-    , "overtime"      .= ot
-    , "goals_for"     .= gf
-    , "goals_against" .= ga
-    ]
-  toEncoding (GameStats w l ot gf ga) = pairs $
-    "wins"          .= w  <>
-    "losses"        .= l  <>
-    "overtime"      .= ot <>
-    "goals_for"     .= gf <>
-    "goals_against" .= ga
-
 -- | Defines a user prompt
 data Prompt = Prompt
   { promptDrawer      :: ProgState -> C.Update ()
@@ -653,6 +558,153 @@ makeLenses ''Goalie
 makeLenses ''GoalieStats
 makeLenses ''GameStats
 
+instance FromJSON Database where
+  parseJSON = withObject "Database" $ \v -> Database
+    <$> v .:? "players"         .!= []
+    <*> v .:? "goalies"         .!= []
+    <*> v .:? "games"           .!= 0
+    <*> v .:? "home_game_stats" .!= newGameStats
+    <*> v .:? "away_game_stats" .!= newGameStats
+
+instance ToJSON Database where
+  toJSON (Database players goalies games hgs ags) = object
+    [ "players"         .= players
+    , "goalies"         .= goalies
+    , "games"           .= games
+    , "home_game_stats" .= hgs
+    , "away_game_stats" .= ags
+    ]
+  toEncoding (Database players goalies games hgs ags) = pairs $
+    "players"         .= players <>
+    "goalies"         .= goalies <>
+    "games"           .= games   <>
+    "home_game_stats" .= hgs     <>
+    "away_game_stats" .= ags
+
+instance FromJSON Player where
+  parseJSON = withObject "Player" $ \v -> Player
+    <$> v .:  "number"
+    <*> v .:  "name"
+    <*> v .:  "position"
+    <*> v .:? "rookie"   .!= False
+    <*> v .:? "active"   .!= True
+    <*> v .:? "ytd"      .!= newPlayerStats
+    <*> v .:? "lifetime" .!= newPlayerStats
+
+instance ToJSON Player where
+  toJSON (Player num name pos rk act ytd lt) = object
+    [ "number"   .= num
+    , "name"     .= name
+    , "position" .= pos
+    , "rookie"   .= rk
+    , "active"   .= act
+    , "ytd"      .= ytd
+    , "lifetime" .= lt
+    ]
+  toEncoding (Player num name pos rk act ytd lt) = pairs $
+    "number"   .= num  <>
+    "name"     .= name <>
+    "position" .= pos  <>
+    "rookie"   .= rk   <>
+    "active"   .= act  <>
+    "ytd"      .= ytd  <>
+    "lifetime" .= lt
+
+instance FromJSON PlayerStats where
+  parseJSON = withObject "PlayerStats" $ \v -> PlayerStats
+    <$> v .:? "goals"        .!= 0
+    <*> v .:? "assists"      .!= 0
+    <*> v .:? "penalty_mins" .!= 0
+
+instance ToJSON PlayerStats where
+  toJSON (PlayerStats g a pm) = object
+    [ "goals"        .= g
+    , "assists"      .= a
+    , "penalty_mins" .= pm
+    ]
+  toEncoding (PlayerStats g a pm) = pairs $
+    "goals"        .= g  <>
+    "assists"      .= a  <>
+    "penalty_mins" .= pm
+
+instance FromJSON Goalie where
+  parseJSON = withObject "Goalie" $ \v -> Goalie
+    <$> v .:  "number"
+    <*> v .:  "name"
+    <*> v .:? "rookie"   .!= False
+    <*> v .:? "active"   .!= True
+    <*> v .:? "ytd"      .!= newGoalieStats
+    <*> v .:? "lifetime" .!= newGoalieStats
+
+instance ToJSON Goalie where
+  toJSON (Goalie num name rk act ytd lt) = object
+    [ "number"   .= num
+    , "name"     .= name
+    , "ytd"      .= ytd
+    , "rookie"   .= rk
+    , "active"   .= act
+    , "lifetime" .= lt
+    ]
+  toEncoding (Goalie num name rk act ytd lt) = pairs $
+    "number"   .= num  <>
+    "name"     .= name <>
+    "rookie"   .= rk   <>
+    "active"   .= act  <>
+    "ytd"      .= ytd  <>
+    "lifetime" .= lt
+
+instance FromJSON GoalieStats where
+  parseJSON = withObject "GoalieStats" $ \v -> GoalieStats
+    <$> v .:? "games"         .!= 0
+    <*> v .:? "mins_played"   .!= 0
+    <*> v .:? "goals_allowed" .!= 0
+    <*> v .:? "shutouts"      .!= 0
+    <*> v .:? "wins"          .!= 0
+    <*> v .:? "losses"        .!= 0
+    <*> v .:? "ties"          .!= 0
+
+instance ToJSON GoalieStats where
+  toJSON (GoalieStats g m a s w l t) = object
+    [ "games"         .= g
+    , "mins_played"   .= m
+    , "goals_allowed" .= a
+    , "shutouts"      .= s
+    , "wins"          .= w
+    , "losses"        .= l
+    , "ties"          .= t
+    ]
+  toEncoding (GoalieStats g m a s w l t) = pairs $
+      "games"         .= g <>
+      "mins_played"   .= m <>
+      "goals_allowed" .= a <>
+      "shutouts"      .= s <>
+      "wins"          .= w <>
+      "losses"        .= l <>
+      "ties"          .= t
+
+instance FromJSON GameStats where
+  parseJSON = withObject "GameStats" $ \v -> GameStats
+    <$> v .: "wins"
+    <*> v .: "losses"
+    <*> v .: "overtime"
+    <*> v .: "goals_for"
+    <*> v .: "goals_against"
+
+instance ToJSON GameStats where
+  toJSON (GameStats w l ot gf ga) = object
+    [ "wins"          .= w
+    , "losses"        .= l
+    , "overtime"      .= ot
+    , "goals_for"     .= gf
+    , "goals_against" .= ga
+    ]
+  toEncoding (GameStats w l ot gf ga) = pairs $
+    "wins"          .= w  <>
+    "losses"        .= l  <>
+    "overtime"      .= ot <>
+    "goals_for"     .= gf <>
+    "goals_against" .= ga
+
 gameStateL :: Lens' ProgMode GameState
 gameStateL = lens
   (\case
@@ -688,11 +740,29 @@ editGoalieStateL = lens
     _              -> newEditGoalieState)
   (\_ egs -> EditGoalie egs)
 
+editStandingsModeL :: Lens' ProgMode EditStandingsMode
+editStandingsModeL = lens
+  (\case
+    EditStandings esm -> esm
+    _                 -> ESMMenu)
+  (\_ esm -> EditStandings esm)
+
+esmSubModeL :: Lens' EditStandingsMode ESMSubMode
+esmSubModeL = lens
+  (\case
+    ESMMenu   -> ESMSubMenu
+    ESMHome m -> m
+    ESMAway m -> m)
+  (\mode subMode -> case mode of
+    ESMMenu   -> ESMMenu
+    ESMHome _ -> ESMHome subMode
+    ESMAway _ -> ESMAway subMode)
+
 -- | Constructor for a 'ProgState'
 newProgState :: ProgState
 newProgState = ProgState
   { _database     = newDatabase
-  , _progMode     = MainMenu
+  , _progMode     = TitleScreen
   , _inputBuffer  = ""
   , _scrollOffset = 0
   }
@@ -780,6 +850,8 @@ newPlayer num name pos = Player
   { _pNumber   = num
   , _pName     = name
   , _pPosition = pos
+  , _pRookie   = True
+  , _pActive   = True
   , _pYtd      = newPlayerStats
   , _pLifetime = newPlayerStats
   }
@@ -802,6 +874,8 @@ newGoalie
 newGoalie num name = Goalie
   { _gNumber   = num
   , _gName     = name
+  , _gRookie   = True
+  , _gActive   = True
   , _gYtd      = newGoalieStats
   , _gLifetime = newGoalieStats
   }

src/Mtlstats/Types/Menu.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -39,6 +39,7 @@ module Mtlstats.Types.Menu (
 import Lens.Micro ((^.))
 import Lens.Micro.TH (makeLenses)
 
+import Mtlstats.Format
 import Mtlstats.Types
 
 -- | Defines a menu
@@ -65,8 +66,15 @@ makeLenses ''Menu
 makeLenses ''MenuItem
 
 instance Show (Menu a) where
-  show m = m ^. menuTitle ++ "\n" ++ items
-    where items = unlines $ map show $ m ^. menuItems
+  show m = unlines
+    $  [ m^.menuTitle
+       , ""
+       ]
+    ++ body
+    where
+      body  = map (left width) items
+      width = maximum $ map length items
+      items = map show $ m^.menuItems
 
 instance Show (MenuItem a) where
-  show i = [i ^. miKey] ++ ") " ++ i ^. miDescription
+  show i = [i ^. miKey] ++ ": " ++ i ^. miDescription

src/Mtlstats/Util.hs

@@ -1,7 +1,7 @@
 {- |
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/Actions/EditGoalieSpec.hs

@@ -1,537 +0,0 @@
-{-
-
-mtlstats
-Copyright (C) 2019 Rhéal Lamothe
-<rheal.lamothe@gmail.com>
-
-This program is free software: you can redistribute it and/or modify
-it under the terms of the GNU General Public License as published by
-the Free Software Foundation, either version 3 of the License, or (at
-your option) any later version.
-
-This program is distributed in the hope that it will be useful, but
-WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-General Public License for more details.
-
-You should have received a copy of the GNU General Public License
-along with this program.  If not, see <https://www.gnu.org/licenses/>.
-
--}
-
-module Actions.EditGoalieSpec (spec) where
-
-import Data.Maybe (fromJust)
-import Lens.Micro ((^.), (&), (.~))
-import Test.Hspec (Spec, context, describe, it, shouldBe)
-
-import Mtlstats.Actions.EditGoalie
-import Mtlstats.Types
-import Mtlstats.Util
-
-spec :: Spec
-spec = describe "EditGoalie" $ do
-  editGoalieNumberSpec
-  editGoalieNameSpec
-  editGoalieYtdGamesSpec
-  editGoalieYtdMinsSpec
-  editGoalieYtdGoalsSpec
-  editGoalieYtdWinsSpec
-  editGoalieYtdLossesSpec
-  editGoalieYtdTiesSpec
-  editGoalieLtGamesSpec
-  editGoalieLtMinsSpec
-  editGoalieLtGoalsSpec
-  editGoalieLtWinsSpec
-  editGoalieLtLossesSpec
-  editGoalieLtTiesSpec
-
-editGoalieNumberSpec :: Spec
-editGoalieNumberSpec = describe "editGoalieNumber" $ editTest
-  (editGoalieNumber 5)
-  EGNumber
-  (uncurry newGoalie)
-  [ ( "set Joe"
-    , Just 0
-    , (5, "Joe")
-    , (3, "Bob")
-    , EGMenu
-    )
-  , ( "set Bob"
-    , Just 1
-    , (2, "Joe")
-    , (5, "Bob")
-    , EGMenu
-    )
-  , ( "out of bounds"
-    , Just 2
-    , (2, "Joe")
-    , (3, "Bob")
-    , EGNumber
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , (2, "Joe")
-    , (3, "Bob")
-    , EGNumber
-    )
-  ]
-
-editGoalieNameSpec :: Spec
-editGoalieNameSpec = describe "editGoalieName" $ editTest
-  (editGoalieName "foo")
-  EGName
-  (uncurry newGoalie)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "foo" )
-    , ( 3, "Bob" )
-    , EGMenu
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe" )
-    , ( 3, "foo" )
-    , EGMenu
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe" )
-    , ( 3, "Bob" )
-    , EGName
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe" )
-    , ( 3, "Bob" )
-    , EGName
-    )
-  ]
-
-editGoalieYtdGamesSpec :: Spec
-editGoalieYtdGamesSpec = describe "editGoalieYtdGames" $ editTest
-  (editGoalieYtdGames 1)
-  EGYtdGames
-  (\(num, name, games) -> newGoalie num name & gYtd.gsGames .~ games)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGYtd
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGYtd
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdGames
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdGames
-    )
-  ]
-
-editGoalieYtdMinsSpec :: Spec
-editGoalieYtdMinsSpec = describe "editGoalieYtdMins" $ editTest
-  (editGoalieYtdMins 1)
-  EGYtdMins
-  (\(num, name, mins) -> newGoalie num name & gYtd.gsMinsPlayed .~ mins)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGYtd
-    )
-  , ( "set Bob"
-    , Just 1
-    , (2, "Joe", 0 )
-    , (3, "Bob", 1 )
-    , EGYtd
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdMins
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdMins
-    )
-  ]
-
-editGoalieYtdGoalsSpec :: Spec
-editGoalieYtdGoalsSpec = describe "editGoalieYtdGoals" $ editTest
-  (editGoalieYtdGoals 1)
-  EGYtdGoals
-  (\(num, name, goals) -> newGoalie num name & gYtd.gsGoalsAllowed .~ goals)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGYtd
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGYtd
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdGoals
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdGoals
-    )
-  ]
-
-editGoalieYtdWinsSpec :: Spec
-editGoalieYtdWinsSpec = describe "editGoalieYtdWins" $ editTest
-  (editGoalieYtdWins 1)
-  EGYtdWins
-  (\(num, name, wins) -> newGoalie num name & gYtd.gsWins .~ wins)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGYtd
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGYtd
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdWins
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdWins
-    )
-  ]
-
-editGoalieYtdLossesSpec :: Spec
-editGoalieYtdLossesSpec = describe "editGoalieYtdLosses" $ editTest
-  (editGoalieYtdLosses 1)
-  EGYtdLosses
-  (\(num, name, losses) -> newGoalie num name & gYtd.gsLosses .~ losses)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGYtd
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGYtd
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdLosses
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdLosses
-    )
-  ]
-
-editGoalieYtdTiesSpec :: Spec
-editGoalieYtdTiesSpec = describe "editGoalieYtdTies" $ editTest
-  (editGoalieYtdTies 1)
-  EGYtdTies
-  (\(num, name, ties) -> newGoalie num name & gYtd.gsTies .~ ties)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGYtd
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGYtd
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdTies
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGYtdTies
-    )
-  ]
-
-editGoalieLtGamesSpec :: Spec
-editGoalieLtGamesSpec = describe "editGoalieLtGames" $ editTest
-  (editGoalieLtGames 1)
-  EGLtGames
-  (\(num, name, games) -> newGoalie num name & gLifetime.gsGames .~ games)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGLifetime
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGLifetime
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtGames
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtGames
-    )
-  ]
-
-editGoalieLtMinsSpec :: Spec
-editGoalieLtMinsSpec = describe "editGoalieLtMins" $ editTest
-  (editGoalieLtMins 1)
-  EGLtMins
-  (\(num, name, mins) -> newGoalie num name & gLifetime.gsMinsPlayed .~ mins)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGLifetime
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGLifetime
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtMins
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtMins
-    )
-  ]
-
-editGoalieLtGoalsSpec :: Spec
-editGoalieLtGoalsSpec = describe "editGoalieLtGoals" $ editTest
-  (editGoalieLtGoals 1)
-  EGLtGoals
-  (\(num, name, goals) -> newGoalie num name & gLifetime.gsGoalsAllowed .~ goals)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGLifetime
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGLifetime
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtGoals
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtGoals
-    )
-  ]
-
-editGoalieLtWinsSpec :: Spec
-editGoalieLtWinsSpec = describe "editGoalieLtWins" $ editTest
-  (editGoalieLtWins 1)
-  EGLtWins
-  (\(num, name, wins) -> newGoalie num name & gLifetime.gsWins .~ wins)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGLifetime
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGLifetime
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtWins
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtWins
-    )
-  ]
-
-editGoalieLtLossesSpec :: Spec
-editGoalieLtLossesSpec = describe "editGoalieLtLosses" $ editTest
-  (editGoalieLtLosses 1)
-  EGLtLosses
-  (\(num, name, losses) -> newGoalie num name & gLifetime.gsLosses .~ losses)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGLifetime
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGLifetime
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtLosses
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtLosses
-    )
-  ]
-
-editGoalieLtTiesSpec :: Spec
-editGoalieLtTiesSpec = describe "editGoalieLtTies" $ editTest
-  (editGoalieLtTies 1)
-  EGLtTies
-  (\(num, name, ties) -> newGoalie num name & gLifetime.gsTies .~ ties)
-  [ ( "set Joe"
-    , Just 0
-    , ( 2, "Joe", 1 )
-    , ( 3, "Bob", 0 )
-    , EGLifetime
-    )
-  , ( "set Bob"
-    , Just 1
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 1 )
-    , EGLifetime
-    )
-  , ( "out of bounds"
-    , Just 2
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtTies
-    )
-  , ( "no goalie selected"
-    , Nothing
-    , ( 2, "Joe", 0 )
-    , ( 3, "Bob", 0 )
-    , EGLtTies
-    )
-  ]
-
-editTest
-  :: (ProgState -> ProgState)
-  -> EditGoalieMode
-  -> (a -> Goalie)
-  -> [(String, Maybe Int, a, a, EditGoalieMode)]
-  -> Spec
-editTest func setMode mkGoalie params = do
-  mapM_
-    (\(setLabel, setGid, joeData, bobData, expectMode) -> context setLabel $ do
-      let
-        egs = newEditGoalieState
-          & egsSelectedGoalie .~ setGid
-          & egsMode           .~ setMode
-
-        ps = func $ progState $ EditGoalie egs
-
-      mapM_
-        (\(chkLabel, chkGid, goalieData) -> context chkLabel $ let
-          actual   = fromJust $ nth chkGid $ ps^.database.dbGoalies
-          expected = mkGoalie goalieData
-          in it ("should be " ++ show expected) $
-            actual `shouldBe` expected)
-        --  label,       goalie ID, goalie data
-        [ ( "check Joe", 0,         joeData     )
-        , ( "check Bob", 1,         bobData     )
-        ]
-
-      context "check mode" $
-        it ("should be " ++ show expectMode) $
-          ps^.progMode.editGoalieStateL.egsMode `shouldBe` expectMode)
-
-    params
-
-  context "wrong progMode" $ do
-    let ps = func $ progState MainMenu
-
-    it "should not change the database" $
-      ps^.database `shouldBe` db
-
-    it "should not change the progMode" $
-      show (ps^.progMode) `shouldBe` "MainMenu"
-
-joe :: Goalie
-joe = newGoalie 2 "Joe"
-
-bob :: Goalie
-bob = newGoalie 3 "Bob"
-
-db :: Database
-db = newDatabase & dbGoalies .~ [joe, bob]
-
-progState :: ProgMode -> ProgState
-progState mode = newProgState
-  & progMode .~ mode
-  & database .~ db

test/Actions/EditStandingsSpec.hs

@@ -0,0 +1,83 @@
+{-
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+{-# LANGUAGE LambdaCase #-}
+
+module Actions.EditStandingsSpec (spec) where
+
+import Lens.Micro ((^.), (&), (.~))
+import Test.Hspec
+  ( Spec
+  , context
+  , describe
+  , it
+  , shouldBe
+  , shouldSatisfy
+  )
+
+import Mtlstats.Actions.EditStandings
+import Mtlstats.Types
+
+spec :: Spec
+spec = describe "EditStandings" $ do
+  mapM_
+    (\(label, f, expected) -> describe label $ do
+      let
+        ps  = newProgState
+        ps' = f ps
+
+      it "should set progMode to EditStandings" $
+        ps'^.progMode `shouldSatisfy` \case
+          (EditStandings _) -> True
+          _                 -> False
+
+      it ("should set editStandingsMode to " ++ show expected) $
+        ps'^.progMode.editStandingsModeL `shouldBe` expected)
+
+    --  label,               function,          expected mode
+    [ ( "editStandings",     editStandings,     ESMMenu            )
+    , ( "editHomeStandings", editHomeStandings, ESMHome ESMSubMenu )
+    , ( "editAwayStandings", editAwayStandings, ESMAway ESMSubMenu )
+    ]
+
+  mapM_
+    (\(label, f, expected) -> describe label $ do
+      mapM_
+        (\prefix -> context ("mode: " ++ show (prefix ESMSubMenu)) $ let
+          ps  = newProgState & progMode.editStandingsModeL .~ prefix ESMSubMenu
+          ps' = f ps
+          in it ("should set the mode to " ++ show expected) $
+            ps'^.progMode.editStandingsModeL `shouldBe` prefix expected)
+        [ESMHome, ESMAway]
+
+      context "mode: ESMMenu" $ let
+        ps  = newProgState & progMode.editStandingsModeL .~ ESMMenu
+        ps' = f ps
+        in it "should not change the mode" $
+          ps'^.progMode.editStandingsModeL `shouldBe` ESMMenu)
+
+    --  label,              function,         expected
+    [ ( "editWins",         editWins,         ESMEditWins         )
+    , ( "editLosses",       editLosses,       ESMEditLosses       )
+    , ( "editOvertime",     editOvertime,     ESMEditOvertime     )
+    , ( "editGoalsFor",     editGoalsFor,     ESMEditGoalsFor     )
+    , ( "editGoalsAgainst", editGoalsAgainst, ESMEditGoalsAgainst )
+    ]

test/Actions/NewGame/GoalieInputSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/Actions/NewGameSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/ActionsSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -24,7 +24,7 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 module ActionsSpec (spec) where
 
 import Control.Monad (replicateM)
-import Lens.Micro ((^.), (&), (.~), (?~))
+import Lens.Micro ((^.), (&), (.~), (?~), (%~))
 import Test.Hspec
   ( Spec
   , context
@@ -38,8 +38,8 @@ import Test.Hspec
 import Mtlstats.Actions
 import Mtlstats.Types
 
-import qualified Actions.EditGoalieSpec as EditGoalie
 import qualified Actions.NewGameSpec as NewGame
+import qualified Actions.EditStandingsSpec as EditStandings
 import qualified TypesSpec as TS
 
 spec :: Spec
@@ -47,6 +47,7 @@ spec = describe "Mtlstats.Actions" $ do
   startNewSeasonSpec
   startNewGameSpec
   resetYtdSpec
+  clearRookiesSpec
   resetStandingsSpec
   addCharSpec
   removeCharSpec
@@ -54,7 +55,9 @@ spec = describe "Mtlstats.Actions" $ do
   createGoalieSpec
   editSpec
   editPlayerSpec
+  editSelectedPlayerSpec
   editGoalieSpec
+  editSelectedGoalieSpec
   addPlayerSpec
   addGoalieSpec
   resetCreatePlayerStateSpec
@@ -63,7 +66,7 @@ spec = describe "Mtlstats.Actions" $ do
   scrollUpSpec
   scrollDownSpec
   NewGame.spec
-  EditGoalie.spec
+  EditStandings.spec
 
 startNewSeasonSpec :: Spec
 startNewSeasonSpec = describe "startNewSeason" $ do
@@ -129,6 +132,45 @@ resetYtdSpec = describe "resetYtd" $
         lt ^. gsTies          `shouldNotBe` 0) $
       s ^. database . dbGoalies
 
+clearRookiesSpec :: Spec
+clearRookiesSpec = describe "clearRookies" $ do
+  let
+
+    players =
+      [ newPlayer 1 "Joe" "centre" & pRookie .~ True
+      , newPlayer 2 "Bob" "centre" & pRookie .~ False
+      ]
+
+    goalies =
+      [ newGoalie 3 "Bill" & gRookie .~ True
+      , newGoalie 4 "Doug" & gRookie .~ False
+      ]
+
+    ps = newProgState
+      & database
+        %~ (dbPlayers .~ players)
+        .  (dbGoalies .~ goalies)
+
+    ps' = clearRookies ps
+
+  context "Players" $ mapM_
+    (\p -> let
+      name  = p^.pName
+      rFlag = p^.pRookie
+      in context name $
+        it "should not be a rookie" $
+          rFlag `shouldBe` False)
+    (ps'^.database.dbPlayers)
+
+  context "Goalies" $ mapM_
+    (\g -> let
+      name  = g^.gName
+      rFlag = g^.gRookie
+      in context name $
+        it "should not be a rookie" $
+          rFlag `shouldBe` False)
+    (ps'^.database.dbGoalies)
+
 resetStandingsSpec :: Spec
 resetStandingsSpec = describe "resetStandings" $ do
   let
@@ -211,12 +253,64 @@ editPlayerSpec = describe "editPlayer" $
     s = editPlayer newProgState
     in show (s^.progMode) `shouldBe` "EditPlayer"
 
+editSelectedPlayerSpec :: Spec
+editSelectedPlayerSpec = describe "editSelectedPlayer" $ mapM_
+  (\(label, pState, expected) -> context label $
+    it "should edit the players appropriately" $ let
+      pState'  = editSelectedPlayer (pName .~ "foo") pState
+      players' = pState'^.database.dbPlayers
+      in players' `shouldBe` expected)
+
+  --  label,           initial state,         expected
+  [ ( "wrong mode",    baseState,             players  )
+  , ( "not selected",  changePlayer Nothing,  players  )
+  , ( "player 0",      changePlayer $ Just 0, changed0 )
+  , ( "player 1",      changePlayer $ Just 1, changed1 )
+  , ( "out of bounds", changePlayer $ Just 2, players  )
+  ]
+
+  where
+    baseState      = newProgState & database.dbPlayers .~ players
+    changePlayer n = baseState
+      & (progMode.editPlayerStateL.epsSelectedPlayer .~ n)
+    players        = [ player 0,  player 1  ]
+    changed0       = [ player' 0, player 1  ]
+    changed1       = [ player 0,  player' 1 ]
+    player n       = newPlayer n ("Player " ++ show n) "pos"
+    player' n      = newPlayer n "foo" "pos"
+
 editGoalieSpec :: Spec
 editGoalieSpec = describe "editGoalie" $
   it "should change the mode appropriately" $ let
     s = editGoalie newProgState
     in show (s^.progMode) `shouldBe` "EditGoalie"
 
+editSelectedGoalieSpec :: Spec
+editSelectedGoalieSpec = describe "editSelectedGoalie" $ mapM_
+  (\(label, pState, expected) -> context label $
+    it "should edit the goalies appropriately" $ let
+      pState'  = editSelectedGoalie (gName .~ "foo") pState
+      goalies' = pState'^.database.dbGoalies
+      in goalies' `shouldBe` expected)
+
+  --  label,           initial state,         expected
+  [ ( "wrong mode",    baseState,             goalies  )
+  , ( "not selected",  changeGoalie Nothing,  goalies  )
+  , ( "goalie 0",      changeGoalie $ Just 0, changed0 )
+  , ( "goalie 1",      changeGoalie $ Just 1, changed1 )
+  , ( "out of bounds", changeGoalie $ Just 2, goalies  )
+  ]
+
+  where
+    baseState      = newProgState & database.dbGoalies .~ goalies
+    changeGoalie n = baseState
+      & (progMode.editGoalieStateL.egsSelectedGoalie .~ n)
+    goalies        = [ goalie 0,  goalie 1  ]
+    changed0       = [ goalie' 0, goalie 1  ]
+    changed1       = [ goalie 0,  goalie' 1 ]
+    goalie n       = newGoalie n ("Player " ++ show n)
+    goalie' n      = newGoalie n "foo"
+
 addPlayerSpec :: Spec
 addPlayerSpec = describe "addPlayer" $ do
   let

test/FormatSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/HandlersSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/Helpers/GoalieSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -22,45 +22,70 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 module Helpers.GoalieSpec (spec) where
 
 import Lens.Micro ((&), (.~), (%~))
-import Test.Hspec (Spec, describe, it, shouldBe)
+import Test.Hspec (Spec, context, describe, it, shouldBe)
 
 import Mtlstats.Helpers.Goalie
 import Mtlstats.Types
 
 spec :: Spec
-spec = describe "Goalie"
+spec = describe "Goalie" $ do
   goalieDetailsSpec
+  goalieNameSpec
 
 goalieDetailsSpec :: Spec
 goalieDetailsSpec = describe "goalieDetails" $ let
   input = newGoalie 1 "Joe"
+    & gRookie .~ True
     & gYtd
       %~ ( gsGames        .~ 2  )
       .  ( gsMinsPlayed   .~ 3  )
       .  ( gsGoalsAllowed .~ 4  )
-      .  ( gsWins         .~ 5  )
-      .  ( gsLosses       .~ 6  )
-      .  ( gsTies         .~ 7  )
+      .  ( gsShutouts     .~ 5  )
+      .  ( gsWins         .~ 6  )
+      .  ( gsLosses       .~ 7  )
+      .  ( gsTies         .~ 8  )
     & gLifetime
-      %~ ( gsGames        .~ 8  )
-      .  ( gsMinsPlayed   .~ 9  )
-      .  ( gsGoalsAllowed .~ 10 )
-      .  ( gsWins         .~ 11 )
-      .  ( gsLosses       .~ 12 )
-      .  ( gsTies         .~ 13 )
+      %~ ( gsGames        .~ 9  )
+      .  ( gsMinsPlayed   .~ 10 )
+      .  ( gsGoalsAllowed .~ 11 )
+      .  ( gsShutouts     .~ 12 )
+      .  ( gsWins         .~ 13 )
+      .  ( gsLosses       .~ 14 )
+      .  ( gsTies         .~ 15 )
 
   expected = unlines
     [ "Number: 1"
-    , "  Name: Joe"
+    , "  Name: Joe*"
     , ""
     , "              YTD Lifetime"
-    , " Games played   2        8"
-    , "  Mins played   3        9"
-    , "Goals allowed   4       10"
-    , "         Wins   5       11"
-    , "       Losses   6       12"
-    , "         Ties   7       13"
+    , " Games played   2        9"
+    , "  Mins played   3       10"
+    , "Goals allowed   4       11"
+    , "     Shutouts   5       12"
+    , "         Wins   6       13"
+    , "       Losses   7       14"
+    , "         Ties   8       15"
     ]
 
   in it "should format the output correctly" $
     goalieDetails input `shouldBe` expected
+
+goalieNameSpec :: Spec
+goalieNameSpec = describe "goalieName" $ mapM_
+  (\(label, g, expected) -> context label $
+    it ("should be " ++ expected) $
+      goalieName g `shouldBe` expected)
+
+  --  label,        goalie,  expected
+  [ ( "rookie",     rookie,  "foo*"   )
+  , ( "non-rookie", active,  "foo"    )
+  , ( "retired",    retired, "*foo"   )
+  ]
+
+  where
+    rookie     = goalie True True
+    active     = goalie False True
+    retired    = goalie False False
+    goalie r a = newGoalie 1 "foo"
+      & gRookie .~ r
+      & gActive .~ a

test/Helpers/PlayerSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -22,20 +22,22 @@ along with this program.  If not, see <https://www.gnu.org/licenses/>.
 module Helpers.PlayerSpec (spec) where
 
 import Lens.Micro ((&), (.~))
-import Test.Hspec (Spec, describe, it, shouldBe)
+import Test.Hspec (Spec, context, describe, it, shouldBe)
 
 import Mtlstats.Helpers.Player
 import Mtlstats.Types
 
 spec :: Spec
-spec = describe "Player"
+spec = describe "Player" $ do
   playerDetailsSpec
+  playerNameSpec
 
 playerDetailsSpec :: Spec
 playerDetailsSpec = describe "playerDetails" $
   it "should give a detailed description" $ let
 
     p = newPlayer 1 "Joe" "centre"
+      & pRookie .~ True
       & pYtd .~ PlayerStats
         { _psGoals   = 2
         , _psAssists = 3
@@ -49,7 +51,7 @@ playerDetailsSpec = describe "playerDetails" $
 
     expected = unlines
       [ "  Number: 1"
-      , "    Name: Joe"
+      , "    Name: Joe*"
       , "Position: centre"
       , ""
       , "             YTD Lifetime"
@@ -59,3 +61,23 @@ playerDetailsSpec = describe "playerDetails" $
       ]
 
     in playerDetails p `shouldBe` expected
+
+playerNameSpec :: Spec
+playerNameSpec = describe "playerName" $ mapM_
+  (\(label, p, expected) -> context label $
+    it ("should be " ++ expected) $
+      playerName p `shouldBe` expected)
+
+  --  label,        player,    expected
+  [ ( "rookie",     rookie,    "foo*"   )
+  , ( "non-rookie", nonRookie, "foo"    )
+  , ( "retired",    retired,   "*foo"  )
+  ]
+
+  where
+    rookie    = player True True
+    nonRookie = player False True
+    retired   = player False False
+    player r a = newPlayer 1 "foo" "centre"
+      & pRookie .~ r
+      & pActive .~ a

test/HelpersSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/ReportSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/Spec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify

test/Types/MenuSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -37,11 +37,11 @@ showSpec :: Spec
 showSpec = describe "show" $
   it "should display correctly" $ let
     menu = Menu "Foo" ()
-      [ MenuItem '1' "Item 1" $ return ()
-      , MenuItem '2' "Item 2" $ return ()
+      [ MenuItem '1' "foo" $ return ()
+      , MenuItem '2' "bar baz" $ return ()
       ]
     expected =
-      "Foo\n\
-      \1) Item 1\n\
-      \2) Item 2\n"
+      "Foo\n\n\
+      \1: foo    \n\
+      \2: bar baz\n"
     in show menu `shouldBe` expected

test/TypesSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify
@@ -36,7 +36,7 @@ import Data.Aeson.Types (Value (Object))
 import qualified Data.HashMap.Strict as HM
 import Data.Ratio ((%))
 import Lens.Micro (Lens', (&), (^.), (.~), (?~))
-import System.Random (randomRIO)
+import System.Random (randomIO, randomRIO)
 import Test.Hspec (Spec, context, describe, it, shouldBe)
 
 import Mtlstats.Config
@@ -58,6 +58,8 @@ spec = describe "Mtlstats.Types" $ do
   createGoalieStateLSpec
   editPlayerStateLSpec
   editGoalieStateLSpec
+  editStandingsModeLSpec
+  esmSubModeLSpec
   teamScoreSpec
   otherScoreSpec
   homeTeamSpec
@@ -191,6 +193,47 @@ editGoalieStateLSpec = describe "editGoalieStateL" $
     egs2 = newEditGoalieState
       & egsSelectedGoalie ?~ 2
 
+editStandingsModeLSpec :: Spec
+editStandingsModeLSpec = describe "editStandingsModeL" $
+  lensSpec editStandingsModeL
+  -- getters
+  [ ( "missing mode", MainMenu,           menu )
+  , ( "with mode",    EditStandings home, home )
+  ]
+  -- setters
+  [ ( "set mode",    MainMenu,           home )
+  , ( "change mode", EditStandings home, away )
+  ]
+  where
+    menu = ESMMenu
+    home = ESMHome ESMSubMenu
+    away = ESMAway ESMSubMenu
+
+esmSubModeLSpec :: Spec
+esmSubModeLSpec = describe "esmSubModeL" $ do
+
+  context "getters" $ mapM_
+    (\(label, mode, expected) -> context label $
+      it ("should be " ++ show expected) $
+        mode^.esmSubModeL `shouldBe` expected)
+
+    --  label,        mode,                expected
+    [ ( "no state",   ESMMenu,             ESMSubMenu  )
+    , ( "with state", ESMHome ESMEditWins, ESMEditWins )
+    ]
+
+  context "setters" $ mapM_
+    (\(label, mode, expected) -> context label $
+      it ("should be " ++ show expected) $ let
+        mode' = mode & esmSubModeL .~ ESMEditWins
+        in mode' `shouldBe` expected)
+
+    --  label,       mode,               expected
+    [ ( "no state",  ESMMenu,            ESMMenu             )
+    , ( "home mode", ESMHome ESMSubMenu, ESMHome ESMEditWins )
+    , ( "away mode", ESMAway ESMSubMenu, ESMAway ESMEditWins )
+    ]
+
 teamScoreSpec :: Spec
 teamScoreSpec = describe "teamScore" $ do
   let
@@ -271,6 +314,7 @@ lensSpec lens getters setters = do
 
 player :: Player
 player = newPlayer 1 "Joe" "centre"
+  & pRookie   .~ False
   & pYtd      .~ playerStats 1
   & pLifetime .~ playerStats 2
 
@@ -279,6 +323,8 @@ playerJSON = Object $ HM.fromList
   [ ( "number",   toJSON (1 :: Int)           )
   , ( "name",     toJSON ("Joe" :: String)    )
   , ( "position", toJSON ("centre" :: String) )
+  , ( "rookie",   toJSON False                )
+  , ( "active",   toJSON True                 )
   , ( "ytd",      playerStatsJSON 1           )
   , ( "lifetime", playerStatsJSON 2           )
   ]
@@ -298,6 +344,7 @@ playerStatsJSON n = Object $ HM.fromList
 
 goalie :: Goalie
 goalie = newGoalie 1 "Joe"
+  & gRookie   .~ False
   & gYtd      .~ goalieStats 1
   & gLifetime .~ goalieStats 2
 
@@ -305,6 +352,8 @@ goalieJSON :: Value
 goalieJSON = Object $ HM.fromList
   [ ( "number",   toJSON (1 :: Int)         )
   , ( "name",     toJSON ("Joe" :: String ) )
+  , ( "rookie",   toJSON False              )
+  , ( "active",   toJSON True               )
   , ( "ytd",      goalieStatsJSON 1         )
   , ( "lifetime", goalieStatsJSON 2         )
   ]
@@ -843,6 +892,8 @@ makePlayer = Player
   <$> makeNum
   <*> makeName
   <*> makeName
+  <*> makeBool
+  <*> makeBool
   <*> makePlayerStats
   <*> makePlayerStats
 
@@ -851,6 +902,8 @@ makeGoalie :: IO Goalie
 makeGoalie = Goalie
   <$> makeNum
   <*> makeName
+  <*> makeBool
+  <*> makeBool
   <*> makeGoalieStats
   <*> makeGoalieStats
 
@@ -875,6 +928,9 @@ makeGoalieStats = GoalieStats
 makeNum :: IO Int
 makeNum = randomRIO (1, 10)
 
+makeBool :: IO Bool
+makeBool = randomIO
+
 makeName :: IO String
 makeName = replicateM 10 $ randomRIO ('A', 'Z')
 
@@ -944,3 +1000,8 @@ instance Comparable CreateGoalieState where
     describe "cgsName" $
       it ("should be " ++ expected^.cgsName) $
         actual^.cgsName `shouldBe` expected^.cgsName
+
+instance Comparable EditStandingsMode where
+  compareTest actual expected =
+    it ("should be " ++ show expected) $
+      actual `shouldBe` expected

test/UtilSpec.hs

@@ -1,7 +1,7 @@
 {-
 
 mtlstats
-Copyright (C) 2019 Rhéal Lamothe
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
 <rheal.lamothe@gmail.com>
 
 This program is free software: you can redistribute it and/or modify