From 7bb0981ed68d63b04ea2459c43c6a804e42a7312 Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jlamothe1980@gmail.com>
Date: Sat, 28 Dec 2019 21:24:45 -0500
Subject: [PATCH] make player edit prompts return to the appropriate menus

---
 ChangeLog.md                      |  1 +
 src/Mtlstats/Prompt/EditPlayer.hs | 24 ++++++++++++------------
 2 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/ChangeLog.md b/ChangeLog.md
index 9287a0e..fead390 100644
--- a/ChangeLog.md
+++ b/ChangeLog.md
@@ -3,6 +3,7 @@
 ## current
 - Don't show player number zero in reports
 - Fixed player/goalie name capitalisation on edit
+- Return to correct edit menus after editing player stats
 
 ## 0.9.0
 - Bugfix: Display lifetime stats in report, not YTD
diff --git a/src/Mtlstats/Prompt/EditPlayer.hs b/src/Mtlstats/Prompt/EditPlayer.hs
index 329d0e1..8a52f73 100644
--- a/src/Mtlstats/Prompt/EditPlayer.hs
+++ b/src/Mtlstats/Prompt/EditPlayer.hs
@@ -42,51 +42,51 @@ import Mtlstats.Util
 -- | Prompt to edit a player's number
 editPlayerNumPrompt :: Prompt
 editPlayerNumPrompt = numPrompt "Player number: " $
-  editPlayer . (pNumber .~)
+  editPlayer EPMenu . (pNumber .~)
 
 -- | Prompt to edit a player's name
 editPlayerNamePrompt :: Prompt
 editPlayerNamePrompt = namePrompt "Player name: " $
-  editPlayer . (pName .~)
+  editPlayer EPMenu . (pName .~)
 
 -- | Prompt to edit a player's position
 editPlayerPosPrompt :: Prompt
 editPlayerPosPrompt = ucStrPrompt "Player position: " $
-  editPlayer . (pPosition .~)
+  editPlayer EPMenu . (pPosition .~)
 
 -- | Prompt to edit a player's year-to-date goals
 editPlayerYtdGoalsPrompt :: Prompt
 editPlayerYtdGoalsPrompt = numPrompt "Year-to-date goals: " $
-  editPlayer . (pYtd.psGoals .~)
+  editPlayer EPYtd . (pYtd.psGoals .~)
 
 -- | Prompt to edit a player's year-to-date assists
 editPlayerYtdAssistsPrompt :: Prompt
 editPlayerYtdAssistsPrompt = numPrompt "Year-to-date assists: " $
-  editPlayer . (pYtd.psAssists .~)
+  editPlayer EPYtd . (pYtd.psAssists .~)
 
 -- | Prompt to edit a player's year-to-date penalty minutes
 editPlayerYtdPMinPrompt :: Prompt
 editPlayerYtdPMinPrompt = numPrompt "Year-to-date penalty minutes: " $
-  editPlayer . (pYtd.psPMin .~)
+  editPlayer EPYtd . (pYtd.psPMin .~)
 
 -- | Prompt to edit a player's lifetime goals
 editPlayerLtGoalsPrompt :: Prompt
 editPlayerLtGoalsPrompt = numPrompt "Lifetime goals: " $
-  editPlayer . (pLifetime.psGoals .~)
+  editPlayer EPLifetime . (pLifetime.psGoals .~)
 
 -- | Prompt to edit a player's lifetime assists
 editPlayerLtAssistsPrompt :: Prompt
 editPlayerLtAssistsPrompt = numPrompt "Lifetime assists: " $
-  editPlayer . (pLifetime.psAssists .~)
+  editPlayer EPLifetime . (pLifetime.psAssists .~)
 
 -- | Prompt to edit a player's lifetime penalty minutes
 editPlayerLtPMinPrompt :: Prompt
 editPlayerLtPMinPrompt = numPrompt "Lifetime penalty minutes: " $
-  editPlayer . (pLifetime.psPMin .~)
+  editPlayer EPLifetime . (pLifetime.psPMin .~)
 
-editPlayer :: (Player -> Player) -> Action ()
-editPlayer f =
+editPlayer :: EditPlayerMode -> (Player -> Player) -> Action ()
+editPlayer mode f =
   whenJustM (gets (^.progMode.editPlayerStateL.epsSelectedPlayer)) $ \pid ->
     modify
       $ (database.dbPlayers %~ modifyNth pid f)
-      . (progMode.editPlayerStateL.epsMode .~ EPMenu)
+      . (progMode.editPlayerStateL.epsMode .~ mode)