@@ -2,116 +2,41 @@
2Module: Reflex.Vty.Widget
2Module: Reflex.Vty.Widget
3Description: Basic set of widgets and building blocks for reflex-vty applications
3Description: Basic set of widgets and building blocks for reflex-vty applications
5โ{-# LANGUAGE FlexibleContexts #-}
5+{-# Language UndecidableInstances #-}
6โ{-# LANGUAGE FlexibleInstances #-}
6+
7โ{-# LANGUAGE FunctionalDependencies #-}
8โ{-# LANGUAGE GeneralizedNewtypeDeriving #-}
9โ{-# LANGUAGE LambdaCase #-}
10โ{-# LANGUAGE MultiParamTypeClasses #-}
11โ{-# LANGUAGE OverloadedStrings #-}
12โ{-# LANGUAGE RankNTypes #-}
13โ{-# LANGUAGE RecursiveDo #-}
14โ{-# LANGUAGE TypeFamilies #-}
15โmodule Reflex.Vty.Widget
7+module Reflex.Vty.Widget where
21โ , mainWidgetWithHandle
22โ , HasDisplaySize(..)
51import Control.Applicative (liftA2)
9import Control.Applicative (liftA2)
52import Control.Monad.Fix (MonadFix)
10import Control.Monad.Fix (MonadFix)
53โimport Control.Monad.Trans (lift)
11+import Control.Monad.IO.Class (MonadIO)
54โimport Control.Monad.Trans.Reader (ReaderT, runReaderT, asks, ask)
12+import Control.Monad.Morph
13+import Control.Monad.NodeId
14+import Control.Monad.Reader (ReaderT, ask, local, runReaderT)
55โimport Control.Monad.Trans.Writer (WriterT, runWriterT, censor, tell)
15+import Control.Monad.Ref
56โimport Data.Default (Default(..))
16+import Control.Monad.Trans (MonadTrans, lift)
57โimport Data.Text (Text)
17+import Data.Set (Set)
58โimport qualified Data.Text as T
18+import qualified Data.Set as Set
59โimport qualified Data.Text.Zipper as TZ
60import Graphics.Vty (Image)
19import Graphics.Vty (Image)
61import qualified Graphics.Vty as V
20import qualified Graphics.Vty as V
62import Reflex
21import Reflex
63โ
22+import Reflex.Class ()
23+import Reflex.Host.Class (MonadReflexCreateTrigger)
64import Reflex.Vty.Host
24import Reflex.Vty.Host
66โ-- | The context within which a 'VtyWidget' runs
26+-- * Running a vty application
67โdata VtyWidgetCtx t = VtyWidgetCtx
68โ { _vtyWidgetCtx_size :: Dynamic t (Int,Int)
69โ -- ^ The width and height of the region allocated to the widget.
70โ , _vtyWidgetCtx_focus :: Dynamic t Bool
71โ -- ^ Whether the widget should behave as if it has focus for keyboard input.
72โ , _vtyWidgetCtx_input :: Event t VtyEvent
73โ -- ^ User input events that the widget's parent chooses to share. These will generally
74โ -- be filtered for relevance:
75โ -- * Keyboard inputs are restricted to focused widgets
76โ -- * Mouse inputs are restricted to the region in which the widget resides and are
77โ -- translated into its internal coordinates.
80โ-- | The output of a 'VtyWidget'
81โdata VtyWidgetOut t = VtyWidgetOut
82โ { _vtyWidgetOut_images :: Behavior t [Image]
83โ -- ^ The visual output of the 'VtyWidget'
84โ , _vtyWidgetOut_shutdown :: Event t ()
87โinstance Reflex t => Semigroup (VtyWidgetOut t) where
88โ wo <> wo' = VtyWidgetOut
89โ { _vtyWidgetOut_images = _vtyWidgetOut_images wo <> _vtyWidgetOut_images wo'
90โ , _vtyWidgetOut_shutdown = _vtyWidgetOut_shutdown wo <> _vtyWidgetOut_shutdown wo'
93โinstance (Reflex t) => Monoid (VtyWidgetOut t) where
94โ mempty = VtyWidgetOut mempty mempty
95โ mappend wo wo' = wo <> wo'
97โ-- | A widget that can read its context and produce image output
98โnewtype VtyWidget t m a = VtyWidget
99โ { unVtyWidget :: WriterT (VtyWidgetOut t) (ReaderT (VtyWidgetCtx t) m) a
101โ deriving (Functor, Applicative, Monad, MonadSample t, MonadHold t, MonadFix)
103โ-- | Runs a 'VtyWidget' with a given context
107โ -> VtyWidget t m a
108โ -> m (a, VtyWidgetOut t)
109โrunVtyWidget ctx w = runReaderT (runWriterT (unVtyWidget w)) ctx
111โ-- | Sets up the top-level context for a 'VtyWidget' and runs it with that context
28+-- | Sets up the top-level context for a vty widget and runs it with that context
112mainWidgetWithHandle
29mainWidgetWithHandle
114โ -> (forall t m. MonadVtyApp t m => VtyWidget t m ())
32+ ( MonadVtyApp t m
35+ , HasDisplayRegion t m
116mainWidgetWithHandle vty child =
41mainWidgetWithHandle vty child =
117 runVtyAppWithHandle vty $ \dr0 inp -> do
42 runVtyAppWithHandle vty $ \dr0 inp -> do
@@ -121,69 +46,178 @@ mainWidgetWithHandle vty child =
121 let inp' = fforMaybe inp $ \case
46 let inp' = fforMaybe inp $ \case
122 V.EvResize {} -> Nothing
47 V.EvResize {} -> Nothing
123 x -> Just x
48 x -> Just x
124โ let ctx = VtyWidgetCtx
49+ (shutdown, images) <- runThemeReader (constant V.defAttr) $
125โ { _vtyWidgetCtx_size = size
50+ runFocusReader (pure True) $
126โ , _vtyWidgetCtx_input = inp'
51+ runDisplayRegion (fmap (\(w, h) -> Region 0 0 w h) size) $
127โ , _vtyWidgetCtx_focus = constDyn True
52+ runImageWriter $
129โ ((), wo) <- runVtyWidget ctx child
54+ runInput inp' $ do
55+ tellImages . ffor (current size) $ \(w, h) -> [V.charFill V.defAttr ' ' w h]
130 return $ VtyResult
57 return $ VtyResult
131โ { _vtyResult_picture = fmap (V.picForLayers . reverse) (_vtyWidgetOut_images wo)
58+ { _vtyResult_picture = fmap (V.picForLayers . reverse) images
132โ , _vtyResult_shutdown = _vtyWidgetOut_shutdown wo
59+ , _vtyResult_shutdown = shutdown
62+-- | The output of a vty widget
63+data VtyWidgetOut t = VtyWidgetOut
64+ { _vtyWidgetOut_shutdown :: Event t ()
135-- | Like 'mainWidgetWithHandle', but uses a default vty configuration
67-- | Like 'mainWidgetWithHandle', but uses a default vty configuration
136mainWidget
68mainWidget
137โ :: (forall t m. MonadVtyApp t m => VtyWidget t m ())
70+ ( MonadVtyApp t m
73+ , HasDisplayRegion t m
139mainWidget child = do
79mainWidget child = do
140 vty <- getDefaultVty
80 vty <- getDefaultVty
141 mainWidgetWithHandle vty child
81 mainWidgetWithHandle vty child
143โ-- | A class for things that know their own display size dimensions
83+-- * Input Events
144โclass (Reflex t, Monad m) => HasDisplaySize t m | m -> t where
145โ displaySize :: m (Dynamic t (Int, Int))
147โinstance (Reflex t, Monad m) => HasDisplaySize t (VtyWidget t m) where
148โ displaySize = VtyWidget . lift $ asks _vtyWidgetCtx_size
150โ-- | Retrieve the display width (columns)
151โdisplayWidth :: HasDisplaySize t m => m (Dynamic t Int)
152โdisplayWidth = fmap fst <$> displaySize
154โ-- | Retrieve the display height (rows)
155โdisplayHeight :: HasDisplaySize t m => m (Dynamic t Int)
156โdisplayHeight = fmap snd <$> displaySize
158-- | A class for things that can receive vty events as input
85-- | A class for things that can receive vty events as input
159โclass HasVtyInput t m | m -> t where
86+class HasInput t m | m -> t where
160 input :: m (Event t VtyEvent)
87 input :: m (Event t VtyEvent)
88+ default input :: (f m' ~ m, Monad m', MonadTrans f, HasInput t m') => m (Event t VtyEvent)
90+ -- | User input events that the widget's parent chooses to share. These will generally
91+ -- be filtered for relevance.
92+ localInput :: (Event t VtyEvent -> Event t VtyEvent) -> m a -> m a
93+ default localInput :: (f m' ~ m, Monad m', MFunctor f, HasInput t m') => (Event t VtyEvent -> Event t VtyEvent) -> m a -> m a
94+ localInput f = hoist (localInput f)
96+instance (Reflex t, Monad m) => HasInput t (Input t m) where
98+ localInput f (Input m) = Input $ local f m
100+-- | A widget that can receive input events. See 'Graphics.Vty.Event'
101+newtype Input t m a = Input
102+ { unInput :: ReaderT (Event t VtyEvent) m a
114+instance (Adjustable t m, MonadHold t m, Reflex t) => Adjustable t (Input t m) where
115+ runWithReplace a0 a' = Input $ runWithReplace (unInput a0) $ fmap unInput a'
116+ traverseIntMapWithKeyWithAdjust f dm0 dm' = Input $
117+ traverseIntMapWithKeyWithAdjust (\k v -> unInput (f k v)) dm0 dm'
118+ traverseDMapWithKeyWithAdjust f dm0 dm' = Input $ do
119+ traverseDMapWithKeyWithAdjust (\k v -> unInput (f k v)) dm0 dm'
120+ traverseDMapWithKeyWithAdjustWithMove f dm0 dm' = Input $ do
121+ traverseDMapWithKeyWithAdjustWithMove (\k v -> unInput (f k v)) dm0 dm'
123+deriving instance MonadReflexCreateTrigger t m => MonadReflexCreateTrigger t (Input t m)
124+deriving instance NotReady t m => NotReady t (Input t m)
125+deriving instance PerformEvent t m => PerformEvent t (Input t m)
126+deriving instance PostBuild t m => PostBuild t (Input t m)
127+deriving instance TriggerEvent t m => TriggerEvent t (Input t m)
128+instance HasImageWriter t m => HasImageWriter t (Input t m)
129+instance HasDisplayRegion t m => HasDisplayRegion t (Input t m)
130+instance HasFocusReader t m => HasFocusReader t (Input t m)
132+instance MonadTrans (Input t) where
133+ lift f = Input $ lift f
135+instance MFunctor (Input t) where
136+ hoist f = Input . hoist f . unInput
138+instance MonadNodeId m => MonadNodeId (Input t m)
140+-- | Runs an 'Input' with a given context
146+runInput e w = runReaderT (unInput w) e
148+-- ** Filtering input
150+-- | Type synonym for a key and modifier combination
151+type KeyCombo = (V.Key, [V.Modifier])
153+-- | Emits an event that fires on a particular key press (without modifiers)
154+key :: (Monad m, Reflex t, HasInput t m) => V.Key -> m (Event t KeyCombo)
155+key = keyCombos . Set.singleton . (,[])
157+-- | Emits an event that fires on particular key presses (without modifiers)
158+keys :: (Monad m, Reflex t, HasInput t m) => [V.Key] -> m (Event t KeyCombo)
159+keys = keyCombos . Set.fromList . fmap (,[])
161+-- | Emit an event that fires whenever the provided key combination occurs
163+ :: (Reflex t, Monad m, HasInput t m)
165+ -> m (Event t KeyCombo)
166+keyCombo = keyCombos . Set.singleton
168+-- | Emit an event that fires whenever any of the provided key combinations occur
170+ :: (Reflex t, Monad m, HasInput t m)
172+ -> m (Event t KeyCombo)
175+ return $ fforMaybe i $ \case
176+ V.EvKey k m -> if Set.member (k, m) ks
162โinstance (Reflex t, Monad m) => HasVtyInput t (VtyWidget t m) where
181+-- | Filter the keyboard input that a child widget may receive
163โ input = VtyWidget . lift $ asks _vtyWidgetCtx_input
182+filterKeys :: (Reflex t, HasInput t m) => (KeyCombo -> Bool) -> m a -> m a
164โ
183+filterKeys f x = localInput (ffilter (\case
165โ-- | A class for things that can dynamically gain and lose focus
184+ V.EvKey k mods -> f (k, mods)
166โclass HasFocus t m | m -> t where
185+ _ -> True)) x
167โ focus :: m (Dynamic t Bool)
186+
168โ
187+-- | Filter mouse input events based on whether they target a particular region
169โinstance (Reflex t, Monad m) => HasFocus t (VtyWidget t m) where
188+-- and translate them to the internal coordinate system of that region.
170โ focus = VtyWidget . lift $ asks _vtyWidgetCtx_focus
189+--
171โ
190+-- NB: Non-mouse events are passed through unfiltered and unchanged
172โ-- | A class for widgets that can produce images to draw to the display
191+mouseInRegion :: Region -> VtyEvent -> Maybe VtyEvent
173โclass (Reflex t, Monad m) => ImageWriter t m | m -> t where
192+mouseInRegion (Region l t w h) e = case e of
174โ -- | Send images upstream for rendering
193+ V.EvMouseDown x y btn m -> mouse (\u v -> V.EvMouseDown u v btn m) x y
175โ tellImages :: Behavior t [Image] -> m ()
194+ V.EvMouseUp x y btn -> mouse (\u v -> V.EvMouseUp u v btn) x y
177โinstance (Reflex t, Monad m) => ImageWriter t (VtyWidget t m) where
196+ where
178โ tellImages imgs = VtyWidget $ tell (mempty { _vtyWidgetOut_images = imgs })
197+ mouse con x y
180โ-- | A class for things that can shut down the vty application
199+ , y < t
181โclass (Reflex t, Monad m) => Shutdown t m where
200+ , x >= l + w
182โ -- | Signals that the vty application should shut down
201+ , y >= t + h ] = Nothing
183โ tellShutdown :: Event t () -> m ()
202+ | otherwise =
203+ Just (con (x - l) (y - t))
205+-- | Filter mouse input outside the current display region and
206+-- all input if the region is not focused
208+ :: (HasDisplayRegion t m, HasFocusReader t m, HasInput t m)
209+ => m (Event t VtyEvent)
210+inputInFocusedRegion = do
212+ reg <- current <$> askRegion
213+ foc <- current <$> focus
214+ pure $ fmapMaybe id $ attachWith filterInput ((,) <$> reg <*> foc) inp
216+ filterInput (r, f) = \case
217+ V.EvKey {} | not f -> Nothing
218+ x -> mouseInRegion r x
185โinstance (Reflex t, Monad m) => Shutdown t (VtyWidget t m) where
220+-- * Getting and setting the display region
186โ tellShutdown sd = VtyWidget $ tell (mempty { _vtyWidgetOut_shutdown = sd })
188-- | A chunk of the display area
222-- | A chunk of the display area
189data Region = Region
223data Region = Region
@@ -194,15 +228,293 @@ data Region = Region
195 deriving (Show, Read, Eq, Ord)
229 deriving (Show, Read, Eq, Ord)
231+-- | A region that occupies no space.
233+nilRegion = Region 0 0 0 0
197-- | The width and height of a 'Region'
235-- | The width and height of a 'Region'
198regionSize :: Region -> (Int, Int)
236regionSize :: Region -> (Int, Int)
199regionSize (Region _ _ w h) = (w, h)
237regionSize (Region _ _ w h) = (w, h)
202-- | Produces an 'Image' that fills a region with space characters
239-- | Produces an 'Image' that fills a region with space characters
203โregionBlankImage :: Region -> Image
240+regionBlankImage :: V.Attr -> Region -> Image
204โregionBlankImage r@(Region _ _ width height) =
241+regionBlankImage attr r@(Region _ _ width height) =
205โ withinImage r $ V.charFill V.defAttr ' ' width height
242+ withinImage r $ V.charFill attr ' ' width height
244+-- | A class for things that know their own display size dimensions
245+class (Reflex t, Monad m) => HasDisplayRegion t m | m -> t where
246+ -- | Retrieve the display region
247+ askRegion :: m (Dynamic t Region)
248+ default askRegion :: (f m' ~ m, MonadTrans f, HasDisplayRegion t m') => m (Dynamic t Region)
249+ askRegion = lift askRegion
250+ -- | Run an action in a local region, by applying a transformation to the region
251+ localRegion :: (Dynamic t Region -> Dynamic t Region) -> m a -> m a
252+ default localRegion :: (f m' ~ m, Monad m', MFunctor f, HasDisplayRegion t m') => (Dynamic t Region -> Dynamic t Region) -> m a -> m a
253+ localRegion f = hoist (localRegion f)
255+-- | Retrieve the display width
256+displayWidth :: HasDisplayRegion t m => m (Dynamic t Int)
257+displayWidth = fmap _region_width <$> askRegion
259+-- | Retrieve the display height
260+displayHeight :: HasDisplayRegion t m => m (Dynamic t Int)
261+displayHeight = fmap _region_height <$> askRegion
263+instance HasDisplayRegion t m => HasDisplayRegion t (ReaderT x m)
264+instance HasDisplayRegion t m => HasDisplayRegion t (BehaviorWriterT t x m)
265+instance HasDisplayRegion t m => HasDisplayRegion t (DynamicWriterT t x m)
266+instance HasDisplayRegion t m => HasDisplayRegion t (EventWriterT t x m)
267+instance HasDisplayRegion t m => HasDisplayRegion t (NodeIdT m)
269+-- | A widget that has access to a particular region of the vty display
270+newtype DisplayRegion t m a = DisplayRegion
271+ { unDisplayRegion :: ReaderT (Dynamic t Region) m a }
283+instance (Monad m, Reflex t) => HasDisplayRegion t (DisplayRegion t m) where
284+ askRegion = DisplayRegion ask
285+ localRegion f = DisplayRegion . local f . unDisplayRegion
287+deriving instance MonadReflexCreateTrigger t m => MonadReflexCreateTrigger t (DisplayRegion t m)
288+deriving instance NotReady t m => NotReady t (DisplayRegion t m)
289+deriving instance PerformEvent t m => PerformEvent t (DisplayRegion t m)
290+deriving instance PostBuild t m => PostBuild t (DisplayRegion t m)
291+deriving instance TriggerEvent t m => TriggerEvent t (DisplayRegion t m)
292+instance HasImageWriter t m => HasImageWriter t (DisplayRegion t m)
293+instance HasFocusReader t m => HasFocusReader t (DisplayRegion t m)
295+instance (Adjustable t m, MonadFix m, MonadHold t m) => Adjustable t (DisplayRegion t m) where
296+ runWithReplace (DisplayRegion a) e = DisplayRegion $ runWithReplace a $ fmap unDisplayRegion e
297+ traverseIntMapWithKeyWithAdjust f m e = DisplayRegion $ traverseIntMapWithKeyWithAdjust (\k v -> unDisplayRegion $ f k v) m e
298+ traverseDMapWithKeyWithAdjust f m e = DisplayRegion $ traverseDMapWithKeyWithAdjust (\k v -> unDisplayRegion $ f k v) m e
299+ traverseDMapWithKeyWithAdjustWithMove f m e = DisplayRegion $ traverseDMapWithKeyWithAdjustWithMove (\k v -> unDisplayRegion $ f k v) m e
301+instance MonadTrans (DisplayRegion t) where
302+ lift = DisplayRegion . lift
304+instance MFunctor (DisplayRegion t) where
305+ hoist f = DisplayRegion . hoist f . unDisplayRegion
307+instance MonadNodeId m => MonadNodeId (DisplayRegion t m)
309+-- | Run a 'DisplayRegion' action with a given 'Region'
311+ :: (Reflex t, Monad m)
313+ -> DisplayRegion t m a
315+runDisplayRegion r = flip runReaderT r . unDisplayRegion
317+-- * Getting focus state
319+-- | A class for things that can dynamically gain and lose focus
320+class (Reflex t, Monad m) => HasFocusReader t m | m -> t where
321+ focus :: m (Dynamic t Bool)
322+ default focus :: (f m' ~ m, Monad m', MonadTrans f, HasFocusReader t m') => m (Dynamic t Bool)
324+ localFocus :: (Dynamic t Bool -> Dynamic t Bool) -> m a -> m a
325+ default localFocus :: (f m' ~ m, Monad m', MFunctor f, HasFocusReader t m') => (Dynamic t Bool -> Dynamic t Bool) -> m a -> m a
326+ localFocus f = hoist (localFocus f)
328+instance HasFocusReader t m => HasFocusReader t (ReaderT x m)
329+instance HasFocusReader t m => HasFocusReader t (BehaviorWriterT t x m)
330+instance HasFocusReader t m => HasFocusReader t (DynamicWriterT t x m)
331+instance HasFocusReader t m => HasFocusReader t (EventWriterT t x m)
332+instance HasFocusReader t m => HasFocusReader t (NodeIdT m)
334+-- | A widget that has access to information about whether it is focused
335+newtype FocusReader t m a = FocusReader
336+ { unFocusReader :: ReaderT (Dynamic t Bool) m a }
348+instance (Monad m, Reflex t) => HasFocusReader t (FocusReader t m) where
349+ focus = FocusReader ask
350+ localFocus f = FocusReader . local f . unFocusReader
352+deriving instance MonadReflexCreateTrigger t m => MonadReflexCreateTrigger t (FocusReader t m)
353+deriving instance NotReady t m => NotReady t (FocusReader t m)
354+deriving instance PerformEvent t m => PerformEvent t (FocusReader t m)
355+deriving instance PostBuild t m => PostBuild t (FocusReader t m)
356+deriving instance TriggerEvent t m => TriggerEvent t (FocusReader t m)
357+instance HasImageWriter t m => HasImageWriter t (FocusReader t m)
359+instance (Adjustable t m, MonadFix m, MonadHold t m) => Adjustable t (FocusReader t m) where
360+ runWithReplace (FocusReader a) e = FocusReader $ runWithReplace a $ fmap unFocusReader e
361+ traverseIntMapWithKeyWithAdjust f m e = FocusReader $ traverseIntMapWithKeyWithAdjust (\k v -> unFocusReader $ f k v) m e
362+ traverseDMapWithKeyWithAdjust f m e = FocusReader $ traverseDMapWithKeyWithAdjust (\k v -> unFocusReader $ f k v) m e
363+ traverseDMapWithKeyWithAdjustWithMove f m e = FocusReader $ traverseDMapWithKeyWithAdjustWithMove (\k v -> unFocusReader $ f k v) m e
365+instance MonadTrans (FocusReader t) where
366+ lift = FocusReader . lift
368+instance MFunctor (FocusReader t) where
369+ hoist f = FocusReader . hoist f . unFocusReader
371+instance MonadNodeId m => MonadNodeId (FocusReader t m)
373+-- | Run a 'FocusReader' action with the given focus value
375+ :: (Reflex t, Monad m)
377+ -> FocusReader t m a
379+runFocusReader b = flip runReaderT b . unFocusReader
383+-- | A class for widgets that can produce images to draw to the display
384+class (Reflex t, Monad m) => HasImageWriter t m | m -> t where
385+ -- | Send images upstream for rendering
386+ tellImages :: Behavior t [Image] -> m ()
387+ default tellImages :: (f m' ~ m, Monad m', MonadTrans f, HasImageWriter t m') => Behavior t [Image] -> m ()
388+ tellImages = lift . tellImages
389+ -- | Apply a transformation to the images produced by the child actions
390+ mapImages :: (Behavior t [Image] -> Behavior t [Image]) -> m a -> m a
391+ default mapImages :: (f m' ~ m, Monad m', MFunctor f, HasImageWriter t m') => (Behavior t [Image] -> Behavior t [Image]) -> m a -> m a
392+ mapImages f = hoist (mapImages f)
394+-- | A widget that can produce images to draw onto the display
395+newtype ImageWriter t m a = ImageWriter
396+ { unImageWriter :: BehaviorWriterT t [Image] m a }
405+ , MonadReflexCreateTrigger t
413+instance MonadTrans (ImageWriter t) where
414+ lift = ImageWriter . lift
416+instance MFunctor (ImageWriter t) where
417+ hoist f = ImageWriter . (hoist f) . unImageWriter
419+instance (Adjustable t m, MonadFix m, MonadHold t m) => Adjustable t (ImageWriter t m) where
420+ runWithReplace (ImageWriter a) e = ImageWriter $ runWithReplace a $ fmap unImageWriter e
421+ traverseIntMapWithKeyWithAdjust f m e = ImageWriter $ traverseIntMapWithKeyWithAdjust (\k v -> unImageWriter $ f k v) m e
422+ traverseDMapWithKeyWithAdjust f m e = ImageWriter $ traverseDMapWithKeyWithAdjust (\k v -> unImageWriter $ f k v) m e
423+ traverseDMapWithKeyWithAdjustWithMove f m e = ImageWriter $ traverseDMapWithKeyWithAdjustWithMove (\k v -> unImageWriter $ f k v) m e
425+instance HasImageWriter t m => HasImageWriter t (ReaderT x m)
426+instance HasImageWriter t m => HasImageWriter t (BehaviorWriterT t x m)
427+instance HasImageWriter t m => HasImageWriter t (DynamicWriterT t x m)
428+instance HasImageWriter t m => HasImageWriter t (EventWriterT t x m)
429+instance HasImageWriter t m => HasImageWriter t (NodeIdT m)
431+instance (Monad m, Reflex t) => HasImageWriter t (ImageWriter t m) where
432+ tellImages = ImageWriter . tellBehavior
433+ mapImages f (ImageWriter x) = ImageWriter $ do
434+ (a, images) <- lift $ runBehaviorWriterT x
435+ tellBehavior $ f images
438+instance HasDisplayRegion t m => HasDisplayRegion t (ImageWriter t m)
439+instance HasFocusReader t m => HasFocusReader t (ImageWriter t m)
441+-- | Run a widget that can produce images
443+ :: (Reflex t, Monad m)
444+ => ImageWriter t m a
445+ -> m (a, Behavior t [Image])
446+runImageWriter = runBehaviorWriterT . unImageWriter
450+-- | A class for things that can be visually styled
451+class (Reflex t, Monad m) => HasTheme t m | m -> t where
452+ theme :: m (Behavior t V.Attr)
453+ default theme :: (f m' ~ m, Monad m', MonadTrans f, HasTheme t m') => m (Behavior t V.Attr)
455+ localTheme :: (Behavior t V.Attr -> Behavior t V.Attr) -> m a -> m a
456+ default localTheme :: (f m' ~ m, Monad m', MFunctor f, HasTheme t m') => (Behavior t V.Attr -> Behavior t V.Attr) -> m a -> m a
457+ localTheme f = hoist (localTheme f)
459+instance HasTheme t m => HasTheme t (ReaderT x m)
460+instance HasTheme t m => HasTheme t (BehaviorWriterT t x m)
461+instance HasTheme t m => HasTheme t (DynamicWriterT t x m)
462+instance HasTheme t m => HasTheme t (EventWriterT t x m)
463+instance HasTheme t m => HasTheme t (NodeIdT m)
464+instance HasTheme t m => HasTheme t (Input t m)
465+instance HasTheme t m => HasTheme t (ImageWriter t m)
466+instance HasTheme t m => HasTheme t (DisplayRegion t m)
467+instance HasTheme t m => HasTheme t (FocusReader t m)
469+-- | A widget that has access to theme information
470+newtype ThemeReader t m a = ThemeReader
471+ { unThemeReader :: ReaderT (Behavior t V.Attr) m a }
483+instance (Monad m, Reflex t) => HasTheme t (ThemeReader t m) where
484+ theme = ThemeReader ask
485+ localTheme f = ThemeReader . local f . unThemeReader
487+deriving instance MonadReflexCreateTrigger t m => MonadReflexCreateTrigger t (ThemeReader t m)
488+deriving instance NotReady t m => NotReady t (ThemeReader t m)
489+deriving instance PerformEvent t m => PerformEvent t (ThemeReader t m)
490+deriving instance PostBuild t m => PostBuild t (ThemeReader t m)
491+deriving instance TriggerEvent t m => TriggerEvent t (ThemeReader t m)
492+instance HasImageWriter t m => HasImageWriter t (ThemeReader t m)
494+instance (Adjustable t m, MonadFix m, MonadHold t m) => Adjustable t (ThemeReader t m) where
495+ runWithReplace (ThemeReader a) e = ThemeReader $ runWithReplace a $ fmap unThemeReader e
496+ traverseIntMapWithKeyWithAdjust f m e = ThemeReader $ traverseIntMapWithKeyWithAdjust (\k v -> unThemeReader $ f k v) m e
497+ traverseDMapWithKeyWithAdjust f m e = ThemeReader $ traverseDMapWithKeyWithAdjust (\k v -> unThemeReader $ f k v) m e
498+ traverseDMapWithKeyWithAdjustWithMove f m e = ThemeReader $ traverseDMapWithKeyWithAdjustWithMove (\k v -> unThemeReader $ f k v) m e
500+instance MonadTrans (ThemeReader t) where
501+ lift = ThemeReader . lift
503+instance MFunctor (ThemeReader t) where
504+ hoist f = ThemeReader . hoist f . unThemeReader
506+instance MonadNodeId m => MonadNodeId (ThemeReader t m)
508+-- | Run a 'ThemeReader' action with the given focus value
510+ :: (Reflex t, Monad m)
511+ => Behavior t V.Attr
512+ -> ThemeReader t m a
514+runThemeReader b = flip runReaderT b . unThemeReader
517+-- ** Manipulating images
207-- | Translates and crops an 'Image' so that it is contained by
519-- | Translates and crops an 'Image' so that it is contained by
208-- the given 'Region'.
520-- the given 'Region'.
@@ -214,7 +526,17 @@ withinImage (Region left top width height)
214 | width < 0 || height < 0 = withinImage (Region left top 0 0)
526 | width < 0 || height < 0 = withinImage (Region left top 0 0)
215 | otherwise = V.translate left top . V.crop width height
527 | otherwise = V.translate left top . V.crop width height
529+-- | Crop a behavior of images to a behavior of regions. See 'withinImage'.
532+ => Behavior t Region
533+ -> Behavior t [Image]
534+ -> Behavior t [Image]
535+imagesInRegion reg = liftA2 (\r is -> map (withinImage r) is) reg
537+-- * Running sub-widgets
217โ-- | Low-level widget combinator that runs a child 'VtyWidget' within
539+-- | Low-level widget combinator that runs a child widget within
218-- a given region and context. This widget filters and modifies the input
540-- a given region and context. This widget filters and modifies the input
219-- that the child widget receives such that:
541-- that the child widget receives such that:
220-- * unfocused widgets receive no key events
542-- * unfocused widgets receive no key events
@@ -222,288 +544,18 @@ withinImage (Region left top width height)
222-- * mouse inputs inside the region have their coordinates translated such
544-- * mouse inputs inside the region have their coordinates translated such
223-- that (0,0) is the top-left corner of the region
545-- that (0,0) is the top-left corner of the region
225โ :: (Reflex t, Monad m)
547+ :: (Reflex t, Monad m, HasInput t m, HasImageWriter t m, HasDisplayRegion t m, HasFocusReader t m)
226โ => Dynamic t Region -- ^ Region into which we should draw the widget (in coordinates relative to our own)
548+ => Dynamic t Region
227 -> Dynamic t Bool -- ^ Whether the widget should be focused when the parent is.
549 -> Dynamic t Bool -- ^ Whether the widget should be focused when the parent is.
228โ -> VtyWidget t m a
550+ -> m a
229โ -> VtyWidget t m a
551+ -> m a
230โpane reg foc child = VtyWidget $ do
552+pane dr foc child = localRegion (const dr) $
231โ ctx <- lift ask
553+ mapImages (imagesInRegion $ current dr) $
232โ let ctx' = VtyWidgetCtx
554+ localFocus (const foc) $
233โ { _vtyWidgetCtx_input = leftmost -- TODO: think about this leftmost more.
555+ inputInFocusedRegion >>= \e -> localInput (const e) child
234โ [ fmapMaybe id $
556+
235โ attachWith (\(r,f) e -> filterInput r f e)
557+-- * Misc
236โ (liftA2 (,) (current reg) (current foc))
558+
237โ (_vtyWidgetCtx_input ctx)
559+-- | A widget that draws nothing
239โ , _vtyWidgetCtx_focus = liftA2 (&&) (_vtyWidgetCtx_focus ctx) foc
240โ , _vtyWidgetCtx_size = fmap regionSize reg }
241โ (result, wo) <- lift . lift $ runVtyWidget ctx' child
242โ let images = _vtyWidgetOut_images wo
243โ images' = liftA2 (\r is -> map (withinImage r) is) (current reg) images
244โ wo' = wo { _vtyWidgetOut_images = images' }
248โ filterInput :: Region -> Bool -> VtyEvent -> Maybe VtyEvent
249โ filterInput (Region l t w h) focused e = case e of
250โ V.EvKey _ _ | not focused -> Nothing
251โ V.EvMouseDown x y btn m -> mouse (\u v -> V.EvMouseDown u v btn m) x y
252โ V.EvMouseUp x y btn -> mouse (\u v -> V.EvMouseUp u v btn) x y
259โ , y >= t + h ] = Nothing
261โ Just (con (x - l) (y - t))
263โ-- | Information about a drag operation
265โ { _drag_from :: (Int, Int) -- ^ Where the drag began
266โ , _drag_to :: (Int, Int) -- ^ Where the mouse currently is
267โ , _drag_button :: V.Button -- ^ Which mouse button is dragging
268โ , _drag_modifiers :: [V.Modifier] -- ^ What modifiers are held
269โ , _drag_end :: Bool -- ^ Whether the drag ended (the mouse button was released)
271โ deriving (Eq, Ord, Show)
273โ-- | Converts raw vty mouse drag events into an event stream of 'Drag's
275โ :: (Reflex t, MonadFix m, MonadHold t m)
277โ -> VtyWidget t m (Event t Drag)
280โ let f :: Maybe Drag -> V.Event -> Maybe Drag
282โ V.EvMouseDown x y btn' mods
283โ | btn == btn' -> Just $ Drag (x,y) (x,y) btn' mods False
284โ | otherwise -> Nothing
286โ f (Just (Drag from _ _ mods end)) = \case
287โ V.EvMouseDown x y btn' mods'
288โ | end -> Just $ Drag (x,y) (x,y) btn' mods' False
289โ | btn == btn' -> Just $ Drag from (x,y) btn mods' False
290โ | otherwise -> Nothing -- Ignore other buttons.
291โ V.EvMouseUp x y (Just btn')
293โ | btn == btn' -> Just $ Drag from (x,y) btn mods True
294โ | otherwise -> Nothing
295โ V.EvMouseUp x y Nothing -- Terminal doesn't specify mouse up button,
296โ -- assume it's the right one.
298โ | otherwise -> Just $ Drag from (x,y) btn mods True
300โ rec let newDrag = attachWithMaybe f (current dragD) inp
301โ dragD <- holdDyn Nothing $ Just <$> newDrag
302โ return (fmapMaybe id $ updated dragD)
304โ-- | Mouse down events for a particular mouse button
306โ :: (Reflex t, Monad m)
308โ -> VtyWidget t m (Event t MouseDown)
311โ return $ fforMaybe i $ \case
312โ V.EvMouseDown x y btn' mods -> if btn == btn'
313โ then Just $ MouseDown btn' (x, y) mods
317โ-- | Information about a mouse down event
318โdata MouseDown = MouseDown
319โ { _mouseDown_button :: V.Button
320โ , _mouseDown_coordinates :: (Int, Int)
321โ , _mouseDown_modifiers :: [V.Modifier]
323โ deriving (Eq, Ord, Show)
325โ-- | Information about a mouse up event
326โdata MouseUp = MouseUp
327โ { _mouseUp_button :: Maybe V.Button
328โ , _mouseUp_coordinates :: (Int, Int)
330โ deriving (Eq, Ord, Show)
332โ-- | A plain split of the available space into vertically stacked panes.
333โ-- No visual separator is built in here.
334โsplitV :: (Reflex t, Monad m)
335โ => Dynamic t (Int -> Int)
336โ -- ^ Function used to determine size of first pane based on available size
337โ -> Dynamic t (Bool, Bool)
338โ -- ^ How to focus the two sub-panes, given that we are focused.
339โ -> VtyWidget t m a
340โ -- ^ Widget for first pane
341โ -> VtyWidget t m b
342โ -- ^ Widget for second pane
343โ -> VtyWidget t m (a,b)
344โsplitV sizeFunD focD wA wB = do
346โ let regA = (\f (w,h) -> Region 0 0 w (f h)) <$> sizeFunD <*> sz
347โ regB = (\(w,h) (Region _ _ _ hA) -> Region 0 hA w (h - hA)) <$> sz <*> regA
348โ ra <- pane regA (fst <$> focD) wA
349โ rb <- pane regB (snd <$> focD) wB
352โ-- | A split of the available space into two parts with a draggable separator.
353โ-- Starts with half the space allocated to each, and the first pane has focus.
354โ-- Clicking in a pane switches focus.
355โsplitVDrag :: (Reflex t, MonadFix m, MonadHold t m)
356โ => VtyWidget t m ()
357โ -> VtyWidget t m a
358โ -> VtyWidget t m b
359โ -> VtyWidget t m (a,b)
360โsplitVDrag wS wA wB = do
362โ (_, h0) <- sample $ current sz
363โ dragE <- drag V.BLeft
364โ let splitter0 = h0 `div` 2
365โ rec splitterCheckpoint <- holdDyn splitter0 $ leftmost [fst <$> ffilter snd dragSplitter, resizeSplitter]
366โ splitterPos <- holdDyn splitter0 $ leftmost [fst <$> dragSplitter, resizeSplitter]
367โ splitterFrac <- holdDyn ((1::Double) / 2) $ ffor (attach (current sz) (fst <$> dragSplitter)) $ \((_,h),x) ->
368โ fromIntegral x / fromIntegral h
369โ let dragSplitter = fforMaybe (attach (current splitterCheckpoint) dragE) $
370โ \(splitterY, Drag (_, fromY) (_, toY) _ _ end) ->
371โ if splitterY == fromY then Just (toY, end) else Nothing
372โ regA = (\(w,_) sp -> Region 0 0 w sp) <$> sz <*> splitterPos
373โ regS = (\(w,_) sp -> Region 0 sp w 1) <$> sz <*> splitterPos
374โ regB = (\(w,h) sp -> Region 0 (sp + 1) w (h - sp - 1)) <$> sz <*> splitterPos
375โ resizeSplitter = ffor (attach (current splitterFrac) (updated sz)) $
376โ \(frac, (_,h)) -> round (frac * fromIntegral h)
377โ focA <- holdDyn True $ leftmost
381โ (mA, rA) <- pane regA focA $ withMouseDown wA
382โ pane regS (pure False) wS
383โ (mB, rB) <- pane regB (not <$> focA) $ withMouseDown wB
386โ withMouseDown x = do
387โ m <- mouseDown V.BLeft
391โ-- | Fill the background with a particular character.
392โfill :: (Reflex t, Monad m) => Char -> VtyWidget t m ()
395โ let fillImg = ffor (current sz) $ \(w,h) -> [V.charFill V.defAttr c w h]
396โ tellImages fillImg
398โ-- | Fill the background with the bottom
399โhRule :: (Reflex t, Monad m) => BoxStyle -> VtyWidget t m ()
400โhRule boxStyle = fill (_boxStyle_s boxStyle)
402โ-- | Transform the images produces by a widget
404โ :: (Reflex t, MonadHold t m, MonadFix m)
405โ => Behavior t ([Image] -> [Image])
406โ -> VtyWidget t m a
407โ -> VtyWidget t m a
408โmodifyImages f (VtyWidget w) = VtyWidget $ flip censor w $ \wo ->
409โ wo { _vtyWidgetOut_images = f <*> (_vtyWidgetOut_images wo) }
411โ-- | Defines a set of symbols to use to draw the outlines of boxes
412โ-- C.f. https://en.wikipedia.org/wiki/Box-drawing_character
413โdata BoxStyle = BoxStyle
414โ { _boxStyle_nw :: Char
415โ , _boxStyle_n :: Char
416โ , _boxStyle_ne :: Char
417โ , _boxStyle_e :: Char
418โ , _boxStyle_se :: Char
419โ , _boxStyle_s :: Char
420โ , _boxStyle_sw :: Char
421โ , _boxStyle_w :: Char
424โinstance Default BoxStyle where
425โ def = singleBoxStyle
427โ-- | A box style that uses hyphens and pipe characters. Doesn't handle
428โ-- corners very well.
429โhyphenBoxStyle :: BoxStyle
430โhyphenBoxStyle = BoxStyle '-' '-' '-' '|' '-' '-' '-' '|'
432โ-- | A single line box style
433โsingleBoxStyle :: BoxStyle
434โsingleBoxStyle = BoxStyle 'โ' 'โ' 'โ' 'โ' 'โ' 'โ' 'โ' 'โ'
436โ-- | A thick single line box style
437โthickBoxStyle :: BoxStyle
438โthickBoxStyle = BoxStyle 'โ' 'โ' 'โ' 'โ' 'โ' 'โ' 'โ' 'โ'
440โ-- | A double line box style
441โdoubleBoxStyle :: BoxStyle
442โdoubleBoxStyle = BoxStyle 'โ' 'โ' 'โ' 'โ' 'โ' 'โ' 'โ' 'โ'
444โ-- | A single line box style with rounded corners
445โroundedBoxStyle :: BoxStyle
446โroundedBoxStyle = BoxStyle 'โญ' 'โ' 'โฎ' 'โ' 'โฏ' 'โ' 'โฐ' 'โ'
448โ-- | Draws a box in the provided style and a child widget inside of that box
449โbox :: (Monad m, Reflex t)
560+blank :: Monad m => m ()
450โ => BoxStyle
561+blank = return ()
451โ -> VtyWidget t m a
452โ -> VtyWidget t m a
453โbox style child = do
455โ let boxReg = ffor (current sz) $ \(w,h) -> Region 0 0 w h
456โ innerReg = ffor sz $ \(w,h) -> Region 1 1 (w - 2) (h - 2)
457โ tellImages (fmap boxImages boxReg)
458โ tellImages (fmap (\r -> [regionBlankImage r]) (current innerReg))
459โ pane innerReg (pure True) child
461โ boxImages :: Region -> [Image]
462โ boxImages (Region left top width height) =
463โ let right = left + width - 1
464โ bottom = top + height - 1
466โ [ withinImage (Region (left + 1) top (width - 2) 1) $
467โ V.charFill V.defAttr (_boxStyle_n style) (width - 2) 1
468โ , withinImage (Region right (top + 1) 1 (height - 2)) $
469โ V.charFill V.defAttr (_boxStyle_e style) 1 (height - 2)
470โ , withinImage (Region (left + 1) bottom (width - 2) 1) $
471โ V.charFill V.defAttr (_boxStyle_s style) (width - 2) 1
472โ , withinImage (Region left (top + 1) 1 (height - 2)) $
473โ V.charFill V.defAttr (_boxStyle_w style) 1 (height - 2)
476โ [ withinImage (Region left top 1 1) $
477โ V.char V.defAttr (_boxStyle_nw style)
478โ , withinImage (Region right top 1 1) $
479โ V.char V.defAttr (_boxStyle_ne style)
480โ , withinImage (Region right bottom 1 1) $
481โ V.char V.defAttr (_boxStyle_se style)
482โ , withinImage (Region left bottom 1 1) $
483โ V.char V.defAttr (_boxStyle_sw style)
485โ in sides ++ if width > 1 && height > 1 then corners else []
487โ-- | Renders text, wrapped to the container width
489โ :: (Reflex t, Monad m)
490โ => Behavior t Text
491โ -> VtyWidget t m ()
493โ dw <- displayWidth
494โ let img = (\w s -> [wrapText w V.defAttr s]) <$> current dw <*> msg
497โ wrapText maxWidth attrs = V.vertCat
498โ . concatMap (fmap (V.string attrs . T.unpack) . TZ.wrapWithOffset maxWidth 0)
499โ . T.split (=='\n')
501โ-- | Renders any behavior whose value can be converted to
502โ-- 'String' as text
504โ :: (Reflex t, Monad m, Show a)
506โ -> VtyWidget t m ()
507โdisplay a = text $ T.pack . show <$> a