{-# LANGUAGE AllowAmbiguousTypes #-}
{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE DerivingStrategies #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE InstanceSigs #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE NoStarIsType #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE PolyKinds #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeFamilyDependencies #-}
{-# LANGUAGE UndecidableInstances #-}
{-# LANGUAGE UndecidableSuperClasses #-}

-- | A version of 'GraphulaT' that @'remove'@s all @'insert'@ed data afterward
module Graphula.Idempotent
  ( GraphulaIdempotentT
  , runGraphulaIdempotentT
  ) where

import Prelude

import Control.Monad.IO.Unlift
import Control.Monad.Reader (MonadReader, ReaderT, ask, runReaderT)
import Control.Monad.Trans (MonadTrans, lift)
import Data.Foldable (for_)
import Data.IORef (IORef, modifyIORef', newIORef, readIORef)
import Database.Persist (Entity(..))
import Graphula.Class
import UnliftIO.Exception (SomeException, catch, mask, throwIO)

newtype GraphulaIdempotentT m a = GraphulaIdempotentT
  { GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
runGraphulaIdempotentT' :: ReaderT (IORef (m ())) m a
  }
  deriving newtype
    ( a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
(forall a b.
 (a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b)
-> (forall a b.
    a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a)
-> Functor (GraphulaIdempotentT m)
forall a b. a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall a b.
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
forall (m :: Type -> Type) a b.
Functor m =>
a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall (m :: Type -> Type) a b.
Functor m =>
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
forall (f :: Type -> Type).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
$c<$ :: forall (m :: Type -> Type) a b.
Functor m =>
a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
fmap :: (a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
$cfmap :: forall (m :: Type -> Type) a b.
Functor m =>
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
Functor
    , Functor (GraphulaIdempotentT m)
a -> GraphulaIdempotentT m a
Functor (GraphulaIdempotentT m)
-> (forall a. a -> GraphulaIdempotentT m a)
-> (forall a b.
    GraphulaIdempotentT m (a -> b)
    -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b)
-> (forall a b c.
    (a -> b -> c)
    -> GraphulaIdempotentT m a
    -> GraphulaIdempotentT m b
    -> GraphulaIdempotentT m c)
-> (forall a b.
    GraphulaIdempotentT m a
    -> GraphulaIdempotentT m b -> GraphulaIdempotentT m b)
-> (forall a b.
    GraphulaIdempotentT m a
    -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a)
-> Applicative (GraphulaIdempotentT m)
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
forall a. a -> GraphulaIdempotentT m a
forall a b.
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall a b.
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall a b.
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
forall a b c.
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
forall (f :: Type -> Type).
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 :: Type -> Type).
Applicative m =>
Functor (GraphulaIdempotentT m)
forall (m :: Type -> Type) a.
Applicative m =>
a -> GraphulaIdempotentT m a
forall (m :: Type -> Type) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall (m :: Type -> Type) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall (m :: Type -> Type) a b.
Applicative m =>
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
forall (m :: Type -> Type) a b c.
Applicative m =>
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
<* :: GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
$c<* :: forall (m :: Type -> Type) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
*> :: GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
$c*> :: forall (m :: Type -> Type) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
liftA2 :: (a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
$cliftA2 :: forall (m :: Type -> Type) a b c.
Applicative m =>
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
<*> :: GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
$c<*> :: forall (m :: Type -> Type) a b.
Applicative m =>
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
pure :: a -> GraphulaIdempotentT m a
$cpure :: forall (m :: Type -> Type) a.
Applicative m =>
a -> GraphulaIdempotentT m a
$cp1Applicative :: forall (m :: Type -> Type).
Applicative m =>
Functor (GraphulaIdempotentT m)
Applicative
    , Applicative (GraphulaIdempotentT m)
a -> GraphulaIdempotentT m a
Applicative (GraphulaIdempotentT m)
-> (forall a b.
    GraphulaIdempotentT m a
    -> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b)
-> (forall a b.
    GraphulaIdempotentT m a
    -> GraphulaIdempotentT m b -> GraphulaIdempotentT m b)
-> (forall a. a -> GraphulaIdempotentT m a)
-> Monad (GraphulaIdempotentT m)
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall a. a -> GraphulaIdempotentT m a
forall a b.
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall a b.
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
forall (m :: Type -> Type).
Monad m =>
Applicative (GraphulaIdempotentT m)
forall (m :: Type -> Type) a.
Monad m =>
a -> GraphulaIdempotentT m a
forall (m :: Type -> Type) a b.
Monad m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall (m :: Type -> Type) a b.
Monad m =>
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
forall (m :: Type -> Type).
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 -> GraphulaIdempotentT m a
$creturn :: forall (m :: Type -> Type) a.
Monad m =>
a -> GraphulaIdempotentT m a
>> :: GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
$c>> :: forall (m :: Type -> Type) a b.
Monad m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
>>= :: GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
$c>>= :: forall (m :: Type -> Type) a b.
Monad m =>
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
$cp1Monad :: forall (m :: Type -> Type).
Monad m =>
Applicative (GraphulaIdempotentT m)
Monad
    , Monad (GraphulaIdempotentT m)
Monad (GraphulaIdempotentT m)
-> (forall a. IO a -> GraphulaIdempotentT m a)
-> MonadIO (GraphulaIdempotentT m)
IO a -> GraphulaIdempotentT m a
forall a. IO a -> GraphulaIdempotentT m a
forall (m :: Type -> Type).
Monad m -> (forall a. IO a -> m a) -> MonadIO m
forall (m :: Type -> Type).
MonadIO m =>
Monad (GraphulaIdempotentT m)
forall (m :: Type -> Type) a.
MonadIO m =>
IO a -> GraphulaIdempotentT m a
liftIO :: IO a -> GraphulaIdempotentT m a
$cliftIO :: forall (m :: Type -> Type) a.
MonadIO m =>
IO a -> GraphulaIdempotentT m a
$cp1MonadIO :: forall (m :: Type -> Type).
MonadIO m =>
Monad (GraphulaIdempotentT m)
MonadIO
    , MonadReader (IORef (m ()))
    )

instance MonadUnliftIO m => MonadUnliftIO (GraphulaIdempotentT m) where
  {-# INLINE withRunInIO #-}
  withRunInIO :: ((forall a. GraphulaIdempotentT m a -> IO a) -> IO b)
-> GraphulaIdempotentT m b
withRunInIO (forall a. GraphulaIdempotentT m a -> IO a) -> IO b
inner = ReaderT (IORef (m ())) m b -> GraphulaIdempotentT m b
forall (m :: Type -> Type) a.
ReaderT (IORef (m ())) m a -> GraphulaIdempotentT m a
GraphulaIdempotentT (ReaderT (IORef (m ())) m b -> GraphulaIdempotentT m b)
-> ReaderT (IORef (m ())) m b -> GraphulaIdempotentT m b
forall a b. (a -> b) -> a -> b
$ ((forall a. ReaderT (IORef (m ())) m a -> IO a) -> IO b)
-> ReaderT (IORef (m ())) m b
forall (m :: Type -> Type) b.
MonadUnliftIO m =>
((forall a. m a -> IO a) -> IO b) -> m b
withRunInIO (((forall a. ReaderT (IORef (m ())) m a -> IO a) -> IO b)
 -> ReaderT (IORef (m ())) m b)
-> ((forall a. ReaderT (IORef (m ())) m a -> IO a) -> IO b)
-> ReaderT (IORef (m ())) m b
forall a b. (a -> b) -> a -> b
$ \forall a. ReaderT (IORef (m ())) m a -> IO a
run ->
    (forall a. GraphulaIdempotentT m a -> IO a) -> IO b
inner ((forall a. GraphulaIdempotentT m a -> IO a) -> IO b)
-> (forall a. GraphulaIdempotentT m a -> IO a) -> IO b
forall a b. (a -> b) -> a -> b
$ ReaderT (IORef (m ())) m a -> IO a
forall a. ReaderT (IORef (m ())) m a -> IO a
run (ReaderT (IORef (m ())) m a -> IO a)
-> (GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a)
-> GraphulaIdempotentT m a
-> IO a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
forall (m :: Type -> Type) a.
GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
runGraphulaIdempotentT'

instance MonadTrans GraphulaIdempotentT where
  lift :: m a -> GraphulaIdempotentT m a
lift = ReaderT (IORef (m ())) m a -> GraphulaIdempotentT m a
forall (m :: Type -> Type) a.
ReaderT (IORef (m ())) m a -> GraphulaIdempotentT m a
GraphulaIdempotentT (ReaderT (IORef (m ())) m a -> GraphulaIdempotentT m a)
-> (m a -> ReaderT (IORef (m ())) m a)
-> m a
-> GraphulaIdempotentT m a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. m a -> ReaderT (IORef (m ())) m a
forall (t :: (Type -> Type) -> Type -> Type) (m :: Type -> Type) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift

instance (MonadIO m, MonadGraphulaFrontend m) => MonadGraphulaFrontend (GraphulaIdempotentT m) where
  insert :: Maybe (Key a) -> a -> GraphulaIdempotentT m (Maybe (Entity a))
insert Maybe (Key a)
mKey a
n = do
    IORef (m ())
finalizersRef <- GraphulaIdempotentT m (IORef (m ()))
forall r (m :: Type -> Type). MonadReader r m => m r
ask
    Maybe (Entity a)
mEnt <- m (Maybe (Entity a)) -> GraphulaIdempotentT m (Maybe (Entity a))
forall (t :: (Type -> Type) -> Type -> Type) (m :: Type -> Type) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (m (Maybe (Entity a)) -> GraphulaIdempotentT m (Maybe (Entity a)))
-> m (Maybe (Entity a)) -> GraphulaIdempotentT m (Maybe (Entity a))
forall a b. (a -> b) -> a -> b
$ Maybe (Key a) -> a -> m (Maybe (Entity a))
forall (m :: Type -> Type) a.
(MonadGraphulaFrontend m, PersistEntityBackend a ~ SqlBackend,
 PersistEntity a, Monad m) =>
Maybe (Key a) -> a -> m (Maybe (Entity a))
insert Maybe (Key a)
mKey a
n
    Maybe (Key a)
-> (Key a -> GraphulaIdempotentT m ()) -> GraphulaIdempotentT m ()
forall (t :: Type -> Type) (f :: Type -> Type) a b.
(Foldable t, Applicative f) =>
t a -> (a -> f b) -> f ()
for_ (Entity a -> Key a
forall record. Entity record -> Key record
entityKey (Entity a -> Key a) -> Maybe (Entity a) -> Maybe (Key a)
forall (f :: Type -> Type) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe (Entity a)
mEnt)
      ((Key a -> GraphulaIdempotentT m ()) -> GraphulaIdempotentT m ())
-> (Key a -> GraphulaIdempotentT m ()) -> GraphulaIdempotentT m ()
forall a b. (a -> b) -> a -> b
$ \Key a
key -> IO () -> GraphulaIdempotentT m ()
forall (m :: Type -> Type) a. MonadIO m => IO a -> m a
liftIO (IO () -> GraphulaIdempotentT m ())
-> IO () -> GraphulaIdempotentT m ()
forall a b. (a -> b) -> a -> b
$ IORef (m ()) -> (m () -> m ()) -> IO ()
forall a. IORef a -> (a -> a) -> IO ()
modifyIORef' IORef (m ())
finalizersRef (Key a -> m ()
forall (m :: Type -> Type) a.
(MonadGraphulaFrontend m, PersistEntityBackend a ~ SqlBackend,
 PersistEntity a, Monad m) =>
Key a -> m ()
remove Key a
key m () -> m () -> m ()
forall (m :: Type -> Type) a b. Monad m => m a -> m b -> m b
>>)
    Maybe (Entity a) -> GraphulaIdempotentT m (Maybe (Entity a))
forall (f :: Type -> Type) a. Applicative f => a -> f a
pure Maybe (Entity a)
mEnt
  remove :: Key a -> GraphulaIdempotentT m ()
remove = m () -> GraphulaIdempotentT m ()
forall (t :: (Type -> Type) -> Type -> Type) (m :: Type -> Type) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift (m () -> GraphulaIdempotentT m ())
-> (Key a -> m ()) -> Key a -> GraphulaIdempotentT m ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Key a -> m ()
forall (m :: Type -> Type) a.
(MonadGraphulaFrontend m, PersistEntityBackend a ~ SqlBackend,
 PersistEntity a, Monad m) =>
Key a -> m ()
remove

runGraphulaIdempotentT :: MonadUnliftIO m => GraphulaIdempotentT m a -> m a
runGraphulaIdempotentT :: GraphulaIdempotentT m a -> m a
runGraphulaIdempotentT GraphulaIdempotentT m a
action = ((forall a. m a -> m a) -> m a) -> m a
forall (m :: Type -> Type) b.
MonadUnliftIO m =>
((forall a. m a -> m a) -> m b) -> m b
mask (((forall a. m a -> m a) -> m a) -> m a)
-> ((forall a. m a -> m a) -> m a) -> m a
forall a b. (a -> b) -> a -> b
$ \forall a. m a -> m a
unmasked -> do
  IORef (m ())
finalizersRef <- IO (IORef (m ())) -> m (IORef (m ()))
forall (m :: Type -> Type) a. MonadIO m => IO a -> m a
liftIO (IO (IORef (m ())) -> m (IORef (m ())))
-> (m () -> IO (IORef (m ()))) -> m () -> m (IORef (m ()))
forall b c a. (b -> c) -> (a -> b) -> a -> c
. m () -> IO (IORef (m ()))
forall a. a -> IO (IORef a)
newIORef (m () -> m (IORef (m ()))) -> m () -> m (IORef (m ()))
forall a b. (a -> b) -> a -> b
$ () -> m ()
forall (f :: Type -> Type) a. Applicative f => a -> f a
pure ()
  a
x <-
    m a -> m a
forall a. m a -> m a
unmasked
    (m a -> m a) -> m a -> m a
forall a b. (a -> b) -> a -> b
$ ReaderT (IORef (m ())) m a -> IORef (m ()) -> m a
forall r (m :: Type -> Type) a. ReaderT r m a -> r -> m a
runReaderT (GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
forall (m :: Type -> Type) a.
GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
runGraphulaIdempotentT' GraphulaIdempotentT m a
action) IORef (m ())
finalizersRef
    m a -> (SomeException -> m a) -> m a
forall (m :: Type -> Type) e a.
(MonadUnliftIO m, Exception e) =>
m a -> (e -> m a) -> m a
`catch` IORef (m ()) -> SomeException -> m a
forall (m :: Type -> Type) a b.
MonadIO m =>
IORef (m a) -> SomeException -> m b
rollbackRethrow IORef (m ())
finalizersRef
  IORef (m ()) -> m a -> m a
forall (m :: Type -> Type) a b.
MonadIO m =>
IORef (m a) -> m b -> m b
rollback IORef (m ())
finalizersRef (m a -> m a) -> m a -> m a
forall a b. (a -> b) -> a -> b
$ a -> m a
forall (f :: Type -> Type) a. Applicative f => a -> f a
pure a
x
 where
  rollback :: MonadIO m => IORef (m a) -> m b -> m b
  rollback :: IORef (m a) -> m b -> m b
rollback IORef (m a)
finalizersRef m b
x = do
    m a
finalizers <- IO (m a) -> m (m a)
forall (m :: Type -> Type) a. MonadIO m => IO a -> m a
liftIO (IO (m a) -> m (m a)) -> IO (m a) -> m (m a)
forall a b. (a -> b) -> a -> b
$ IORef (m a) -> IO (m a)
forall a. IORef a -> IO a
readIORef IORef (m a)
finalizersRef
    m a
finalizers m a -> m b -> m b
forall (m :: Type -> Type) a b. Monad m => m a -> m b -> m b
>> m b
x

  rollbackRethrow :: MonadIO m => IORef (m a) -> SomeException -> m b
  rollbackRethrow :: IORef (m a) -> SomeException -> m b
rollbackRethrow IORef (m a)
finalizersRef (SomeException
e :: SomeException) =
    IORef (m a) -> m b -> m b
forall (m :: Type -> Type) a b.
MonadIO m =>
IORef (m a) -> m b -> m b
rollback IORef (m a)
finalizersRef (SomeException -> m b
forall (m :: Type -> Type) e a.
(MonadIO m, Exception e) =>
e -> m a
throwIO SomeException
e)