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

use std::{future::Future, pin::Pin, task::Poll};

use crate::HookPollNextUpdate;

/// A future which polls [`HookPollNextUpdate::poll_next_update`].
/// See [`HookExt::next_update`](crate::HookPollNextUpdateExt::next_update).
#[must_use = "futures do nothing unless you `.await` or poll them"]
pub struct NextUpdate<'a, H: ?Sized> {
    hook: &'a mut H,
}

impl<'a, H: ?Sized> NextUpdate<'a, H> {
    pub fn new(hook: &'a mut H) -> Self {
        Self { hook }
    }
}

impl<'a, H: HookPollNextUpdate + Unpin + ?Sized> Future for NextUpdate<'a, H> {
    type Output = bool;

    #[inline]
    fn poll(self: Pin<&mut Self>, cx: &mut std::task::Context<'_>) -> Poll<Self::Output> {
        let hook = &mut self.get_mut().hook;
        Pin::new(hook).poll_next_update(cx)
    }
}