use color_eyre::Result;
use crossterm::event::{self, Event as CrosstermEvent, KeyEvent};
use std::sync::mpsc;
use std::thread;
use std::time::{Duration, Instant};
#[derive(Debug)]
pub enum Event {
Key(KeyEvent),
Mouse(event::MouseEvent),
Resize(u16, u16),
Tick,
}
pub struct EventHandler {
receiver: mpsc::Receiver<Event>,
#[allow(dead_code)]
handler: thread::JoinHandle<()>,
}
impl EventHandler {
#[must_use]
pub fn new(tick_rate_ms: u64) -> Self {
let tick_rate = Duration::from_millis(tick_rate_ms);
let (sender, receiver) = mpsc::channel();
let handler = thread::spawn(move || {
let mut last_tick = Instant::now();
loop {
let timeout = tick_rate
.checked_sub(last_tick.elapsed())
.unwrap_or(Duration::ZERO);
if event::poll(timeout).unwrap_or(false) {
if let Ok(evt) = event::read() {
match evt {
CrosstermEvent::Key(key) => {
if sender.send(Event::Key(key)).is_err() {
return;
}
}
CrosstermEvent::Resize(w, h) => {
if sender.send(Event::Resize(w, h)).is_err() {
return;
}
}
CrosstermEvent::Mouse(mouse) => {
if sender.send(Event::Mouse(mouse)).is_err() {
return;
}
}
_ => {}
}
}
}
if last_tick.elapsed() >= tick_rate {
if sender.send(Event::Tick).is_err() {
return;
}
last_tick = Instant::now();
}
}
});
Self { receiver, handler }
}
pub fn next(&self) -> Result<Event> {
Ok(self.receiver.recv()?)
}
pub fn try_next(&self) -> Result<Option<Event>> {
match self.receiver.try_recv() {
Ok(event) => Ok(Some(event)),
Err(mpsc::TryRecvError::Empty) => Ok(None),
Err(mpsc::TryRecvError::Disconnected) => {
Err(color_eyre::eyre::eyre!("Event channel disconnected"))
}
}
}
}