State instances: Functor, Applicative, Monad

Estado.hs

{-# LANGUAGE InstanceSigs #-}

module State where

import Control.Monad

data Estado s a = Estado { corre :: s -> (a, s) }


instance Functor (Estado s) where
  -- fmap :: (a -> b) -> Estado s a -> Estado s b
  fmap f (Estado g) = Estado $ \s -> case g s of (a, s') -> (f a, s')
          
 
instance Applicative (Estado s) where
  pure :: a -> Estado s a
  pure a = Estado $ \s -> (a, s)
  
  (<*>) :: Estado s (a -> b) -> Estado s a -> Estado s b
  (Estado f) <*> (Estado g) = Estado corre'
    where 
      corre' s = (f' a, s')
        where
          (f', _) = f s
          (a, s') = g s


      
instance Monad (Estado s) where
  return = pure
  
  (>>=) :: Estado s a -> (a -> Estado s b) -> Estado s b
  -- g :: a -> { s -> (b, s) }
  (Estado f) >>= g = Estado $ \s -> case f s of (a, s') -> (corre $ g a) s'  
  
--  (Estado f) >>= g = Estado corre'
--    where
--      corre' s = (corre $ g a) s'
--        where
--           (a, s') = f s