{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{- |
Here we implement a monad transformer
which adds exception handling and
labelling of actions (using "Control.Monad.Label")
in order to extend exceptions with a kind of call stack.
-}
module Control.Monad.Exception.Label where

import qualified Control.Monad.Exception.Synchronous as Exception
import qualified Control.Monad.Label as Label

import Control.Monad.Exception.Synchronous (ExceptionalT, mapExceptionT, )
import Control.Monad.Label (LabelT, )
import Control.Applicative (Applicative, )

import Control.Monad (liftM, )
import Control.Monad.Fix (MonadFix, )
import Control.Monad.Trans.Class (MonadTrans, lift, )


data LabeledException l e =
   LabeledException {forall l e. LabeledException l e -> [l]
labels :: [l], forall l e. LabeledException l e -> e
exception :: e}

newtype LabeledExceptionalT l e m a =
   LabeledExceptionalT
      {forall l e (m :: * -> *) a.
LabeledExceptionalT l e m a
-> LabelT l (ExceptionalT (LabeledException l e) m) a
runLabeledExceptionalT :: LabelT l (ExceptionalT (LabeledException l e) m) a}
      deriving (forall a b.
a -> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
forall a b.
(a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
forall l e (m :: * -> *) a b.
Functor m =>
a -> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
forall l e (m :: * -> *) a b.
Functor m =>
(a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: forall a b.
a -> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
$c<$ :: forall l e (m :: * -> *) a b.
Functor m =>
a -> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
fmap :: forall a b.
(a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
$cfmap :: forall l e (m :: * -> *) a b.
Functor m =>
(a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
Functor, forall a. a -> LabeledExceptionalT l e m a
forall a b.
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
forall a b.
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
forall a b.
LabeledExceptionalT l e m (a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
forall a b c.
(a -> b -> c)
-> LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b
-> LabeledExceptionalT l e m c
forall {l} {e} {m :: * -> *}.
Applicative m =>
Functor (LabeledExceptionalT l e m)
forall l e (m :: * -> *) a.
Applicative m =>
a -> LabeledExceptionalT l e m a
forall l e (m :: * -> *) a b.
Applicative m =>
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
forall l e (m :: * -> *) a b.
Applicative m =>
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
forall l e (m :: * -> *) a b.
Applicative m =>
LabeledExceptionalT l e m (a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
forall l e (m :: * -> *) a b c.
Applicative m =>
(a -> b -> c)
-> LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b
-> LabeledExceptionalT l e 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 a b.
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
$c<* :: forall l e (m :: * -> *) a b.
Applicative m =>
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m a
*> :: forall a b.
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
$c*> :: forall l e (m :: * -> *) a b.
Applicative m =>
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
liftA2 :: forall a b c.
(a -> b -> c)
-> LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b
-> LabeledExceptionalT l e m c
$cliftA2 :: forall l e (m :: * -> *) a b c.
Applicative m =>
(a -> b -> c)
-> LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b
-> LabeledExceptionalT l e m c
<*> :: forall a b.
LabeledExceptionalT l e m (a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
$c<*> :: forall l e (m :: * -> *) a b.
Applicative m =>
LabeledExceptionalT l e m (a -> b)
-> LabeledExceptionalT l e m a -> LabeledExceptionalT l e m b
pure :: forall a. a -> LabeledExceptionalT l e m a
$cpure :: forall l e (m :: * -> *) a.
Applicative m =>
a -> LabeledExceptionalT l e m a
Applicative, forall a. a -> LabeledExceptionalT l e m a
forall a b.
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
forall a b.
LabeledExceptionalT l e m a
-> (a -> LabeledExceptionalT l e m b)
-> LabeledExceptionalT l e m b
forall {l} {e} {m :: * -> *}.
Monad m =>
Applicative (LabeledExceptionalT l e m)
forall l e (m :: * -> *) a.
Monad m =>
a -> LabeledExceptionalT l e m a
forall l e (m :: * -> *) a b.
Monad m =>
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
forall l e (m :: * -> *) a b.
Monad m =>
LabeledExceptionalT l e m a
-> (a -> LabeledExceptionalT l e m b)
-> LabeledExceptionalT l e 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 :: forall a. a -> LabeledExceptionalT l e m a
$creturn :: forall l e (m :: * -> *) a.
Monad m =>
a -> LabeledExceptionalT l e m a
>> :: forall a b.
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
$c>> :: forall l e (m :: * -> *) a b.
Monad m =>
LabeledExceptionalT l e m a
-> LabeledExceptionalT l e m b -> LabeledExceptionalT l e m b
>>= :: forall a b.
LabeledExceptionalT l e m a
-> (a -> LabeledExceptionalT l e m b)
-> LabeledExceptionalT l e m b
$c>>= :: forall l e (m :: * -> *) a b.
Monad m =>
LabeledExceptionalT l e m a
-> (a -> LabeledExceptionalT l e m b)
-> LabeledExceptionalT l e m b
Monad, forall a.
(a -> LabeledExceptionalT l e m a) -> LabeledExceptionalT l e m a
forall {l} {e} {m :: * -> *}.
MonadFix m =>
Monad (LabeledExceptionalT l e m)
forall l e (m :: * -> *) a.
MonadFix m =>
(a -> LabeledExceptionalT l e m a) -> LabeledExceptionalT l e m a
forall (m :: * -> *).
Monad m -> (forall a. (a -> m a) -> m a) -> MonadFix m
mfix :: forall a.
(a -> LabeledExceptionalT l e m a) -> LabeledExceptionalT l e m a
$cmfix :: forall l e (m :: * -> *) a.
MonadFix m =>
(a -> LabeledExceptionalT l e m a) -> LabeledExceptionalT l e m a
MonadFix)


runLabelT :: (Monad m) =>
   LabeledExceptionalT l e m a ->
   [l] ->
   ExceptionalT (LabeledException l e) m a
runLabelT :: forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT =
   forall (m :: * -> *) l a. Monad m => LabelT l m a -> [l] -> m a
Label.runLabelT forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall l e (m :: * -> *) a.
LabeledExceptionalT l e m a
-> LabelT l (ExceptionalT (LabeledException l e) m) a
runLabeledExceptionalT

labelT :: (Monad m) =>
   ExceptionalT (LabeledException l e) m a ->
   LabeledExceptionalT l e m a
labelT :: forall (m :: * -> *) l e a.
Monad m =>
ExceptionalT (LabeledException l e) m a
-> LabeledExceptionalT l e m a
labelT =
   forall l e (m :: * -> *) a.
LabelT l (ExceptionalT (LabeledException l e) m) a
-> LabeledExceptionalT l e m a
LabeledExceptionalT forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift -- Label.LabelT . ReaderT


stripLabelT :: (Monad m) =>
   LabeledExceptionalT l e m a -> ExceptionalT e m a
stripLabelT :: forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a -> ExceptionalT e m a
stripLabelT LabeledExceptionalT l e m a
action =
   forall (m :: * -> *) e0 e1 a.
Monad m =>
(e0 -> e1) -> ExceptionalT e0 m a -> ExceptionalT e1 m a
mapExceptionT forall l e. LabeledException l e -> e
exception (forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT LabeledExceptionalT l e m a
action [])

decorateLabelT :: (Monad m) =>
   ExceptionalT e m a -> LabeledExceptionalT l e m a
decorateLabelT :: forall (m :: * -> *) e a l.
Monad m =>
ExceptionalT e m a -> LabeledExceptionalT l e m a
decorateLabelT =
   forall (m :: * -> *) l e a.
Monad m =>
ExceptionalT (LabeledException l e) m a
-> LabeledExceptionalT l e m a
labelT forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (m :: * -> *) e0 e1 a.
Monad m =>
(e0 -> e1) -> ExceptionalT e0 m a -> ExceptionalT e1 m a
mapExceptionT (forall l e. [l] -> e -> LabeledException l e
LabeledException [])

getLabels :: (Monad m) =>
   LabeledExceptionalT l e m [l]
getLabels :: forall (m :: * -> *) l e. Monad m => LabeledExceptionalT l e m [l]
getLabels = forall l e (m :: * -> *) a.
LabelT l (ExceptionalT (LabeledException l e) m) a
-> LabeledExceptionalT l e m a
LabeledExceptionalT forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *) l. Monad m => LabelT l m [l]
Label.askT


throwT :: (Monad m) =>
   e -> LabeledExceptionalT l e m a
throwT :: forall (m :: * -> *) e l a.
Monad m =>
e -> LabeledExceptionalT l e m a
throwT e
e =
   do [l]
l <- forall (m :: * -> *) l e. Monad m => LabeledExceptionalT l e m [l]
getLabels
      forall (m :: * -> *) l e a.
Monad m =>
ExceptionalT (LabeledException l e) m a
-> LabeledExceptionalT l e m a
labelT forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *) e a. Monad m => e -> ExceptionalT e m a
Exception.throwT (forall l e. [l] -> e -> LabeledException l e
LabeledException [l]
l e
e)


{- |
Currently 'catchT' calls the exception handler with a full call stack.
Since 'catchT' handles exceptions locally
it may however clear the call stack before calling the inner action
and a re-throw should append the inner call stack to the outer one.
For this semantics, a difference list would be more efficient for labels.
-}
catchT :: (Monad m) =>
   LabeledExceptionalT l e0 m a ->
   ([l] -> e0 -> LabeledExceptionalT l e1 m a) ->
   LabeledExceptionalT l e1 m a
catchT :: forall (m :: * -> *) l e0 a e1.
Monad m =>
LabeledExceptionalT l e0 m a
-> ([l] -> e0 -> LabeledExceptionalT l e1 m a)
-> LabeledExceptionalT l e1 m a
catchT LabeledExceptionalT l e0 m a
action [l] -> e0 -> LabeledExceptionalT l e1 m a
handler =
   do [l]
ls <- forall (m :: * -> *) l e. Monad m => LabeledExceptionalT l e m [l]
getLabels
      forall (m :: * -> *) l e a.
Monad m =>
ExceptionalT (LabeledException l e) m a
-> LabeledExceptionalT l e m a
labelT forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *) e0 a e1.
Monad m =>
ExceptionalT e0 m a
-> (e0 -> ExceptionalT e1 m a) -> ExceptionalT e1 m a
Exception.catchT
         (forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT LabeledExceptionalT l e0 m a
action [l]
ls)
         (\(LabeledException [l]
l e0
e) -> forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT ([l] -> e0 -> LabeledExceptionalT l e1 m a
handler [l]
l e0
e) [l]
ls)


{- |
If the enclosed monad has custom exception facilities,
they could skip the cleanup code.
Make sure, that this cannot happen by choosing an appropriate monad.
-}
bracketT :: (Monad m) =>
   l ->
   LabeledExceptionalT l e m h ->
   (h -> LabeledExceptionalT l e m ()) ->
   (h -> LabeledExceptionalT l e m a) ->
   LabeledExceptionalT l e m a
bracketT :: forall (m :: * -> *) l e h a.
Monad m =>
l
-> LabeledExceptionalT l e m h
-> (h -> LabeledExceptionalT l e m ())
-> (h -> LabeledExceptionalT l e m a)
-> LabeledExceptionalT l e m a
bracketT l
label LabeledExceptionalT l e m h
open h -> LabeledExceptionalT l e m ()
close h -> LabeledExceptionalT l e m a
action =
   do [l]
ls <- forall (m :: * -> *) a1 r. Monad m => (a1 -> r) -> m a1 -> m r
liftM (l
labelforall a. a -> [a] -> [a]
:) forall (m :: * -> *) l e. Monad m => LabeledExceptionalT l e m [l]
getLabels
      forall (m :: * -> *) l e a.
Monad m =>
ExceptionalT (LabeledException l e) m a
-> LabeledExceptionalT l e m a
labelT forall a b. (a -> b) -> a -> b
$
         forall (m :: * -> *) e h a.
Monad m =>
ExceptionalT e m h
-> (h -> ExceptionalT e m ())
-> (h -> ExceptionalT e m a)
-> ExceptionalT e m a
Exception.bracketT
            (forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT LabeledExceptionalT l e m h
open [l]
ls)
            (\h
h -> forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT (h -> LabeledExceptionalT l e m ()
close h
h) [l]
ls)
            (\h
h -> forall (m :: * -> *) l e a.
Monad m =>
LabeledExceptionalT l e m a
-> [l] -> ExceptionalT (LabeledException l e) m a
runLabelT (h -> LabeledExceptionalT l e m a
action h
h) [l]
ls)


instance MonadTrans (LabeledExceptionalT l e) where
   lift :: forall (m :: * -> *) a.
Monad m =>
m a -> LabeledExceptionalT l e m a
lift m a
m = forall (m :: * -> *) l e a.
Monad m =>
ExceptionalT (LabeledException l e) m a
-> LabeledExceptionalT l e m a
labelT forall a b. (a -> b) -> a -> b
$ forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift m a
m