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}