ai_usagebar/kiro/
vendor.rs

1//! Kiro CLI renderer — a single credit pool (used/limit + reset), closer in
2//! shape to `anthropic_api::vendor` (one headline %) than to Cursor's two
3//! pools, plus the reset countdown/tooltip layout `cursor::vendor` uses.
4
5use std::collections::HashMap;
6
7use chrono::{DateTime, Utc};
8
9use crate::countdown;
10use crate::format::{placeholders, substitute, updated_at_hm};
11use crate::pacing::PaceSeverity;
12use crate::pango::{color_span, escape, severity_color, severity_for};
13use crate::theme::Theme;
14use crate::tooltip::{Line as TooltipLine, render_bordered};
15use crate::usage::KiroSnapshot;
16use crate::vendor::{RenderOpts, VendorId, VendorOutcome};
17use crate::waybar::{Class, WaybarOutput};
18
19use super::fetch::FetchOutcome;
20
21pub const DEFAULT_FORMAT: &str = "{kiro_pct}%";
22
23/// No confirmed Kiro glyph in the common Nerd Font sets — a plain diamond,
24/// matching Cursor's "no dedicated icon" treatment. Override with `--icon`.
25const DEFAULT_ICON: &str = "◆";
26
27fn credits(v: f64) -> String {
28    // Two decimals only when the wire value actually carries a fraction
29    // (kiro-cli's own `/usage` shows "9943.38 of 10000", not "9943.00").
30    if (v.fract()).abs() < f64::EPSILON {
31        format!("{v:.0}")
32    } else {
33        format!("{v:.2}")
34    }
35}
36
37pub fn build_placeholders(
38    snap: &KiroSnapshot,
39    now: DateTime<Utc>,
40) -> HashMap<&'static str, String> {
41    let pct = snap.pct();
42    let reset = countdown::format(snap.reset_at, now);
43    placeholders(vec![
44        ("icon", DEFAULT_ICON.to_string()),
45        ("vendor_short", VendorId::Kiro.short_name().to_string()),
46        // Cross-vendor aliases: one pool, so it fills both generic slots.
47        ("plan", snap.plan.clone()),
48        ("session_pct", pct.to_string()),
49        ("session_reset", reset.clone()),
50        ("weekly_pct", pct.to_string()),
51        ("weekly_reset", reset.clone()),
52        // Kiro-specific placeholders.
53        ("kiro_plan", snap.plan.clone()),
54        ("kiro_pct", pct.to_string()),
55        ("kiro_used", credits(snap.used)),
56        ("kiro_limit", credits(snap.limit)),
57        ("kiro_reset", reset),
58    ])
59}
60
61pub fn severity(snap: &KiroSnapshot) -> PaceSeverity {
62    severity_for(snap.pct())
63}
64
65pub fn render(
66    outcome: &VendorOutcome,
67    snap: &KiroSnapshot,
68    theme: &Theme,
69    opts: &RenderOpts,
70    now: DateTime<Utc>,
71) -> WaybarOutput {
72    let class = Class::from(severity(snap));
73    let format = opts
74        .format
75        .clone()
76        .unwrap_or_else(|| DEFAULT_FORMAT.to_string());
77    let mut values = build_placeholders(snap, now);
78    // Both sinks fed by this map (bar text and --tooltip-format) are Pango
79    // markup. The plan label is API-controlled, so escape its aliases at the
80    // projection boundary. The default tooltip escapes the raw snapshot.
81    for key in ["plan", "kiro_plan"] {
82        if let Some(value) = values.get_mut(key) {
83            *value = escape(value);
84        }
85    }
86
87    let mut text = substitute(&format, &values);
88    if outcome.stale {
89        text.push_str(" ⏸");
90    }
91
92    let wrapper_color = severity_color(severity(snap), theme).to_string();
93    let icon_prefix = match opts.icon.as_deref() {
94        Some(ic) if !ic.is_empty() => format!("{ic} "),
95        _ => String::new(),
96    };
97    let bar_text = color_span(&wrapper_color, &format!("{icon_prefix}{text}"));
98
99    let tooltip = if let Some(fmt) = opts.tooltip_format.as_deref() {
100        substitute(fmt, &values)
101    } else {
102        render_tooltip(outcome, snap, theme, now)
103    };
104
105    WaybarOutput {
106        text: bar_text,
107        tooltip,
108        class,
109    }
110}
111
112fn render_tooltip(
113    outcome: &VendorOutcome,
114    snap: &KiroSnapshot,
115    theme: &Theme,
116    now: DateTime<Utc>,
117) -> String {
118    let blue = &theme.blue;
119    let dim = &theme.dim;
120    let fg = &theme.fg;
121    let pct = snap.pct();
122    let color = severity_color(severity(snap), theme);
123
124    let mut lines: Vec<TooltipLine> = Vec::new();
125    lines.push(TooltipLine::Center(format!(
126        "<span font_weight='bold' foreground='{blue}'>Kiro {}</span>",
127        escape(&snap.plan)
128    )));
129    lines.push(TooltipLine::Sep);
130    lines.push(TooltipLine::Body("".into()));
131    lines.push(TooltipLine::Body(format!(
132        " <span foreground='{fg}'>  󰠮  Credits</span>"
133    )));
134    lines.push(TooltipLine::Body(format!(
135        "   <span font_weight='bold' foreground='{color}'>{pct}%</span> used \
136         <span foreground='{dim}'>({} of {})</span>",
137        credits(snap.used),
138        credits(snap.limit)
139    )));
140
141    lines.push(TooltipLine::Body("".into()));
142    lines.push(TooltipLine::Body(format!(
143        " <span foreground='{dim}'>  󰃰  Resets {}</span>",
144        escape(&countdown::format(snap.reset_at, now))
145    )));
146
147    if let Some((code, msg)) = outcome.last_error.as_ref()
148        && *code != 0
149    {
150        let (icon, ecolor) = if *code >= 500 {
151            ("󰅚", theme.red.as_str())
152        } else {
153            ("󰀪", theme.orange.as_str())
154        };
155        lines.push(TooltipLine::Body("".into()));
156        lines.push(TooltipLine::Sep);
157        lines.push(TooltipLine::Body(format!(
158            " <span foreground='{ecolor}'>  {icon}  HTTP {code}</span>"
159        )));
160        lines.push(TooltipLine::Body(format!(
161            "     <span foreground='{dim}'>{}</span>",
162            escape(msg)
163        )));
164    }
165
166    let updated = updated_at_hm(now, outcome.cache_age);
167    lines.push(TooltipLine::Body("".into()));
168    lines.push(TooltipLine::Sep);
169    lines.push(TooltipLine::Body(format!(
170        " <span foreground='{dim}'>  󰅐  Updated {updated}</span>"
171    )));
172
173    render_bordered(&lines, theme)
174}
175
176impl From<FetchOutcome> for VendorOutcome {
177    fn from(o: FetchOutcome) -> Self {
178        o.map(crate::usage::VendorSnapshot::Kiro)
179    }
180}
181
182#[cfg(test)]
183mod tests {
184    use super::*;
185    use chrono::TimeZone;
186
187    fn now() -> DateTime<Utc> {
188        Utc.with_ymd_and_hms(2026, 7, 26, 12, 0, 0).unwrap()
189    }
190
191    fn sample_snap() -> KiroSnapshot {
192        KiroSnapshot {
193            plan: "KIRO POWER".into(),
194            used: 9943.38,
195            limit: 10000.0,
196            reset_at: Some(now() + chrono::Duration::days(1)),
197        }
198    }
199
200    fn sample_outcome(snap: KiroSnapshot) -> VendorOutcome {
201        VendorOutcome {
202            snapshot: crate::usage::VendorSnapshot::Kiro(snap),
203            stale: false,
204            last_error: None,
205            cache_age: Some(std::time::Duration::from_secs(10)),
206        }
207    }
208
209    fn opts() -> RenderOpts {
210        RenderOpts {
211            format: None,
212            tooltip_format: None,
213            icon: None,
214            pace_tolerance: 5,
215            format_pace_color: false,
216            tooltip_pace_pts: false,
217        }
218    }
219
220    #[test]
221    fn default_bar_shows_the_percentage() {
222        let snap = sample_snap();
223        let out = render(
224            &sample_outcome(snap.clone()),
225            &snap,
226            &Theme::default(),
227            &opts(),
228            now(),
229        );
230        assert!(out.text.contains("99%"), "text: {}", out.text);
231    }
232
233    #[test]
234    fn tooltip_shows_plan_credits_and_reset() {
235        let snap = sample_snap();
236        let out = render(
237            &sample_outcome(snap.clone()),
238            &snap,
239            &Theme::default(),
240            &opts(),
241            now(),
242        );
243        assert!(out.tooltip.contains("Kiro KIRO POWER"));
244        assert!(out.tooltip.contains("99%"));
245        assert!(out.tooltip.contains("9943.38"));
246        assert!(out.tooltip.contains("10000"));
247        assert!(out.tooltip.contains("1d"));
248    }
249
250    #[test]
251    fn whole_number_credits_have_no_decimals() {
252        assert_eq!(credits(10000.0), "10000");
253        assert_eq!(credits(9943.38), "9943.38");
254    }
255
256    #[test]
257    fn severity_tracks_the_credit_percentage() {
258        let mut snap = sample_snap();
259        snap.used = 99.0;
260        snap.limit = 100.0;
261        assert_eq!(severity(&snap), PaceSeverity::Critical);
262    }
263
264    #[test]
265    fn stale_appends_pause() {
266        let snap = sample_snap();
267        let mut outcome = sample_outcome(snap.clone());
268        outcome.stale = true;
269        let out = render(&outcome, &snap, &Theme::default(), &opts(), now());
270        assert!(out.text.contains("⏸"));
271    }
272
273    #[test]
274    fn custom_tooltip_uses_placeholders() {
275        let snap = sample_snap();
276        let mut o = opts();
277        o.tooltip_format = Some("{kiro_pct}% of {kiro_limit} · {kiro_plan}".into());
278        let out = render(
279            &sample_outcome(snap.clone()),
280            &snap,
281            &Theme::default(),
282            &o,
283            now(),
284        );
285        assert_eq!(out.tooltip, "99% of 10000 · KIRO POWER");
286    }
287
288    #[test]
289    fn api_plan_is_inert_in_custom_pango_formats() {
290        let mut snap = sample_snap();
291        snap.plan = "<b>not markup</b> & control\u{1b}".into();
292        let mut o = opts();
293        o.format = Some("{kiro_plan}".into());
294        o.tooltip_format = Some("{plan}".into());
295
296        let out = render(
297            &sample_outcome(snap.clone()),
298            &snap,
299            &Theme::default(),
300            &o,
301            now(),
302        );
303
304        assert!(!out.text.contains("<b>"));
305        assert!(!out.tooltip.contains("<b>"));
306        assert!(
307            out.text
308                .contains("&lt;b&gt;not markup&lt;/b&gt; &amp; control")
309        );
310        assert_eq!(out.tooltip, "&lt;b&gt;not markup&lt;/b&gt; &amp; control");
311        assert!(!out.text.contains('\u{1b}'));
312    }
313
314    #[test]
315    fn generic_windows_map_to_the_single_pool() {
316        let values = build_placeholders(&sample_snap(), now());
317        assert_eq!(values["session_pct"], "99");
318        assert_eq!(values["weekly_pct"], "99");
319        assert_eq!(values["plan"], "KIRO POWER");
320    }
321
322    #[test]
323    fn placeholder_set_contains_all_keys() {
324        let values = build_placeholders(&sample_snap(), now());
325        for key in [
326            "icon",
327            "vendor_short",
328            "plan",
329            "session_pct",
330            "session_reset",
331            "weekly_pct",
332            "weekly_reset",
333            "kiro_plan",
334            "kiro_pct",
335            "kiro_used",
336            "kiro_limit",
337            "kiro_reset",
338        ] {
339            assert!(values.contains_key(key), "missing placeholder {key}");
340        }
341    }
342
343    #[test]
344    fn fetch_outcome_conversion_preserves_metadata() {
345        let fetch = FetchOutcome {
346            snapshot: sample_snap(),
347            stale: true,
348            last_error: Some((401, "bad".into())),
349            cache_age: Some(std::time::Duration::from_secs(42)),
350        };
351        let vendor: VendorOutcome = fetch.into();
352        assert!(matches!(
353            vendor.snapshot,
354            crate::usage::VendorSnapshot::Kiro(_)
355        ));
356        assert!(vendor.stale);
357    }
358}
ai_usagebar/kiro/vendor.rs

ai_usagebar/kiro/
vendor.rs
