1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use hooks_core::HookExt;

use super::{StateUpdater, State, STAGING_STATES_DEFAULT_STACK_COUNT};

pub struct StateEq<'a, T: 'a + PartialEq, const N: usize = STAGING_STATES_DEFAULT_STACK_COUNT>(
    State<'a, T, N>,
);

impl<'a, T: 'a + PartialEq, const N: usize> Unpin for StateEq<'a, T, N> {}

pub struct StateEqWith<'a, T: 'a + PartialEq, const N: usize = STAGING_STATES_DEFAULT_STACK_COUNT>(
    State<'a, T, N>,
);

impl<'a, T: 'a + PartialEq, const N: usize> Unpin for StateEqWith<'a, T, N> {}

crate::utils::impl_hook! {
    impl ['a, T: 'a + PartialEq, const N: usize] for StateEq<'a, T, N> {
        #[inline]
        poll_next_update(mut self, cx) {
            self.0.poll_next_update_if_not_equal(cx, PartialEq::eq)
        }

        #[inline]
        use_hook(self, initial_state: T) -> (&'hook mut T, &'hook StateUpdater<'a, T, N>) {
            self.get_mut().0.use_hook((initial_state,))
        }
    }
}

crate::utils::impl_hook! {
    impl ['a, T: 'a + PartialEq, const N: usize] for StateEqWith<'a, T, N> {
        #[inline]
        poll_next_update(mut self, cx) {
            self.0.poll_next_update_if_not_equal(cx, PartialEq::eq)
        }

        #[inline]
        use_hook[F: FnOnce() -> T](self, get_initial_state: F) -> (&'hook mut T, &'hook StateUpdater<'a, T, N>) {
            ::core::pin::Pin::new(&mut self.get_mut().0).use_hook_with(get_initial_state)
        }
    }
}