ai_usagebar/

usage.rs

//! Canonical in-memory representation of "how much have I used my plan".
//!
//! Each vendor's snapshot lives in its own variant — this is deliberate.
//! Anthropic exposes three windows + extra credits; OpenAI Codex exposes up to
//! two windows + credit balance + message-count ranges; OpenRouter is a single
//! credit-balance number with daily/weekly/monthly totals; Z.AI is a list of
//! token + MCP buckets; DeepSeek is a credit balance; Kimi is a weekly quota
//! plus a 5h rolling rate-limit window. Forcing them into a shared shape would
//! either drop information or paper over genuine differences.
//!
//! Renderers (widget tooltip, TUI tab) consume a `VendorSnapshot` directly,
//! not a flattened shape — so each vendor controls its own presentation while
//! sharing the pacing math, color thresholds, and Pango primitives.

use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};

use crate::error::{AppError, Result};

/// Reject a non-finite monetary value. A NaN or infinity reaching a balance
/// field means the payload was not what we think it is; displaying it as money
/// (or caching it as authoritative) is worse than failing loudly.
pub fn finite_amount(vendor: &str, field: &str, v: f64) -> Result<f64> {
    if v.is_finite() {
        Ok(v)
    } else {
        Err(AppError::Schema(format!(
            "{vendor}: `{field}` is not a finite number"
        )))
    }
}

/// Parse a monetary field that the wire encodes as a string. A malformed or
/// empty value is a schema error, **not** a zero balance — silently reporting
/// $0.00 for an error envelope is the failure mode this guards against.
pub fn parse_amount(vendor: &str, field: &str, s: &str) -> Result<f64> {
    let t = s.trim();
    if t.is_empty() {
        return Err(AppError::Schema(format!("{vendor}: `{field}` is empty")));
    }
    let v: f64 = t
        .parse()
        .map_err(|_| AppError::Schema(format!("{vendor}: `{field}` is not numeric (got {t:?})")))?;
    finite_amount(vendor, field, v)
}

/// A single usage window — generic enough that every vendor with a notion of
/// "% used vs. when does it reset" can express itself with it.
///
/// `utilization_pct` is `0..=100` (integer percent, matching claudebar's units).
/// `resets_at` is `None` when the vendor doesn't report a reset time.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct UsageWindow {
    pub utilization_pct: i32,
    pub resets_at: Option<DateTime<Utc>>,
    /// Window length (used for pacing math).
    pub window_duration: chrono::Duration,
}

/// Money in minor currency units (historically always cents; see
/// `ExtraUsage::decimal_places` for the actual scale) to dodge float roundoff.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct Cents(pub i64);

/// Anthropic-specific snapshot — three rolling windows plus optional
/// pay-as-you-go credit balance.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct AnthropicSnapshot {
    /// "Pro", "Max", "Ultra", etc. — renderers add the "Claude " prefix.
    pub plan: String,
    /// The five-hour window. `None` when the account has no five-hour limit
    /// at all — a verified Ultra account: there is no window to show, which
    /// is not the same thing as an idle one at 0%.
    pub session: Option<UsageWindow>,
    pub weekly: UsageWindow,
    /// Some vendors of Claude (Pro, some Max tiers) don't have a separate
    /// Sonnet bucket — in which case this is None.
    pub sonnet: Option<UsageWindow>,
    /// Model-scoped weekly windows from the newer `limits[]` array
    /// (`kind == "weekly_scoped"`), e.g. the Fable weekly cap. Labels come
    /// from the API (`scope.model.display_name`), so new models show up
    /// without a code change. Empty when the account has none.
    pub scoped: Vec<ScopedWindow>,
    /// `None` when the block is absent, the user disabled extra usage, or the
    /// response has no spend worth reporting. Provider-disabled exhausted
    /// usage stays visible with an explicit status.
    pub extra: Option<ExtraUsage>,
    /// Banked limit resets from the `cedar_ember` block — the same idea as
    /// Codex's rate-limit reset credits and SuperGrok's remaining resets.
    /// Empty when the account has no grant or the endpoint withheld the block.
    pub reset_credits: ResetCredits,
    /// Claude Code's cloud session credits, from the `iguana_necktie` block:
    /// the dollar limit its cloud sessions draw from, what is left, when it
    /// expires, and whether the grant is locked. `None` unless the response
    /// reports a usable limit.
    pub cloud_credits: Option<CloudCredits>,
    /// The prepaid usage-credit balance, from `spend.balance`. `None` unless
    /// the response reports a balance we can express — a balance the
    /// response does not report is never shown, never as $0.
    pub usage_credits: Option<UsageCredits>,
}

/// Cloud session credits (the usage response's `iguana_necktie` block).
/// Both amounts are whole cents in USD — the wire is dollars, converted at
/// the parse boundary so the snapshot, which derives `Eq`, stays integer.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CloudCredits {
    pub limit: Cents,
    pub remaining: Cents,
    /// `resets_at` as RFC 3339. `None` when the wire omitted it or it does
    /// not parse.
    pub expires_at: Option<DateTime<Utc>>,
    /// `locked_reason` was a non-empty string: the grant exists but cannot
    /// currently be drawn on.
    pub locked: bool,
}

/// The prepaid usage-credit balance (the usage response's `spend.balance`).
/// The filled-in shape of `balance` has never been observed on the wire, so
/// the parse interprets it after the fact and only then lifts it here.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct UsageCredits {
    /// Balance in minor units of `currency`, at the scale of
    /// `decimal_places`.
    pub balance: Cents,
    pub decimal_places: u32,
    pub currency: Option<String>,
    /// Whether the balance auto-reloads, when the response says so —
    /// `None` means "not reported", not "off".
    pub auto_reload: Option<bool>,
}

/// A usage window scoped to a specific model, labeled by the API
/// (e.g. "Fable"). Weekly (7d) duration.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ScopedWindow {
    pub label: String,
    pub window: UsageWindow,
}

/// "Extra usage" pay-as-you-go block (claudebar's `extra_usage`).
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ExtraUsage {
    /// `None` when the payload carries no usable `monthly_limit` — an
    /// explicit null (observed for plans without a spending cap, e.g. Claude
    /// Pro, #30) or an absent field. Either way the spend is real and stays
    /// visible; only the limit is unreported, and the renderers say exactly
    /// that rather than inferring a plan tier from it.
    pub limit: Option<Cents>,
    pub spent: Cents,
    /// ISO code from the block (`"BRL"`, `"USD"`, …). `None` on older payloads
    /// that predate the field — formatted as `$` for back-compat, which was
    /// the only behaviour before the field existed.
    pub currency: Option<String>,
    /// Minor-unit digits from the block's `decimal_places` (BRL/USD = 2,
    /// JPY/KRW = 0). `None` means the wire did not report the scale. We keep
    /// that absence instead of guessing from an incomplete currency table.
    pub decimal_places: Option<u32>,
    /// Why extra usage cannot currently continue. Enabled usage has `Active`.
    pub status: ExtraUsageStatus,
}

#[derive(Debug, Default, Clone, Copy, PartialEq, Eq)]
pub enum ExtraUsageStatus {
    #[default]
    Active,
    OutOfCredits,
    MonthlyLimitReached,
    Unavailable,
}

impl ExtraUsage {
    /// Integer percentage of the monthly limit consumed (0..=100, saturating
    /// at 0 when limit is non-positive — matches claudebar:540-542).
    ///
    /// With no cap there is no denominator, so no meaningful percentage
    /// exists; 0 keeps the bar and severity calm rather than inventing one.
    pub fn percent(&self) -> i32 {
        match self.limit {
            Some(l) if l.0 > 0 => ((self.spent.0 * 100) / l.0) as i32,
            _ => 0,
        }
    }

    pub fn fmt_spent(&self) -> String {
        self.fmt_amount(self.spent)
    }

    pub fn fmt_limit(&self) -> Option<String> {
        self.limit.map(|l| self.fmt_amount(l))
    }

    pub fn status_note(&self) -> Option<&'static str> {
        match self.status {
            ExtraUsageStatus::Active => None,
            ExtraUsageStatus::OutOfCredits => Some("out of credits"),
            ExtraUsageStatus::MonthlyLimitReached => Some("monthly limit reached"),
            ExtraUsageStatus::Unavailable => Some("currently unavailable"),
        }
    }

    fn fmt_amount(&self, amount: Cents) -> String {
        match (self.decimal_places, self.currency.as_deref()) {
            (Some(decimal_places), currency) => fmt_minor(amount.0, decimal_places, currency),
            // Legacy payloads predate both fields and were always cents/USD.
            // Preserve that established behaviour only when neither field can
            // tell us otherwise.
            (None, None) => fmt_minor(amount.0, 2, None),
            // A currency code alone does not determine its ISO minor-unit
            // exponent. Keep the amount truthful instead of silently dividing
            // zero-, three-, or four-decimal currencies by the wrong scale.
            (None, Some(currency)) => fmt_minor_units(amount.0, currency),
        }
    }
}

fn fmt_minor_units(minor: i64, currency: &str) -> String {
    let sign = if minor < 0 { "-" } else { "" };
    format!("{sign}{} minor units {currency}", minor.unsigned_abs())
}

/// Format an amount in minor units with its own currency and scale.
///
/// The scale is this function's own — `money` cannot express a zero- or
/// three-decimal currency — but the *symbol* comes from
/// [`crate::format::with_currency`], so the two cannot disagree about what a
/// given code looks like.
pub fn fmt_minor(minor: i64, decimal_places: u32, currency: Option<&str>) -> String {
    let scale = 10_u64.pow(decimal_places);
    // `unsigned_abs`, not negation: `-i64::MIN` overflows. Unreachable from
    // the wire (the parse gate rejects negatives) but this is a pub fn.
    let sign = if minor < 0 { "-" } else { "" };
    let abs = minor.unsigned_abs();
    let number = if decimal_places == 0 {
        format!("{abs}")
    } else {
        format!(
            "{}.{:0width$}",
            abs / scale,
            abs % scale,
            width = decimal_places as usize
        )
    };
    crate::format::with_currency(sign, &number, currency)
}

/// DeepSeek — credit balance from `/user/balance`.
#[derive(Debug, Clone, PartialEq)]
pub struct DeepseekSnapshot {
    pub is_available: bool,
    /// Current balance (prefer USD, fallback to CNY).
    pub balance: f64,
    /// Free-granted credits component.
    pub granted: f64,
    /// Topped-up (purchased) credits component.
    pub topped_up: f64,
    /// The currency of the above amounts (currently "USD" or "CNY").
    pub currency: String,
}

impl Eq for DeepseekSnapshot {}

impl Default for DeepseekSnapshot {
    fn default() -> Self {
        Self {
            is_available: false,
            balance: 0.0,
            granted: 0.0,
            topped_up: 0.0,
            currency: String::new(),
        }
    }
}

/// DeepInfra prepaid balance and current-month usage from the documented
/// `/payment/checklist` and `/payment/usage` billing endpoints.
#[derive(Debug, Clone, PartialEq)]
pub struct DeepInfraSnapshot {
    /// General-purpose prepaid credit remaining, in US dollars.
    pub balance: f64,
    /// Current calendar-month spend, in US dollars.
    pub monthly_spend: f64,
    /// Optional monthly spending limit, in US dollars.
    pub monthly_limit: Option<f64>,
    /// Usage period in the API's `YYYY.MM` form.
    pub period: String,
}

impl Eq for DeepInfraSnapshot {}

impl DeepInfraSnapshot {
    pub fn monthly_consumed_pct(&self) -> Option<i32> {
        let limit = self.monthly_limit.filter(|limit| *limit > 0.0)?;
        Some(
            ((self.monthly_spend / limit) * 100.0)
                .round()
                .clamp(0.0, 9999.0) as i32,
        )
    }
}

/// One promo balance from Cursor's spending page
/// (`GetClientVisibleCreditGrants`). Amounts are USD cents.
/// `remaining_cents` is the figure to the left of the slash (`$21/$25`).
///
/// The grant may cover only one product (Cloud Agents, Bugbot, a single
/// model). `display_name` is that product title when Cursor sent one. The
/// row still says "Credits" when the name is not a title Cursor shows.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CursorCreditGrant {
    pub remaining_cents: i64,
    pub total_cents: i64,
    /// When the grant lapses. `None` when the payload has no expiry — some
    /// account credits never do.
    pub expires_at: Option<DateTime<Utc>>,
    /// Vendor label. Empty when the grant has none.
    pub display_name: String,
}

/// Cursor — the two included-usage pools the dashboard shows, from the
/// undocumented `cursor.com/api/usage-summary` endpoint (the same one the
/// dashboard's own frontend calls), authenticated with the session token the
/// Cursor IDE wrote to its local `state.vscdb`.
///
/// Since Cursor's mid-2026 pricing, a plan's included compute is split into two
/// quota pools, each shown as a percentage: **Cursor Models** (Auto + Composer,
/// `autoPercentUsed`) and **Other Models** (named / third-party, `apiPercentUsed`).
/// Overflow past either pool falls to on-demand spend. Percentages are integers
/// (rounded from the wire floats) to match the dashboard and every other
/// vendor's integer-percent convention; they can exceed 100 when a pool is over
/// its included allowance.
///
/// [`CursorSnapshot::credits`] is a second, best-effort call
/// (`GetClientVisibleCreditGrants`). An empty list means the account has no
/// visible grant, or that call failed; either way the pools above still stand.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CursorSnapshot {
    /// Membership label, title-cased from `membershipType` (e.g. "Ultra").
    pub plan: String,
    /// "Cursor Models" pool — Auto + Composer (`autoPercentUsed`, rounded).
    pub auto_pct: i32,
    /// "Other Models" pool — named / third-party (`apiPercentUsed`, rounded).
    pub api_pct: i32,
    /// Overall included usage (`totalPercentUsed`, rounded) — the dashboard's
    /// "you've used N% of your included total usage" headline.
    pub total_pct: i32,
    /// `true` when the plan reports `isUnlimited` — the pools don't cap and the
    /// percentages are not meaningful.
    pub unlimited: bool,
    /// Whether on-demand (overage) spend is turned on (`onDemand.enabled`).
    pub on_demand_enabled: bool,
    /// On-demand spend in cents (`onDemand.used`), when Cursor reports it.
    pub on_demand_used_cents: Option<i64>,
    /// Configured on-demand spending limit in cents (`onDemand.limit`).
    pub on_demand_limit_cents: Option<i64>,
    /// End of the current billing cycle (`billingCycleEnd`) — when the pools
    /// reset.
    pub reset_at: Option<DateTime<Utc>>,
    /// Start of the current billing cycle (`billingCycleStart`), when the API
    /// sends it. With `reset_at` it gives the exact window length the pace
    /// projection needs; absent, no window length is reported at all.
    pub cycle_start: Option<DateTime<Utc>>,
    /// Spending-page credit grants, soonest expiry first. Empty when the
    /// account has none or the grant call did not succeed.
    pub credits: Vec<CursorCreditGrant>,
}

impl CursorSnapshot {
    /// Length of the current billing cycle, but only when the API stated both
    /// ends and they are ordered. `None` otherwise — older responses and every
    /// snapshot cached before `billingCycleStart` existed omit the start, and a
    /// guessed month would reach a frontend as an exact window and be paced as
    /// one. `window_secs` is absent instead; the reset time still shows.
    pub fn cycle_window(&self) -> Option<chrono::Duration> {
        match (self.cycle_start, self.reset_at) {
            (Some(start), Some(end)) if end > start => Some(end - start),
            _ => None,
        }
    }

    /// The binding pool — whichever is closest to (or furthest past) its cap.
    /// Drives the bar color and the single generic `session_pct` alias.
    pub fn worst_pct(&self) -> i32 {
        self.auto_pct.max(self.api_pct)
    }
}

/// Kiro CLI (AWS CodeWhisperer / Q Developer backend) — a single credit pool
/// from `AmazonCodeWhispererService.GetUsageLimits`, the same call kiro-cli's
/// own `/usage` slash command makes. Authenticated with the AWS SSO OIDC
/// bearer token kiro-cli already cached locally, refreshed with the paired
/// refresh token when it's close to expiry — see `kiro::db` and `kiro::oauth`.
#[derive(Debug, Clone, PartialEq)]
pub struct KiroSnapshot {
    /// Subscription tier label (`subscriptionInfo.subscriptionTitle`, e.g.
    /// "KIRO POWER").
    pub plan: String,
    /// Credits consumed this cycle (`currentUsageWithPrecision`).
    pub used: f64,
    /// Credits included in the plan (`usageLimitWithPrecision`).
    pub limit: f64,
    /// When the credit pool resets (`nextDateReset`).
    pub reset_at: Option<DateTime<Utc>>,
}

impl Eq for KiroSnapshot {}

impl KiroSnapshot {
    /// Percentage of the credit pool consumed, rounded. `0` when `limit` is
    /// not positive — defensive; the API has not been observed to send that.
    pub fn pct(&self) -> i32 {
        if self.limit <= 0.0 {
            return 0;
        }
        ((self.used / self.limit) * 100.0)
            .round()
            .clamp(0.0, 9999.0) as i32
    }
}

/// Kimi Code — weekly subscription quota plus a 5h rolling rate-limit window.
///
/// Accounts on the newer `/coding/v1/usages` response shape have no weekly
/// counters at all: the top-level `usage` block is replaced by a `usages` map
/// of ratios, of which only `limit_month_total` — the combined monthly pool —
/// is read. On such accounts `has_weekly` is false, the weekly counters stay
/// zero (never fabricated), and the monthly pool arrives as
/// `monthly_pct`/`monthly_reset_at`.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct KimiSnapshot {
    pub plan: Option<String>,
    pub weekly_limit: u64,
    pub weekly_used: u64,
    pub weekly_remaining: u64,
    pub weekly_reset_at: Option<DateTime<Utc>>,
    /// `false` on the newer `usages`-map shape, which exposes no weekly
    /// bucket; renderers must drop the weekly row rather than draw zeros.
    pub has_weekly: bool,
    /// Combined monthly pool usage (0..=100) on the newer shape. The
    /// `limit_month_code` entry is the Code slice *inside* that pool, never
    /// its own allowance, so it is not carried here.
    pub monthly_pct: Option<i32>,
    pub monthly_reset_at: Option<DateTime<Utc>>,
    pub window_limit: u64,
    pub window_used: u64,
    pub window_remaining: u64,
    pub window_reset_at: Option<DateTime<Utc>>,
}

impl KimiSnapshot {
    fn pct(used: u64, limit: u64) -> i32 {
        if limit == 0 {
            0
        } else {
            // Keep all quota values exact: f64 loses integer precision above
            // 2^53. This is the integer equivalent of round(used / limit *
            // 100), with saturation for inconsistent upstream counters.
            let pct = ((used as u128 * 100) + (limit as u128 / 2)) / limit as u128;
            pct.min(100) as i32
        }
    }

    /// Percentage of the weekly subscription quota consumed (0..=100).
    pub fn weekly_pct(&self) -> i32 {
        Self::pct(self.weekly_used, self.weekly_limit)
    }

    /// Percentage of the rolling rate-limit window consumed (0..=100).
    pub fn window_pct(&self) -> i32 {
        Self::pct(self.window_used, self.window_limit)
    }

    /// Worst percentage across the windows this snapshot actually has: the
    /// rolling window, the weekly quota when present, and the monthly pool
    /// when present.
    pub fn worst_pct(&self) -> i32 {
        let mut worst = self.window_pct();
        if self.has_weekly {
            worst = worst.max(self.weekly_pct());
        }
        if let Some(monthly) = self.monthly_pct {
            worst = worst.max(monthly);
        }
        worst
    }
}

/// Discriminated union of vendor-specific snapshots. The widget and TUI match
/// on this to pick a renderer.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum VendorSnapshot {
    Anthropic(AnthropicSnapshot),
    Openai(OpenAiSnapshot),
    Copilot(crate::copilot::types::Snapshot),
    Zai(ZaiSnapshot),
    Openrouter(OpenRouterSnapshot),
    Deepseek(DeepseekSnapshot),
    Deepinfra(DeepInfraSnapshot),
    Kimi(KimiSnapshot),
    Kilo(KiloSnapshot),
    Novita(NovitaSnapshot),
    Synthetic(SyntheticSnapshot),
    Moonshot(MoonshotSnapshot),
    Grok(GrokSnapshot),
    SuperGrok(SuperGrokSnapshot),
    Grokbot(GrokbotSnapshot),
    AnthropicApi(AnthropicApiSnapshot),
    Antigravity(AntigravitySnapshot),
    Cursor(CursorSnapshot),
    Minimax(MinimaxSnapshot),
    Kiro(KiroSnapshot),
    NousResearch(crate::nous::types::AccountSnapshot),
    OpenCodeGo(crate::opencode_go::types::Usage),
    CommandCode(crate::commandcode::types::Snapshot),
    Ollama(OllamaSnapshot),
    OrcaRouter(OrcaRouterSnapshot),
    ModelStudio(ModelStudioSnapshot),
    Lyceum(LyceumSnapshot),
    Devin(DevinSnapshot),
    Droid(DroidSnapshot),
    DeepRelay(DeepRelaySnapshot),
    Mistral(MistralSnapshot),
    MuseCode(MuseSnapshot),
    /// A `[[custom]]` provider. Which one is not in the snapshot: the caller
    /// that fetched it holds the `CustomProviderConfig`, and the cache
    /// directory is keyed by its `id`.
    Custom(crate::custom::types::CustomSnapshot),
    Cline(crate::cline::types::Snapshot),
}

impl VendorSnapshot {
    /// Banked, user-redeemable resets, for the vendors that have them.
    ///
    /// The one place this table lives. It had already been written twice —
    /// once for the report's `reset_credits` field and once for the
    /// expiry notifications — and adding a third provider to only one of them
    /// is a silent half-feature: the sidebar lists a grant the notifier never
    /// warns about. `None` is the honest answer for every other vendor, and
    /// for Claude and SuperGrok while they hold none. Codex's usage response
    /// states its count, so an empty [`ResetCredits`] there means "this
    /// account banks resets and currently holds none" and is shown as such;
    /// `None` for Codex means the response omitted the block.
    pub fn reset_credits(&self) -> Option<&ResetCredits> {
        match self {
            Self::Anthropic(snapshot) => Some(&snapshot.reset_credits).filter(|c| !c.is_empty()),
            Self::Openai(snapshot) => snapshot.reset_credits.as_ref(),
            Self::SuperGrok(snapshot) => Some(&snapshot.reset_credits).filter(|c| !c.is_empty()),
            _ => None,
        }
    }
}

/// Google Antigravity 2.0 / CLI snapshot. The API groups models into Gemini
/// and third-party (Claude/GPT) buckets, and each group may carry a 5-hour and
/// a weekly window — up to four, and not every product or plan offers all of
/// them. Antigravity CLI 1.1.22 returns weekly buckets only, so every window is
/// optional and a snapshot is valid when at least one arrived.
#[derive(Debug, Clone, PartialEq)]
pub struct AntigravitySnapshot {
    pub plan: String,
    /// Fingerprint of the signed-in account. Never displayed — it exists so a
    /// cache written for one Google account is not served for another.
    pub account: String,
    /// Where the figures came from: a running local product, or the Cloud
    /// Code API reached with the saved Google session when no local server
    /// can answer — none is running, or `agy` withholds its CSRF token.
    pub source: AntigravitySource,
    /// Gemini group, 5-hour window.
    pub session: Option<UsageWindow>,
    /// Gemini group, weekly window.
    pub weekly: Option<UsageWindow>,
    /// Claude/GPT group, 5-hour window.
    pub third_party_session: Option<UsageWindow>,
    /// Claude/GPT group, weekly window.
    pub third_party_weekly: Option<UsageWindow>,
    /// Pools kept apart from the two shared groups, such as the weekly
    /// "Gemini 4 Argon" limit on Pro plans, in the order the response listed
    /// them. Empty for a product or plan that reports none.
    pub model_pools: Vec<ModelPoolWindow>,
}

impl Eq for AntigravitySnapshot {}

/// One window of an Antigravity per-model pool, named by the response's group
/// display name. Its cadence is `window.window_duration`.
#[derive(Debug, Clone, PartialEq)]
pub struct ModelPoolWindow {
    pub name: String,
    pub window: UsageWindow,
}

impl Eq for ModelPoolWindow {}

impl ModelPoolWindow {
    /// Pools report the shared windows' two cadences, 5-hour and weekly.
    pub fn is_weekly(&self) -> bool {
        self.window.window_duration > chrono::Duration::hours(5)
    }
}

/// Which path produced an [`AntigravitySnapshot`]. The local language server
/// is the primary source; the remote API is the fallback for when no product
/// is running, and the panel says so because the two can disagree briefly.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum AntigravitySource {
    #[default]
    Local,
    Remote,
    /// The running `agy` CLI's official status-line payload.
    Statusline,
    /// `agy`'s own reading, through `agy -p "/quota"`: the figures its Models
    /// & Quota page shows, so the panel has nothing to qualify.
    Agy,
}

impl AntigravitySource {
    pub fn as_str(self) -> &'static str {
        match self {
            AntigravitySource::Local => "local",
            AntigravitySource::Remote => "remote",
            AntigravitySource::Statusline => "statusline",
            AntigravitySource::Agy => "agy",
        }
    }

    /// Inverse of [`as_str`](Self::as_str); anything unrecognised is `None` so
    /// a cache reader can fall back to the default rather than guess.
    pub fn parse(s: &str) -> Option<Self> {
        match s {
            "local" => Some(AntigravitySource::Local),
            "remote" => Some(AntigravitySource::Remote),
            "statusline" => Some(AntigravitySource::Statusline),
            "agy" => Some(AntigravitySource::Agy),
            _ => None,
        }
    }
}

/// MiniMax Token Plan — `/v1/token_plan/remains` returns one row per model
/// bucket (`general` for text/coding, `video`), and each row carries its own
/// rolling interval window plus a weekly window.
///
/// Two things the payload dictates rather than convention: the interval length
/// is **not fixed** (`general` rolls every 5h, `video` every 24h), so the
/// duration is derived from the row's own start/end rather than assumed; and
/// the API reports the percentage **remaining**, which is inverted on the way
/// in so these windows carry consumed-% like every other vendor's.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct MinimaxSnapshot {
    pub plan: String,
    /// `general` bucket — rolling interval window (5h on the observed plans).
    pub session: UsageWindow,
    /// `general` bucket — weekly window.
    pub weekly: UsageWindow,
    /// `video` bucket, `None` on plans that carry no video quota.
    pub video_session: Option<UsageWindow>,
    pub video_weekly: Option<UsageWindow>,
}

/// Anthropic Admin API — month-to-date spend (USD) from the cost report. The
/// monthly `limit` is supplied from config (the API exposes neither the limit
/// nor the remaining prepaid credit balance).
#[derive(Debug, Clone, PartialEq)]
pub struct AnthropicApiSnapshot {
    pub spent: f64,
    pub limit: Option<f64>,
    /// Last UTC calendar day fully covered by the daily cost buckets.
    pub covered_through: Option<chrono::NaiveDate>,
}

impl Eq for AnthropicApiSnapshot {}

impl AnthropicApiSnapshot {
    /// Spend as an integer percentage of the configured limit; `None` when no
    /// positive limit is set.
    pub fn pct(&self) -> Option<i32> {
        self.limit
            .filter(|l| l.is_finite() && *l > 0.0)
            .map(|l| ((self.spent / l) * 100.0).round().clamp(0.0, 9999.0) as i32)
    }

    pub fn coverage_label(&self) -> String {
        match self.covered_through_day() {
            Some(day) => format!("Month to date (through {day})"),
            None => "Completed daily buckets this month".into(),
        }
    }

    /// The last fully covered UTC day as the card prints it, `Oct 17`.
    pub fn covered_through_day(&self) -> Option<String> {
        let date = self.covered_through?;
        const MONTHS: [&str; 12] = [
            "Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec",
        ];
        let month = MONTHS[chrono::Datelike::month0(&date) as usize];
        Some(format!("{month} {}", chrono::Datelike::day(&date)))
    }
}

/// Kilo Code — remaining credit balance from `/api/profile/balance` (USD).
/// No purchased-total is exposed on that endpoint, so there's no consumed-%.
#[derive(Debug, Clone, PartialEq)]
pub struct KiloSnapshot {
    pub label: String,
    pub balance: f64,
}

impl Eq for KiloSnapshot {}

/// Novita AI — account balance from `/openapi/v1/billing/balance/detail`, with
/// all amounts already converted from the API's 1/10000-USD integers to USD.
#[derive(Debug, Clone, PartialEq)]
pub struct NovitaSnapshot {
    /// Spendable credit balance (`availableBalance`).
    pub available: f64,
    /// Remaining top-up (`cashBalance`).
    pub cash: f64,
    /// Credit limit — max you can owe (`creditLimit`).
    pub credit_limit: f64,
    /// Amount currently owed (`outstandingInvoices`).
    pub outstanding: f64,
}

impl Eq for NovitaSnapshot {}

/// Synthetic's sole limit from `/v2/quotas`: the regenerating weekly credit
/// budget (`planWeeklyLimit`, percentage-only) or its credit-denominated
/// variant (`weeklyTokenLimit`). The legacy five-hour `subscription` counter
/// stopped being tracked when the weekly window became the only metric.
#[derive(Debug, Clone, PartialEq)]
pub struct SyntheticSnapshot {
    /// Configured plan tier label (go/standard/big) — the quotas API has no
    /// tier identifier, so this is display-only config.
    pub plan: Option<String>,
    pub weekly: UsageWindow,
    /// Credit spend figures when the plan reports a credit budget.
    pub credits: Option<WeeklyCredits>,
    /// Percentage points restored at the next regeneration tick.
    pub next_regen_percent: Option<f64>,
    /// The used share with the API's decimal precision preserved (the
    /// integer `weekly.utilization_pct` is what meters and severity colors
    /// use); frontends that print a figure read this so "70.5" does not
    /// collapse to "70".
    pub utilization_precise: f64,
}

impl Eq for SyntheticSnapshot {}

/// Credit figures for Synthetic's weekly token budget.
#[derive(Debug, Clone, PartialEq)]
pub struct WeeklyCredits {
    pub used: f64,
    pub max: f64,
    /// Credits restored at the next regeneration tick.
    pub next_regen: f64,
}

impl Eq for WeeklyCredits {}

/// Lyceum's platform credit balance is denominated in USD, not a percentage quota.
#[derive(Debug, Clone, PartialEq, Serialize)]
pub struct LyceumSnapshot {
    pub available_credits: f64,
    pub used_credits: f64,
    pub total_credits_used: f64,
    pub remaining_credits: f64,
    pub monthly_free_credits: f64,
    pub purchased_credits: f64,
}
impl Eq for LyceumSnapshot {}

/// Moonshot / Kimi — account balance from `/v1/users/me/balance`. Currency is
/// USD (`api.moonshot.ai`) or CNY (`api.moonshot.cn`); there's no currency
/// field in the response, so it's carried here from the region config.
#[derive(Debug, Clone, PartialEq)]
pub struct MoonshotSnapshot {
    /// Spendable balance (`available_balance` = cash + voucher). `<= 0` blocks
    /// the inference API.
    pub available: f64,
    /// Voucher credit (`voucher_balance`).
    pub voucher: f64,
    /// Cash balance (`cash_balance`); can be negative (debt).
    pub cash: f64,
    /// "USD" or "CNY", implied by the host.
    pub currency: String,
}

impl Eq for MoonshotSnapshot {}

/// xAI (Grok) — prepaid credit balance in USD, derived from the Management
/// API's `total.val` (USD cents, inverted-ledger; see `grok::types`).
#[derive(Debug, Clone, PartialEq)]
pub struct GrokSnapshot {
    pub balance: f64,
}

impl Eq for GrokSnapshot {}

/// SuperGrok subscription usage from Grok Build's billing endpoint (ACP as
/// fallback) plus banked remaining-resets. Distinct from [`GrokSnapshot`]
/// (Management API prepaid balance).
#[derive(Debug, Clone, PartialEq)]
pub struct SuperGrokSnapshot {
    /// Subscription tier label when the billing response supplies one
    /// (e.g. "SuperGrok", "SuperGrok Heavy"); otherwise `"SuperGrok"`.
    pub plan: String,
    /// Opaque digest of Grok auth/config state. Never displayed — cache
    /// isolation only.
    pub account: String,
    /// Current included-credit usage percent. The field name is retained as a
    /// compatibility alias for format/render code; [`Self::period`] says
    /// whether the server's actual window is weekly or monthly.
    pub weekly_pct: i32,
    pub period: SuperGrokPeriod,
    /// When the current usage period ends.
    pub reset_at: Option<DateTime<Utc>>,
    /// Remaining prepaid (purchased) API credit in USD, when present.
    pub prepaid_balance: Option<f64>,
    pub reset_credits: ResetCredits,
    /// Per-product slices of the same included-credit pool (`GrokBuild`,
    /// `GrokChat`, `GrokImagine`, …). Empty when the billing document omits
    /// `productUsage`.
    pub products: Vec<SuperGrokProduct>,
    /// On-demand spend beyond the included pool, in USD cents. From the Grok
    /// CLI billing document's `onDemandUsed`. `None` when the field is absent.
    pub on_demand_used_cents: Option<i64>,
    /// On-demand spending cap for that same pool, in USD cents
    /// (`onDemandCap`). `None` when the field is absent.
    pub on_demand_cap_cents: Option<i64>,
}

impl SuperGrokSnapshot {
    /// Additional usage beyond the included pool, as dollars.
    ///
    /// A cap renders as `$used / $cap`. Spend with no cap renders as the
    /// amount alone. Both absent or zero is the proto default — nothing was
    /// configured — so the line stays hidden.
    pub fn additional_usage_text(&self) -> Option<String> {
        let used = self.on_demand_used_cents.unwrap_or(0);
        let cap = self.on_demand_cap_cents;
        if used == 0 && cap.unwrap_or(0) == 0 {
            return None;
        }
        let used_s = crate::format::usd(used as f64 / 100.0);
        match cap {
            Some(cap) if cap > 0 => Some(format!(
                "{used_s} / {}",
                crate::format::usd(cap as f64 / 100.0)
            )),
            _ => Some(used_s),
        }
    }
}

/// One SuperGrok product's share of the current included-credit window.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct SuperGrokProduct {
    pub label: String,
    pub percent: i32,
}

impl Eq for SuperGrokSnapshot {}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum SuperGrokPeriod {
    Weekly,
    Monthly,
    Unknown,
}

impl SuperGrokPeriod {
    pub fn label(self) -> &'static str {
        match self {
            Self::Weekly => "Weekly",
            Self::Monthly => "Monthly",
            Self::Unknown => "Current period",
        }
    }

    pub fn short(self) -> &'static str {
        match self {
            Self::Weekly => "wk",
            Self::Monthly => "mo",
            Self::Unknown => "period",
        }
    }
}

/// Grok Bot desktop app — the weekly included-usage pool from
/// `aiserver.v1.DashboardService/GetSandUsageStatus` (Connect-RPC), read with
/// the app's own OAuth session. Distinct from [`GrokSnapshot`] (Management
/// API prepaid dollars) and [`SuperGrokSnapshot`] (Grok Build subscription).
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct GrokbotSnapshot {
    /// `grokPlanLabel`, falling back to `cursorPlanName`, then "Grok Bot".
    pub plan: String,
    /// The subscription that bills the pool, from `billingBrand` and the plan
    /// reported for it ("Cursor Ultra"). `None` for a brand not recognized yet,
    /// which is left unnamed rather than guessed.
    pub billed_by: Option<String>,
    /// `hasNonZeroIncludedLimit`. When false the account carries no included
    /// allowance at all — a distinct "no included allowance" state, never a
    /// fabricated 0% meter.
    pub has_included_allowance: bool,
    /// `usagePercent` of the included pool (0..=100). Meaningful only when
    /// `has_included_allowance` is set.
    pub weekly_pct: i32,
    /// `hasAvailableUsage` — the account can still serve requests, which at
    /// 100% of the included pool means on-demand is picking up the rest.
    pub has_available_usage: bool,
    /// `onDemandSettings.enabled` — pay-as-you-go past the included pool.
    pub on_demand_enabled: bool,
    /// `currentPeriodStart`.
    pub period_start: Option<DateTime<Utc>>,
    /// `nextResetTimestampUtc`.
    pub reset_at: Option<DateTime<Utc>>,
    /// `reset_at − period_start` when both are reported (7 days on the
    /// captured account) — computed, never assumed.
    pub window: Option<chrono::Duration>,
}

impl GrokbotSnapshot {
    /// The plan a frontend shows: the subscription that bills the pool
    /// ("Cursor Ultra") over the app's own label, which reads "Grok Bot Plan"
    /// on every account.
    pub fn display_plan(&self) -> &str {
        self.billed_by.as_deref().unwrap_or(&self.plan)
    }

    /// At 100% of the included pool, `hasAvailableUsage` can still be true
    /// because on-demand keeps serving — say so, but only when the account
    /// actually has on-demand switched on.
    pub fn on_demand_note(&self) -> Option<&'static str> {
        (self.has_included_allowance
            && self.weekly_pct >= 100
            && self.has_available_usage
            && self.on_demand_enabled)
            .then_some("included pool exhausted — on-demand may still be serving usage")
    }
}

/// OpenAI Codex OAuth — exposes whichever rolling windows the API reports.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OpenAiSnapshot {
    pub plan: String,
    /// 5h window, identified by its duration rather than its wire position.
    pub session: Option<UsageWindow>,
    /// 7d window, identified by its duration rather than its wire position.
    pub weekly: Option<UsageWindow>,
    /// Optional 7d code-review bucket.
    pub code_review: Option<UsageWindow>,
    /// Named limits beside the main one, each with its own windows. Empty for
    /// an account that has none.
    pub additional_limits: Vec<OpenAiNamedLimit>,
    /// Models the account currently cannot dispatch to, with the time they
    /// return when the API states one. Only unavailable models are kept: a
    /// list of everything that *is* working is noise, and the reason this
    /// exists is to explain a refusal no percentage accounts for.
    pub unavailable_models: Vec<OpenAiUnavailableModel>,
    /// Optional credit balance + approximate message-count ranges.
    pub credits: Option<OpenAiCredits>,
    /// `None` when the usage response omits the block (a Codex build that
    /// predates banked resets); `Some` with `available == 0` when it states
    /// that the account holds none, which the report shows as such.
    pub reset_credits: Option<ResetCredits>,
    /// Source of the snapshot — Codex OAuth vs admin-key fallback. Drives
    /// the placeholder set and the "OpenAI does not expose this for Plus"
    /// tooltip when the OAuth path isn't available.
    pub source: OpenAiSource,
}

/// A named limit that sits beside Codex's main window — a reserved pool or a
/// model-specific allowance. It can be exhausted while the headline window is
/// nearly untouched, which is the case it exists to make visible.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OpenAiNamedLimit {
    /// The API's own name for it, shown as given.
    pub name: String,
    pub session: Option<UsageWindow>,
    pub weekly: Option<UsageWindow>,
}

/// A model the account cannot currently dispatch to.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OpenAiUnavailableModel {
    pub model: String,
    /// When the API says it returns. `None` means it did not say.
    pub available_at: Option<DateTime<Utc>>,
}

/// Banked, user-redeemable quota resets — Codex's "rate limit reset credits"
/// and SuperGrok's "remaining resets" are the same idea under two names: a
/// count you have earned, each with its own expiry, redeemed by hand rather
/// than arriving on the window's own schedule. Distinct from a
/// [`UsageWindow::resets_at`], which needs no action and cannot be banked.
///
/// The redemption identifier each provider returns alongside these
/// (`credits[].id`, `tokens[].token_id`) is deliberately *not* carried here:
/// it is the handle that spends the credit, and nothing that renders a status
/// bar needs it.
#[derive(Debug, Clone, Default, PartialEq, Eq, Serialize, Deserialize)]
pub struct ResetCredits {
    pub available: u32,
    /// One row per credit the provider described. May be shorter than
    /// `available` — Codex's usage endpoint gives the count without the
    /// per-credit detail, and the detail call is allowed to fail on its own.
    #[serde(default)]
    pub credits: Vec<ResetCredit>,
}

#[derive(Debug, Clone, Default, PartialEq, Eq, Serialize, Deserialize)]
pub struct ResetCredit {
    /// Provider label when one exists ("Full reset (Weekly + 5 hr)"). SuperGrok
    /// tokens have no title.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub title: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub expires_at: Option<DateTime<Utc>>,
}

/// A provider's own label for a banked reset, rendered verbatim in Pango bar
/// markup and in the `;;`-delimited desktop FORMAT protocol. Both vendors that
/// carry one gate it here rather than each keeping a copy: an over-long or
/// control-character-bearing title is dropped, leaving the expiry line alone,
/// which still says everything the user has to act on.
pub fn checked_reset_title(value: Option<String>) -> Option<String> {
    const MAX_RESET_TITLE_CHARS: usize = 80;
    let value = value
        .map(|s| s.trim().to_string())
        .filter(|s| !s.is_empty())?;
    if value.chars().count() > MAX_RESET_TITLE_CHARS || value.chars().any(char::is_control) {
        None
    } else {
        Some(value)
    }
}

impl ResetCredits {
    pub fn is_empty(&self) -> bool {
        self.available == 0
    }

    pub fn next_expiry(&self) -> Option<DateTime<Utc>> {
        self.credits
            .iter()
            .filter_map(|credit| credit.expires_at)
            .min()
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum OpenAiSource {
    CodexOauth,
    AdminKeyMtd,
    Unavailable,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OpenAiCredits {
    /// Credit balance as a bare count of Codex credits ("62500", "12.5"),
    /// not dollars. A value the vendor sends that is not a number is kept as
    /// it came, so this stays a string.
    pub balance: String,
    pub has_credits: bool,
    pub unlimited: bool,
    /// OpenAI's estimate of how many messages the balance buys. Only the
    /// `{oai_local_msgs}`/`{oai_cloud_msgs}` placeholders carry them now.
    pub approx_local_messages: Option<(i64, i64)>,
    pub approx_cloud_messages: Option<(i64, i64)>,
}

/// Z.AI / BigModel — list of buckets with discriminated types. We project the
/// two we care about into named fields (5h tokens, weekly tokens, MCP).
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ZaiSnapshot {
    pub plan: String,
    pub session: Option<UsageWindow>,
    pub weekly: Option<UsageWindow>,
    pub mcp: Option<UsageWindow>,
}

/// Ollama Cloud — credit balances or legacy quota windows from
/// `ollama.com/api/balance`. Older cached `/api/usage` payloads can also
/// carry per-model counts and activity cost.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OllamaSnapshot {
    /// Display label, taken from the `[ollama] plan` config field. The API
    /// itself does not report a plan name.
    pub plan: String,
    /// 5h rolling window (`limits.session`). `None` when the account has not
    /// touched the cloud tier yet (the window is omitted from the response,
    /// not reported as zero).
    pub session: Option<UsageWindow>,
    /// 7d rolling window (`limits.weekly`).
    pub weekly: Option<UsageWindow>,
    /// Calendar-month window (`limits.monthly`). Some Pro accounts report
    /// this in place of `session`/`weekly` instead of alongside them.
    pub monthly: Option<UsageWindow>,
    /// Per-model request counts inside the session window, in the order the
    /// API returned them. Renderers sort and truncate this for the tooltip.
    pub session_models: Vec<OllamaModelUsage>,
    /// Per-model request counts inside the weekly window.
    pub weekly_models: Vec<OllamaModelUsage>,
    /// Per-model request counts inside the monthly window.
    pub monthly_models: Vec<OllamaModelUsage>,
    /// `activity.cost` as a pre-formatted dollar string (`"0.00000"`,
    /// `"1.23456"`). Already a string on the wire — the renderer decides
    /// whether to keep it verbatim or reformat.
    pub activity_cost: Option<String>,
    /// `activity.period.type` (`"last_4_weeks"` and friends). A short
    /// human-readable label the renderer can show next to the cost.
    pub activity_period: Option<String>,
    /// Monthly included credits and optional purchased credits, formatted
    /// through the shared money formatter for all desktop surfaces.
    pub credits: Option<OllamaCredits>,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OllamaCredits {
    pub balance: String,
    pub allowance: String,
    pub purchased: Option<String>,
}

/// One row of `OllamaSnapshot::{session,weekly}_models`. The API carries the
/// per-model request count; the percentage of the window that this single
/// model represents is not reported, so the renderer derives it locally.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OllamaModelUsage {
    pub name: String,
    pub request_count: u64,
}

/// One DeepRelay plan meter: weighted tokens used against the allowance.
///
/// DeepRelay multiplies served tokens by each model's usage factor before they
/// are metered, so every figure here is already in allowance units and is
/// comparable with its own limit and with the other three meters.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct DeepRelayMeter {
    pub used: i64,
    pub limit: i64,
}

impl DeepRelayMeter {
    /// Used share of the allowance. `None` only for a zero ceiling, which is
    /// "uncapped", not "0% used".
    pub fn pct(&self) -> Option<i32> {
        if self.limit <= 0 {
            return None;
        }
        let pct = (self.used as f64 / self.limit as f64 * 100.0).round();
        Some(pct.clamp(0.0, 100.0) as i32)
    }

    /// `"812.4M / 2.5B"` — the raw figure next to its ceiling.
    pub fn used_of_limit(&self) -> String {
        format!(
            "{} / {}",
            crate::format::tokens(self.used),
            crate::format::tokens(self.limit)
        )
    }
}

/// One row of [`DeepRelaySnapshot::models`].
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct DeepRelayModelUsage {
    pub model: String,
    pub weighted_input_tokens: i64,
    pub weighted_output_tokens: i64,
}

impl DeepRelayModelUsage {
    pub fn weighted_total(&self) -> i64 {
        self.weighted_input_tokens
            .saturating_add(self.weighted_output_tokens)
    }
}

/// DeepRelay — a monthly plan with four weighted-token meters, a dollar usage
/// ceiling, and an optional pay-as-you-go wallet.
///
/// The monthly meters reset at the end of the billing period. The weekly caps
/// **roll** (seven days, no reset instant), so a `None` reset there is the
/// vendor's answer, not missing data.
#[derive(Debug, Clone, PartialEq)]
pub struct DeepRelaySnapshot {
    /// `plan.name`, or "Pay as you go" when the organization has no plan.
    pub plan: String,
    pub plan_price_cents: Option<i64>,
    /// `"month"`, `"3 months"`, … — already pluralized from `interval_count`.
    pub plan_interval: Option<String>,
    pub subscribed: bool,
    pub cancel_at_period_end: bool,
    pub monthly_input: Option<DeepRelayMeter>,
    pub monthly_output: Option<DeepRelayMeter>,
    pub weekly_input: Option<DeepRelayMeter>,
    pub weekly_output: Option<DeepRelayMeter>,
    /// Plan spend against the per-period usage ceiling, in dollars.
    pub backstop_used_usd: Option<f64>,
    pub backstop_limit_usd: Option<f64>,
    /// `(used, included)` web searches this period.
    pub searches: Option<(u64, u64)>,
    /// Calls billed to the wallet in the last 24 h while subscribed.
    pub payg_calls_24h: Option<u64>,
    /// Wallet balance in dollars. Absent unless the balance request was made
    /// and answered.
    pub balance_usd: Option<f64>,
    pub period_start: Option<chrono::DateTime<chrono::Utc>>,
    pub period_end: Option<chrono::DateTime<chrono::Utc>>,
    pub models: Vec<DeepRelayModelUsage>,
}

impl DeepRelaySnapshot {
    /// The monthly meter closest to its ceiling — the one that will cap the
    /// account. Drives the bar headline, its severity class and the TUI
    /// overview cell.
    pub fn monthly_worst(&self) -> Option<&DeepRelayMeter> {
        worst(&self.monthly_input, &self.monthly_output)
    }

    /// The rolling-weekly meter closest to its ceiling.
    pub fn weekly_worst(&self) -> Option<&DeepRelayMeter> {
        worst(&self.weekly_input, &self.weekly_output)
    }

    /// Monthly allowance as the shared window shape, for `--format` pacing.
    ///
    /// Both period ends come from the server, so the length is exact rather
    /// than a guessed "30 days" — but the report still exposes no `window_secs`
    /// for these rows (they go through `push_metric`), because a frontend reads
    /// that field as a rolling window. Pacing here is local to a placeholder.
    pub fn monthly_window(&self) -> Option<UsageWindow> {
        let meter = self.monthly_worst()?;
        Some(UsageWindow {
            utilization_pct: meter.pct()?,
            resets_at: self.period_end,
            window_duration: match (self.period_start, self.period_end) {
                (Some(start), Some(end)) if end > start => end - start,
                _ => chrono::Duration::zero(),
            },
        })
    }

    /// Rolling weekly caps as the shared window shape. The window rolls, so it
    /// carries no reset instant; pacing stays neutral rather than guessing.
    pub fn weekly_window(&self) -> Option<UsageWindow> {
        let meter = self.weekly_worst()?;
        Some(UsageWindow {
            utilization_pct: meter.pct()?,
            resets_at: None,
            window_duration: chrono::Duration::days(7),
        })
    }
}

// The dollar figures are finite by construction (they are integer cents and
// micro-cents divided by a constant), so the snapshot is total in the sense
// `Eq` asks for — the same reasoning as `OpenRouterSnapshot`.
impl Eq for DeepRelaySnapshot {}

fn worst<'a>(
    left: &'a Option<DeepRelayMeter>,
    right: &'a Option<DeepRelayMeter>,
) -> Option<&'a DeepRelayMeter> {
    [left.as_ref(), right.as_ref()]
        .into_iter()
        .flatten()
        .max_by_key(|meter| meter.pct().unwrap_or(0))
}

/// Mistral — the two monthly budgets the plan shares across Studio and Vibe
/// Code. Both reset on the calendar-month boundary the billing response
/// reports; the exact cycle length is not stated, so `window_duration` is zero
/// and the renderers do not pace against a guessed month.
#[derive(Debug, Clone, PartialEq)]
pub struct MistralSnapshot {
    /// `plan_name` from `/v1/billing/subscription`; `None` when that optional
    /// read failed.
    pub plan: Option<String>,
    /// Studio / API usage allowance. `None` when the plan does not expose the
    /// API budget — the official `mistral-vibe` client only requires
    /// `vibe_budget`, and some accounts report no `api_budget` at all.
    pub api: Option<UsageWindow>,
    /// Vibe Code usage allowance.
    pub vibe: UsageWindow,
    /// Monthly API allowance in `currency`; `None` when [`Self::api`] is.
    pub api_budget: Option<f64>,
    /// Monthly Vibe Code allowance in `currency`.
    pub vibe_budget: f64,
    /// ISO 4217 code the two allowances are denominated in.
    pub currency: String,
    /// Pay-as-you-go covers API usage past the included allowance.
    pub api_payg: bool,
    /// Pay-as-you-go covers Vibe Code usage past the included allowance.
    pub vibe_payg: bool,
}

// The allowances are validated finite and non-negative at the parse boundary,
// so the snapshot is total in the sense `Eq` asks for — the same reasoning as
// `DeepRelaySnapshot` and `OpenRouterSnapshot`.
impl Eq for MistralSnapshot {}

/// OpenRouter — credit balance + lifetime/daily/weekly/monthly usage from
/// `/api/v1/credits` and `/api/v1/key`.
#[derive(Debug, Clone, PartialEq)]
pub struct OpenRouterSnapshot {
    pub label: String,
    pub total_credits: f64,
    pub total_usage: f64,
    pub usage_daily: f64,
    pub usage_weekly: f64,
    pub usage_monthly: f64,
    pub is_free_tier: bool,
    pub limit: Option<f64>,
    pub limit_remaining: Option<f64>,
    pub recent_models: Vec<String>,
}

impl Eq for OpenRouterSnapshot {}

impl OpenRouterSnapshot {
    /// Spendable credit, **which can be negative**: OpenRouter lets an account
    /// run into debt, and clamping that to zero would report a healthy-looking
    /// `$0.00` to someone who has to top up before anything works again. The
    /// wire fields are each non-negative (see `openrouter::types`), so a
    /// negative result only ever means usage has overrun credits.
    pub fn balance(&self) -> f64 {
        self.total_credits - self.total_usage
    }
    /// Percentage of total_credits consumed (0..=100). Returns 0 when
    /// `total_credits` is 0 (free-tier-only accounts) — there is no
    /// denominator to be a percentage of. Severity does not come from this
    /// number alone: see [`crate::openrouter::vendor::severity`], which treats
    /// a negative [`Self::balance`] as critical regardless of the percentage.
    pub fn consumed_pct(&self) -> i32 {
        if self.total_credits <= 0.0 {
            return 0;
        }
        i32::from(crate::format::clamp_pct(
            (self.total_usage / self.total_credits) * 100.0,
        ))
    }
}

/// OrcaRouter — prepaid credit card from the one-api compatible dashboard
/// billing endpoints (`/v1/dashboard/billing/usage` + `/subscription`), over an
/// API key. Usage arrives in **US cents** (`total_usage: 275` = $2.75); the
/// subscription's limit fields are USD and mean the *total* credit limit
/// (remaining + used), with `100000000` as the unlimited sentinel.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct OrcaRouterSnapshot {
    /// Cumulative spend, exact US cents (`total_usage`).
    pub spent_cents: i64,
    /// Total credit limit in exact US cents. `None` for unlimited keys (the
    /// `100000000` sentinel) or when the subscription response carried no
    /// limit field at all — either way the card is spend-only.
    pub limit_cents: Option<i64>,
    /// Key expiry (`access_until`, Unix seconds); `None` = no expiry (a wire
    /// `0` means the same thing).
    pub access_until: Option<DateTime<Utc>>,
}

impl OrcaRouterSnapshot {
    pub fn spent_usd(&self) -> f64 {
        self.spent_cents as f64 / 100.0
    }

    pub fn limit_usd(&self) -> Option<f64> {
        self.limit_cents.map(|c| c as f64 / 100.0)
    }

    /// Remaining credit in exact cents. Can be negative (spend past the
    /// limit) — the sign belongs outside the symbol, like OpenRouter debt.
    pub fn remaining_cents(&self) -> Option<i64> {
        self.limit_cents.map(|limit| limit - self.spent_cents)
    }

    pub fn remaining_usd(&self) -> Option<f64> {
        self.remaining_cents().map(|c| c as f64 / 100.0)
    }

    /// Integer-percentage of the limit consumed, computed in cents so no
    /// float division is involved. `None` when there is no limit — an
    /// unlimited key has no percentage to be exact *about*.
    pub fn consumed_pct(&self) -> Option<i32> {
        self.limit_cents.filter(|l| *l > 0).map(|limit| {
            let pct = (self.spent_cents.saturating_mul(100)) / limit;
            pct.clamp(0, 100) as i32
        })
    }
}

/// Alibaba Cloud Model Studio Token Plan — a 5-hour, a weekly and a monthly
/// ratio window, any of which the console account may not report. An absent
/// window is no-data (possibly unlimited), never 0%.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ModelStudioSnapshot {
    /// 5-hour window. `None` when `per5HourPercentage` was absent.
    pub session: Option<UsageWindow>,
    /// Weekly window. `None` when `per1WeekPercentage` was absent.
    pub weekly: Option<UsageWindow>,
    /// Monthly window. `None` when `per1MonthPercentage` was absent. The
    /// reset is exact; the cycle length is plan-dependent, so no exact
    /// window duration is declared for it (same rule as Ollama's calendar
    /// month) and pacing is omitted wherever it would guess a denominator.
    pub monthly: Option<UsageWindow>,
}

/// Muse Code subscription windows. Missing observations stay absent.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct MuseSnapshot {
    pub plan: String,
    pub session: Option<UsageWindow>,
    pub weekly: Option<UsageWindow>,
}

/// Devin CLI quota snapshot. The CLI reports remaining percentages, while
/// renderers and report consumers use consumed percentages consistently.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct DevinSnapshot {
    pub daily: Option<UsageWindow>,
    pub weekly: Option<UsageWindow>,
    /// Exact value from the CLI's overageBalanceMicros field. Its USD meaning
    /// is empirically observed for the tested account, not a universal API
    /// contract; only renderers perform that display conversion.
    pub overage_balance_micros: Option<i64>,
}

/// Factory `droid` CLI subscription quota. The limits endpoint reports a
/// `usedPercent` per window in two buckets — Standard and Droid Core — each
/// with a 5-hour, a weekly and a monthly window. A window the account has not
/// started still reports a percentage with `windowEnd: null`, so it is present
/// with no reset; an absent window stays absent.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct DroidSnapshot {
    /// Standard quota, 5-hour window.
    pub session: Option<UsageWindow>,
    /// Standard quota, weekly (7d) window.
    pub weekly: Option<UsageWindow>,
    /// Standard quota, monthly (30d) window.
    pub monthly: Option<UsageWindow>,
    /// Droid Core quota, 5-hour window.
    pub core_session: Option<UsageWindow>,
    /// Droid Core quota, weekly window.
    pub core_weekly: Option<UsageWindow>,
    /// Droid Core quota, monthly window.
    pub core_monthly: Option<UsageWindow>,
    /// `extraUsageBalanceCents`, verbatim; renderers convert cents to dollars.
    pub extra_balance_cents: Option<i64>,
}

/// Worst-of severity class for the Waybar bar text color. Mirrors
/// claudebar:606-620 — "extra usage only matters when a rate limit hits 100%".
pub fn anthropic_severity(snap: &AnthropicSnapshot) -> crate::pacing::PaceSeverity {
    let mut max = snap.weekly.utilization_pct;
    if let Some(s) = &snap.session
        && s.utilization_pct > max
    {
        max = s.utilization_pct;
    }
    if let Some(s) = &snap.sonnet
        && s.utilization_pct > max
    {
        max = s.utilization_pct;
    }
    for sw in &snap.scoped {
        if sw.window.utilization_pct > max {
            max = sw.window.utilization_pct;
        }
    }
    // Extra usage only promotes severity if a rate-limit window is at 100%.
    let any_at_cap = snap
        .session
        .as_ref()
        .is_some_and(|s| s.utilization_pct >= 100)
        || snap.weekly.utilization_pct >= 100
        || snap
            .sonnet
            .as_ref()
            .is_some_and(|s| s.utilization_pct >= 100)
        || snap.scoped.iter().any(|s| s.window.utilization_pct >= 100);
    if any_at_cap && let Some(extra) = snap.extra.as_ref() {
        let p = extra.percent();
        if p > max {
            max = p;
        }
    }
    crate::pango::severity_for(max)
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::pacing::PaceSeverity;
    use chrono::Duration;

    fn w(pct: i32) -> UsageWindow {
        UsageWindow {
            utilization_pct: pct,
            resets_at: None,
            window_duration: Duration::hours(5),
        }
    }

    fn snap(s: i32, w_: i32, sonnet: Option<i32>, extra: Option<(i64, i64)>) -> AnthropicSnapshot {
        AnthropicSnapshot {
            plan: "Max 5x".into(),
            session: Some(w(s)),
            weekly: w(w_),
            sonnet: sonnet.map(w),
            scoped: vec![],
            extra: extra.map(|(limit, spent)| ExtraUsage {
                limit: Some(Cents(limit)),
                spent: Cents(spent),
                currency: None,
                decimal_places: Some(2),
                status: ExtraUsageStatus::Active,
            }),
            reset_credits: Default::default(),
            cloud_credits: None,
            usage_credits: None,
        }
    }

    #[test]
    fn fmt_minor_honors_currency_and_scale() {
        // No currency (older payloads) keeps the historical `$`.
        assert_eq!(fmt_minor(250, 2, None), "$2.50");
        // The #30 reporter's actual figures: BRL must not be claimed as `$`.
        assert_eq!(fmt_minor(14157, 2, Some("BRL")), "R$141.57");
        assert_eq!(fmt_minor(14157, 2, Some("USD")), "$141.57");
        // Zero-exponent currency: no decimal point, no /100.
        assert_eq!(fmt_minor(500, 0, Some("JPY")), "¥500");
        // Sign precedes the symbol, as `format::money` spells it.
        assert_eq!(fmt_minor(-150, 2, Some("BRL")), "-R$1.50");
        // Unknown code stays truthful as a suffix rather than guessing a symbol.
        assert_eq!(fmt_minor(1234, 2, Some("CHF")), "12.34 CHF");
    }

    #[test]
    fn extra_usage_formats_in_its_own_currency() {
        let e = ExtraUsage {
            limit: None,
            spent: Cents(14157),
            currency: Some("BRL".into()),
            decimal_places: Some(2),
            status: ExtraUsageStatus::Active,
        };
        assert_eq!(e.fmt_spent(), "R$141.57");
        assert_eq!(e.fmt_limit(), None);

        let capped = ExtraUsage {
            limit: Some(Cents(5000)),
            spent: Cents(250),
            currency: None,
            decimal_places: Some(2),
            status: ExtraUsageStatus::Active,
        };
        assert_eq!(capped.fmt_spent(), "$2.50");
        assert_eq!(capped.fmt_limit().as_deref(), Some("$50.00"));
    }

    #[test]
    fn extra_percent_with_zero_limit_is_zero() {
        assert_eq!(
            ExtraUsage {
                limit: Some(Cents(0)),
                spent: Cents(100),
                currency: None,
                decimal_places: Some(2),
                status: ExtraUsageStatus::Active,
            }
            .percent(),
            0
        );
    }

    #[test]
    fn extra_percent_truncates() {
        // Bash integer division — 33/100 -> 33%, 50/100 -> 50%.
        assert_eq!(
            ExtraUsage {
                limit: Some(Cents(10000)),
                spent: Cents(3333),
                currency: None,
                decimal_places: Some(2),
                status: ExtraUsageStatus::Active,
            }
            .percent(),
            33
        );
    }

    #[test]
    fn severity_picks_worst_of_three_windows() {
        let s = snap(40, 60, Some(80), None);
        assert_eq!(anthropic_severity(&s), PaceSeverity::High); // 80 → high
    }

    #[test]
    fn severity_ignores_extra_when_no_cap_hit() {
        // Extra at 95% but no rate-limit at 100% → extra is NOT promoted.
        let s = snap(50, 60, None, Some((10000, 9500)));
        assert_eq!(anthropic_severity(&s), PaceSeverity::Mid); // capped at 60
    }

    #[test]
    fn severity_promotes_extra_when_session_at_100() {
        let s = snap(100, 50, None, Some((10000, 9500)));
        assert_eq!(anthropic_severity(&s), PaceSeverity::Critical); // 100 → critical
    }

    /// A verified Ultra account has no five-hour window: the worst of the
    /// windows it does have decides.
    #[test]
    fn severity_without_a_session_window_uses_the_others() {
        let mut s = snap(0, 60, Some(10), None);
        s.session = None;
        assert_eq!(anthropic_severity(&s), PaceSeverity::Mid); // weekly 60

        let mut s = with_scoped(snap(0, 20, None, None), 84);
        s.session = None;
        assert_eq!(anthropic_severity(&s), PaceSeverity::High); // Fable 84
    }

    #[test]
    fn severity_falls_through_to_extra_when_extra_higher_than_capped_window() {
        // session = 100, weekly = 50, extra = 100% → max should be 100.
        let s = snap(100, 50, None, Some((10000, 10000)));
        assert_eq!(anthropic_severity(&s), PaceSeverity::Critical);
    }

    fn with_scoped(mut s: AnthropicSnapshot, pct: i32) -> AnthropicSnapshot {
        s.scoped.push(ScopedWindow {
            label: "Fable".into(),
            window: w(pct),
        });
        s
    }

    #[test]
    fn severity_includes_scoped_windows() {
        // The PR #19 scenario: overall weekly at 55 (Mid) but a scoped Fable
        // week at 84 → the bar class must escalate to High.
        let s = with_scoped(snap(10, 55, None, None), 84);
        assert_eq!(anthropic_severity(&s), PaceSeverity::High);
    }

    #[test]
    fn severity_promotes_extra_when_scoped_at_100() {
        // A scoped window at cap counts as a rate-limit cap hit, so extra
        // usage above the window max is promoted — same rule as session/weekly.
        let s = with_scoped(snap(10, 50, None, Some((10000, 9900))), 100);
        assert_eq!(anthropic_severity(&s), PaceSeverity::Critical);
    }

    #[test]
    fn kimi_percent_is_exact_above_f64_precision() {
        let snap = KimiSnapshot {
            plan: None,
            weekly_limit: (1 << 53) + 1,
            weekly_used: 1 << 52,
            weekly_remaining: 0,
            weekly_reset_at: None,
            has_weekly: true,
            monthly_pct: None,
            monthly_reset_at: None,
            window_limit: u64::MAX,
            window_used: u64::MAX - 1,
            window_remaining: 0,
            window_reset_at: None,
        };
        assert_eq!(snap.weekly_pct(), 50);
        assert_eq!(snap.window_pct(), 100);
    }

    #[test]
    fn kiro_pct_is_zero_without_a_positive_limit() {
        let snap = KiroSnapshot {
            plan: "FREE".into(),
            used: 5.0,
            limit: 0.0,
            reset_at: None,
        };
        assert_eq!(snap.pct(), 0);
    }

    #[test]
    fn kiro_pct_rounds_the_credit_ratio() {
        let snap = KiroSnapshot {
            plan: "KIRO POWER".into(),
            used: 1.0,
            limit: 3.0,
            reset_at: None,
        };
        assert_eq!(snap.pct(), 33);
    }
}