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
#[cfg(not(windows))]
extern crate termios;
#[cfg(not(windows))]
use std::io::Read;
#[cfg(windows)]
pub struct Getch{}
#[cfg(not(windows))]
pub enum Getch {
Termios(termios::Termios),
None
}
#[cfg(windows)]
extern crate libc;
#[cfg(windows)]
use libc::c_int;
#[cfg(windows)]
extern "C" {
fn _getch()->c_int;
}
#[cfg(not(windows))]
use termios::{tcsetattr,ICANON,ECHO};
impl Getch {
#[cfg(windows)]
pub fn new() -> Getch {
Getch{}
}
#[cfg(not(windows))]
pub fn new() -> Getch {
if let Ok(mut termios) = termios::Termios::from_fd(0) {
let c_lflag = termios.c_lflag;
termios.c_lflag &= !(ICANON|ECHO);
if let Ok(()) = tcsetattr(0, termios::TCSADRAIN, &termios) {
termios.c_lflag = c_lflag;
return Getch::Termios(termios)
}
}
Getch::None
}
#[cfg(windows)]
pub fn getch(&self) -> Result<u8, std::io::Error> {
loop {
unsafe {
let k= _getch();
if k==0 {
_getch();
} else {
return Ok(k as u8)
}
}
}
}
#[cfg(not(windows))]
pub fn getch(&self) -> Result<u8, std::io::Error> {
let mut r:[u8;1]=[0];
let mut stdin = std::io::stdin();
loop {
if try!(stdin.read(&mut r[..])) == 0 { return Ok(0) }
else {
if r[0]==27 {
if try!(stdin.read(&mut r[..])) == 0 { return Ok(0) }
else {
if r[0]==91 {
if try!(stdin.read(&mut r[..])) == 0 { return Ok(0) }
}
}
} else {
return Ok(r[0])
}
}
}
}
}
impl Drop for Getch {
#[cfg(not(windows))]
fn drop(&mut self) {
if let Getch::Termios(ref mut termios) = *self {
tcsetattr(0,termios::TCSADRAIN, &termios).unwrap_or(())
}
}
#[cfg(windows)]
fn drop(&mut self) {
}
}