module Data.Conduit.Util.Source
( sourceState
, sourceStateIO
, SourceStateResult (..)
, sourceIO
, SourceIOResult (..)
) where
import Control.Monad.Trans.Resource
import Data.Conduit.Internal
data SourceStateResult state output = StateOpen state output | StateClosed
sourceState
:: Monad m
=> state
-> (state -> m (SourceStateResult state output))
-> Source m output
sourceState state0 pull0 =
src state0
where
src state = PipeM (pull state) (return ())
pull state = do
res <- pull0 state
return $ case res of
StateOpen state' val -> HaveOutput (src state') (return ()) val
StateClosed -> Done Nothing ()
data SourceIOResult output = IOOpen output | IOClosed
sourceIO :: MonadResource m
=> IO state
-> (state -> IO ())
-> (state -> m (SourceIOResult output))
-> Source m output
sourceIO alloc cleanup pull0 =
PipeM (do
(key, state) <- allocate alloc cleanup
pull key state) (return ())
where
src key state = PipeM (pull key state) (release key)
pull key state = do
res <- pull0 state
case res of
IOClosed -> do
release key
return $ Done Nothing ()
IOOpen val -> return $ HaveOutput (src key state) (release key) val
sourceStateIO :: MonadResource m
=> IO state
-> (state -> IO ())
-> (state -> m (SourceStateResult state output))
-> Source m output
sourceStateIO alloc cleanup pull0 =
PipeM (do
(key, state) <- allocate alloc cleanup
pull key state) (return ())
where
src key state = PipeM (pull key state) (release key)
pull key state = do
res <- pull0 state
case res of
StateClosed -> do
release key
return $ Done Nothing ()
StateOpen state' val -> return $ HaveOutput (src key state') (release key) val