punfetch 0.3.6

A blazingly fast system fetch program to pair with onefetch
Documentation 
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

use std::fmt;
use num_enum::FromPrimitive;
use owo_colors::{
    AnsiColors,
    DynColors::{self, Ansi, Rgb},
};

struct Colors {
    basic_colors: Vec<DynColors>,
    true_colors: Option<Vec<DynColors>>,
}

/// The big list of distros! Provides ascii art and colors for each, and a way to search for a match
#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, FromPrimitive)]
#[allow(clippy::upper_case_acronyms)]
#[repr(u64)]
pub enum Distro {
    #[num_enum(default)]
    {% for distro, attrs in distros -%}
    {{ attrs.type_name }},
    {% endfor %}
}

impl Distro {
    fn colors(&self, color_mode: Option<bool>) -> Vec<DynColors> {
        let colors = match self {
            {% for distro, attrs in distros -%}
                Distro::{{ attrs.type_name }} => Colors {
                    basic_colors: vec![{%- for color in attrs.colors.ansi -%}Ansi(AnsiColors::{{ color | pascal_case | replace(from="White", to="Default") }}),{% endfor %}],
                    true_colors: {% if attrs.colors.hex -%}
                        Some(vec![
                            {%- for hex in attrs.colors.hex -%}
                                {% set rgb = hex | hex_to_rgb -%}
                                Rgb({{ rgb.r }}, {{ rgb.g }}, {{ rgb.b }}),
                            {% endfor %}
                        ]),
                    {% else -%}None,
                    {% endif %}
                },
            {% endfor %}
        };

        match color_mode {
            Some(true) => colors.true_colors.unwrap_or(colors.basic_colors),
            Some(false) => colors.basic_colors,
            None => vec![],
        }
    }

    fn regex() -> regex::RegexSet {
        regex::RegexSet::new([
            {% for distro, attrs in distros -%}
            r#"{{ attrs.regex }}"#,
            {% endfor %}
        ]).unwrap()
    }

    fn template(&self) -> &'static str {
        match self {
            {% for distro, attrs in distros -%}
                Distro::{{ attrs.type_name }} => {
"{{ attrs.ascii | addslashes }}"
                },
            {% endfor %}
        }
    }
}

impl fmt::Display for Distro {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            {% for distro, attrs in distros -%}
                Self::{{ attrs.type_name }} => write!(f, "{{ distro }}"),
            {% endfor %}
        }
    }
}

// Ensure there are equal number of basic and true colors
{% for distro, attrs in distros -%}
    {% if attrs.colors.rgb %}
        {% set ansi_length = attrs.colors.ansi | length -%}
        {% set rgb_length = attrs.colors.rgb | length %}
        {% if ansi_length != rgb_length %}
            compile_error!("{{ distro }}: ansi and rgb colors must be the same length");
        {% endif %}
    {% endif -%}
{% endfor -%}