75 lines
2.1 KiB
Haskell
75 lines
2.1 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 ScopedTypeVariables #-}
|
|
|
|
module Mtlstats (initState, mainLoop) where
|
|
|
|
import Control.Exception (IOException, catch)
|
|
import Control.Monad (void)
|
|
import Control.Monad.Extra (whenM)
|
|
import Control.Monad.IO.Class (liftIO)
|
|
import Control.Monad.Trans.Class (lift)
|
|
import Control.Monad.Trans.State (get, gets)
|
|
import Data.Aeson (decodeFileStrict)
|
|
import Data.Maybe (fromJust, fromMaybe)
|
|
import Lens.Micro ((&), (.~))
|
|
import System.EasyFile (getAppUserDataDirectory, (</>))
|
|
import qualified UI.NCurses as C
|
|
|
|
import Mtlstats.Config
|
|
import Mtlstats.Control
|
|
import Mtlstats.Types
|
|
|
|
-- | Initializes the progran
|
|
initState :: C.Curses ProgState
|
|
initState = do
|
|
C.setEcho False
|
|
void $ C.setCursorMode C.CursorInvisible
|
|
db <- liftIO $ do
|
|
dir <- getAppUserDataDirectory appName
|
|
let dbFile = dir </> dbFname
|
|
fromMaybe newDatabase <$> catch
|
|
(decodeFileStrict dbFile)
|
|
(\(_ :: IOException) -> return Nothing)
|
|
return
|
|
$ newProgState
|
|
& database .~ db
|
|
|
|
-- | Main program loop
|
|
mainLoop :: Action ()
|
|
mainLoop = do
|
|
c <- gets dispatch
|
|
get >>= lift . draw . drawController c
|
|
w <- lift C.defaultWindow
|
|
whenM (lift (fromJust <$> C.getEvent w Nothing) >>= handleController c)
|
|
mainLoop
|
|
|
|
draw :: C.Update C.CursorMode -> C.Curses ()
|
|
draw u = do
|
|
void $ C.setCursorMode C.CursorInvisible
|
|
w <- C.defaultWindow
|
|
cm <- C.updateWindow w $ do
|
|
C.clear
|
|
u
|
|
C.render
|
|
void $ C.setCursorMode cm
|