brick/programs/ListDemo.hs

106 lines
3.0 KiB
Haskell
Raw Normal View History

{-# LANGUAGE CPP #-}
2015-06-28 23:06:22 +03:00
module Main where
import Lens.Micro ((^.))
import Lens.Micro.Mtl
2015-06-28 23:06:22 +03:00
import Control.Monad (void)
import Control.Monad.State (modify)
#if !(MIN_VERSION_base(4,11,0))
2015-06-28 23:06:22 +03:00
import Data.Monoid
#endif
import Data.Maybe (fromMaybe)
2015-07-10 23:25:44 +03:00
import qualified Graphics.Vty as V
2015-06-28 23:06:22 +03:00
2015-07-10 23:25:44 +03:00
import qualified Brick.Main as M
import qualified Brick.Types as T
import qualified Brick.Widgets.Border as B
import qualified Brick.Widgets.List as L
import qualified Brick.Widgets.Center as C
import qualified Brick.AttrMap as A
import qualified Data.Vector as Vec
import Brick.Types
2015-07-10 23:25:44 +03:00
( Widget
)
import Brick.Widgets.Core
( (<+>)
2015-07-10 23:25:44 +03:00
, str
, vLimit
, hLimit
, vBox
, withAttr
)
2015-07-12 02:53:06 +03:00
import Brick.Util (fg, on)
2015-06-28 23:06:22 +03:00
drawUI :: (Show a) => L.List () a -> [Widget ()]
2015-06-28 23:22:05 +03:00
drawUI l = [ui]
2015-06-28 23:06:22 +03:00
where
label = str "Item " <+> cur <+> str " of " <+> total
2015-07-10 23:25:44 +03:00
cur = case l^.(L.listSelectedL) of
Nothing -> str "-"
2015-06-28 23:06:22 +03:00
Just i -> str (show (i + 1))
total = str $ show $ Vec.length $ l^.(L.listElementsL)
2015-07-10 23:25:44 +03:00
box = B.borderWithLabel label $
2015-06-28 23:15:15 +03:00
hLimit 25 $
vLimit 15 $
L.renderList listDrawElement True l
2015-07-10 23:25:44 +03:00
ui = C.vCenter $ vBox [ C.hCenter box
, str " "
, C.hCenter $ str "Press +/- to add/remove list elements."
, C.hCenter $ str "Press Esc to exit."
2015-07-10 23:25:44 +03:00
]
2015-06-28 23:06:22 +03:00
appEvent :: T.BrickEvent () e -> T.EventM () (L.List () Char) ()
appEvent (T.VtyEvent e) =
2015-06-28 23:06:22 +03:00
case e of
V.EvKey (V.KChar '+') [] -> do
els <- use L.listElementsL
let el = nextElement els
pos = Vec.length els
modify $ L.listInsert pos el
2015-06-28 23:06:22 +03:00
V.EvKey (V.KChar '-') [] -> do
sel <- use L.listSelectedL
case sel of
Nothing -> return ()
Just i -> modify $ L.listRemove i
2015-06-28 23:15:15 +03:00
V.EvKey V.KEsc [] -> M.halt
2015-06-28 23:06:22 +03:00
ev -> L.handleListEvent ev
where
nextElement :: Vec.Vector Char -> Char
nextElement v = fromMaybe '?' $ Vec.find (flip Vec.notElem v) (Vec.fromList ['a' .. 'z'])
appEvent _ = return ()
2015-06-28 23:06:22 +03:00
listDrawElement :: (Show a) => Bool -> a -> Widget ()
listDrawElement sel a =
2015-06-28 23:06:22 +03:00
let selStr s = if sel
then withAttr customAttr (str $ "<" <> s <> ">")
2015-06-28 23:06:22 +03:00
else str s
in C.hCenter $ str "Item " <+> (selStr $ show a)
2015-06-28 23:06:22 +03:00
initialState :: L.List () Char
2016-05-20 06:29:12 +03:00
initialState = L.list () (Vec.fromList ['a','b','c']) 1
2015-06-28 23:06:22 +03:00
2015-07-10 23:25:44 +03:00
customAttr :: A.AttrName
customAttr = L.listSelectedAttr <> A.attrName "custom"
2015-06-28 23:06:22 +03:00
2015-07-10 23:25:44 +03:00
theMap :: A.AttrMap
theMap = A.attrMap V.defAttr
[ (L.listAttr, V.white `on` V.blue)
, (L.listSelectedAttr, V.blue `on` V.white)
, (customAttr, fg V.cyan)
2015-06-28 23:06:22 +03:00
]
theApp :: M.App (L.List () Char) e ()
2015-06-28 23:06:22 +03:00
theApp =
2015-07-10 23:25:44 +03:00
M.App { M.appDraw = drawUI
, M.appChooseCursor = M.showFirstCursor
, M.appHandleEvent = appEvent
, M.appStartEvent = return ()
2015-07-10 23:25:44 +03:00
, M.appAttrMap = const theMap
}
2015-06-28 23:06:22 +03:00
main :: IO ()
2015-07-10 23:25:44 +03:00
main = void $ M.defaultMain theApp initialState