use {
crate::{
errors::Error,
events::Event,
},
crossbeam::channel::{unbounded, Receiver, Sender},
crossterm::{
self,
terminal,
},
std::{
sync::{
atomic::{AtomicUsize, Ordering},
Arc,
},
thread,
time::{Duration, Instant},
}
};
const DOUBLE_CLICK_MAX_DURATION: Duration = Duration::from_millis(700);
struct TimedEvent {
time: Instant,
event: Event,
}
impl From<Event> for TimedEvent {
fn from(event: Event) -> Self {
TimedEvent {
time: Instant::now(),
event,
}
}
}
pub struct EventSource {
rx_events: Receiver<Event>,
tx_quit: Sender<bool>,
event_count: Arc<AtomicUsize>,
}
impl EventSource {
pub fn new() -> Result<EventSource, Error> {
let (tx_events, rx_events) = unbounded();
let (tx_quit, rx_quit) = unbounded();
let event_count = Arc::new(AtomicUsize::new(0));
let internal_event_count = Arc::clone(&event_count);
terminal::enable_raw_mode()?;
let mut last_event: Option<TimedEvent> = None;
thread::spawn(move || {
loop {
if let Some(mut event) = Event::from_crossterm_event(crossterm::event::read()) {
if let Event::Click(x, y, ..) = event {
if let Some(TimedEvent {
time,
event: Event::Click(last_x, last_y, ..),
}) = last_event
{
if
last_x == x && last_y == y
&& time.elapsed() < DOUBLE_CLICK_MAX_DURATION
{
event = Event::DoubleClick(x, y);
}
}
}
last_event = Some(TimedEvent::from(event));
internal_event_count.fetch_add(1, Ordering::SeqCst);
if let Err(_) = tx_events.send(event) {
return; }
match rx_quit.recv() {
Ok(false) => {},
_ => {
return;
}
}
}
}
});
Ok(EventSource {
rx_events,
tx_quit,
event_count,
})
}
pub fn unblock(&self, quit: bool) {
self.tx_quit.send(quit).unwrap();
}
pub fn shared_event_count(&self) -> Arc<AtomicUsize> {
Arc::clone(&self.event_count)
}
pub fn receiver(&self) -> Receiver<Event> {
self.rx_events.clone()
}
}
impl Drop for EventSource {
fn drop(&mut self) {
terminal::disable_raw_mode().unwrap();
}
}