module Compose where

import Prelude hiding (Monad, return )

-- | TODO 
-- | 
-- | 1. default methods are currently not supported

data ST s a = ST {runState :: s -> (a,s)}

{-@ data ST s a <r :: a -> Bool> 
  = ST (runState :: x:s -> (a<r>, s)) @-}

{-@ runState :: forall <r :: a -> Bool>. ST <r> s a -> x:s -> (a<r>, s) @-}


class Foo m where
  return :: a -> m a


instance Foo (ST s) where
  {-@ instance Foo ST s where
    return :: forall s a. x:a -> ST <{\v -> x == v}> s a
    @-}
  return x     = ST $ \s -> (x, s)
 

{-@ foo :: w:a -> ST <{v:a | v = w}>  Bool a @-}
foo :: a -> ST Bool a
foo x = return x


bar = runState (foo 0) True