switchroom 0.14.92 → 0.15.0

package/dist/auth-broker/index.js

@@ -13560,6 +13560,7 @@ class AuthBroker {
   listeners = new Map;
   refreshTimer = null;
   consumerProbeTimer = null;
+  fleetProbeTimer = null;
   fetchQuotaImpl;
   stateDir;
   socketRoot;
@@ -13641,6 +13642,17 @@ class AuthBroker {
         }, probeMs);
         this.consumerProbeTimer.unref();
       }
+      if (probeMs > 0 && process.env.SWITCHROOM_DISABLE_FLEET_QUOTA_PROBE !== "1") {
+        this.fleetQuotaProbeTick().catch((err) => {
+          this.logErr(`fleet-quota-probe (boot) threw: ${err.message}`);
+        });
+        this.fleetProbeTimer = setInterval(() => {
+          this.fleetQuotaProbeTick().catch((err) => {
+            this.logErr(`fleet-quota-probe threw: ${err.message}`);
+          });
+        }, probeMs);
+        this.fleetProbeTimer.unref();
+      }
     }
     const fanned = this.fanoutAll();
     if (fanned.length > 0) {
@@ -13671,6 +13683,10 @@ class AuthBroker {
       clearInterval(this.consumerProbeTimer);
       this.consumerProbeTimer = null;
     }
+    if (this.fleetProbeTimer) {
+      clearInterval(this.fleetProbeTimer);
+      this.fleetProbeTimer = null;
+    }
     for (const [sock, lis] of this.listeners) {
       try {
         lis.server.close();
@@ -14096,22 +14112,42 @@ class AuthBroker {
         ok: result.ok,
         error: result.ok ? undefined : result.reason
       });
-      if (result.ok) {
-        this.lastQuotaCache[label] = {
-          fiveHourUtilizationPct: result.data.fiveHourUtilizationPct,
-          sevenDayUtilizationPct: result.data.sevenDayUtilizationPct,
-          fiveHourResetAt: result.data.fiveHourResetAt?.toISOString() ?? null,
-          sevenDayResetAt: result.data.sevenDayResetAt?.toISOString() ?? null,
-          representativeClaim: result.data.representativeClaim,
-          overageStatus: result.data.overageStatus,
-          overageDisabledReason: result.data.overageDisabledReason,
-          capturedAt: this.now()
-        };
-      }
+      if (result.ok)
+        this.cacheQuotaSnapshot(label, result);
       return { label, result };
     }));
     socket.write(encodeSuccess(id, { results }));
   }
+  cacheQuotaSnapshot(label, result) {
+    if (!result.ok)
+      return;
+    this.lastQuotaCache[label] = {
+      fiveHourUtilizationPct: result.data.fiveHourUtilizationPct,
+      sevenDayUtilizationPct: result.data.sevenDayUtilizationPct,
+      fiveHourResetAt: result.data.fiveHourResetAt?.toISOString() ?? null,
+      sevenDayResetAt: result.data.sevenDayResetAt?.toISOString() ?? null,
+      representativeClaim: result.data.representativeClaim,
+      overageStatus: result.data.overageStatus,
+      overageDisabledReason: result.data.overageDisabledReason,
+      capturedAt: this.now()
+    };
+  }
+  async fleetQuotaProbeTick() {
+    for (const label of listAccounts(this.home)) {
+      const creds = readAccountCredentials(label, this.home);
+      const token = creds?.claudeAiOauth?.accessToken;
+      if (!token)
+        continue;
+      let result;
+      try {
+        result = await this.fetchQuotaImpl({ accessToken: token });
+      } catch (err) {
+        this.logErr(`fleet-quota-probe ${label}: ${err.message}`);
+        continue;
+      }
+      this.cacheQuotaSnapshot(label, result);
+    }
+  }
   async consumerQuotaProbeTick() {
     const accounts = Array.from(new Set((this.config.auth?.consumers ?? []).map((c) => c.account)));
     for (const label of accounts) {