78 lines
2.0 KiB
Rust
78 lines
2.0 KiB
Rust
// This file was copied from https://github.com/fdehau/tui-rs/blob/master/examples/util/event.rs
|
|
use std::io;
|
|
use std::sync::mpsc;
|
|
use std::sync::{
|
|
atomic::{AtomicBool, Ordering},
|
|
Arc,
|
|
};
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
use termion::event::Key;
|
|
use termion::input::TermRead;
|
|
|
|
pub enum Event<I> {
|
|
Input(I),
|
|
}
|
|
|
|
/// A small event handler that wrap termion input and tick events. Each event
|
|
/// type is handled in its own thread and returned to a common `Receiver`
|
|
pub struct Events {
|
|
rx: mpsc::Receiver<Event<Key>>,
|
|
input_handle: thread::JoinHandle<()>,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy)]
|
|
pub struct Config {
|
|
pub exit_key: Key,
|
|
pub tick_rate: Duration,
|
|
}
|
|
|
|
impl Default for Config {
|
|
fn default() -> Config {
|
|
Config {
|
|
exit_key: Key::Char('q'),
|
|
tick_rate: Duration::from_millis(250),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Events {
|
|
pub fn new() -> Events {
|
|
Events::with_config(Config::default())
|
|
}
|
|
|
|
pub fn with_config(config: Config) -> Events {
|
|
let (tx, rx) = mpsc::channel();
|
|
let ignore_exit_key = Arc::new(AtomicBool::new(false));
|
|
let input_handle = {
|
|
let tx = tx.clone();
|
|
let ignore_exit_key = ignore_exit_key.clone();
|
|
thread::spawn(move || {
|
|
let stdin = io::stdin();
|
|
for evt in stdin.keys() {
|
|
match evt {
|
|
Ok(key) => {
|
|
if let Err(_) = tx.send(Event::Input(key)) {
|
|
return;
|
|
}
|
|
if !ignore_exit_key.load(Ordering::Relaxed) && key == config.exit_key {
|
|
return;
|
|
}
|
|
}
|
|
Err(_) => {}
|
|
}
|
|
}
|
|
})
|
|
};
|
|
Events {
|
|
rx,
|
|
input_handle,
|
|
}
|
|
}
|
|
|
|
pub fn next(&self) -> Result<Event<Key>, mpsc::RecvError> {
|
|
self.rx.recv()
|
|
}
|
|
}
|