copilot-powerline 0.3.2

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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use crate::config::IconSet;

pub fn tokens_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "Tokens:",
        IconSet::Nerd => "󰮚",
        IconSet::Emoji => "🪙",
    }
}

pub fn session_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "Session:",
        IconSet::Nerd => "󰄬",
        IconSet::Emoji => "💰",
    }
}

pub fn month_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "Month:",
        IconSet::Nerd => "󰠠",
        IconSet::Emoji => "📅",
    }
}

pub fn cache_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "Cache:",
        IconSet::Nerd => "󰘸",
        IconSet::Emoji => "⚡",
    }
}

pub fn reasoning_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "Think:",
        IconSet::Nerd => "󰚩",
        IconSet::Emoji => "🧠",
    }
}

pub fn total_tokens_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "Total:",
        IconSet::Nerd => "󰓅",
        IconSet::Emoji => "📊",
    }
}

pub fn pr_icon(icon_set: IconSet, custom: Option<&str>) -> &str {
    if let Some(c) = custom {
        return c;
    }
    match icon_set {
        IconSet::Plain => "PR",
        IconSet::Nerd => "",
        IconSet::Emoji => "🔀",
    }
}

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

    #[test]
    fn test_icon_set_plain() {
        assert_eq!(tokens_icon(IconSet::Plain, None), "Tokens:");
        assert_eq!(cache_icon(IconSet::Plain, None), "Cache:");
        assert_eq!(reasoning_icon(IconSet::Plain, None), "Think:");
        assert_eq!(total_tokens_icon(IconSet::Plain, None), "Total:");
        assert_eq!(pr_icon(IconSet::Plain, None), "PR");
    }

    #[test]
    fn test_icon_set_nerd() {
        assert_eq!(tokens_icon(IconSet::Nerd, None), "󰮚");
        assert_eq!(cache_icon(IconSet::Nerd, None), "󰘸");
        assert_eq!(reasoning_icon(IconSet::Nerd, None), "󰚩");
        assert_eq!(total_tokens_icon(IconSet::Nerd, None), "󰓅");
        assert_eq!(pr_icon(IconSet::Nerd, None), "");
    }

    #[test]
    fn test_icon_set_emoji() {
        assert_eq!(tokens_icon(IconSet::Emoji, None), "🪙");
        assert_eq!(session_icon(IconSet::Emoji, None), "💰");
        assert_eq!(month_icon(IconSet::Emoji, None), "📅");
        assert_eq!(cache_icon(IconSet::Emoji, None), "⚡");
        assert_eq!(reasoning_icon(IconSet::Emoji, None), "🧠");
        assert_eq!(total_tokens_icon(IconSet::Emoji, None), "📊");
        assert_eq!(pr_icon(IconSet::Emoji, None), "🔀");
    }

    #[test]
    fn test_custom_override() {
        assert_eq!(
            cache_icon(IconSet::Emoji, Some("CustomCache:")),
            "CustomCache:"
        );
        assert_eq!(pr_icon(IconSet::Plain, Some("Pull:")), "Pull:");
    }
}