From 57ef0caae26b0c7417ce82b04506596ef737acba Mon Sep 17 00:00:00 2001
From: KooshaPari <42529354+KooshaPari@users.noreply.github.com>
Date: Mon, 22 Jun 2026 13:12:31 -0700
Subject: [PATCH] fix(quota): expose Bailian quota windows (#4610)

Integrated into release/v3.8.34 (rebuilt onto tip)
---
 open-sse/services/bailianQuotaFetcher.ts | 19 ++++++++++++++++++-
 tests/unit/bailian-quota-fetcher.test.ts |  6 ++++++
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/open-sse/services/bailianQuotaFetcher.ts b/open-sse/services/bailianQuotaFetcher.ts
index 3eb0c11847..33a85664fc 100644
--- a/open-sse/services/bailianQuotaFetcher.ts
+++ b/open-sse/services/bailianQuotaFetcher.ts
@@ -19,7 +19,7 @@
  * Registration: call registerBailianCodingPlanQuotaFetcher() once at server startup.
  */
 
-import { registerQuotaFetcher, type QuotaInfo } from "./quotaPreflight.ts";
+import { registerQuotaFetcher, registerQuotaWindows, type QuotaInfo } from "./quotaPreflight.ts";
 import { registerMonitorFetcher } from "./quotaMonitor.ts";
 
 // Bailian quota hosts (international / china fallback)
@@ -34,9 +34,15 @@ const BAILIAN_QUOTA_PATH =
 // Cache TTL — short enough to be reactive, long enough to avoid rate limits
 const CACHE_TTL_MS = 60_000; // 60 seconds
 
+// Window keys as surfaced to the dashboard and quota-window registry
+export const BAILIAN_WINDOW_5H = "window_5h";
+export const BAILIAN_WINDOW_WEEKLY = "window_weekly";
+export const BAILIAN_WINDOW_MONTHLY = "window_monthly";
+
 // Triple-window quota info (richer than QuotaInfo — includes all 3 windows)
 // [Oracle CONDITIONAL] bailian-coding-plan only — do not reuse for other providers
 export interface BailianTripleWindowQuota extends QuotaInfo {
+  windows: Record<string, { percentUsed: number; resetAt: string | null }>;
   window5h: { percentUsed: number; resetAt: string | null };
   windowWeekly: { percentUsed: number; resetAt: string | null };
   windowMonthly: { percentUsed: number; resetAt: string | null };
@@ -181,6 +187,11 @@ function parseBailianQuotaResponse(data: unknown): BailianTripleWindowQuota | nu
     percentUsed: pctMonthly,
     resetAt: resetAtMonthly > 0 ? new Date(resetAtMonthly * 1000).toISOString() : null,
   };
+  const windows = {
+    [BAILIAN_WINDOW_5H]: window5h,
+    [BAILIAN_WINDOW_WEEKLY]: windowWeekly,
+    [BAILIAN_WINDOW_MONTHLY]: windowMonthly,
+  };
 
   // Dominant reset = reset time of the most restrictive window
   const dominantResetAt =
@@ -195,6 +206,7 @@ function parseBailianQuotaResponse(data: unknown): BailianTripleWindowQuota | nu
     total: 100,
     percentUsed: worstPercentUsed,
     resetAt: dominantResetAt,
+    windows,
     window5h,
     windowWeekly,
     windowMonthly,
@@ -313,4 +325,9 @@ export function invalidateBailianQuotaCache(connectionId: string): void {
 export function registerBailianCodingPlanQuotaFetcher(): void {
   registerQuotaFetcher("bailian-coding-plan", fetchBailianQuota);
   registerMonitorFetcher("bailian-coding-plan", fetchBailianQuota);
+  registerQuotaWindows("bailian-coding-plan", [
+    BAILIAN_WINDOW_5H,
+    BAILIAN_WINDOW_WEEKLY,
+    BAILIAN_WINDOW_MONTHLY,
+  ]);
 }
diff --git a/tests/unit/bailian-quota-fetcher.test.ts b/tests/unit/bailian-quota-fetcher.test.ts
index f9c9d5baaf..77e5d10945 100644
--- a/tests/unit/bailian-quota-fetcher.test.ts
+++ b/tests/unit/bailian-quota-fetcher.test.ts
@@ -2,6 +2,9 @@ import test from "node:test";
 import assert from "node:assert/strict";
 
 import {
+  BAILIAN_WINDOW_5H,
+  BAILIAN_WINDOW_MONTHLY,
+  BAILIAN_WINDOW_WEEKLY,
   fetchBailianQuota,
   invalidateBailianQuotaCache,
   registerBailianCodingPlanQuotaFetcher,
@@ -216,6 +219,9 @@ test("fetchBailianQuota parses triple-window and returns percentUsed = max(5h%,
   assert.equal(quota?.window5h?.percentUsed, 0.6);
   assert.equal(quota?.windowWeekly?.percentUsed, 0.8);
   assert.equal(quota?.windowMonthly?.percentUsed, 0.4);
+  assert.equal(quota?.windows?.[BAILIAN_WINDOW_5H]?.percentUsed, 0.6);
+  assert.equal(quota?.windows?.[BAILIAN_WINDOW_WEEKLY]?.percentUsed, 0.8);
+  assert.equal(quota?.windows?.[BAILIAN_WINDOW_MONTHLY]?.percentUsed, 0.4);
 
   invalidateBailianQuotaCache(connectionId);
 });