From 4c13cc9103c365f06b66bcef5fc40a84d4f05862 Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jlamothe1980@gmail.com>
Date: Fri, 30 Aug 2019 01:21:17 -0400
Subject: [PATCH] implemented updateGameStats

---
 src/Mtlstats/Actions.hs | 29 +++++++++++++-
 src/Mtlstats/Events.hs  |  1 +
 test/ActionsSpec.hs     | 86 +++++++++++++++++++++++++++++++++++++++++
 3 files changed, 115 insertions(+), 1 deletion(-)

diff --git a/src/Mtlstats/Actions.hs b/src/Mtlstats/Actions.hs
index 482db98..32c1c64 100644
--- a/src/Mtlstats/Actions.hs
+++ b/src/Mtlstats/Actions.hs
@@ -28,10 +28,11 @@ module Mtlstats.Actions
   , addChar
   , removeChar
   , overtimeCheck
+  , updateGameStats
   ) where
 
 import Data.Maybe (fromMaybe)
-import Lens.Micro (over, (^.), (&), (.~), (?~), (%~))
+import Lens.Micro (over, (^.), (&), (.~), (?~), (%~), (+~))
 
 import Mtlstats.Types
 
@@ -71,3 +72,29 @@ overtimeCheck s
   | fromMaybe False $ gameWon $ s^.progMode.gameStateL =
     s & progMode.gameStateL.overtimeFlag ?~ False
   | otherwise  = s
+
+-- | Adjusts the game stats based on the results of the current game
+updateGameStats :: ProgState -> ProgState
+updateGameStats s = fromMaybe s result
+  where
+    result = do
+      gType <- s^.progMode.gameStateL.gameType
+      won   <- gameWon $ s^.progMode.gameStateL
+      lost  <- gameLost $ s^.progMode.gameStateL
+      ot    <- s^.progMode.gameStateL.overtimeFlag
+      let
+        hw  = if gType == HomeGame && won then 1 else 0
+        hl  = if gType == HomeGame && lost then 1 else 0
+        hot = if gType == HomeGame && ot then 1 else 0
+        aw  = if gType == AwayGame && won then 1 else 0
+        al  = if gType == AwayGame && lost then 1 else 0
+        aot = if gType == AwayGame && ot then 1 else 0
+      Just $ s
+        & database.dbHomeGameStats
+          %~ (gmsWins +~ hw)
+          .  (gmsLosses +~ hl)
+          .  (gmsOvertime +~ hot)
+        & database.dbAwayGameStats
+          %~ (gmsWins +~ aw)
+          .  (gmsLosses +~ al)
+          .  (gmsOvertime +~ aot)
diff --git a/src/Mtlstats/Events.hs b/src/Mtlstats/Events.hs
index 6dfd6a8..cc3efe2 100644
--- a/src/Mtlstats/Events.hs
+++ b/src/Mtlstats/Events.hs
@@ -59,6 +59,7 @@ handleEvent e = gets (view progMode) >>= \case
     | null $ gs ^. overtimeFlag -> do
       overtimePrompt e
         >>= modify . (progMode.gameStateL.overtimeFlag .~)
+      modify updateGameStats
       return True
     | otherwise -> undefined
 
diff --git a/test/ActionsSpec.hs b/test/ActionsSpec.hs
index 50595e3..0ec851c 100644
--- a/test/ActionsSpec.hs
+++ b/test/ActionsSpec.hs
@@ -37,6 +37,7 @@ spec = describe "Mtlstats.Actions" $ do
   addCharSpec
   removeCharSpec
   overtimeCheckSpec
+  updateGameStatsSpec
 
 startNewSeasonSpec :: Spec
 startNewSeasonSpec = describe "startNewSeason" $ do
@@ -183,6 +184,91 @@ overtimeCheckSpec = describe "overtimeCheck" $ do
     it "should leave the overtimeCheck flag blank" $
       s^.progMode.gameStateL.overtimeFlag `shouldBe` Nothing
 
+updateGameStatsSpec :: Spec
+updateGameStatsSpec = describe "updateGameStats" $ do
+  let
+
+    baseStats = newGameStats
+      & gmsWins     .~ 1
+      & gmsLosses   .~ 1
+      & gmsOvertime .~ 1
+
+    s t h a o = newProgState
+      & progMode.gameStateL
+        %~ (gameType     .~ t)
+        .  (homeScore    .~ h)
+        .  (awayScore    .~ a)
+        .  (overtimeFlag .~ o)
+      & database
+        %~ (dbHomeGameStats .~ baseStats)
+        .  (dbAwayGameStats .~ baseStats)
+
+    db hw hl ho aw al ao = newDatabase
+      & dbHomeGameStats
+        %~ (gmsWins     .~ hw)
+        .  (gmsLosses   .~ hl)
+        .  (gmsOvertime .~ ho)
+      & dbAwayGameStats
+        %~ (gmsWins     .~ aw)
+        .  (gmsLosses   .~ al)
+        .  (gmsOvertime .~ ao)
+
+  context "home win" $
+    it "should record a home win" $ let
+      s'  = s (Just HomeGame) (Just 2) (Just 1) (Just False)
+      db' = updateGameStats s' ^. database
+      in db' `shouldBe` db 2 1 1 1 1 1
+
+  context "home loss" $
+    it "should record a home loss" $ let
+      s'  = s (Just HomeGame) (Just 1) (Just 2) (Just False)
+      db' = updateGameStats s' ^. database
+      in db' `shouldBe` db 1 2 1 1 1 1
+
+  context "home overtime loss" $
+    it "should record a home loss and overtime" $ let
+      s'  = s (Just HomeGame) (Just 1) (Just 2) (Just True)
+      db' = updateGameStats s' ^. database
+      in db' `shouldBe` db 1 2 2 1 1 1
+
+  context "away win" $
+    it "should record an away win" $ let
+      s'  = s (Just AwayGame) (Just 1) (Just 2) (Just False)
+      db' = updateGameStats s' ^. database
+      in db' `shouldBe` db 1 1 1 2 1 1
+
+  context "away loss" $
+    it "should record an away loss" $ let
+      s'  = s (Just AwayGame) (Just 2) (Just 1) (Just False)
+      db' = updateGameStats s' ^. database
+      in db' `shouldBe` db 1 1 1 1 2 1
+
+  context "away overtime loss" $
+    it "should record an away loss and overtime" $ let
+      s'  = s (Just AwayGame) (Just 2) (Just 1) (Just True)
+      db' = updateGameStats s' ^. database
+      in db' `shouldBe` db 1 1 1 1 2 2
+
+  context "missing game type" $
+    it "should not change anything" $ let
+      s' = s Nothing (Just 1) (Just 2) (Just True)
+      in updateGameStats s' `shouldBe` s'
+
+  context "missing home score" $
+    it "should not change anything" $ let
+      s' = s (Just HomeGame) Nothing (Just 1) (Just True)
+      in updateGameStats s' `shouldBe` s'
+
+  context "missing away score" $
+    it "should not change anything" $ let
+      s' = s (Just HomeGame) (Just 1) Nothing (Just True)
+      in updateGameStats s' `shouldBe` s'
+
+  context "missing overtime flag" $
+    it "should not change anything" $ let
+      s' = s (Just HomeGame) (Just 1) (Just 2) Nothing
+      in updateGameStats s' `shouldBe` s'
+
 makePlayer :: IO Player
 makePlayer = Player
   <$> makeNum