claude-usage-tray 1.0.0

A simple tray application for Linux to show Claude's usage
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

use std::sync::LazyLock;

use image::GenericImageView;
use tracing::warn;

use crate::state::UsageStatus;

struct IconAsset {
    png: Vec<u8>,
    pixmap: ksni::Icon,
}

static LOADING: LazyLock<Option<IconAsset>> =
    LazyLock::new(|| load(include_bytes!("../assets/loading-icon.png")));
static DEFAULT: LazyLock<Option<IconAsset>> =
    LazyLock::new(|| load(include_bytes!("../assets/default-icon.png")));
static LOGIN: LazyLock<Option<IconAsset>> =
    LazyLock::new(|| load(include_bytes!("../assets/login-icon.png")));
static ERROR: LazyLock<Option<IconAsset>> =
    LazyLock::new(|| load(include_bytes!("../assets/error-icon.png")));

fn asset(status: &UsageStatus) -> Option<&'static IconAsset> {
    match status {
        UsageStatus::Loading => LOADING.as_ref(),
        UsageStatus::Ok => DEFAULT.as_ref(),
        UsageStatus::NeedsLogin => LOGIN.as_ref(),
        UsageStatus::Error(_) => ERROR.as_ref(),
    }
}

pub fn theme_name(status: &UsageStatus) -> &'static str {
    match status {
        UsageStatus::Loading => "image-loading",
        UsageStatus::Ok => "utilities-system-monitor",
        UsageStatus::NeedsLogin => "dialog-password",
        UsageStatus::Error(_) => "dialog-error",
    }
}

/// Returns an empty name when a custom pixmap is available, causing tray hosts
/// to prefer the supplied image. Otherwise, falls back to the system icon theme.
pub fn icon_name(status: &UsageStatus) -> &'static str {
    if asset(status).is_some() {
        ""
    } else {
        theme_name(status)
    }
}

pub fn pixmap(status: &UsageStatus) -> Vec<ksni::Icon> {
    asset(status)
        .map(|asset| vec![asset.pixmap.clone()])
        .unwrap_or_default()
}

pub fn png(status: &UsageStatus) -> Vec<u8> {
    asset(status)
        .map(|asset| asset.png.clone())
        .unwrap_or_default()
}

fn load(png: &'static [u8]) -> Option<IconAsset> {
    let image = match image::load_from_memory_with_format(png, image::ImageFormat::Png) {
        Ok(image) => image,
        Err(error) => {
            warn!(%error, "invalid custom icon; using theme icon");
            return None;
        }
    };

    let (width, height) = image.dimensions();
    let mut data = image.into_rgba8().into_raw();

    // StatusNotifierItem pixmaps use ARGB32 in network byte order; image gives RGBA.
    for pixel in data.chunks_exact_mut(4) {
        pixel.rotate_right(1);
    }

    Some(IconAsset {
        png: png.to_vec(),
        pixmap: ksni::Icon {
            width: width as i32,
            height: height as i32,
            data,
        },
    })
}