copilot-powerline 0.3.1

A fast, customizable powerline status line for GitHub Copilot CLI
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

pub struct Palette {
    pub reset: &'static str,
    pub dim: &'static str,
    pub label: &'static str,
    pub sep: &'static str,
    pub spend: &'static str,
    pub tokens_normal: &'static str,
    pub tokens_alert: &'static str,
    #[allow(dead_code)]
    pub badge: &'static str,
    pub pr: &'static str,
}

impl Palette {
    pub fn for_theme(theme_name: &str) -> Self {
        match theme_name.to_lowercase().as_str() {
            "github" | "default" => Self {
                reset: "\x1b[0m",
                dim: "\x1b[90m",
                label: "\x1b[90m",
                sep: "\x1b[90m  │  \x1b[0m",
                spend: "\x1b[92m", // Bright green
                tokens_normal: "\x1b[97m", // Bright white
                tokens_alert: "\x1b[1;31m", // Bold red
                badge: "\x1b[93m", // Bright yellow
                pr: "\x1b[33m",    // Yellow
            },
            "nord" => Self {
                reset: "\x1b[0m",
                dim: "\x1b[90m",
                label: "\x1b[90m",
                sep: "\x1b[90m  │  \x1b[0m",
                spend: "\x1b[36m", // Cyan (Frost)
                tokens_normal: "\x1b[97m",
                tokens_alert: "\x1b[1;31m",
                badge: "\x1b[93m",
                pr: "\x1b[33m", // Yellow
            },
            "tokyo-night" => Self {
                reset: "\x1b[0m",
                dim: "\x1b[90m",
                label: "\x1b[90m",
                sep: "\x1b[90m  │  \x1b[0m",
                spend: "\x1b[95m", // Bright magenta / violet
                tokens_normal: "\x1b[97m",
                tokens_alert: "\x1b[1;31m",
                badge: "\x1b[93m",
                pr: "\x1b[93m", // Bright yellow
            },
            "plain" => Self {
                reset: "",
                dim: "",
                label: "",
                sep: "  |  ",
                spend: "",
                tokens_normal: "",
                tokens_alert: "",
                badge: "",
                pr: "",
            },
            _ /* "colorblind" or any other */ => Self {
                reset: "\x1b[0m",
                dim: "\x1b[90m",
                label: "\x1b[90m",
                sep: "\x1b[90m  │  \x1b[0m",
                spend: "\x1b[94m", // Bright blue
                tokens_normal: "\x1b[97m", // Bright white
                tokens_alert: "\x1b[1;31m", // Bold red
                badge: "\x1b[93m", // Bright yellow
                pr: "\x1b[93m",    // Bright yellow
            },
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_theme_selection() {
        let cb = Palette::for_theme("colorblind");
        assert_eq!(cb.spend, "\x1b[94m");

        let gh = Palette::for_theme("github");
        assert_eq!(gh.spend, "\x1b[92m");

        let plain = Palette::for_theme("plain");
        assert_eq!(plain.reset, "");
        assert_eq!(plain.sep, "  |  ");
        assert_eq!(plain.pr, "");

        assert_eq!(cb.pr, "\x1b[93m");
    }
}