module ST where

import MyState

newtype ST s a = ST (s -> (a, s))

unST (ST m) = m

instance Monad (ST s)  where
  return a = ST (\ s -> (a, s))
  (ST m) >>= k = ST (\ s0 -> let { (a, s1) = m s0 } in unST (k a) s1)

instance MyState ST where
  get p   = ST (\ s -> (fst (p s), s))
  set p v = ST (\ s -> ((), snd (p s) v))