hush 0.1.4

Hush is a unix shell scripting language based on the Lua programming language
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
123

use std::{
	io,
	fmt::{self, Debug, Display},
};

pub use termion::color::{Black, Blue, Green, Red, Yellow};


thread_local! {
	static IS_TTY: bool = termion::is_tty(&io::stdout())
		&& termion::is_tty(&io::stderr());
}


macro_rules! tty_fmt {
	($f: expr, $open: expr, $value: expr, $close: expr) => {
		IS_TTY.with(
			|&is_tty| if is_tty {
				write!($f, "{}", $open)?;
				$value.fmt($f)?;
				write!($f, "{}", $close)
			} else {
				$value.fmt($f)
			}
		)
	}
}


/// Paint the foreground with a given color when formatting the value.
pub struct Fg<C, T>(pub C, pub T);


impl<C, T> Debug for Fg<C, T>
where
	C: termion::color::Color + Copy,
	T: Debug,
{
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		tty_fmt!(
			f,
			termion::color::Fg(self.0),
			self.1,
			termion::color::Fg(termion::color::Reset)
		)
	}
}


impl<C, T> Display for Fg<C, T>
where
	C: termion::color::Color + Copy,
	T: Display,
{
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		tty_fmt!(
			f,
			termion::color::Fg(self.0),
			self.1,
			termion::color::Fg(termion::color::Reset)
		)
	}
}


/// Paint with a given style.
pub struct Style<S, T>(pub S, pub T);


impl<S, T> Debug for Style<S, T>
where
	S: Display,
	T: Debug,
{
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		tty_fmt!(
			f,
			self.0,
			self.1,
			termion::style::Reset
		)
	}
}


impl<S, T> Display for Style<S, T>
where
	S: Display,
	T: Display,
{
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		tty_fmt!(
			f,
			self.0,
			self.1,
			termion::style::Reset
		)
	}
}


/// Paint with a bold style.
pub struct Bold<T>(pub T);


impl<T> Debug for Bold<T>
where
	T: Debug,
{
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		Style(termion::style::Bold, &self.0).fmt(f)
	}
}


impl<T> Display for Bold<T>
where
	T: Display,
{
	fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
		Style(termion::style::Bold, &self.0).fmt(f)
	}
}