implemented edit functions in Mtlstats.Actions.EditState module

src/Mtlstats/Actions/EditStandings.hs

@@ -40,28 +40,31 @@ editStandings = progMode .~ EditStandings ESMMenu
 
 -- | Edits the home standings
 editHomeStandings :: ProgState -> ProgState
-editHomeStandings = progMode .~ EditStandings ESMHome
+editHomeStandings = progMode .~ EditStandings (ESMHome ESMSubMenu)
 
 -- | Edits the road standings
 editAwayStandings :: ProgState -> ProgState
-editAwayStandings = progMode .~ EditStandings ESMAway
+editAwayStandings = progMode .~ EditStandings (ESMAway ESMSubMenu)
 
 -- | Changes to edit wins mode
 editWins :: ProgState -> ProgState
-editWins = undefined
+editWins = doEdit ESMEditWins
 
 -- | Changes to edit losses mode
 editLosses :: ProgState -> ProgState
-editLosses = undefined
+editLosses = doEdit ESMEditLosses
 
 -- | Changes to edit overtime mode
 editOvertime :: ProgState -> ProgState
-editOvertime = undefined
+editOvertime = doEdit ESMEditOvertime
 
 -- | Changes to edit goals for mode
 editGoalsFor :: ProgState -> ProgState
-editGoalsFor = undefined
+editGoalsFor = doEdit ESMEditGoalsFor
 
 -- | Changes to edit goals against mode
 editGoalsAgainst :: ProgState -> ProgState
-editGoalsAgainst = undefined
+editGoalsAgainst = doEdit ESMEditGoalsAgainst
+
+doEdit :: ESMSubMode -> ProgState -> ProgState
+doEdit = (progMode.editStandingsModeL.esmSubModeL .~)

src/Mtlstats/Control/EditStandings.hs

@@ -34,15 +34,19 @@ import Mtlstats.Types
 -- | Controller for the edit standings menu
 editStandingsC :: EditStandingsMode -> Controller
 editStandingsC = \case
-  ESMMenu -> menuControllerWith header editStandingsMenu
-  ESMHome -> editHomeStandingsC
-  ESMAway -> editAwayStandingsC
+  ESMMenu   -> menuControllerWith header editStandingsMenu
+  ESMHome m -> editHomeStandingsC m
+  ESMAway m -> editAwayStandingsC m
 
-editHomeStandingsC :: Controller
-editHomeStandingsC = menuControllerWith header editHomeStandingsMenu
+editHomeStandingsC :: ESMSubMode -> Controller
+editHomeStandingsC = \case
+  ESMSubMenu -> menuControllerWith header editHomeStandingsMenu
+  _          -> undefined
 
-editAwayStandingsC :: Controller
-editAwayStandingsC = menuControllerWith header editAwayStandingsMenu
+editAwayStandingsC :: ESMSubMode -> Controller
+editAwayStandingsC = \case
+  ESMSubMenu -> menuControllerWith header editAwayStandingsMenu
+  _          -> undefined
 
 header :: ProgState -> C.Update ()
 header = do

src/Mtlstats/Types.hs

@@ -36,6 +36,7 @@ module Mtlstats.Types (
   EditGoalieState (..),
   EditGoalieMode (..),
   EditStandingsMode (..),
+  ESMSubMode (..),
   Database (..),
   Player (..),
   PlayerStats (..),
@@ -58,6 +59,8 @@ module Mtlstats.Types (
   editPlayerStateL,
   editGoalieStateL,
   editStandingsModeL,
+  -- ** EditStandingsMode Lenses
+  esmSubModeL,
   -- ** GameState Lenses
   gameYear,
   gameMonth,
@@ -396,8 +399,18 @@ data EditGoalieMode
 -- | Represents the standings edit mode
 data EditStandingsMode
   = ESMMenu
-  | ESMHome
-  | ESMAway
+  | 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
@@ -732,6 +745,17 @@ editStandingsModeL = lens
     _                 -> 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