Populate windowMinutes for Zai, Kimi, Grok, and Antigravity rate windows (#2431)

Sources/CodexBarCore/Providers/Antigravity/AntigravityStatusProbe.swift

@@ -447,9 +447,22 @@ public struct AntigravityStatusSnapshot: Sendable {
     }
 
     private static func rateWindow(for quota: AntigravityModelQuota) -> RateWindow {
-        RateWindow(
+        let windowMinutes: Int? = if let desc = quota.resetDescription?.lowercased() {
+            if desc.contains("weekly") || desc.contains("7-day") || desc.contains("7d") {
+                10080
+            } else if desc.contains("session") || desc.contains("5h") || desc.contains("5-hour") || desc
+                .contains("five hour")
+            {
+                300
+            } else {
+                nil
+            }
+        } else {
+            nil
+        }
+        return RateWindow(
             usedPercent: 100 - quota.remainingPercent,
-            windowMinutes: nil,
+            windowMinutes: windowMinutes,
             resetsAt: quota.resetTime,
             resetDescription: quota.resetDescription)
     }

Sources/CodexBarCore/Providers/Grok/GrokStatusProbe.swift

@@ -44,7 +44,7 @@ public struct GrokUsageSnapshot: Sendable {
         {
             primary = RateWindow(
                 usedPercent: percent,
-                windowMinutes: nil,
+                windowMinutes: ProviderPaceCapability.monthlyWindowSentinelMinutes,
                 resetsAt: webBilling.resetsAt,
                 resetDescription: nil)
         }

Sources/CodexBarCore/Providers/Kimi/KimiUsageSnapshot.swift

@@ -126,7 +126,7 @@ extension KimiUsageSnapshot {
             guard let ratio = balance.amountUsedRatio, ratio.isFinite else { return nil }
             let window = RateWindow(
                 usedPercent: Self.clampedPercent(ratio * 100),
-                windowMinutes: nil, // Calendar-month duration varies; do not fabricate a fixed 30-day window.
+                windowMinutes: ProviderPaceCapability.monthlyWindowSentinelMinutes,
                 resetsAt: Self.parseDate(balance.expireTime),
                 resetDescription: nil)
             return NamedRateWindow(id: "kimi-monthly", title: "Monthly", window: window)

Sources/CodexBarCore/Providers/Zai/ZaiUsageStats.swift

@@ -224,9 +224,16 @@ extension ZaiUsageSnapshot {
     }
 
     private static func rateWindow(for limit: ZaiLimitEntry) -> RateWindow {
-        RateWindow(
+        let windowMinutes: Int? = if limit.type == .timeLimit || limit.isMCPMonthlyMarker {
+            ProviderPaceCapability.monthlyWindowSentinelMinutes
+        } else if let minutes = limit.windowMinutes {
+            minutes
+        } else {
+            nil
+        }
+        return RateWindow(
             usedPercent: limit.usedPercent,
-            windowMinutes: limit.type == .tokensLimit ? limit.windowMinutes : nil,
+            windowMinutes: windowMinutes,
             resetsAt: limit.nextResetTime,
             resetDescription: self.resetDescription(for: limit))
     }

Tests/CodexBarTests/AntigravityWindowMinutesTests.swift

@@ -0,0 +1,32 @@
+import CodexBarCore
+import Foundation
+import Testing
+
+@Suite(.serialized)
+struct AntigravityWindowMinutesTests {
+    @Test
+    func `antigravity model quota derives windowMinutes from reset description`() throws {
+        let snapshot = AntigravityStatusSnapshot(
+            modelQuotas: [
+                AntigravityModelQuota(
+                    label: "Claude 3.5 Sonnet",
+                    modelId: "claude-3-5-sonnet",
+                    remainingFraction: 0.8,
+                    resetTime: Date(),
+                    resetDescription: "5-hour session reset"),
+                AntigravityModelQuota(
+                    label: "Gemini 2.5 Flash",
+                    modelId: "gemini-2-5-flash",
+                    remainingFraction: 0.5,
+                    resetTime: Date(),
+                    resetDescription: "Weekly reset"),
+            ],
+            accountEmail: "test@example.com",
+            accountPlan: "Pro",
+            source: .remote)
+
+        let usage = try snapshot.toUsageSnapshot()
+        #expect(usage.primary?.windowMinutes == 10080)
+        #expect(usage.secondary?.windowMinutes == 300)
+    }
+}

Tests/CodexBarTests/KimiProviderTests.swift

@@ -1023,7 +1023,7 @@ struct KimiUsageSnapshotConversionTests {
         #expect(monthly.id == "kimi-monthly")
         #expect(monthly.title == "Monthly")
         #expect(monthly.window.usedPercent == 100)
-        #expect(monthly.window.windowMinutes == nil)
+        #expect(monthly.window.windowMinutes == ProviderPaceCapability.monthlyWindowSentinelMinutes)
         #expect(monthly.window.resetsAt == Self.date("2026-07-23T00:00:00Z"))
     }
 

Tests/CodexBarTests/ZaiProviderTests.swift

@@ -253,7 +253,7 @@ struct ZaiUsageParsingTests {
         #expect(snapshot.tokenLimit?.percentage == 34.0)
 
         let usage = snapshot.toUsageSnapshot()
-        #expect(usage.secondary?.windowMinutes == nil)
+        #expect(usage.secondary?.windowMinutes == ProviderPaceCapability.monthlyWindowSentinelMinutes)
         #expect(usage.secondary?.resetDescription == "Monthly")
     }
 
@@ -292,7 +292,7 @@ struct ZaiUsageParsingTests {
         let usage = snapshot.toUsageSnapshot()
 
         #expect(snapshot.timeLimit?.windowDescription == "1 minute")
-        #expect(usage.secondary?.windowMinutes == nil)
+        #expect(usage.secondary?.windowMinutes == ProviderPaceCapability.monthlyWindowSentinelMinutes)
         #expect(usage.secondary?.resetDescription == "Monthly")
     }