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

#[cfg(not(windows))]
extern crate termios;
#[cfg(not(windows))]
use std::io::Read;
#[cfg(windows)]
pub struct Getch{}

#[cfg(not(windows))]
pub struct Getch {
    termios: termios::Termios
}

#[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() -> Result<Getch, std::io::Error> {
        Ok(Getch{})
    }
    #[cfg(not(windows))]
    pub fn new() -> Result<Getch, std::io::Error> {
        let mut termios=try!(termios::Termios::from_fd(0));
        let c_lflag = termios.c_lflag;
        termios.c_lflag &= !(ICANON|ECHO);

        try!(tcsetattr(0, termios::TCSADRAIN, &termios));

        termios.c_lflag = c_lflag;
        Ok(Getch { termios: termios })
    }

    #[cfg(windows)]
    pub fn getch(&self) -> Result<u8, std::io::Error> {
        loop {
            unsafe {
                let k= _getch();
                if k==0 {
                    // Ignore next input.
                    _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 {
                    // TODO: accept utf-8
                    return Ok(r[0])
                }
            }
        }
    }
}

impl Drop for Getch {
    #[cfg(not(windows))]
    fn drop(&mut self) {
        tcsetattr(0,termios::TCSADRAIN, &self.termios).unwrap_or(())
    }
    #[cfg(windows)]
    fn drop(&mut self) {
    }
}