{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE TypeOperators #-}
{-# LANGUAGE UndecidableInstances #-}

{- | A carrier for 'Writer' effects. This carrier performs its append operations strictly and thus avoids the space leaks inherent in lazy writer monads. These appends are left-associative; as such, @[]@ is a poor choice of monoid for computations that entail many calls to 'tell'. The [Seq](http://hackage.haskell.org/package/containersdocs/Data-Sequence.html) or [DList](http://hackage.haskell.org/package/dlist) monoids may be a superior choice.

This implementation is based on a post Gabriel Gonzalez made to the Haskell mailing list: <https://mail.haskell.org/pipermail/libraries/2013-March/019528.html>

@since 1.0.0.0
-}

module Control.Carrier.Writer.Strict
( -- * Writer carrier
  runWriter
, execWriter
, WriterC(WriterC)
  -- * Writer effect
, module Control.Effect.Writer
) where

import Control.Algebra
import Control.Applicative (Alternative)
import Control.Carrier.State.Strict
import Control.Effect.Writer
import Control.Monad (MonadPlus)
import Control.Monad.Fail as Fail
import Control.Monad.Fix
import Control.Monad.IO.Class
import Control.Monad.Trans.Class

-- | Run a 'Writer' effect with a 'Monoid'al log, producing the final log alongside the result value.
--
-- @
-- 'runWriter' ('tell' w) = 'pure' (w, ())
-- @
-- @
-- 'runWriter' ('pure' a) = 'pure' ('mempty', a)
-- @
runWriter :: Monoid w => WriterC w m a -> m (w, a)
runWriter :: WriterC w m a -> m (w, a)
runWriter (WriterC StateC w m a
m) = w -> StateC w m a -> m (w, a)
forall s (m :: * -> *) a. s -> StateC s m a -> m (s, a)
runState w
forall a. Monoid a => a
mempty StateC w m a
m
{-# INLINE runWriter #-}

-- | Run a 'Writer' effect with a 'Monoid'al log, producing the final log and discarding the result value.
--
-- @
-- 'execWriter' m = 'fmap' 'fst' ('runWriter' m)
-- @
execWriter :: (Monoid w, Functor m) => WriterC w m a -> m w
execWriter :: WriterC w m a -> m w
execWriter = ((w, a) -> w) -> m (w, a) -> m w
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (w, a) -> w
forall a b. (a, b) -> a
fst (m (w, a) -> m w)
-> (WriterC w m a -> m (w, a)) -> WriterC w m a -> m w
forall b c a. (b -> c) -> (a -> b) -> a -> c
. WriterC w m a -> m (w, a)
forall w (m :: * -> *) a. Monoid w => WriterC w m a -> m (w, a)
runWriter
{-# INLINE execWriter #-}


-- | A space-efficient carrier for 'Writer' effects, implemented atop "Control.Carrier.State.Strict".
--
-- @since 1.0.0.0
newtype WriterC w m a = WriterC { WriterC w m a -> StateC w m a
runWriterC :: StateC w m a }
  deriving (Applicative (WriterC w m)
WriterC w m a
Applicative (WriterC w m)
-> (forall a. WriterC w m a)
-> (forall a. WriterC w m a -> WriterC w m a -> WriterC w m a)
-> (forall a. WriterC w m a -> WriterC w m [a])
-> (forall a. WriterC w m a -> WriterC w m [a])
-> Alternative (WriterC w m)
WriterC w m a -> WriterC w m a -> WriterC w m a
WriterC w m a -> WriterC w m [a]
WriterC w m a -> WriterC w m [a]
forall a. WriterC w m a
forall a. WriterC w m a -> WriterC w m [a]
forall a. WriterC w m a -> WriterC w m a -> WriterC w m a
forall w (m :: * -> *).
(Alternative m, Monad m) =>
Applicative (WriterC w m)
forall w (m :: * -> *) a. (Alternative m, Monad m) => WriterC w m a
forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m [a]
forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m a -> WriterC w m a
forall (f :: * -> *).
Applicative f
-> (forall a. f a)
-> (forall a. f a -> f a -> f a)
-> (forall a. f a -> f [a])
-> (forall a. f a -> f [a])
-> Alternative f
many :: WriterC w m a -> WriterC w m [a]
$cmany :: forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m [a]
some :: WriterC w m a -> WriterC w m [a]
$csome :: forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m [a]
<|> :: WriterC w m a -> WriterC w m a -> WriterC w m a
$c<|> :: forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m a -> WriterC w m a
empty :: WriterC w m a
$cempty :: forall w (m :: * -> *) a. (Alternative m, Monad m) => WriterC w m a
$cp1Alternative :: forall w (m :: * -> *).
(Alternative m, Monad m) =>
Applicative (WriterC w m)
Alternative, Functor (WriterC w m)
a -> WriterC w m a
Functor (WriterC w m)
-> (forall a. a -> WriterC w m a)
-> (forall a b.
    WriterC w m (a -> b) -> WriterC w m a -> WriterC w m b)
-> (forall a b c.
    (a -> b -> c) -> WriterC w m a -> WriterC w m b -> WriterC w m c)
-> (forall a b. WriterC w m a -> WriterC w m b -> WriterC w m b)
-> (forall a b. WriterC w m a -> WriterC w m b -> WriterC w m a)
-> Applicative (WriterC w m)
WriterC w m a -> WriterC w m b -> WriterC w m b
WriterC w m a -> WriterC w m b -> WriterC w m a
WriterC w m (a -> b) -> WriterC w m a -> WriterC w m b
(a -> b -> c) -> WriterC w m a -> WriterC w m b -> WriterC w m c
forall a. a -> WriterC w m a
forall a b. WriterC w m a -> WriterC w m b -> WriterC w m a
forall a b. WriterC w m a -> WriterC w m b -> WriterC w m b
forall a b. WriterC w m (a -> b) -> WriterC w m a -> WriterC w m b
forall a b c.
(a -> b -> c) -> WriterC w m a -> WriterC w m b -> WriterC w m c
forall w (m :: * -> *). Monad m => Functor (WriterC w m)
forall w (m :: * -> *) a. Monad m => a -> WriterC w m a
forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> WriterC w m b -> WriterC w m a
forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> WriterC w m b -> WriterC w m b
forall w (m :: * -> *) a b.
Monad m =>
WriterC w m (a -> b) -> WriterC w m a -> WriterC w m b
forall w (m :: * -> *) a b c.
Monad m =>
(a -> b -> c) -> WriterC w m a -> WriterC w m b -> WriterC w m c
forall (f :: * -> *).
Functor f
-> (forall a. a -> f a)
-> (forall a b. f (a -> b) -> f a -> f b)
-> (forall a b c. (a -> b -> c) -> f a -> f b -> f c)
-> (forall a b. f a -> f b -> f b)
-> (forall a b. f a -> f b -> f a)
-> Applicative f
<* :: WriterC w m a -> WriterC w m b -> WriterC w m a
$c<* :: forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> WriterC w m b -> WriterC w m a
*> :: WriterC w m a -> WriterC w m b -> WriterC w m b
$c*> :: forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> WriterC w m b -> WriterC w m b
liftA2 :: (a -> b -> c) -> WriterC w m a -> WriterC w m b -> WriterC w m c
$cliftA2 :: forall w (m :: * -> *) a b c.
Monad m =>
(a -> b -> c) -> WriterC w m a -> WriterC w m b -> WriterC w m c
<*> :: WriterC w m (a -> b) -> WriterC w m a -> WriterC w m b
$c<*> :: forall w (m :: * -> *) a b.
Monad m =>
WriterC w m (a -> b) -> WriterC w m a -> WriterC w m b
pure :: a -> WriterC w m a
$cpure :: forall w (m :: * -> *) a. Monad m => a -> WriterC w m a
$cp1Applicative :: forall w (m :: * -> *). Monad m => Functor (WriterC w m)
Applicative, a -> WriterC w m b -> WriterC w m a
(a -> b) -> WriterC w m a -> WriterC w m b
(forall a b. (a -> b) -> WriterC w m a -> WriterC w m b)
-> (forall a b. a -> WriterC w m b -> WriterC w m a)
-> Functor (WriterC w m)
forall a b. a -> WriterC w m b -> WriterC w m a
forall a b. (a -> b) -> WriterC w m a -> WriterC w m b
forall w (m :: * -> *) a b.
Functor m =>
a -> WriterC w m b -> WriterC w m a
forall w (m :: * -> *) a b.
Functor m =>
(a -> b) -> WriterC w m a -> WriterC w m b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: a -> WriterC w m b -> WriterC w m a
$c<$ :: forall w (m :: * -> *) a b.
Functor m =>
a -> WriterC w m b -> WriterC w m a
fmap :: (a -> b) -> WriterC w m a -> WriterC w m b
$cfmap :: forall w (m :: * -> *) a b.
Functor m =>
(a -> b) -> WriterC w m a -> WriterC w m b
Functor, Applicative (WriterC w m)
a -> WriterC w m a
Applicative (WriterC w m)
-> (forall a b.
    WriterC w m a -> (a -> WriterC w m b) -> WriterC w m b)
-> (forall a b. WriterC w m a -> WriterC w m b -> WriterC w m b)
-> (forall a. a -> WriterC w m a)
-> Monad (WriterC w m)
WriterC w m a -> (a -> WriterC w m b) -> WriterC w m b
WriterC w m a -> WriterC w m b -> WriterC w m b
forall a. a -> WriterC w m a
forall a b. WriterC w m a -> WriterC w m b -> WriterC w m b
forall a b. WriterC w m a -> (a -> WriterC w m b) -> WriterC w m b
forall w (m :: * -> *). Monad m => Applicative (WriterC w m)
forall w (m :: * -> *) a. Monad m => a -> WriterC w m a
forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> WriterC w m b -> WriterC w m b
forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> (a -> WriterC w m b) -> WriterC w m b
forall (m :: * -> *).
Applicative m
-> (forall a b. m a -> (a -> m b) -> m b)
-> (forall a b. m a -> m b -> m b)
-> (forall a. a -> m a)
-> Monad m
return :: a -> WriterC w m a
$creturn :: forall w (m :: * -> *) a. Monad m => a -> WriterC w m a
>> :: WriterC w m a -> WriterC w m b -> WriterC w m b
$c>> :: forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> WriterC w m b -> WriterC w m b
>>= :: WriterC w m a -> (a -> WriterC w m b) -> WriterC w m b
$c>>= :: forall w (m :: * -> *) a b.
Monad m =>
WriterC w m a -> (a -> WriterC w m b) -> WriterC w m b
$cp1Monad :: forall w (m :: * -> *). Monad m => Applicative (WriterC w m)
Monad, Monad (WriterC w m)
Monad (WriterC w m)
-> (forall a. String -> WriterC w m a) -> MonadFail (WriterC w m)
String -> WriterC w m a
forall a. String -> WriterC w m a
forall w (m :: * -> *). MonadFail m => Monad (WriterC w m)
forall w (m :: * -> *) a. MonadFail m => String -> WriterC w m a
forall (m :: * -> *).
Monad m -> (forall a. String -> m a) -> MonadFail m
fail :: String -> WriterC w m a
$cfail :: forall w (m :: * -> *) a. MonadFail m => String -> WriterC w m a
$cp1MonadFail :: forall w (m :: * -> *). MonadFail m => Monad (WriterC w m)
Fail.MonadFail, Monad (WriterC w m)
Monad (WriterC w m)
-> (forall a. (a -> WriterC w m a) -> WriterC w m a)
-> MonadFix (WriterC w m)
(a -> WriterC w m a) -> WriterC w m a
forall a. (a -> WriterC w m a) -> WriterC w m a
forall w (m :: * -> *). MonadFix m => Monad (WriterC w m)
forall w (m :: * -> *) a.
MonadFix m =>
(a -> WriterC w m a) -> WriterC w m a
forall (m :: * -> *).
Monad m -> (forall a. (a -> m a) -> m a) -> MonadFix m
mfix :: (a -> WriterC w m a) -> WriterC w m a
$cmfix :: forall w (m :: * -> *) a.
MonadFix m =>
(a -> WriterC w m a) -> WriterC w m a
$cp1MonadFix :: forall w (m :: * -> *). MonadFix m => Monad (WriterC w m)
MonadFix, Monad (WriterC w m)
Monad (WriterC w m)
-> (forall a. IO a -> WriterC w m a) -> MonadIO (WriterC w m)
IO a -> WriterC w m a
forall a. IO a -> WriterC w m a
forall w (m :: * -> *). MonadIO m => Monad (WriterC w m)
forall w (m :: * -> *) a. MonadIO m => IO a -> WriterC w m a
forall (m :: * -> *).
Monad m -> (forall a. IO a -> m a) -> MonadIO m
liftIO :: IO a -> WriterC w m a
$cliftIO :: forall w (m :: * -> *) a. MonadIO m => IO a -> WriterC w m a
$cp1MonadIO :: forall w (m :: * -> *). MonadIO m => Monad (WriterC w m)
MonadIO, Monad (WriterC w m)
Alternative (WriterC w m)
WriterC w m a
Alternative (WriterC w m)
-> Monad (WriterC w m)
-> (forall a. WriterC w m a)
-> (forall a. WriterC w m a -> WriterC w m a -> WriterC w m a)
-> MonadPlus (WriterC w m)
WriterC w m a -> WriterC w m a -> WriterC w m a
forall a. WriterC w m a
forall a. WriterC w m a -> WriterC w m a -> WriterC w m a
forall w (m :: * -> *).
(Alternative m, Monad m) =>
Monad (WriterC w m)
forall w (m :: * -> *).
(Alternative m, Monad m) =>
Alternative (WriterC w m)
forall w (m :: * -> *) a. (Alternative m, Monad m) => WriterC w m a
forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m a -> WriterC w m a
forall (m :: * -> *).
Alternative m
-> Monad m
-> (forall a. m a)
-> (forall a. m a -> m a -> m a)
-> MonadPlus m
mplus :: WriterC w m a -> WriterC w m a -> WriterC w m a
$cmplus :: forall w (m :: * -> *) a.
(Alternative m, Monad m) =>
WriterC w m a -> WriterC w m a -> WriterC w m a
mzero :: WriterC w m a
$cmzero :: forall w (m :: * -> *) a. (Alternative m, Monad m) => WriterC w m a
$cp2MonadPlus :: forall w (m :: * -> *).
(Alternative m, Monad m) =>
Monad (WriterC w m)
$cp1MonadPlus :: forall w (m :: * -> *).
(Alternative m, Monad m) =>
Alternative (WriterC w m)
MonadPlus, m a -> WriterC w m a
(forall (m :: * -> *) a. Monad m => m a -> WriterC w m a)
-> MonadTrans (WriterC w)
forall w (m :: * -> *) a. Monad m => m a -> WriterC w m a
forall (m :: * -> *) a. Monad m => m a -> WriterC w m a
forall (t :: (* -> *) -> * -> *).
(forall (m :: * -> *) a. Monad m => m a -> t m a) -> MonadTrans t
lift :: m a -> WriterC w m a
$clift :: forall w (m :: * -> *) a. Monad m => m a -> WriterC w m a
MonadTrans)

instance (Monoid w, Algebra sig m) => Algebra (Writer w :+: sig) (WriterC w m) where
  alg :: Handler ctx n (WriterC w m)
-> (:+:) (Writer w) sig n a -> ctx () -> WriterC w m (ctx a)
alg Handler ctx n (WriterC w m)
hdl (:+:) (Writer w) sig n a
sig ctx ()
ctx = StateC w m (ctx a) -> WriterC w m (ctx a)
forall w (m :: * -> *) a. StateC w m a -> WriterC w m a
WriterC (StateC w m (ctx a) -> WriterC w m (ctx a))
-> StateC w m (ctx a) -> WriterC w m (ctx a)
forall a b. (a -> b) -> a -> b
$ case (:+:) (Writer w) sig n a
sig of
    L Writer w n a
writer -> (w -> m (w, ctx a)) -> StateC w m (ctx a)
forall s (m :: * -> *) a. (s -> m (s, a)) -> StateC s m a
StateC ((w -> m (w, ctx a)) -> StateC w m (ctx a))
-> (w -> m (w, ctx a)) -> StateC w m (ctx a)
forall a b. (a -> b) -> a -> b
$ \ w
w -> case Writer w n a
writer of
      Tell w
w'    -> do
        let !w'' :: w
w'' = w -> w -> w
forall a. Monoid a => a -> a -> a
mappend w
w w
w'
        (w, ctx ()) -> m (w, ctx ())
forall (f :: * -> *) a. Applicative f => a -> f a
pure (w
w'', ctx ()
ctx)
      Listen   n a
m -> do
        (w
w', ctx a
a) <- WriterC w m (ctx a) -> m (w, ctx a)
forall w (m :: * -> *) a. Monoid w => WriterC w m a -> m (w, a)
runWriter (ctx (n a) -> WriterC w m (ctx a)
Handler ctx n (WriterC w m)
hdl (n a
m n a -> ctx () -> ctx (n a)
forall (f :: * -> *) a b. Functor f => a -> f b -> f a
<$ ctx ()
ctx))
        let !w'' :: w
w'' = w -> w -> w
forall a. Monoid a => a -> a -> a
mappend w
w w
w'
        (w, ctx (w, a)) -> m (w, ctx (w, a))
forall (f :: * -> *) a. Applicative f => a -> f a
pure (w
w'', (,) w
w' (a -> (w, a)) -> ctx a -> ctx (w, a)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> ctx a
a)
      Censor w -> w
f n a
m -> do
        (w
w', ctx a
a) <- WriterC w m (ctx a) -> m (w, ctx a)
forall w (m :: * -> *) a. Monoid w => WriterC w m a -> m (w, a)
runWriter (ctx (n a) -> WriterC w m (ctx a)
Handler ctx n (WriterC w m)
hdl (n a
m n a -> ctx () -> ctx (n a)
forall (f :: * -> *) a b. Functor f => a -> f b -> f a
<$ ctx ()
ctx))
        let !w'' :: w
w'' = w -> w -> w
forall a. Monoid a => a -> a -> a
mappend w
w (w -> w
f w
w')
        (w, ctx a) -> m (w, ctx a)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (w
w'', ctx a
a)
    R sig n a
other  -> Handler ctx n (StateC w m)
-> (:+:) (State w) sig n a -> ctx () -> StateC w m (ctx a)
forall (sig :: (* -> *) -> * -> *) (m :: * -> *) (ctx :: * -> *)
       (n :: * -> *) a.
(Algebra sig m, Functor ctx) =>
Handler ctx n m -> sig n a -> ctx () -> m (ctx a)
alg (WriterC w m (ctx x) -> StateC w m (ctx x)
forall w (m :: * -> *) a. WriterC w m a -> StateC w m a
runWriterC (WriterC w m (ctx x) -> StateC w m (ctx x))
-> (ctx (n x) -> WriterC w m (ctx x))
-> ctx (n x)
-> StateC w m (ctx x)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ctx (n x) -> WriterC w m (ctx x)
Handler ctx n (WriterC w m)
hdl) (sig n a -> (:+:) (State w) sig n a
forall (f :: (* -> *) -> * -> *) (g :: (* -> *) -> * -> *)
       (m :: * -> *) k.
g m k -> (:+:) f g m k
R sig n a
other) ctx ()
ctx
  {-# INLINE alg #-}