diff --git a/src/Mtlstats/Menu.hs b/src/Mtlstats/Menu.hs index aad2274..7c42a91 100644 --- a/src/Mtlstats/Menu.hs +++ b/src/Mtlstats/Menu.hs @@ -37,7 +37,6 @@ module Mtlstats.Menu ( 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 @@ -87,7 +86,7 @@ menuStateController menuFunc = Controller drawMenu :: Menu a -> Widget () drawMenu m = let menuLines = lines $ show m - in hCenter $ vBox $ map str menuLines + in hCenter $ linesToWidget menuLines -- | The event handler for a 'Menu' menuHandler :: Menu a -> Handler a diff --git a/src/Mtlstats/Util.hs b/src/Mtlstats/Util.hs index c06e520..6aaae46 100644 --- a/src/Mtlstats/Util.hs +++ b/src/Mtlstats/Util.hs @@ -19,6 +19,8 @@ along with this program. If not, see . -} +{-# LANGUAGE LambdaCase #-} + module Mtlstats.Util ( nth , modifyNth @@ -123,4 +125,8 @@ capitalizeName ch s = s ++ [ch'] | otherwise = False linesToWidget :: [String] -> Widget () -linesToWidget = vBox . map str +linesToWidget = vBox . map + ( str . \case + "" -> " " + s -> s + )