ai_usagebar/widget/
cli.rs

1//! Command-line interface — claudebar-compatible flags plus the new
2//! local-testing additions (`--pretty`, `--watch`, `--json`).
3//!
4//! Mirrors claudebar:54-93. The defaults are identical so existing waybar
5//! configs that invoke `claudebar ...` can be retargeted to
6//! `ai-usagebar --vendor anthropic ...` without changing any flags.
7
8use clap::{Parser, ValueEnum};
9
10#[derive(Parser, Debug, Clone)]
11#[command(
12    name = "ai-usagebar",
13    about = "Waybar widget for AI plan usage (Anthropic / OpenAI / Z.AI / OpenRouter)",
14    long_about = "\
15Drop-in replacement for `claudebar` with multi-vendor support.
16
17Output modes:
18  - Default: Waybar JSON ({text, tooltip, class}). Used when stdout is piped.
19  - --pretty: human-readable terminal output for local testing. Auto-enabled
20    when stdout is a TTY, so just running `ai-usagebar --vendor anthropic`
21    in a terminal Does The Right Thing.
22  - --watch N: like --pretty but refreshes every N seconds, clearing the screen
23    between ticks. Useful while iterating on `--format` or `--tooltip-format`.
24  - --json: force JSON output even when stdout is a TTY (for scripting)."
25)]
26pub struct Cli {
27    /// Which vendor to query. When omitted, reads `[ui] primary` from
28    /// `~/.config/ai-usagebar/config.toml`; falls back to `anthropic` if
29    /// neither is set.
30    #[arg(long, value_enum)]
31    pub vendor: Option<Vendor>,
32
33    /// Optional icon prepended to the bar text (Nerd Font glyph / emoji /
34    /// Pango span). claudebar `--icon`.
35    #[arg(long)]
36    pub icon: Option<String>,
37
38    /// Bar-text format string with `{placeholder}` substitutions.
39    /// Defaults to `{session_pct}% · {session_reset}`.
40    #[arg(long)]
41    pub format: Option<String>,
42
43    /// Custom tooltip format. Overrides the default bordered tooltip when
44    /// set; identical placeholder set as `--format`.
45    #[arg(long)]
46    pub tooltip_format: Option<String>,
47
48    /// Tolerance band (in percentage points) for ratio-based pacing icons.
49    #[arg(long, default_value_t = 5)]
50    pub pace_tolerance: u32,
51
52    /// Color pace placeholders individually per window (instead of the
53    /// global usage-based color). Claudebar `--format-pace-color`.
54    #[arg(long)]
55    pub format_pace_color: bool,
56
57    /// Use point-based pacing in the tooltip's pace column (vs ratio-based).
58    /// Also enables an elapsed-position marker on the tooltip progress bars.
59    /// Claudebar `--tooltip-pace-pts`.
60    #[arg(long)]
61    pub tooltip_pace_pts: bool,
62
63    /// Override the low-usage color (#RRGGBB).
64    #[arg(long)]
65    pub color_low: Option<String>,
66    /// Override the mid-usage color (#RRGGBB).
67    #[arg(long)]
68    pub color_mid: Option<String>,
69    /// Override the high-usage color (#RRGGBB).
70    #[arg(long)]
71    pub color_high: Option<String>,
72    /// Override the critical-usage color (#RRGGBB).
73    #[arg(long)]
74    pub color_critical: Option<String>,
75
76    /// Render human-readable terminal output (ANSI colors + box drawing)
77    /// instead of Waybar JSON. Auto-on when stdout is a TTY.
78    #[arg(long)]
79    pub pretty: bool,
80
81    /// Force JSON output even on a TTY (useful when piping into `jq` from
82    /// an interactive shell).
83    #[arg(long, conflicts_with = "pretty")]
84    pub json: bool,
85
86    /// Re-render every N seconds, clearing the screen between ticks. Implies
87    /// `--pretty`. Press Ctrl-C to exit.
88    #[arg(long, value_name = "SECS")]
89    pub watch: Option<u64>,
90
91    /// Cycle the persisted "active vendor" forward and exit. Wire to
92    /// Waybar's `on-scroll-up` to scroll-cycle through enabled vendors.
93    /// Sends SIGRTMIN+13 to waybar afterwards so the bar refreshes
94    /// immediately rather than waiting for the next interval tick.
95    #[arg(long, conflicts_with_all = ["cycle_prev", "watch", "pretty", "json"])]
96    pub cycle_next: bool,
97
98    /// Cycle backwards. Wire to `on-scroll-down`.
99    #[arg(long, conflicts_with_all = ["cycle_next", "watch", "pretty", "json"])]
100    pub cycle_prev: bool,
101
102    /// Override the cache directory (for tests / debugging).
103    #[arg(long, hide = true)]
104    pub cache_dir: Option<std::path::PathBuf>,
105
106    /// Override the credentials file path (for tests / debugging).
107    #[arg(long, hide = true)]
108    pub creds_path: Option<std::path::PathBuf>,
109}
110
111#[derive(Debug, Clone, Copy, ValueEnum, PartialEq, Eq)]
112pub enum Vendor {
113    Anthropic,
114    Openai,
115    Zai,
116    Openrouter,
117}
118
119impl Vendor {
120    pub fn to_id(self) -> crate::vendor::VendorId {
121        match self {
122            Vendor::Anthropic => crate::vendor::VendorId::Anthropic,
123            Vendor::Openai => crate::vendor::VendorId::Openai,
124            Vendor::Zai => crate::vendor::VendorId::Zai,
125            Vendor::Openrouter => crate::vendor::VendorId::Openrouter,
126        }
127    }
128}
129
130impl Cli {
131    /// Resolve the vendor with full precedence:
132    ///   1. explicit `--vendor` (highest)
133    ///   2. persisted scroll-cycle state (`~/.cache/ai-usagebar/active_vendor`)
134    ///   3. `[ui] primary` from config
135    ///   4. anthropic (lowest)
136    pub fn resolved_vendor(&self, config: &crate::config::Config) -> Vendor {
137        if let Some(v) = self.vendor {
138            return v;
139        }
140        if let Some(id) = crate::active::read() {
141            if config.is_enabled(id) {
142                return id_to_vendor(id);
143            }
144        }
145        match config.ui.primary {
146            Some(id) => id_to_vendor(id),
147            None => Vendor::Anthropic,
148        }
149    }
150}
151
152fn id_to_vendor(id: crate::vendor::VendorId) -> Vendor {
153    match id {
154        crate::vendor::VendorId::Anthropic => Vendor::Anthropic,
155        crate::vendor::VendorId::Openai => Vendor::Openai,
156        crate::vendor::VendorId::Zai => Vendor::Zai,
157        crate::vendor::VendorId::Openrouter => Vendor::Openrouter,
158    }
159}
160
161impl Cli {
162    /// True when we should emit Waybar JSON. Default behavior: JSON when
163    /// stdout is piped, pretty when on a TTY (unless `--json` is set).
164    pub fn output_json(&self) -> bool {
165        if self.json {
166            return true;
167        }
168        if self.pretty || self.watch.is_some() {
169            return false;
170        }
171        // Auto-detect: emit pretty when stdout is a TTY.
172        !is_stdout_tty()
173    }
174}
175
176fn is_stdout_tty() -> bool {
177    use std::io::IsTerminal;
178    std::io::stdout().is_terminal()
179}
180
181#[cfg(test)]
182mod tests {
183    use super::*;
184    use clap::Parser;
185
186    #[test]
187    fn defaults_match_claudebar() {
188        let cli = Cli::parse_from(["ai-usagebar"]);
189        assert_eq!(cli.vendor, None);
190        // Without explicit --vendor and with default config, resolve to anthropic.
191        let cfg = crate::config::Config::default();
192        assert_eq!(cli.resolved_vendor(&cfg), Vendor::Anthropic);
193        assert_eq!(cli.pace_tolerance, 5);
194        assert!(cli.format.is_none());
195        assert!(cli.tooltip_format.is_none());
196        assert!(cli.icon.is_none());
197        assert!(!cli.format_pace_color);
198        assert!(!cli.tooltip_pace_pts);
199        assert!(!cli.pretty);
200        assert!(!cli.json);
201        assert!(cli.watch.is_none());
202    }
203
204    #[test]
205    fn primary_from_config_wins_when_vendor_unset() {
206        let cli = Cli::parse_from(["ai-usagebar"]);
207        let mut cfg = crate::config::Config::default();
208        cfg.ui.primary = Some(crate::vendor::VendorId::Openrouter);
209        assert_eq!(cli.resolved_vendor(&cfg), Vendor::Openrouter);
210    }
211
212    #[test]
213    fn explicit_vendor_overrides_config_primary() {
214        let cli = Cli::parse_from(["ai-usagebar", "--vendor", "zai"]);
215        let mut cfg = crate::config::Config::default();
216        cfg.ui.primary = Some(crate::vendor::VendorId::Openrouter);
217        assert_eq!(cli.resolved_vendor(&cfg), Vendor::Zai);
218    }
219
220    #[test]
221    fn claudebar_compatible_flag_surface() {
222        let cli = Cli::parse_from([
223            "ai-usagebar",
224            "--icon",
225            "󰚩",
226            "--format",
227            "{session_pct}% · {session_reset}",
228            "--tooltip-format",
229            "S:{session_pct}",
230            "--pace-tolerance",
231            "10",
232            "--format-pace-color",
233            "--tooltip-pace-pts",
234            "--color-low",
235            "#50fa7b",
236            "--color-mid",
237            "#f1fa8c",
238            "--color-high",
239            "#ffb86c",
240            "--color-critical",
241            "#ff5555",
242        ]);
243        assert_eq!(cli.icon.as_deref(), Some("󰚩"));
244        assert_eq!(
245            cli.format.as_deref(),
246            Some("{session_pct}% · {session_reset}")
247        );
248        assert_eq!(cli.tooltip_format.as_deref(), Some("S:{session_pct}"));
249        assert_eq!(cli.pace_tolerance, 10);
250        assert!(cli.format_pace_color);
251        assert!(cli.tooltip_pace_pts);
252        assert_eq!(cli.color_low.as_deref(), Some("#50fa7b"));
253        assert_eq!(cli.color_critical.as_deref(), Some("#ff5555"));
254    }
255
256    #[test]
257    fn pretty_and_json_conflict() {
258        let res = Cli::try_parse_from(["ai-usagebar", "--pretty", "--json"]);
259        assert!(res.is_err());
260    }
261
262    #[test]
263    fn watch_disables_json_output() {
264        let cli = Cli::parse_from(["ai-usagebar", "--watch", "5"]);
265        assert_eq!(cli.watch, Some(5));
266        assert!(!cli.output_json());
267    }
268}
ai_usagebar/widget/cli.rs

ai_usagebar/widget/
cli.rs
