diff --git a/src/Mtlstats/Control/EditPlayer.hs b/src/Mtlstats/Control/EditPlayer.hs index 46218ba..0691af8 100644 --- a/src/Mtlstats/Control/EditPlayer.hs +++ b/src/Mtlstats/Control/EditPlayer.hs @@ -95,7 +95,12 @@ positionC = Controller } ytdGoalsC :: Controller -ytdGoalsC = undefined +ytdGoalsC = Controller + { drawController = drawPrompt editPlayerYtdGoalsPrompt + , handleController = \e -> do + promptHandler editPlayerYtdGoalsPrompt e + return True + } ytdAssistsC :: Controller ytdAssistsC = undefined diff --git a/src/Mtlstats/Prompt/EditPlayer.hs b/src/Mtlstats/Prompt/EditPlayer.hs index 376c9f9..f466ddc 100644 --- a/src/Mtlstats/Prompt/EditPlayer.hs +++ b/src/Mtlstats/Prompt/EditPlayer.hs @@ -23,6 +23,7 @@ module Mtlstats.Prompt.EditPlayer ( editPlayerNumPrompt , editPlayerNamePrompt , editPlayerPosPrompt + , editPlayerYtdGoalsPrompt ) where import Control.Monad.Extra (whenJustM) @@ -48,6 +49,11 @@ editPlayerPosPrompt :: Prompt editPlayerPosPrompt = strPrompt "Player position: " $ editPlayer . (pPosition .~) +-- | Prompt to edit a player's year-to-date goals +editPlayerYtdGoalsPrompt :: Prompt +editPlayerYtdGoalsPrompt = numPrompt "Year-to-date goals: " $ + editPlayer . (pYtd.psGoals .~) + editPlayer :: (Player -> Player) -> Action () editPlayer f = whenJustM (gets (^.progMode.editPlayerStateL.epsSelectedPlayer)) $ \pid ->