1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
use libc::{termios, winsize};
use std::error::Error;
use std::io::{Read, Stdin};
pub fn size() -> Result<winsize, Box<dyn Error>> {
unsafe {
let mut window: winsize = std::mem::zeroed();
let status = libc::ioctl(1, libc::TIOCGWINSZ, &mut window);
if status < 0 {
return Err(Box::from(format!("terminal size: ioctl failed with code {}", status)));
}
return Ok(window);
}
}
fn get_attr() -> Result<termios, Box<dyn Error>> {
unsafe {
let mut attr: termios = std::mem::zeroed();
let status = libc::tcgetattr(1, &mut attr);
if status < 0 {
return Err(Box::from(format!("tcgetattr failed with code {}", status)));
}
return Ok(attr);
}
}
fn set_attr(attrs: &termios) -> Result<(), Box<dyn Error>> {
unsafe {
let status = libc::tcsetattr(1, libc::TCSANOW, attrs);
if status < 0 {
return Err(Box::from(format!("set attributes: code {}", status)));
}
return Ok(());
}
}
fn make_raw() -> termios {
unsafe {
let mut attr: termios = std::mem::zeroed();
libc::cfmakeraw(&mut attr);
return attr;
}
}
pub struct RawHandle {
old: termios,
}
impl Drop for RawHandle {
fn drop(&mut self) {
let _ = set_attr(&self.old);
}
}
pub fn raw() -> Result<RawHandle, Box<dyn Error>> {
let old = get_attr()?;
set_attr(&make_raw())?;
return Ok(RawHandle { old: old });
}
pub enum Key {
CtrlQ,
CtrlS,
Up,
Down,
Left,
Right,
Char(u8),
}
pub fn read_key(stdin: &mut Stdin) -> Result<Key, Box<dyn Error>> {
const CTRL_Q: u8 = 17;
const CTRL_S: u8 = 19;
const ESC: u8 = 27;
loop {
let mut buf = vec![0; 1];
stdin.read_exact(&mut buf)?;
if buf[0] == CTRL_Q {
return Ok(Key::CtrlQ);
}
if buf[0] == CTRL_S {
return Ok(Key::CtrlS);
}
if buf[0] != ESC {
return Ok(Key::Char(buf[0]));
}
// Try to handle an escape sequence.
let mut buf = vec![0; 2];
stdin.read_exact(&mut buf)?;
if buf[0] != b'[' {
// Unknown escape sequence, just read another key.
continue;
}
match buf[1] {
b'A' => {
return Ok(Key::Up);
}
b'B' => {
return Ok(Key::Down);
}
b'C' => {
return Ok(Key::Right);
}
b'D' => {
return Ok(Key::Left);
}
_ => {
// Unknown key, just ignore it.
continue;
}
}
}
}
|
