parent
99759854e2
commit
35c1818320
@ -0,0 +1,136 @@ |
|||||||
|
use std::io; |
||||||
|
use std::io::{Read, stdin, stdout, Write}; |
||||||
|
use std::os::unix::io::RawFd; |
||||||
|
use std::thread::JoinHandle; |
||||||
|
use std::time::Duration; |
||||||
|
use termios::{tcgetattr, tcsetattr, Termios}; |
||||||
|
use timeout_readwrite::TimeoutReader; |
||||||
|
use crate::{InputEvent, TerminalInterface}; |
||||||
|
|
||||||
|
const STDIN_FILENO : RawFd = 0; |
||||||
|
|
||||||
|
struct StdinNonblockingGuard { |
||||||
|
original_termios : Termios, |
||||||
|
} |
||||||
|
|
||||||
|
impl StdinNonblockingGuard { |
||||||
|
pub fn new() -> std::io::Result<Self> { |
||||||
|
let mut old_tio = unsafe { std::mem::zeroed::<Termios>() }; |
||||||
|
|
||||||
|
/* get the terminal settings for stdin */ |
||||||
|
tcgetattr(STDIN_FILENO, &mut old_tio)?; |
||||||
|
|
||||||
|
/* we want to keep the old setting to restore them a the end */ |
||||||
|
let mut new_tio = old_tio; |
||||||
|
|
||||||
|
/* disable canonical mode (buffered i/o) and local echo */ |
||||||
|
new_tio.c_lflag &= !termios::ICANON & !termios::ECHO; |
||||||
|
|
||||||
|
/* set the new settings immediately */ |
||||||
|
tcsetattr(STDIN_FILENO, termios::TCSANOW, &new_tio)?; |
||||||
|
|
||||||
|
Ok(Self { |
||||||
|
original_termios: old_tio, |
||||||
|
}) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl Drop for StdinNonblockingGuard { |
||||||
|
fn drop(&mut self) { |
||||||
|
/* restore the former settings */ |
||||||
|
let _ = tcsetattr(STDIN_FILENO, termios::TCSANOW, &self.original_termios); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
#[derive(SmartDefault)] |
||||||
|
pub struct StdinInterfaceOptions { |
||||||
|
#[default(Duration::from_millis(250))] |
||||||
|
poll_interval: Duration, |
||||||
|
} |
||||||
|
|
||||||
|
pub struct StdinInterface { |
||||||
|
receiver : Option<crossbeam_channel::Receiver<InputEvent>>, |
||||||
|
shutdown_sender : crossbeam_channel::Sender<()>, |
||||||
|
recv_thread_handle: Option<JoinHandle<()>>, |
||||||
|
} |
||||||
|
|
||||||
|
impl StdinInterface { |
||||||
|
pub fn new(opts : StdinInterfaceOptions) -> Self { |
||||||
|
let (sender, receiver) = crossbeam_channel::bounded(0); |
||||||
|
let (shutdown_sender, shutdown_receiver) = crossbeam_channel::bounded(0); |
||||||
|
let handle = std::thread::spawn(move || { |
||||||
|
let stdin_guard = StdinNonblockingGuard::new().unwrap(); |
||||||
|
let mut sin = TimeoutReader::new(stdin().lock(), opts.poll_interval); |
||||||
|
let mut parser = anes::parser::Parser::default(); |
||||||
|
|
||||||
|
'input: loop { |
||||||
|
match shutdown_receiver.try_recv() { |
||||||
|
Ok(_) | Err(crossbeam_channel::TryRecvError::Disconnected) => { |
||||||
|
break 'input; |
||||||
|
} |
||||||
|
Err(crossbeam_channel::TryRecvError::Empty) => { |
||||||
|
//
|
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
let mut buf = [0u8; 64]; |
||||||
|
// TODO add a way to kill the thread from outside
|
||||||
|
let numch = match sin.read(&mut buf[..]) { |
||||||
|
Err(e) => { |
||||||
|
if e.kind() != io::ErrorKind::TimedOut { |
||||||
|
let _ = sender.send(InputEvent::EndOfStream); |
||||||
|
break 'input; |
||||||
|
} |
||||||
|
continue 'input; |
||||||
|
} |
||||||
|
Ok(0) => { |
||||||
|
let _ = sender.send(InputEvent::EndOfStream); |
||||||
|
break 'input; |
||||||
|
}, |
||||||
|
Ok(numch) => numch |
||||||
|
}; |
||||||
|
|
||||||
|
let received_slice = &buf[0..numch]; |
||||||
|
parser.advance(received_slice, numch == buf.len()); |
||||||
|
|
||||||
|
while let Some(item) = parser.next() { |
||||||
|
if sender.send(InputEvent::Key(item)).is_err() { |
||||||
|
break 'input; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
drop(stdin_guard); |
||||||
|
}); |
||||||
|
|
||||||
|
Self { |
||||||
|
receiver: Some(receiver), |
||||||
|
shutdown_sender, |
||||||
|
recv_thread_handle: Some(handle), |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl Write for StdinInterface { |
||||||
|
fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> { |
||||||
|
stdout().write(buf) |
||||||
|
} |
||||||
|
|
||||||
|
fn flush(&mut self) -> std::io::Result<()> { |
||||||
|
stdout().flush() |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl TerminalInterface for StdinInterface { |
||||||
|
fn get_receiver(&mut self) -> Option<crossbeam_channel::Receiver<InputEvent>> { |
||||||
|
self.receiver.take() |
||||||
|
} |
||||||
|
|
||||||
|
fn shutdown(&mut self) { |
||||||
|
if self.shutdown_sender.send(()).is_ok() { |
||||||
|
if let Some(handle) = self.recv_thread_handle.take() { |
||||||
|
let _ = handle.join(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue