Extra utils

1 files changed, 32 insertions, 2 deletions

diff --git a/src/Prolog.hs b/src/Prolog.hs
index 145f2d3..94a6eb6 100644
--- a/src/Prolog.hs
+++ b/src/Prolog.hs
@@ -43,7 +43,9 @@ module Prolog
     , renderTo
     , fromStringlike
     -- * Utils
+    , ffmap
     , (<<$>>)
+    , fffmap
     , (<<<$>>>)
     , (<<&>>)
     , (-->)
@@ -54,6 +56,9 @@ module Prolog
     , squashMaybe
     , throwEither
     , squashEither
+    , rebase
+    , rebaseWith
+    , rebaseWithM
     ) where
 
 import Control.Applicative
@@ -100,14 +105,26 @@ import Text.LaTeX.Packages.Hyperref qualified as LH
 type StrictText = T.Text
 type LazyText = TL.Text
 
+-- | 'fmap' over nested functors
+ffmap :: (Functor f, Functor g) => (a -> b) -> f (g a) -> f (g b)
+ffmap = fmap fmap fmap
+{-# INLINE ffmap #-}
+
+-- | Infix version of 'ffmap'
 (<<$>>) :: (Functor f, Functor g) => (a -> b) -> f (g a) -> f (g b)
 infixl 4 <<$>>
-(<<$>>) = fmap fmap fmap
+(<<$>>) = ffmap
 {-# INLINE (<<$>>) #-}
 
+-- | 'fmap' over nested functors
+fffmap ::  (Functor f, Functor g, Functor h) => (a -> b) -> f (g (h a)) -> f (g (h b))
+fffmap = fmap . fmap . fmap
+{-# INLINE fffmap #-}
+
+-- | Infix version of 'fffmap'
 (<<<$>>>) :: (Functor f, Functor g, Functor h) => (a -> b) -> f (g (h a)) -> f (g (h b))
 infixl 4 <<<$>>>
-(<<<$>>>) = fmap . fmap . fmap
+(<<<$>>>) = fffmap
 {-# INLINE (<<<$>>>) #-}
 
 (<<&>>) :: (Functor f, Functor g) => f (g a) -> (a -> b) -> f (g b)
@@ -160,6 +177,19 @@ fromStringlike :: (IsString s, StringConv t String) => t -> s
 fromStringlike = fromString . toSL
 {-# INLINE fromStringlike #-}
 
+
+rebase :: (Applicative m, Comonad w) => w a -> m a
+rebase = rebaseWith id
+{-# INLINE rebase #-}
+
+rebaseWith :: (Applicative m, Comonad w) => (a -> b) -> w a -> m b
+rebaseWith f = rebaseWithM $ pure . f
+{-# INLINE rebaseWith #-}
+
+rebaseWithM :: (Applicative m, Comonad w) => (a -> m b) -> w a -> m b
+rebaseWithM f = f . extract
+{-# INLINE rebaseWithM #-}
+
 -- Orphan instances