118 lines
3.6 KiB
Haskell
118 lines
3.6 KiB
Haskell
{- |
|
|
|
|
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/>.
|
|
|
|
-}
|
|
|
|
{-# LANGUAGE LambdaCase #-}
|
|
|
|
module Mtlstats.Actions
|
|
( startNewSeason
|
|
, resetYtd
|
|
, startNewGame
|
|
, addChar
|
|
, removeChar
|
|
, overtimeCheck
|
|
, updateGameStats
|
|
, validateGameDate
|
|
) where
|
|
|
|
import Data.Maybe (fromMaybe)
|
|
import Data.Time.Calendar (fromGregorianValid)
|
|
import Lens.Micro (over, (^.), (&), (.~), (?~), (%~), (+~))
|
|
|
|
import Mtlstats.Types
|
|
|
|
-- | Starts a new season
|
|
startNewSeason :: ProgState -> ProgState
|
|
startNewSeason = (progMode .~ NewSeason) . (database . dbGames .~ 0)
|
|
|
|
-- | Resets all players year-to-date stats
|
|
resetYtd :: ProgState -> ProgState
|
|
resetYtd
|
|
= (database . dbPlayers %~ map (pYtd .~ newPlayerStats))
|
|
. (database . dbGoalies %~ map (gYtd .~ newGoalieStats))
|
|
|
|
-- | Starts a new game
|
|
startNewGame :: ProgState -> ProgState
|
|
startNewGame
|
|
= (progMode .~ NewGame newGameState)
|
|
. (database . dbGames %~ succ)
|
|
|
|
-- | Adds a character to the input buffer
|
|
addChar :: Char -> ProgState -> ProgState
|
|
addChar c = inputBuffer %~ (++[c])
|
|
|
|
-- | Removes a character from the input buffer (if possible)
|
|
removeChar :: ProgState -> ProgState
|
|
removeChar = inputBuffer %~ \case
|
|
"" -> ""
|
|
str -> init str
|
|
|
|
-- | Determines whether or not to perform a check for overtime
|
|
overtimeCheck :: ProgState -> ProgState
|
|
overtimeCheck s
|
|
| fromMaybe False $ gameTied $ s^.progMode.gameStateL =
|
|
s & progMode.gameStateL
|
|
%~ (homeScore .~ Nothing)
|
|
. (awayScore .~ Nothing)
|
|
| 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)
|
|
|
|
-- | Validates the game date
|
|
validateGameDate :: ProgState -> ProgState
|
|
validateGameDate s = fromMaybe s result
|
|
where
|
|
result = do
|
|
y <- toInteger <$> s^.progMode.gameStateL.gameYear
|
|
m <- s^.progMode.gameStateL.gameMonth
|
|
d <- s^.progMode.gameStateL.gameDay
|
|
Just $ if null $ fromGregorianValid y m d
|
|
then s & progMode.gameStateL
|
|
%~ (gameYear .~ Nothing)
|
|
. (gameMonth .~ Nothing)
|
|
. (gameDay .~ Nothing)
|
|
else s
|