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
117
118
119
120
121
122
#[cfg(not(feature = "color"))]
use crate::util::termcolor::{Color, ColorChoice};
#[cfg(feature = "color")]
use termcolor::{Color, ColorChoice};
use std::{
fmt::{self, Display, Formatter},
io::{self, Write},
};
#[cfg(feature = "color")]
fn is_a_tty(stderr: bool) -> bool {
debug!("is_a_tty: stderr={:?}", stderr);
let stream = if stderr {
atty::Stream::Stderr
} else {
atty::Stream::Stdout
};
atty::is(stream)
}
#[derive(Debug)]
pub(crate) struct Colorizer {
use_stderr: bool,
color_when: ColorChoice,
pieces: Vec<(String, Option<Color>)>,
}
impl Colorizer {
#[inline]
pub(crate) fn new(use_stderr: bool, color_when: ColorChoice) -> Self {
Colorizer {
use_stderr,
color_when,
pieces: vec![],
}
}
#[inline]
pub(crate) fn good(&mut self, msg: impl Into<String>) {
self.pieces.push((msg.into(), Some(Color::Green)));
}
#[inline]
pub(crate) fn warning(&mut self, msg: impl Into<String>) {
self.pieces.push((msg.into(), Some(Color::Yellow)));
}
#[inline]
pub(crate) fn error(&mut self, msg: impl Into<String>) {
self.pieces.push((msg.into(), Some(Color::Red)));
}
#[inline]
pub(crate) fn none(&mut self, msg: impl Into<String>) {
self.pieces.push((msg.into(), None));
}
}
impl Colorizer {
#[cfg(feature = "color")]
pub(crate) fn print(&self) -> io::Result<()> {
use termcolor::{BufferWriter, ColorSpec, WriteColor};
let color_when = if is_a_tty(self.use_stderr) {
self.color_when
} else {
ColorChoice::Never
};
let writer = if self.use_stderr {
BufferWriter::stderr(color_when)
} else {
BufferWriter::stdout(color_when)
};
let mut buffer = writer.buffer();
for piece in &self.pieces {
let mut color = ColorSpec::new();
color.set_fg(piece.1);
if piece.1 == Some(Color::Red) {
color.set_bold(true);
}
buffer.set_color(&color)?;
buffer.write_all(piece.0.as_bytes())?;
buffer.reset()?;
}
writer.print(&buffer)
}
#[cfg(not(feature = "color"))]
pub(crate) fn print(&self) -> io::Result<()> {
if self.use_stderr {
let stderr = std::io::stderr();
let mut stderr = stderr.lock();
write!(stderr, "{}", self)
} else {
let stdout = std::io::stdout();
let mut stdout = stdout.lock();
write!(stdout, "{}", self)
}
}
}
impl Display for Colorizer {
fn fmt(&self, f: &mut Formatter) -> fmt::Result {
for piece in &self.pieces {
Display::fmt(&piece.0, f)?;
}
Ok(())
}
}