module Glazier.Gadget.Strict where
import Control.Applicative
import Control.Arrow
import qualified Control.Category as C
import Control.Lens
import qualified Control.Monad.Fail as Fail
import Control.Monad.Fix (MonadFix)
import Control.Monad.Morph
import Control.Monad.Reader
import Control.Monad.State.Strict
import Data.Profunctor
import Data.Semigroup
import Glazier.Class
newtype Gadget s m a c = Gadget
{ runGadget :: ReaderT a (StateT s m) c
} deriving ( MonadState s
, MonadReader a
, Monad
, Applicative
, Functor
, Fail.MonadFail
, Alternative
, MonadPlus
, MonadFix
, MonadIO
)
class HasGadget s a | s -> a where
gadget :: Lens' s a
instance HasGadget (Gadget s m a c) (Gadget s m a c) where
gadget = id
makeWrapped ''Gadget
hoistGadget :: (Monad m) => (forall b. m b -> n b) -> Gadget s m a c -> Gadget s n a c
hoistGadget g = _Wrapping Gadget %~ hoist (hoist g)
_Gadget :: Iso (Gadget s m a c) (Gadget s' m' a' c') (a -> s -> m (c, s)) (a' -> s' -> m' (c', s'))
_Gadget = _Wrapping Gadget . iso runReaderT ReaderT . iso (runStateT .) (StateT .)
_Gadget' :: Iso' (Gadget s m a c) (a -> s -> m (c, s))
_Gadget' = _Gadget
instance (Monad m, Semigroup c) => Semigroup (Gadget s m a c) where
(Gadget f) <> (Gadget g) = Gadget $ (<>) <$> f <*> g
instance (Monad m, Monoid c) => Monoid (Gadget s m a c) where
mempty = Gadget $ pure mempty
(Gadget f) `mappend` (Gadget g) = Gadget $ mappend <$> f <*> g
instance Monad m => Profunctor (Gadget s m) where
dimap f g (Gadget (ReaderT m)) = Gadget $ ReaderT $ \a -> StateT $ \s -> undefined
(first g) <$> runStateT (m (f a)) s
instance Monad m => Strong (Gadget s m) where
first' (Gadget (ReaderT bc)) = Gadget $ ReaderT $ \(b, d) -> StateT $ \s ->
(\(c, s') -> ((c, d), s')) <$> runStateT (bc b) s
instance Monad m => C.Category (Gadget s m) where
id = Gadget $ ReaderT $ \a -> StateT $ \s -> pure (a, s)
Gadget (ReaderT bc) . Gadget (ReaderT ab) = Gadget $ ReaderT $ \a -> StateT $ \s -> do
(b, s') <- runStateT (ab a) s
runStateT (bc b) s'
instance Monad m => Arrow (Gadget s m) where
arr f = dimap f id C.id
first = first'
instance Monad m => Choice (Gadget s m) where
left' (Gadget (ReaderT bc)) = Gadget $ ReaderT $ \db -> StateT $ \s -> case db of
Left b -> do
(c, s') <- runStateT (bc b) s
pure (Left c, s')
Right d -> pure (Right d, s)
instance Monad m => ArrowChoice (Gadget s m) where
left = left'
instance Monad m => ArrowApply (Gadget s m) where
app = Gadget $ ReaderT $ \(Gadget (ReaderT bc), b) -> StateT $ \s -> runStateT (bc b) s
instance MonadPlus m => ArrowZero (Gadget s m) where
zeroArrow = Gadget mzero
instance MonadPlus m => ArrowPlus (Gadget s m) where
Gadget a <+> Gadget b = Gadget (a `mplus` b)
type instance Zoomed (Gadget s m a) = Zoomed (StateT s m)
instance Monad m => Zoom (Gadget s m a) (Gadget t m a) s t where
zoom l = Gadget . zoom l . runGadget
type instance Magnified (Gadget s m a) = Magnified (ReaderT a (StateT s m))
instance Monad m => Magnify (Gadget s m a) (Gadget s m b) a b where
magnify l = Gadget . magnify l . runGadget
type instance Implanted (Gadget s m a c) = Zoomed (Gadget s m a) c
instance Monad m => Implant (Gadget s m a c) (Gadget t m a c) s t where
implant = zoom
type instance Dispatched (Gadget s m a c) = Magnified (Gadget s m a) c
instance Monad m => Dispatch (Gadget s m a c) (Gadget s m b c) a b where
dispatch = magnify