{-# options_haddock prune #-}
-- |Description: Mask Effect, Internal
module Polysemy.Conc.Effect.Mask where

import Polysemy (makeSem_)

import Polysemy.Conc.Effect.Scoped (Scoped, scoped)

-- |Part of an effect abstracting 'Control.Exception.mask'.
data RestoreMask :: Effect where
  Restore :: m a -> RestoreMask m a

makeSem_ ''RestoreMask

-- |Restore the previous masking state.
-- Can only be called inside of an action passed to 'mask' or 'uninterruptibleMask'.
restore ::
  ∀ r a .
  Member RestoreMask r =>
  Sem r a ->
  Sem r a

newtype MaskResource resource =
  MaskResource { MaskResource resource -> resource
unMaskResource :: resource }

newtype UninterruptipleMaskResource resource =
  UninterruptipleMaskResource { UninterruptipleMaskResource resource -> resource
unUninterruptipleMaskResource :: resource }

-- |The scoped masking effect.
type Mask resource =
  Scoped (MaskResource resource) RestoreMask

-- |The scoped uninterruptible masking effect.
type UninterruptipleMask resource =
  Scoped (UninterruptipleMaskResource resource) RestoreMask

-- |Mark a region as masked.
-- Uses the 'Scoped' pattern.
mask ::
  ∀ resource r .
  Member (Mask resource) r =>
  InterpreterFor RestoreMask r
mask :: InterpreterFor RestoreMask r
mask =
  forall resource (effect :: Effect) (r :: [Effect]).
Member (Scoped resource effect) r =>
InterpreterFor effect r
forall (effect :: Effect) (r :: [Effect]).
Member (Scoped (MaskResource resource) effect) r =>
InterpreterFor effect r
scoped @(MaskResource resource)

-- |Mark a region as uninterruptibly masked.
-- Uses the 'Scoped' pattern.
uninterruptibleMask ::
  ∀ resource r .
  Member (UninterruptipleMask resource) r =>
  InterpreterFor RestoreMask r
uninterruptibleMask :: InterpreterFor RestoreMask r
uninterruptibleMask =
  forall resource (effect :: Effect) (r :: [Effect]).
Member (Scoped resource effect) r =>
InterpreterFor effect r
forall (effect :: Effect) (r :: [Effect]).
Member (Scoped (UninterruptipleMaskResource resource) effect) r =>
InterpreterFor effect r
scoped @(UninterruptipleMaskResource resource)