{-# LANGUAGE UndecidableInstances #-}

-- | running the @fused-effects@ frontend
module Control.Carrier.Unclog (UnclogC (..), runUnclogging, runUncloggingToList) where

import Chronos (now)
import Control.Algebra (Algebra (..), (:+:) (..))
import Control.Applicative (Alternative)
import Control.Carrier.Reader (ReaderC (..))
import Control.Effect.Unclog (Log (..))
import Control.Monad (MonadPlus)
import Control.Monad.Fix (MonadFix)
import Data.Kind (Type)
import Unclog.Common (LogEntry, PLogChan)
import Unclog.Frontend qualified as Frontend
import Unclog.Subscriber (Subscriber, refSubscriber, withLoggingWithSubscribers)
import UnliftIO (MonadIO (..), MonadUnliftIO, atomically, newIORef, readIORef)

-- | the 'IO' carrier for unclogging
type UnclogC :: (Type -> Type) -> Type -> Type
newtype UnclogC m r = MkUnclogC {forall (m :: * -> *) r. UnclogC m r -> PLogChan -> m r
runUnclogC :: PLogChan -> m r}
  deriving ((forall a b. (a -> b) -> UnclogC m a -> UnclogC m b)
-> (forall a b. a -> UnclogC m b -> UnclogC m a)
-> Functor (UnclogC m)
forall a b. a -> UnclogC m b -> UnclogC m a
forall a b. (a -> b) -> UnclogC m a -> UnclogC m b
forall (m :: * -> *) a b.
Functor m =>
a -> UnclogC m b -> UnclogC m a
forall (m :: * -> *) a b.
Functor m =>
(a -> b) -> UnclogC m a -> UnclogC m b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
$cfmap :: forall (m :: * -> *) a b.
Functor m =>
(a -> b) -> UnclogC m a -> UnclogC m b
fmap :: forall a b. (a -> b) -> UnclogC m a -> UnclogC m b
$c<$ :: forall (m :: * -> *) a b.
Functor m =>
a -> UnclogC m b -> UnclogC m a
<$ :: forall a b. a -> UnclogC m b -> UnclogC m a
Functor, Functor (UnclogC m)
Functor (UnclogC m) =>
(forall a. a -> UnclogC m a)
-> (forall a b. UnclogC m (a -> b) -> UnclogC m a -> UnclogC m b)
-> (forall a b c.
    (a -> b -> c) -> UnclogC m a -> UnclogC m b -> UnclogC m c)
-> (forall a b. UnclogC m a -> UnclogC m b -> UnclogC m b)
-> (forall a b. UnclogC m a -> UnclogC m b -> UnclogC m a)
-> Applicative (UnclogC m)
forall a. a -> UnclogC m a
forall a b. UnclogC m a -> UnclogC m b -> UnclogC m a
forall a b. UnclogC m a -> UnclogC m b -> UnclogC m b
forall a b. UnclogC m (a -> b) -> UnclogC m a -> UnclogC m b
forall a b c.
(a -> b -> c) -> UnclogC m a -> UnclogC m b -> UnclogC 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
forall (m :: * -> *). Applicative m => Functor (UnclogC m)
forall (m :: * -> *) a. Applicative m => a -> UnclogC m a
forall (m :: * -> *) a b.
Applicative m =>
UnclogC m a -> UnclogC m b -> UnclogC m a
forall (m :: * -> *) a b.
Applicative m =>
UnclogC m a -> UnclogC m b -> UnclogC m b
forall (m :: * -> *) a b.
Applicative m =>
UnclogC m (a -> b) -> UnclogC m a -> UnclogC m b
forall (m :: * -> *) a b c.
Applicative m =>
(a -> b -> c) -> UnclogC m a -> UnclogC m b -> UnclogC m c
$cpure :: forall (m :: * -> *) a. Applicative m => a -> UnclogC m a
pure :: forall a. a -> UnclogC m a
$c<*> :: forall (m :: * -> *) a b.
Applicative m =>
UnclogC m (a -> b) -> UnclogC m a -> UnclogC m b
<*> :: forall a b. UnclogC m (a -> b) -> UnclogC m a -> UnclogC m b
$cliftA2 :: forall (m :: * -> *) a b c.
Applicative m =>
(a -> b -> c) -> UnclogC m a -> UnclogC m b -> UnclogC m c
liftA2 :: forall a b c.
(a -> b -> c) -> UnclogC m a -> UnclogC m b -> UnclogC m c
$c*> :: forall (m :: * -> *) a b.
Applicative m =>
UnclogC m a -> UnclogC m b -> UnclogC m b
*> :: forall a b. UnclogC m a -> UnclogC m b -> UnclogC m b
$c<* :: forall (m :: * -> *) a b.
Applicative m =>
UnclogC m a -> UnclogC m b -> UnclogC m a
<* :: forall a b. UnclogC m a -> UnclogC m b -> UnclogC m a
Applicative, Applicative (UnclogC m)
Applicative (UnclogC m) =>
(forall a b. UnclogC m a -> (a -> UnclogC m b) -> UnclogC m b)
-> (forall a b. UnclogC m a -> UnclogC m b -> UnclogC m b)
-> (forall a. a -> UnclogC m a)
-> Monad (UnclogC m)
forall a. a -> UnclogC m a
forall a b. UnclogC m a -> UnclogC m b -> UnclogC m b
forall a b. UnclogC m a -> (a -> UnclogC m b) -> UnclogC m b
forall (m :: * -> *). Monad m => Applicative (UnclogC m)
forall (m :: * -> *) a. Monad m => a -> UnclogC m a
forall (m :: * -> *) a b.
Monad m =>
UnclogC m a -> UnclogC m b -> UnclogC m b
forall (m :: * -> *) a b.
Monad m =>
UnclogC m a -> (a -> UnclogC m b) -> UnclogC 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
$c>>= :: forall (m :: * -> *) a b.
Monad m =>
UnclogC m a -> (a -> UnclogC m b) -> UnclogC m b
>>= :: forall a b. UnclogC m a -> (a -> UnclogC m b) -> UnclogC m b
$c>> :: forall (m :: * -> *) a b.
Monad m =>
UnclogC m a -> UnclogC m b -> UnclogC m b
>> :: forall a b. UnclogC m a -> UnclogC m b -> UnclogC m b
$creturn :: forall (m :: * -> *) a. Monad m => a -> UnclogC m a
return :: forall a. a -> UnclogC m a
Monad, Monad (UnclogC m)
Monad (UnclogC m) =>
(forall a. IO a -> UnclogC m a) -> MonadIO (UnclogC m)
forall a. IO a -> UnclogC m a
forall (m :: * -> *).
Monad m =>
(forall a. IO a -> m a) -> MonadIO m
forall (m :: * -> *). MonadIO m => Monad (UnclogC m)
forall (m :: * -> *) a. MonadIO m => IO a -> UnclogC m a
$cliftIO :: forall (m :: * -> *) a. MonadIO m => IO a -> UnclogC m a
liftIO :: forall a. IO a -> UnclogC m a
MonadIO, MonadIO (UnclogC m)
MonadIO (UnclogC m) =>
(forall b.
 ((forall a. UnclogC m a -> IO a) -> IO b) -> UnclogC m b)
-> MonadUnliftIO (UnclogC m)
forall b. ((forall a. UnclogC m a -> IO a) -> IO b) -> UnclogC m b
forall (m :: * -> *).
MonadIO m =>
(forall b. ((forall a. m a -> IO a) -> IO b) -> m b)
-> MonadUnliftIO m
forall (m :: * -> *). MonadUnliftIO m => MonadIO (UnclogC m)
forall (m :: * -> *) b.
MonadUnliftIO m =>
((forall a. UnclogC m a -> IO a) -> IO b) -> UnclogC m b
$cwithRunInIO :: forall (m :: * -> *) b.
MonadUnliftIO m =>
((forall a. UnclogC m a -> IO a) -> IO b) -> UnclogC m b
withRunInIO :: forall b. ((forall a. UnclogC m a -> IO a) -> IO b) -> UnclogC m b
MonadUnliftIO, Applicative (UnclogC m)
Applicative (UnclogC m) =>
(forall a. UnclogC m a)
-> (forall a. UnclogC m a -> UnclogC m a -> UnclogC m a)
-> (forall a. UnclogC m a -> UnclogC m [a])
-> (forall a. UnclogC m a -> UnclogC m [a])
-> Alternative (UnclogC m)
forall a. UnclogC m a
forall a. UnclogC m a -> UnclogC m [a]
forall a. UnclogC m a -> UnclogC m a -> UnclogC 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
forall (m :: * -> *). Alternative m => Applicative (UnclogC m)
forall (m :: * -> *) a. Alternative m => UnclogC m a
forall (m :: * -> *) a.
Alternative m =>
UnclogC m a -> UnclogC m [a]
forall (m :: * -> *) a.
Alternative m =>
UnclogC m a -> UnclogC m a -> UnclogC m a
$cempty :: forall (m :: * -> *) a. Alternative m => UnclogC m a
empty :: forall a. UnclogC m a
$c<|> :: forall (m :: * -> *) a.
Alternative m =>
UnclogC m a -> UnclogC m a -> UnclogC m a
<|> :: forall a. UnclogC m a -> UnclogC m a -> UnclogC m a
$csome :: forall (m :: * -> *) a.
Alternative m =>
UnclogC m a -> UnclogC m [a]
some :: forall a. UnclogC m a -> UnclogC m [a]
$cmany :: forall (m :: * -> *) a.
Alternative m =>
UnclogC m a -> UnclogC m [a]
many :: forall a. UnclogC m a -> UnclogC m [a]
Alternative, Monad (UnclogC m)
Alternative (UnclogC m)
(Alternative (UnclogC m), Monad (UnclogC m)) =>
(forall a. UnclogC m a)
-> (forall a. UnclogC m a -> UnclogC m a -> UnclogC m a)
-> MonadPlus (UnclogC m)
forall a. UnclogC m a
forall a. UnclogC m a -> UnclogC m a -> UnclogC m a
forall (m :: * -> *). (Alternative m, Monad m) => Monad (UnclogC m)
forall (m :: * -> *).
(Alternative m, Monad m) =>
Alternative (UnclogC m)
forall (m :: * -> *).
(Alternative m, Monad m) =>
(forall a. m a) -> (forall a. m a -> m a -> m a) -> MonadPlus m
forall (m :: * -> *) a. (Alternative m, Monad m) => UnclogC m a
forall (m :: * -> *) a.
(Alternative m, Monad m) =>
UnclogC m a -> UnclogC m a -> UnclogC m a
$cmzero :: forall (m :: * -> *) a. (Alternative m, Monad m) => UnclogC m a
mzero :: forall a. UnclogC m a
$cmplus :: forall (m :: * -> *) a.
(Alternative m, Monad m) =>
UnclogC m a -> UnclogC m a -> UnclogC m a
mplus :: forall a. UnclogC m a -> UnclogC m a -> UnclogC m a
MonadPlus, Monad (UnclogC m)
Monad (UnclogC m) =>
(forall a. String -> UnclogC m a) -> MonadFail (UnclogC m)
forall a. String -> UnclogC m a
forall (m :: * -> *).
Monad m =>
(forall a. String -> m a) -> MonadFail m
forall (m :: * -> *). MonadFail m => Monad (UnclogC m)
forall (m :: * -> *) a. MonadFail m => String -> UnclogC m a
$cfail :: forall (m :: * -> *) a. MonadFail m => String -> UnclogC m a
fail :: forall a. String -> UnclogC m a
MonadFail, Monad (UnclogC m)
Monad (UnclogC m) =>
(forall a. (a -> UnclogC m a) -> UnclogC m a)
-> MonadFix (UnclogC m)
forall a. (a -> UnclogC m a) -> UnclogC m a
forall (m :: * -> *).
Monad m =>
(forall a. (a -> m a) -> m a) -> MonadFix m
forall (m :: * -> *). MonadFix m => Monad (UnclogC m)
forall (m :: * -> *) a.
MonadFix m =>
(a -> UnclogC m a) -> UnclogC m a
$cmfix :: forall (m :: * -> *) a.
MonadFix m =>
(a -> UnclogC m a) -> UnclogC m a
mfix :: forall a. (a -> UnclogC m a) -> UnclogC m a
MonadFix) via ReaderC PLogChan m

instance (Algebra sig m, MonadIO m) => Algebra (Log :+: sig) (UnclogC m) where
  alg :: forall (ctx :: * -> *) (n :: * -> *) a.
Functor ctx =>
Handler ctx n (UnclogC m)
-> (:+:) Log sig n a -> ctx () -> UnclogC m (ctx a)
alg Handler ctx n (UnclogC m)
hdl (:+:) Log sig n a
sig ctx ()
ctx = case (:+:) Log sig n a
sig of
    L (PublishLogEntry Time -> LogEntry
entry) -> (PLogChan -> m (ctx a)) -> UnclogC m (ctx a)
forall (m :: * -> *) r. (PLogChan -> m r) -> UnclogC m r
MkUnclogC \PLogChan
chan -> do
      Time
ts <- IO Time -> m Time
forall a. IO a -> m a
forall (m :: * -> *) a. MonadIO m => IO a -> m a
liftIO IO Time
now
      ctx a
ctx ()
ctx ctx a -> m () -> m (ctx a)
forall a b. a -> m b -> m a
forall (f :: * -> *) a b. Functor f => a -> f b -> f a
<$ STM () -> m ()
forall (m :: * -> *) a. MonadIO m => STM a -> m a
atomically (PLogChan -> LogEntry -> STM ()
Frontend.publishLogEntry PLogChan
chan (LogEntry -> STM ()) -> LogEntry -> STM ()
forall a b. (a -> b) -> a -> b
$ Time -> LogEntry
entry Time
ts)
    R sig n a
other -> (PLogChan -> m (ctx a)) -> UnclogC m (ctx a)
forall (m :: * -> *) r. (PLogChan -> m r) -> UnclogC m r
MkUnclogC \PLogChan
chan -> Handler ctx n m -> sig n a -> ctx () -> m (ctx a)
forall (ctx :: * -> *) (n :: * -> *) a.
Functor ctx =>
Handler ctx n m -> sig n a -> ctx () -> 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 ((UnclogC m (ctx x) -> PLogChan -> m (ctx x))
-> PLogChan -> UnclogC m (ctx x) -> m (ctx x)
forall a b c. (a -> b -> c) -> b -> a -> c
flip UnclogC m (ctx x) -> PLogChan -> m (ctx x)
forall (m :: * -> *) r. UnclogC m r -> PLogChan -> m r
runUnclogC PLogChan
chan (UnclogC m (ctx x) -> m (ctx x))
-> (ctx (n x) -> UnclogC m (ctx x)) -> ctx (n x) -> m (ctx x)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ctx (n x) -> UnclogC m (ctx x)
Handler ctx n (UnclogC m)
hdl) sig n a
other ctx ()
ctx

-- | run the logging effect into multiple subscribers.
-- this is what you would use in actual code
runUnclogging :: MonadUnliftIO m => [Subscriber m] -> UnclogC m a -> m a
runUnclogging :: forall (m :: * -> *) a.
MonadUnliftIO m =>
[Subscriber m] -> UnclogC m a -> m a
runUnclogging [Subscriber m]
subs UnclogC m a
act = [Subscriber m] -> (PLogChan -> m a) -> m a
forall (m :: * -> *) r.
MonadUnliftIO m =>
[Subscriber m] -> (PLogChan -> m r) -> m r
withLoggingWithSubscribers [Subscriber m]
subs (UnclogC m a -> PLogChan -> m a
forall (m :: * -> *) r. UnclogC m r -> PLogChan -> m r
runUnclogC UnclogC m a
act)

-- | run the logging effect and simply collect the log entries.
-- this is useful to check if e.g. a specific log entry was emitted in tests
runUncloggingToList :: MonadUnliftIO m => UnclogC m a -> m ([LogEntry], a)
runUncloggingToList :: forall (m :: * -> *) a.
MonadUnliftIO m =>
UnclogC m a -> m ([LogEntry], a)
runUncloggingToList UnclogC m a
act = do
  IORef [LogEntry]
ref <- [LogEntry] -> m (IORef [LogEntry])
forall (m :: * -> *) a. MonadIO m => a -> m (IORef a)
newIORef []
  a
res <- [Subscriber m] -> (PLogChan -> m a) -> m a
forall (m :: * -> *) r.
MonadUnliftIO m =>
[Subscriber m] -> (PLogChan -> m r) -> m r
withLoggingWithSubscribers [IORef [LogEntry] -> Subscriber m
forall (m :: * -> *). MonadIO m => IORef [LogEntry] -> Subscriber m
refSubscriber IORef [LogEntry]
ref] (UnclogC m a -> PLogChan -> m a
forall (m :: * -> *) r. UnclogC m r -> PLogChan -> m r
runUnclogC UnclogC m a
act)
  [LogEntry]
logs <- IORef [LogEntry] -> m [LogEntry]
forall (m :: * -> *) a. MonadIO m => IORef a -> m a
readIORef IORef [LogEntry]
ref
  ([LogEntry], a) -> m ([LogEntry], a)
forall a. a -> m a
forall (f :: * -> *) a. Applicative f => a -> f a
pure ([LogEntry] -> [LogEntry]
forall a. [a] -> [a]
reverse [LogEntry]
logs, a
res)