gitui/src/poll.rs

use asyncgit::AsyncNotification;
use crossbeam_channel::{unbounded, Receiver};
use crossterm::event::{self, Event};
use std::time::{Duration, Instant};

///
#[derive(Clone, Copy)]
pub enum QueueEvent {
    Tick,
    SpinnerUpdate,
    GitEvent(AsyncNotification),
    InputEvent(Event),
}

static MAX_POLL_DURATION: Duration = Duration::from_secs(2);
static MIN_POLL_DURATION: Duration = Duration::from_millis(5);
static MAX_BATCHING_DURATION: Duration = Duration::from_millis(25);

///
pub fn start_polling_thread() -> Receiver<Vec<QueueEvent>> {
    let (tx, rx) = unbounded();

    rayon_core::spawn(move || {
        let mut last_send = Instant::now();
        let mut batch = Vec::new();

        loop {
            let timeout = if batch.is_empty() {
                MAX_POLL_DURATION
            } else {
                MIN_POLL_DURATION
            };
            if let Some(e) =
                poll(timeout).expect("failed to pull events.")
            {
                batch.push(QueueEvent::InputEvent(e));
            }

            if !batch.is_empty()
                && last_send.elapsed() > MAX_BATCHING_DURATION
            {
                tx.send(batch).expect("send input event failed");
                batch = Vec::new();
                last_send = Instant::now();
            }
        }
    });

    rx
}

///
fn poll(dur: Duration) -> anyhow::Result<Option<Event>> {
    if event::poll(dur)? {
        Ok(Some(event::read()?))
    } else {
        Ok(None)
    }
}