use crate::prelude::Monad;
use super::def::State;
impl<'a, S, A> Monad<'a, A> for State<'a, S, A>
where
S: Clone + 'a,
A: 'a,
{
fn bind<F, B>(self, f: F) -> Self::Type<B>
where
F: Fn(A) -> Self::Type<B> + 'a,
B: 'a,
{
Self::Type::<B> {
run_state: Some(Box::new(move |s| {
let (a, st) = self.run(s);
let new_state = f(a);
let (b, stt) = new_state.run(st);
(b, stt)
})),
}
}
}