1use chrono::{DateTime, Utc};
2use serde::{Deserialize, Serialize};
3
4#[derive(Debug, Deserialize, Serialize)]
5pub struct UsageWindow {
6 pub utilization: f64,
7 pub resets_at: Option<DateTime<Utc>>,
8}
9
10#[derive(Debug, Deserialize, Serialize)]
11pub struct UsageResponse {
12 pub five_hour: Option<UsageWindow>,
13 pub seven_day: Option<UsageWindow>,
14 pub seven_day_sonnet: Option<UsageWindow>,
15}
16
17impl UsageResponse {
18 pub fn next_reset_time(&self) -> Option<chrono::DateTime<Utc>> {
19 [
20 self.five_hour.as_ref(),
21 self.seven_day.as_ref(),
22 self.seven_day_sonnet.as_ref(),
23 ]
24 .into_iter()
25 .flatten()
26 .filter(|w| w.utilization >= 100.0)
27 .filter_map(|w| w.resets_at)
28 .max()
29 }
30}