{-# LANGUAGE AllowAmbiguousTypes #-}
{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE DerivingStrategies #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE InstanceSigs #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# 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
  { forall (m :: * -> *) a.
GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
runGraphulaIdempotentT' :: ReaderT (IORef (m ())) m a
  }
  deriving newtype
    ( forall a b. a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall a b.
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
forall (m :: * -> *) a b.
Functor m =>
a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall (m :: * -> *) a b.
Functor m =>
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT 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 -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
$c<$ :: forall (m :: * -> *) a b.
Functor m =>
a -> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
fmap :: forall a b.
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
$cfmap :: forall (m :: * -> *) a b.
Functor m =>
(a -> b) -> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
Functor
    , 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 :: * -> *).
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 (GraphulaIdempotentT m)
forall (m :: * -> *) a.
Applicative m =>
a -> GraphulaIdempotentT m a
forall (m :: * -> *) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
forall (m :: * -> *) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall (m :: * -> *) a b.
Applicative m =>
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
forall (m :: * -> *) a b c.
Applicative m =>
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
<* :: forall a b.
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
$c<* :: forall (m :: * -> *) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m a
*> :: forall a b.
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
$c*> :: forall (m :: * -> *) a b.
Applicative m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
liftA2 :: forall a b c.
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
$cliftA2 :: forall (m :: * -> *) a b c.
Applicative m =>
(a -> b -> c)
-> GraphulaIdempotentT m a
-> GraphulaIdempotentT m b
-> GraphulaIdempotentT m c
<*> :: forall a b.
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
$c<*> :: forall (m :: * -> *) a b.
Applicative m =>
GraphulaIdempotentT m (a -> b)
-> GraphulaIdempotentT m a -> GraphulaIdempotentT m b
pure :: forall a. a -> GraphulaIdempotentT m a
$cpure :: forall (m :: * -> *) a.
Applicative m =>
a -> GraphulaIdempotentT m a
Applicative
    , 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 :: * -> *}.
Monad m =>
Applicative (GraphulaIdempotentT m)
forall (m :: * -> *) a. Monad m => a -> GraphulaIdempotentT m a
forall (m :: * -> *) a b.
Monad m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
forall (m :: * -> *) a b.
Monad m =>
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT 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 -> GraphulaIdempotentT m a
$creturn :: forall (m :: * -> *) a. Monad m => a -> GraphulaIdempotentT m a
>> :: forall a b.
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
$c>> :: forall (m :: * -> *) a b.
Monad m =>
GraphulaIdempotentT m a
-> GraphulaIdempotentT m b -> GraphulaIdempotentT m b
>>= :: forall a b.
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
$c>>= :: forall (m :: * -> *) a b.
Monad m =>
GraphulaIdempotentT m a
-> (a -> GraphulaIdempotentT m b) -> GraphulaIdempotentT m b
Monad
    , forall a. IO a -> GraphulaIdempotentT m a
forall (m :: * -> *).
Monad m -> (forall a. IO a -> m a) -> MonadIO m
forall {m :: * -> *}. MonadIO m => Monad (GraphulaIdempotentT m)
forall (m :: * -> *) a.
MonadIO m =>
IO a -> GraphulaIdempotentT m a
liftIO :: forall a. IO a -> GraphulaIdempotentT m a
$cliftIO :: forall (m :: * -> *) a.
MonadIO m =>
IO a -> GraphulaIdempotentT m a
MonadIO
    , MonadReader (IORef (m ()))
    )

instance MonadUnliftIO m => MonadUnliftIO (GraphulaIdempotentT m) where
  {-# INLINE withRunInIO #-}
  withRunInIO :: forall b.
((forall a. GraphulaIdempotentT m a -> IO a) -> IO b)
-> GraphulaIdempotentT m b
withRunInIO (forall a. GraphulaIdempotentT m a -> IO a) -> IO b
inner = forall (m :: * -> *) a.
ReaderT (IORef (m ())) m a -> GraphulaIdempotentT m a
GraphulaIdempotentT forall a b. (a -> b) -> a -> b
$ forall (m :: * -> *) b.
MonadUnliftIO m =>
((forall a. m a -> IO a) -> IO b) -> m b
withRunInIO 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 b. (a -> b) -> a -> b
$ forall a. ReaderT (IORef (m ())) m a -> IO a
run forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (m :: * -> *) a.
GraphulaIdempotentT m a -> ReaderT (IORef (m ())) m a
runGraphulaIdempotentT'

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

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

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

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