use serde::{Deserialize, Serialize};
use crate::usage::{AnthropicSnapshot, Cents, ExtraUsage, ScopedWindow, UsageWindow};
#[derive(Debug, Default, Clone, Deserialize, Serialize, PartialEq)]
pub struct UsageResponse {
#[serde(default)]
pub five_hour: Option<Window>,
#[serde(default)]
pub seven_day: Option<Window>,
#[serde(default)]
pub seven_day_sonnet: Option<Window>,
#[serde(default)]
pub extra_usage: Option<ExtraUsageBlock>,
#[serde(default)]
pub limits: Vec<LimitEntry>,
}
#[derive(Debug, Default, Clone, Deserialize, Serialize, PartialEq)]
pub struct LimitEntry {
#[serde(default)]
pub kind: Option<String>,
#[serde(default, deserialize_with = "de_percent_opt")]
pub percent: Option<f64>,
#[serde(default)]
pub resets_at: Option<String>,
#[serde(default)]
pub scope: Option<LimitScope>,
}
#[derive(Debug, Default, Clone, Deserialize, Serialize, PartialEq)]
pub struct LimitScope {
#[serde(default)]
pub model: Option<LimitModel>,
}
#[derive(Debug, Default, Clone, Deserialize, Serialize, PartialEq)]
pub struct LimitModel {
#[serde(default)]
pub display_name: Option<String>,
}
#[derive(Debug, Default, Clone, Deserialize, Serialize, PartialEq)]
pub struct Window {
#[serde(default, deserialize_with = "de_percent")]
pub utilization: f64,
#[serde(default)]
pub resets_at: Option<String>,
}
#[derive(Debug, Default, Clone, Deserialize, Serialize, PartialEq)]
pub struct ExtraUsageBlock {
#[serde(default)]
pub is_enabled: bool,
#[serde(default, deserialize_with = "de_opt_cents")]
pub monthly_limit: Option<i64>,
#[serde(default, deserialize_with = "de_opt_cents")]
pub used_credits: Option<i64>,
#[serde(default, deserialize_with = "de_opt_currency")]
pub currency: Option<String>,
#[serde(default, deserialize_with = "de_opt_decimal_places")]
pub decimal_places: Option<u32>,
}
fn de_opt_decimal_places<'de, D>(d: D) -> std::result::Result<Option<u32>, D::Error>
where
D: serde::Deserializer<'de>,
{
let n = match serde_json::Value::deserialize(d)? {
serde_json::Value::Null => return Ok(None),
serde_json::Value::Number(n) => n
.as_i64()
.or_else(|| n.as_f64().filter(|f| f.fract() == 0.0).map(|f| f as i64)),
_ => None,
};
match n {
Some(n) if (0..=6).contains(&n) => Ok(Some(n as u32)),
_ => Err(serde::de::Error::custom(
"decimal_places must be an integer in 0..=6",
)),
}
}
fn de_opt_currency<'de, D>(d: D) -> std::result::Result<Option<String>, D::Error>
where
D: serde::Deserializer<'de>,
{
match Option::<String>::deserialize(d)? {
None => Ok(None),
Some(s) if s.len() == 3 && s.bytes().all(|b| b.is_ascii_uppercase()) => Ok(Some(s)),
Some(s) => Err(serde::de::Error::custom(format!(
"currency {s:?} is not an ISO 4217 alpha code"
))),
}
}
fn de_opt_cents<'de, D>(d: D) -> std::result::Result<Option<i64>, D::Error>
where
D: serde::Deserializer<'de>,
{
let v = serde_json::Value::deserialize(d)?;
match v {
serde_json::Value::Null => Ok(None),
serde_json::Value::Number(n) => {
if let Some(i) = n.as_i64() {
if i >= 0 {
Ok(Some(i))
} else {
Err(serde::de::Error::custom("cents cannot be negative"))
}
} else if let Some(f) = n.as_f64() {
const MAX_EXACT_F64_INT: f64 = (1_u64 << 53) as f64;
if f.is_finite() && f.fract() == 0.0 && (0.0..=MAX_EXACT_F64_INT).contains(&f) {
Ok(Some(f as i64))
} else {
Err(serde::de::Error::custom(
"cents must be a non-negative integer in range",
))
}
} else {
Err(serde::de::Error::custom("number out of i64 range"))
}
}
other => Err(serde::de::Error::custom(format!(
"expected number or null, got {other:?}"
))),
}
}
const PCT_SLACK: f64 = 1.0;
fn checked_percent<E: serde::de::Error>(v: f64) -> std::result::Result<f64, E> {
if !v.is_finite() {
return Err(E::custom(format!("percentage {v} is not finite")));
}
if !(0.0..=100.0 + PCT_SLACK).contains(&v) {
return Err(E::custom(format!("percentage {v} outside 0..=100")));
}
Ok(v)
}
fn de_percent<'de, D>(d: D) -> std::result::Result<f64, D::Error>
where
D: serde::Deserializer<'de>,
{
checked_percent(f64::deserialize(d)?)
}
fn de_percent_opt<'de, D>(d: D) -> std::result::Result<Option<f64>, D::Error>
where
D: serde::Deserializer<'de>,
{
Option::<f64>::deserialize(d)?
.map(checked_percent)
.transpose()
}
impl UsageResponse {
pub fn into_snapshot(self, plan_label: String) -> AnthropicSnapshot {
const SESSION: chrono::Duration = chrono::Duration::hours(5);
const WEEKLY: chrono::Duration = chrono::Duration::days(7);
fn to_window(w: Option<Window>, dur: chrono::Duration) -> UsageWindow {
let Some(w) = w else {
return UsageWindow {
utilization_pct: 0,
resets_at: None,
window_duration: dur,
};
};
UsageWindow {
utilization_pct: (w.utilization.round() as i32).clamp(0, 100),
resets_at: w
.resets_at
.as_deref()
.and_then(|s| chrono::DateTime::parse_from_rfc3339(s).ok())
.map(|dt| dt.with_timezone(&chrono::Utc)),
window_duration: dur,
}
}
let session = to_window(self.five_hour, SESSION);
let weekly = to_window(self.seven_day, WEEKLY);
let sonnet = self.seven_day_sonnet.map(|w| to_window(Some(w), WEEKLY));
let extra = self.extra_usage.filter(|e| e.is_enabled).and_then(|e| {
Some(ExtraUsage {
limit: e.monthly_limit.map(Cents),
spent: Cents(e.used_credits?),
decimal_places: e.decimal_places,
currency: e.currency,
})
});
let scoped = self
.limits
.into_iter()
.filter(|l| l.kind.as_deref() == Some("weekly_scoped"))
.filter_map(|l| {
let label = l.scope?.model?.display_name?;
let utilization = l.percent?;
let window = to_window(
Some(Window {
utilization,
resets_at: l.resets_at,
}),
WEEKLY,
);
Some(ScopedWindow { label, window })
})
.collect();
AnthropicSnapshot {
plan: plan_label,
session,
weekly,
sonnet,
scoped,
extra,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn parses_full_response() {
let raw = r#"{
"five_hour": {"utilization": 42.7, "resets_at": "2026-05-23T17:30:00Z"},
"seven_day": {"utilization": 27.0, "resets_at": "2026-05-30T12:00:00Z"},
"seven_day_sonnet": {"utilization": 4.2, "resets_at": "2026-05-30T12:00:00Z"},
"extra_usage": {"is_enabled": true, "monthly_limit": 5000, "used_credits": 250}
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Max 5x".into());
assert_eq!(snap.session.utilization_pct, 43); assert_eq!(snap.weekly.utilization_pct, 27);
assert_eq!(snap.sonnet.as_ref().unwrap().utilization_pct, 4);
let extra = snap.extra.as_ref().unwrap();
assert_eq!(extra.limit, Some(Cents(5000)));
assert_eq!(extra.spent.0, 250);
assert!(snap.session.resets_at.is_some());
}
#[test]
fn parses_weekly_scoped_limits() {
let raw = r#"{
"five_hour": {"utilization": 10.0, "resets_at": "2026-07-08T22:59:59Z"},
"seven_day": {"utilization": 55.0, "resets_at": "2026-07-10T10:59:59Z"},
"limits": [
{"kind": "session", "group": "session", "percent": 10,
"severity": "normal", "resets_at": "2026-07-08T22:59:59Z",
"scope": null, "is_active": false},
{"kind": "weekly_all", "group": "weekly", "percent": 55,
"severity": "normal", "resets_at": "2026-07-10T10:59:59Z",
"scope": null, "is_active": false},
{"kind": "weekly_scoped", "group": "weekly", "percent": 84,
"severity": "warning", "resets_at": "2026-07-10T10:59:59Z",
"scope": {"model": {"id": null, "display_name": "Fable"}, "surface": null},
"is_active": true}
]
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert_eq!(snap.scoped.len(), 1);
assert_eq!(snap.scoped[0].label, "Fable");
assert_eq!(snap.scoped[0].window.utilization_pct, 84);
assert!(snap.scoped[0].window.resets_at.is_some());
assert_eq!(snap.weekly.utilization_pct, 55);
}
#[test]
fn missing_limits_array_yields_empty_scoped() {
let raw = r#"{
"five_hour": {"utilization": 0, "resets_at": "2026-05-23T17:30:00Z"},
"seven_day": {"utilization": 0, "resets_at": "2026-05-30T12:00:00Z"}
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert!(snap.scoped.is_empty());
}
#[test]
fn missing_sonnet_and_extra_are_none() {
let raw = r#"{
"five_hour": {"utilization": 0, "resets_at": "2026-05-23T17:30:00Z"},
"seven_day": {"utilization": 0, "resets_at": "2026-05-30T12:00:00Z"}
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert!(snap.sonnet.is_none());
assert!(snap.extra.is_none());
}
#[test]
fn disabled_extra_usage_becomes_none() {
let raw = r#"{
"five_hour": {"utilization": 0},
"seven_day": {"utilization": 0},
"extra_usage": {"is_enabled": false, "monthly_limit": 5000, "used_credits": 0}
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert!(snap.extra.is_none());
}
#[test]
fn enabled_extra_usage_without_spend_is_dropped() {
for raw in [
r#"{"extra_usage":{"is_enabled":true,"monthly_limit":5000}}"#,
r#"{"extra_usage":{"is_enabled":true,"monthly_limit":5000,"used_credits":null}}"#,
] {
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
assert!(resp.into_snapshot("Pro".into()).extra.is_none(), "{raw}");
}
}
#[test]
fn uncapped_plan_keeps_real_spend_visible() {
let resp: UsageResponse = serde_json::from_str(
r#"{"extra_usage":{"is_enabled":true,"monthly_limit":null,
"used_credits":14157.0,"currency":"BRL","decimal_places":2,
"disabled_reason":null}}"#,
)
.unwrap();
let extra = resp.into_snapshot("Pro".into()).extra.unwrap();
assert_eq!(extra.limit, None);
assert_eq!(extra.spent.0, 14157);
assert_eq!(extra.percent(), 0);
assert_eq!(extra.currency.as_deref(), Some("BRL"));
assert_eq!(extra.decimal_places, Some(2));
assert_eq!(extra.fmt_spent(), "R$141.57");
let resp: UsageResponse =
serde_json::from_str(r#"{"extra_usage":{"is_enabled":true,"used_credits":250}}"#)
.unwrap();
let extra = resp.into_snapshot("Pro".into()).extra.unwrap();
assert_eq!(extra.limit, None);
assert_eq!(extra.spent.0, 250);
}
#[test]
fn implausible_decimal_places_is_schema_drift() {
for value in ["7", "-1", "100", "2.5"] {
let raw = format!(
r#"{{"extra_usage":{{"is_enabled":true,"used_credits":250,"decimal_places":{value}}}}}"#
);
assert!(
serde_json::from_str::<UsageResponse>(&raw).is_err(),
"{raw}"
);
}
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":250,"decimal_places":2.0}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
assert_eq!(
resp.into_snapshot("Pro".into())
.extra
.unwrap()
.decimal_places,
Some(2)
);
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":250,"decimal_places":null}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let extra = resp.into_snapshot("Pro".into()).extra.unwrap();
assert_eq!(extra.decimal_places, None);
assert_eq!(extra.fmt_spent(), "$2.50");
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":500,"currency":"KRW"}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let extra = resp.into_snapshot("Pro".into()).extra.unwrap();
assert_eq!(extra.decimal_places, None);
assert_eq!(extra.fmt_spent(), "500 minor units KRW");
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":1234,"currency":"KWD"}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let extra = resp.into_snapshot("Pro".into()).extra.unwrap();
assert_eq!(extra.decimal_places, None);
assert_eq!(extra.fmt_spent(), "1234 minor units KWD");
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":500,"currency":"KRW","decimal_places":0}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
assert_eq!(
resp.into_snapshot("Pro".into()).extra.unwrap().fmt_spent(),
"500 KRW"
);
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":1234,"currency":"KWD","decimal_places":3}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
assert_eq!(
resp.into_snapshot("Pro".into()).extra.unwrap().fmt_spent(),
"1.234 KWD"
);
}
#[test]
fn currency_must_be_an_iso_alpha_code() {
for value in [r#""brl""#, r#""""#, r#""R$""#, r#""USD;;0""#, r#""<b>""#] {
let raw = format!(
r#"{{"extra_usage":{{"is_enabled":true,"used_credits":250,"currency":{value}}}}}"#
);
assert!(
serde_json::from_str::<UsageResponse>(&raw).is_err(),
"{raw}"
);
}
let raw = r#"{"extra_usage":{"is_enabled":true,"used_credits":250,"currency":null}}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
assert_eq!(
resp.into_snapshot("Pro".into()).extra.unwrap().currency,
None
);
}
#[test]
fn malformed_cent_values_are_schema_drift() {
for value in ["-1", "1.5", "1e300", "true", r#""lots""#] {
let raw = format!(
r#"{{"extra_usage":{{"is_enabled":true,"monthly_limit":{value},"used_credits":0}}}}"#
);
assert!(
serde_json::from_str::<UsageResponse>(&raw).is_err(),
"{raw}"
);
}
let resp: UsageResponse = serde_json::from_str(
r#"{"extra_usage":{"is_enabled":true,"monthly_limit":5000.0,"used_credits":250.0}}"#,
)
.unwrap();
let extra = resp.into_snapshot("Pro".into()).extra.unwrap();
assert_eq!(extra.limit, Some(Cents(5000)));
assert_eq!(extra.spent.0, 250);
}
#[test]
fn empty_object_yields_neutral_snapshot() {
let resp: UsageResponse = serde_json::from_str("{}").unwrap();
let snap = resp.into_snapshot("Unknown".into());
assert_eq!(snap.session.utilization_pct, 0);
assert_eq!(snap.weekly.utilization_pct, 0);
assert!(snap.session.resets_at.is_none());
}
#[test]
fn benign_overshoot_saturates_to_hundred() {
let raw = r#"{
"five_hour": {"utilization": 100.4},
"seven_day": {"utilization": 100.6}
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert_eq!(snap.session.utilization_pct, 100);
assert_eq!(snap.weekly.utilization_pct, 100); }
#[test]
fn out_of_range_utilization_is_rejected_not_clamped() {
for raw in [
r#"{"five_hour": {"utilization": 500}}"#,
r#"{"five_hour": {"utilization": -1}}"#,
r#"{"seven_day": {"utilization": 101.5}}"#,
] {
let err = serde_json::from_str::<UsageResponse>(raw).unwrap_err();
assert!(err.to_string().contains("outside 0..=100"), "{raw}: {err}");
}
}
#[test]
fn out_of_range_scoped_percent_is_rejected() {
let raw = r#"{
"limits": [{"kind": "weekly_scoped", "percent": 420,
"scope": {"model": {"display_name": "Fable"}}}]
}"#;
let err = serde_json::from_str::<UsageResponse>(raw).unwrap_err();
assert!(err.to_string().contains("outside 0..=100"), "{err}");
}
#[test]
fn non_finite_percentage_is_rejected() {
for v in [f64::NAN, f64::INFINITY, f64::NEG_INFINITY] {
assert!(checked_percent::<serde_json::Error>(v).is_err(), "{v}");
}
}
#[test]
fn scoped_limit_without_a_percentage_is_dropped_not_zeroed() {
let raw = r#"{
"limits": [{"kind": "weekly_scoped", "percent": null,
"scope": {"model": {"display_name": "Fable"}}}]
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert!(
snap.scoped.is_empty(),
"a scoped limit with no percentage must not render a fabricated bar"
);
let ok = r#"{
"limits": [{"kind": "weekly_scoped", "percent": 84,
"scope": {"model": {"display_name": "Fable"}}}]
}"#;
let resp: UsageResponse = serde_json::from_str(ok).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert_eq!(snap.scoped[0].label, "Fable");
assert_eq!(snap.scoped[0].window.utilization_pct, 84);
}
#[test]
fn unparseable_reset_becomes_none() {
let raw = r#"{
"five_hour": {"utilization": 50, "resets_at": "not a date"},
"seven_day": {"utilization": 0}
}"#;
let resp: UsageResponse = serde_json::from_str(raw).unwrap();
let snap = resp.into_snapshot("Pro".into());
assert!(snap.session.resets_at.is_none());
assert_eq!(snap.session.utilization_pct, 50);
}
}