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
use std::fmt;
use std::io;
pub struct Utf8Writer<W: io::Write> {
inner: W,
result: io::Result<()>,
}
impl<W: io::Write> Utf8Writer<W> {
pub fn new(inner: W) -> Utf8Writer<W> {
Utf8Writer {
inner: inner,
result: Ok(()),
}
}
pub fn into_inner(self) -> (W, io::Result<()>) {
let Utf8Writer { inner, result } = self;
(inner, result)
}
pub fn into_result(self) -> io::Result<()> {
self.result
}
}
impl<W: io::Write> fmt::Write for Utf8Writer<W> {
fn write_str(&mut self, s: &str) -> fmt::Result {
match io::Write::write_all(&mut self.inner, s.as_bytes()) {
Ok(()) => Ok(()),
Err(e) => {
self.result = Err(e);
Err(fmt::Error)
}
}
}
fn write_fmt(&mut self, args: fmt::Arguments) -> fmt::Result {
match io::Write::write_fmt(&mut self.inner, args) {
Ok(()) => Ok(()),
Err(e) => {
self.result = Err(e);
Err(fmt::Error)
}
}
}
}
pub fn escape(s: &str) -> String {
use std::fmt::Write;
let mut buf = String::new();
rt::Escaper { inner: &mut buf }.write_str(s).unwrap();
buf
}
#[doc(hidden)]
pub mod rt {
use std::fmt;
pub struct Escaper<'a, 'b: 'a> {
pub inner: &'a mut (fmt::Write + 'b),
}
impl<'a, 'b> fmt::Write for Escaper<'a, 'b> {
fn write_str(&mut self, s: &str) -> fmt::Result {
for c in s.chars() {
try!(match c {
'&' => self.inner.write_str("&"),
'<' => self.inner.write_str("<"),
'>' => self.inner.write_str(">"),
'"' => self.inner.write_str("""),
'\'' => self.inner.write_str("'"),
_ => write!(self.inner, "{}", c),
});
}
Ok(())
}
}
}