wip: switching from ncurses to brick

src/Mtlstats/Menu.hs

@@ -34,40 +34,39 @@ module Mtlstats.Menu (
   editMenu
 ) where
 
-import Control.Monad.Trans.State (gets, modify)
+import Brick.Main (halt)
+import Brick.Types (BrickEvent (VtyEvent), Widget)
+import Brick.Widgets.Center (hCenter)
+import Brick.Widgets.Core (str, vBox)
+import Control.Monad.State.Class (gets, modify)
 import Data.Char (toUpper)
 import qualified Data.Map as M
 import Data.Maybe (mapMaybe)
+import Graphics.Vty.Input.Events (Event (EvKey), Key (KChar))
 import Lens.Micro ((^.), (?~))
-import qualified UI.NCurses as C
 
 import Mtlstats.Actions
 import qualified Mtlstats.Actions.NewGame.GoalieInput as GI
 import Mtlstats.Actions.EditStandings
-import Mtlstats.Format
 import Mtlstats.Types
 import Mtlstats.Types.Menu
 import Mtlstats.Util
 
 -- | Generates a simple 'Controller' for a Menu
 menuController :: Menu () -> Controller
-menuController = menuControllerWith $ const $ return ()
+menuController = menuControllerWith $ const id
 
 -- | Generate a simple 'Controller' for a 'Menu' with a header
 menuControllerWith
-  :: (ProgState -> C.Update ())
-  -- ^ Generates the header
+  :: (ProgState -> Widget () -> Widget())
+  -- ^ Function to attach the header
   -> Menu ()
   -- ^ The menu
   -> Controller
   -- ^ The resulting controller
 menuControllerWith header menu = Controller
-  { drawController = \s -> do
-    header s
-    drawMenu menu
-  , handleController = \e -> do
-    menuHandler menu e
-    return True
+  { drawController = \s -> header s $ drawMenu menu
+  , handleController = menuHandler menu
   }
 
 -- | Generate and create a controller for a menu based on the current
@@ -82,38 +81,33 @@ menuStateController menuFunc = Controller
   , handleController = \e -> do
     menu <- gets menuFunc
     menuHandler menu e
-    return True
   }
 
 -- | The draw function for a 'Menu'
-drawMenu :: Menu a -> C.Update C.CursorMode
-drawMenu m = do
-  (_, cols) <- C.windowSize
-  let
-    width    = fromIntegral $ pred cols
-    menuText = map (centre width) $ lines $ show m
-  C.drawString $ unlines menuText
-  return C.CursorInvisible
+drawMenu :: Menu a -> Widget ()
+drawMenu m = let
+  menuLines = lines $ show m
+  in hCenter $ vBox $ map str menuLines
 
 -- | The event handler for a 'Menu'
-menuHandler :: Menu a -> C.Event -> Action a
-menuHandler m (C.EventCharacter c) =
+menuHandler :: Menu a -> Handler a
+menuHandler m (VtyEvent (EvKey (KChar c) [])) =
   case filter (\i -> i^.miKey == toUpper c) $ m^.menuItems of
     i:_ -> i^.miAction
     []  -> return $ m^.menuDefault
 menuHandler m _ = return $ m^.menuDefault
 
 -- | The main menu
-mainMenu :: Menu Bool
-mainMenu = Menu "MASTER MENU" True
+mainMenu :: Menu ()
+mainMenu = Menu "MASTER MENU" ()
   [ MenuItem 'A' "NEW SEASON" $
-    modify startNewSeason >> return True
+    modify startNewSeason
   , MenuItem 'B' "NEW GAME" $
-    modify startNewGame >> return True
+    modify startNewGame
   , MenuItem 'C' "EDIT MENU" $
-    modify edit >> return True
+    modify edit
   , MenuItem 'E' "EXIT" $
-    saveDatabase >> return False
+    saveDatabase >> halt
   ]
 
 -- | The new season menu