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
use super::*;
use std::{
io,
ops::{Deref, DerefMut},
os::unix::io::AsRawFd
};
#[derive(Clone, Copy)]
pub struct Termios(sys::Termios);
impl Termios {
pub fn new(inner: sys::Termios) -> Self {
Termios(inner)
}
pub fn inner(&self) -> &sys::Termios {
&self.0
}
pub fn inner_mut(&mut self) -> &mut sys::Termios {
&mut self.0
}
pub fn make_raw(mut self) -> Self {
sys::make_raw(&mut self.0);
self
}
}
pub struct TermiosSetter(sys::TermiosSetter);
impl TermiosSetter {
pub fn new<F: AsRawFd>(fd: &F) -> io::Result<Self> {
sys::TermiosSetter::new(fd.as_raw_fd()).map(TermiosSetter)
}
pub fn get(&mut self) -> io::Result<Termios> {
self.0.get().map(Termios)
}
pub fn set(&mut self, termios: &Termios) -> io::Result<()> {
self.0.set(termios.inner())
}
}
pub struct RawTerminal<F: AsRawFd> {
file: F,
restore: Option<(TermiosSetter, Termios)>
}
impl<F: AsRawFd> RawTerminal<F> {
pub fn new(file: F) -> io::Result<Self> {
let mut setter = TermiosSetter::new(&file)?;
let prev = setter.get()?;
setter.set(&prev.make_raw())?;
Ok(Self {
file,
restore: Some((setter, prev))
})
}
pub fn new_allow_failure(file: F) -> Self {
let mut restore = None;
if let Ok(mut setter) = TermiosSetter::new(&file) {
if let Ok(prev) = setter.get() {
if setter.set(&prev.make_raw()).is_ok() {
restore = Some((setter, prev));
}
}
}
Self { file, restore }
}
}
impl<F: AsRawFd> Deref for RawTerminal<F> {
type Target = F;
fn deref(&self) -> &Self::Target {
&self.file
}
}
impl<F: AsRawFd> DerefMut for RawTerminal<F> {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.file
}
}
impl<F: AsRawFd> Drop for RawTerminal<F> {
fn drop(&mut self) {
if let Some((ref mut setter, ref prev)) = self.restore {
let _ = setter.set(prev);
}
}
}