brick/programs/FileBrowserDemo.hs
2021-04-29 12:14:08 -07:00

106 lines
3.4 KiB
Haskell

{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE CPP #-}
module Main where
import qualified Control.Exception as E
#if !(MIN_VERSION_base(4,11,0))
import Data.Monoid
#endif
import qualified Graphics.Vty as V
import qualified Data.Text as Text
import qualified Brick.Main as M
import qualified Brick.Widgets.List as L
import Brick.AttrMap (AttrName)
import Brick.Types
( Widget
, BrickEvent(..)
)
import Brick.Widgets.Center
( center
, hCenter
)
import Brick.Widgets.Border
( borderWithLabel
)
import Brick.Widgets.Core
( vBox, (<=>), padTop
, hLimit, vLimit, txt
, withDefAttr, emptyWidget
)
import qualified Brick.Widgets.FileBrowser as FB
import qualified Brick.AttrMap as A
import Brick.Util (on, fg)
import qualified Brick.Types as T
data Name = FileBrowser1
deriving (Eq, Show, Ord)
drawUI :: FB.FileBrowser Name -> [Widget Name]
drawUI b = [center $ ui <=> help]
where
ui = hCenter $
vLimit 15 $
hLimit 50 $
borderWithLabel (txt "Choose a file") $
FB.renderFileBrowser True b
help = padTop (T.Pad 1) $
vBox [ case FB.fileBrowserException b of
Nothing -> emptyWidget
Just e -> hCenter $ withDefAttr errorAttr $
txt $ Text.pack $ E.displayException e
, hCenter $ txt "Up/Down: select"
, hCenter $ txt "/: search, Ctrl-C or Esc: cancel search"
, hCenter $ txt "Enter: change directory or select file"
, hCenter $ txt "Esc: quit"
]
appEvent :: FB.FileBrowser Name -> BrickEvent Name e -> T.EventM Name (T.Next (FB.FileBrowser Name))
appEvent b (VtyEvent ev) =
case ev of
V.EvKey V.KEsc [] | not (FB.fileBrowserIsSearching b) ->
M.halt b
_ -> do
b' <- FB.handleFileBrowserEvent ev b
-- If the browser has a selected file after handling the
-- event (because the user pressed Enter), shut down.
case ev of
V.EvKey V.KEnter [] ->
case FB.fileBrowserSelection b' of
[] -> M.continue b'
_ -> M.halt b'
_ -> M.continue b'
appEvent b _ = M.continue b
errorAttr :: AttrName
errorAttr = "error"
theMap :: A.AttrMap
theMap = A.attrMap V.defAttr
[ (L.listSelectedFocusedAttr, V.black `on` V.yellow)
, (FB.fileBrowserCurrentDirectoryAttr, V.white `on` V.blue)
, (FB.fileBrowserSelectionInfoAttr, V.white `on` V.blue)
, (FB.fileBrowserDirectoryAttr, fg V.blue)
, (FB.fileBrowserBlockDeviceAttr, fg V.magenta)
, (FB.fileBrowserCharacterDeviceAttr, fg V.green)
, (FB.fileBrowserNamedPipeAttr, fg V.yellow)
, (FB.fileBrowserSymbolicLinkAttr, fg V.cyan)
, (FB.fileBrowserUnixSocketAttr, fg V.red)
, (FB.fileBrowserSelectedAttr, V.white `on` V.magenta)
, (errorAttr, fg V.red)
]
theApp :: M.App (FB.FileBrowser Name) e Name
theApp =
M.App { M.appDraw = drawUI
, M.appChooseCursor = M.showFirstCursor
, M.appHandleEvent = appEvent
, M.appStartEvent = return
, M.appAttrMap = const theMap
}
main :: IO ()
main = do
b <- M.defaultMain theApp =<< FB.newFileBrowser FB.selectNonDirectories FileBrowser1 Nothing
putStrLn $ "Selected entry: " <> show (FB.fileBrowserSelection b)