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
115
116
|
use libc::{termios, winsize};
use std::io::{Read, Stdin};
pub fn size() -> Option<winsize> {
unsafe {
let mut window: winsize = std::mem::zeroed();
if libc::ioctl(1, libc::TIOCGWINSZ, &mut window) < 0 {
return None;
}
return Some(window);
}
}
fn get_attr() -> Option<termios> {
unsafe {
let mut attr: termios = std::mem::zeroed();
if libc::tcgetattr(1, &mut attr) < 0 {
return None;
}
return Some(attr);
}
}
fn set_attr(attrs: &termios) -> Option<()> {
unsafe {
if libc::tcsetattr(1, libc::TCSANOW, attrs) < 0 {
return None;
}
return Some(());
}
}
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) {
set_attr(&self.old);
}
}
pub fn raw() -> Option<RawHandle> {
let old = match get_attr() {
Some(old) => old,
None => {
return None;
}
};
if let None = set_attr(&make_raw()) {
return None;
}
return Some(RawHandle { old: old });
}
pub enum Key {
CtrlQ,
Up,
Down,
Left,
Right,
Char(char),
}
pub fn read_key(stdin: &mut Stdin) -> Result<Key, String> {
const CTRL_Q: u8 = 17;
const ESC: u8 = 27;
loop {
let mut buf = vec![0; 1];
if let Err(err) = stdin.read_exact(&mut buf) {
return Err(format!("read input: {}", err));
}
if buf[0] == CTRL_Q {
return Ok(Key::CtrlQ);
}
if buf[0] != ESC {
return Ok(Key::Char(char::from(buf[0])));
}
// Try to handle an escape sequence.
let mut buf = vec![0; 2];
if let Err(err) = stdin.read_exact(&mut buf) {
return Err(format!("read input: {}", err));
}
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;
}
}
}
}
|
