-- | This module provides a type and functions for handling focus rings
-- of values.
module Brick.Focus
  ( FocusRing
  , focusRing
  , focusNext
  , focusPrev
  , focusGetCurrent
  , focusSetCurrent
  , focusRingLength
  , focusRingToList
  , focusRingCursor
  , withFocusRing
  , focusRingModify
  )
where

import Lens.Micro ((^.))
import Data.List (find)
import qualified Data.CircularList as C

import Brick.Types
import Brick.Widgets.Core (Named(..))

-- | A focus ring containing a sequence of resource names to focus and a
-- currently-focused name.
newtype FocusRing n = FocusRing (C.CList n)

-- | Construct a focus ring from the list of resource names.
focusRing :: [n] -> FocusRing n
focusRing :: forall n. [n] -> FocusRing n
focusRing = forall n. CList n -> FocusRing n
FocusRing forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. [a] -> CList a
C.fromList

-- | Advance focus to the next value in the ring.
focusNext :: FocusRing n -> FocusRing n
focusNext :: forall n. FocusRing n -> FocusRing n
focusNext r :: FocusRing n
r@(FocusRing CList n
l)
    | forall a. CList a -> Bool
C.isEmpty CList n
l = FocusRing n
r
    | Bool
otherwise = forall n. CList n -> FocusRing n
FocusRing forall a b. (a -> b) -> a -> b
$ forall a. CList a -> CList a
C.rotR CList n
l

-- | Advance focus to the previous value in the ring.
focusPrev :: FocusRing n -> FocusRing n
focusPrev :: forall n. FocusRing n -> FocusRing n
focusPrev r :: FocusRing n
r@(FocusRing CList n
l)
    | forall a. CList a -> Bool
C.isEmpty CList n
l = FocusRing n
r
    | Bool
otherwise = forall n. CList n -> FocusRing n
FocusRing forall a b. (a -> b) -> a -> b
$ forall a. CList a -> CList a
C.rotL CList n
l

-- | This function is a convenience function to look up a widget state
-- value's resource name in a focus ring and set its focus setting
-- according to the focus ring's state. This function determines whether
-- a given widget state value is the focus of the ring and passes the
-- resulting boolean to a rendering function, along with the state value
-- (a), to produce whatever comes next (b).
--
-- Focus-aware widgets have rendering functions that should be
-- usable with this combinator; see 'Brick.Widgets.List.List' and
-- 'Brick.Widgets.Edit.Edit'.
withFocusRing :: (Eq n, Named a n)
              => FocusRing n
              -- ^ The focus ring to use as the source of focus state.
              -> (Bool -> a -> b)
              -- ^ A function that takes a value and its focus state.
              -> a
              -- ^ The widget state value that we need to check for focus.
              -> b
              -- ^ The rest of the computation.
withFocusRing :: forall n a b.
(Eq n, Named a n) =>
FocusRing n -> (Bool -> a -> b) -> a -> b
withFocusRing FocusRing n
ring Bool -> a -> b
f a
a = Bool -> a -> b
f (forall n. FocusRing n -> Maybe n
focusGetCurrent FocusRing n
ring forall a. Eq a => a -> a -> Bool
== forall a. a -> Maybe a
Just (forall a n. Named a n => a -> n
getName a
a)) a
a

-- | Get the currently-focused resource name from the ring. If the ring
-- is empty, return 'Nothing'.
focusGetCurrent :: FocusRing n -> Maybe n
focusGetCurrent :: forall n. FocusRing n -> Maybe n
focusGetCurrent (FocusRing CList n
l) = forall a. CList a -> Maybe a
C.focus CList n
l

-- | Set the currently-focused resource name in the ring, provided the
-- name is in the ring. Otherwise return the ring unmodified.
focusSetCurrent :: (Eq n) => n -> FocusRing n -> FocusRing n
focusSetCurrent :: forall n. Eq n => n -> FocusRing n -> FocusRing n
focusSetCurrent n
n r :: FocusRing n
r@(FocusRing CList n
l) =
    case forall a. Eq a => a -> CList a -> Maybe (CList a)
C.rotateTo n
n CList n
l of
        Maybe (CList n)
Nothing -> FocusRing n
r
        Just CList n
l' -> forall n. CList n -> FocusRing n
FocusRing CList n
l'

-- | Get the size of the FocusRing.
focusRingLength :: FocusRing n -> Int
focusRingLength :: forall n. FocusRing n -> Int
focusRingLength (FocusRing CList n
l) = forall a. CList a -> Int
C.size CList n
l

-- | Return all of the entries in the focus ring, starting with the
-- currently-focused entry and wrapping around the ring.
--
-- For example, if a ring contains A, B, C, and D, and the current entry
-- is B, the result will be [B, C, D, A].
focusRingToList :: FocusRing n -> [n]
focusRingToList :: forall n. FocusRing n -> [n]
focusRingToList (FocusRing CList n
l) = forall a. CList a -> [a]
C.rightElements CList n
l

-- | Modify the internal circular list structure of a focus ring
-- directly. This function permits modification of the circular list
-- using the rich Data.CircularList API.
focusRingModify :: (C.CList n -> C.CList n) -> FocusRing n -> FocusRing n
focusRingModify :: forall n. (CList n -> CList n) -> FocusRing n -> FocusRing n
focusRingModify CList n -> CList n
f (FocusRing CList n
l) = forall n. CList n -> FocusRing n
FocusRing forall a b. (a -> b) -> a -> b
$ CList n -> CList n
f CList n
l

-- | Cursor selection convenience function for use as an
-- 'Brick.Main.appChooseCursor' value.
focusRingCursor :: (Eq n)
                => (a -> FocusRing n)
                -- ^ The function used to get the focus ring out of your
                -- application state.
                -> a
                -- ^ Your application state.
                -> [CursorLocation n]
                -- ^ The list of available cursor positions.
                -> Maybe (CursorLocation n)
                -- ^ The cursor position, if any, that matches the
                -- resource name currently focused by the 'FocusRing'.
focusRingCursor :: forall n a.
Eq n =>
(a -> FocusRing n)
-> a -> [CursorLocation n] -> Maybe (CursorLocation n)
focusRingCursor a -> FocusRing n
getRing a
st = forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Maybe a
find forall a b. (a -> b) -> a -> b
$ \CursorLocation n
cl ->
    CursorLocation n
clforall s a. s -> Getting a s a -> a
^.forall n1 n2.
Lens (CursorLocation n1) (CursorLocation n2) (Maybe n1) (Maybe n2)
cursorLocationNameL forall a. Eq a => a -> a -> Bool
== forall n. FocusRing n -> Maybe n
focusGetCurrent (a -> FocusRing n
getRing a
st)