auraline 0.6.5

utility that renders information into a string suitable to be embedded in the shell prompt.
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

use std::fmt::Display;

use owo_colors::style;
use owo_colors::Style;
use owo_colors::Styled;

pub struct Chunk<T> {
    icon: Option<Styled<&'static str>>,
    info: Option<Styled<T>>,
}

#[derive(Default, Debug)]
pub struct Unit;
impl Display for Unit {
    fn fmt(&self, _: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        Ok(())
    }
}

pub struct Adjoin<T, V>(pub (T, V));
impl<T, V> Display for Adjoin<T, V>
where
    T: Display,
    V: Display,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}{}", self.0 .0, self.0 .1)
    }
}

impl<T, V> Default for Adjoin<T, V>
where
    T: Default,
    V: Default,
{
    fn default() -> Self {
        Self((T::default(), V::default()))
    }
}

impl<T: Default> Chunk<T> {
    pub fn new(icon: &'static str, info: T) -> Self {
        Self {
            icon: Some(style().style(icon)),
            info: Some(style().style(info)),
        }
    }

    pub fn icon(icon: &'static str) -> Self {
        Self {
            icon: Some(style().style(icon)),
            info: None,
        }
    }

    pub fn info(info: T) -> Self {
        Self {
            icon: None,
            info: Some(style().style(info)),
        }
    }

    pub fn with_style(mut self, icon_s: Style, info_s: Style) -> Self {
        let icon = self.icon.as_mut().map(|i| std::mem::take(i.inner_mut()));
        let info = self.info.as_mut().map(|i| std::mem::take(i.inner_mut()));
        Self {
            icon: icon.map(|i| icon_s.style(i)),
            info: info.map(|i| info_s.style(i)),
        }
    }
}

impl<T: Display> Display for Chunk<T> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match (&self.icon, &self.info) {
            (Some(icon), Some(info)) => write!(f, "{icon} {info}"),
            (Some(icon), None) => write!(f, "{icon}"),
            (None, Some(info)) => write!(f, "{info}"),
            (None, None) => Ok(()),
        }
    }
}