predicates/
color.rs
1#[derive(Copy, Clone, Debug, Default)]
2pub(crate) struct Palette {
3 description: anstyle::Style,
4 var: anstyle::Style,
5 expected: anstyle::Style,
6}
7
8impl Palette {
9 pub(crate) fn new(alternate: bool) -> Self {
10 if alternate && cfg!(feature = "color") {
11 Self {
12 description: anstyle::AnsiColor::Blue.on_default() | anstyle::Effects::BOLD,
13 var: anstyle::AnsiColor::Red.on_default() | anstyle::Effects::BOLD,
14 expected: anstyle::AnsiColor::Green.on_default() | anstyle::Effects::BOLD,
15 }
16 } else {
17 Self::plain()
18 }
19 }
20
21 pub(crate) fn plain() -> Self {
22 Self {
23 description: Default::default(),
24 var: Default::default(),
25 expected: Default::default(),
26 }
27 }
28
29 pub(crate) fn description<D: std::fmt::Display>(self, display: D) -> Styled<D> {
30 Styled::new(display, self.description)
31 }
32
33 pub(crate) fn var<D: std::fmt::Display>(self, display: D) -> Styled<D> {
34 Styled::new(display, self.var)
35 }
36
37 pub(crate) fn expected<D: std::fmt::Display>(self, display: D) -> Styled<D> {
38 Styled::new(display, self.expected)
39 }
40}
41
42#[derive(Debug)]
43pub(crate) struct Styled<D> {
44 display: D,
45 style: anstyle::Style,
46}
47
48impl<D: std::fmt::Display> Styled<D> {
49 pub(crate) fn new(display: D, style: anstyle::Style) -> Self {
50 Self { display, style }
51 }
52}
53
54impl<D: std::fmt::Display> std::fmt::Display for Styled<D> {
55 #[inline]
56 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
57 if f.alternate() {
58 write!(f, "{}", self.style.render())?;
59 self.display.fmt(f)?;
60 write!(f, "{}", self.style.render_reset())?;
61 Ok(())
62 } else {
63 self.display.fmt(f)
64 }
65 }
66}