Source file state.ml

module Make (S : sig
  type t
end)
(M : Monad.S) =
struct
  module T = struct
    type 'a t = S.t -> (S.t * 'a) M.t

    let return a s = M.return (s, a)

    let bind x ~f s =
      let open M.O in
      let* s', a = x s in
      (f a) s'
  end

  open M.O
  include T

  let lift m s = m >>| fun a -> (s, a)

  let modify f s = M.return (f s, ())

  let get : S.t T.t = fun s -> M.return (s, s)

  let set s _ = M.return (s, ())

  let run t s = t s

  include Monad.Make (T)
end