94 lines
2.2 KiB
Rust
94 lines
2.2 KiB
Rust
|
// Copyright (c) 2017-2019, scoobybejesus
|
||
|
// Redistributions must include the license: https://github.com/scoobybejesus/cryptools/blob/master/LEGAL.txt
|
||
|
|
||
|
// TODO: cite source?
|
||
|
|
||
|
use std::io;
|
||
|
use std::sync::mpsc;
|
||
|
use std::thread;
|
||
|
use std::time::Duration;
|
||
|
|
||
|
|
||
|
use termion::input::TermRead;
|
||
|
use termion::event::Key;
|
||
|
|
||
|
|
||
|
#[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),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub enum Event<I> {
|
||
|
Input(I),
|
||
|
Tick,
|
||
|
}
|
||
|
|
||
|
/// A small event handler that wraps 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<()>,
|
||
|
tick_handle: thread::JoinHandle<()>,
|
||
|
}
|
||
|
|
||
|
impl Events {
|
||
|
|
||
|
pub fn new() -> Events {
|
||
|
Events::with_config(Config::default())
|
||
|
}
|
||
|
|
||
|
pub fn with_config(config: Config) -> Events {
|
||
|
|
||
|
let (tx, rx) = mpsc::channel();
|
||
|
let input_handle = {
|
||
|
let tx = tx.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 key == config.exit_key {
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
Err(_) => {}
|
||
|
}
|
||
|
}
|
||
|
})
|
||
|
};
|
||
|
|
||
|
let tick_handle = {
|
||
|
let tx = tx.clone();
|
||
|
thread::spawn(move || {
|
||
|
let tx = tx.clone();
|
||
|
loop {
|
||
|
tx.send(Event::Tick).unwrap();
|
||
|
thread::sleep(config.tick_rate);
|
||
|
}
|
||
|
})
|
||
|
};
|
||
|
|
||
|
Events {
|
||
|
rx,
|
||
|
input_handle,
|
||
|
tick_handle,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn next(&self) -> Result<Event<Key>, mpsc::RecvError> {
|
||
|
self.rx.recv()
|
||
|
}
|
||
|
}
|