@expiren/opencode-antigravity-auth 1.6.32 → 1.6.34

package/dist/index.js

@@ -9049,6 +9049,8 @@ var AccountManager = class _AccountManager {
   savePending = false;
   saveTimeout = null;
   savePromiseResolvers = [];
+  sessionStartTime = Date.now();
+  sessionRequestCounts = /* @__PURE__ */ new Map();
   static async loadFromDisk(authFallback) {
     const stored = await loadAccounts();
     return new _AccountManager(authFallback, stored);
@@ -9089,6 +9091,7 @@ var AccountManager = class _AccountManager {
           fingerprintHistory: acc.fingerprintHistory ?? [],
           cachedQuota: acc.cachedQuota,
           cachedQuotaUpdatedAt: acc.cachedQuotaUpdatedAt,
+          dailyRequestCounts: acc.dailyRequestCounts,
           verificationRequired: acc.verificationRequired,
           verificationRequiredAt: acc.verificationRequiredAt,
           verificationRequiredReason: acc.verificationRequiredReason,
@@ -9579,6 +9582,7 @@ var AccountManager = class _AccountManager {
         fingerprintHistory: a.fingerprintHistory?.length ? a.fingerprintHistory : void 0,
         cachedQuota: a.cachedQuota && Object.keys(a.cachedQuota).length > 0 ? a.cachedQuota : void 0,
         cachedQuotaUpdatedAt: a.cachedQuotaUpdatedAt,
+        dailyRequestCounts: a.dailyRequestCounts,
         verificationRequired: a.verificationRequired,
         verificationRequiredAt: a.verificationRequiredAt,
         verificationRequiredReason: a.verificationRequiredReason,
@@ -9701,6 +9705,98 @@ var AccountManager = class _AccountManager {
       account.cachedQuotaUpdatedAt = nowMs();
     }
   }
+  /**
+   * Record a successful API request for an account.
+   * Tracks per model family with daily reset.
+   */
+  recordRequest(accountIndex, family) {
+    const account = this.accounts[accountIndex];
+    if (!account) return;
+    const today = (/* @__PURE__ */ new Date()).toISOString().slice(0, 10);
+    if (!account.dailyRequestCounts || account.dailyRequestCounts.date !== today) {
+      account.dailyRequestCounts = { date: today, claude: 0, gemini: 0 };
+    }
+    account.dailyRequestCounts[family]++;
+    account.lastUsed = nowMs();
+    this.recordSessionRequest(accountIndex, family);
+  }
+  /**
+   * Get request counts for an account for today.
+   */
+  getDailyRequestCounts(accountIndex) {
+    const account = this.accounts[accountIndex];
+    if (!account?.dailyRequestCounts) return null;
+    const today = (/* @__PURE__ */ new Date()).toISOString().slice(0, 10);
+    if (account.dailyRequestCounts.date !== today) return null;
+    return { ...account.dailyRequestCounts };
+  }
+  /**
+   * Get total daily request counts across all accounts for a model family.
+   */
+  getTotalDailyRequests(family) {
+    const today = (/* @__PURE__ */ new Date()).toISOString().slice(0, 10);
+    let total = 0;
+    for (const account of this.accounts) {
+      if (account.dailyRequestCounts?.date === today) {
+        total += account.dailyRequestCounts[family];
+      }
+    }
+    return total;
+  }
+  /**
+   * Get a summary of daily request distribution across accounts.
+   * Returns accounts sorted by request count (descending).
+   */
+  getDailyRequestSummary(family) {
+    const today = (/* @__PURE__ */ new Date()).toISOString().slice(0, 10);
+    const result = [];
+    for (const account of this.accounts) {
+      const count = account.dailyRequestCounts?.date === today ? account.dailyRequestCounts[family] : 0;
+      if (count > 0) {
+        result.push({ index: account.index, email: account.email, count });
+      }
+    }
+    return result.sort((a, b) => b.count - a.count);
+  }
+  /**
+   * Record a request for the current session (in-memory only).
+   */
+  recordSessionRequest(accountIndex, family) {
+    const key = String(accountIndex);
+    const current = this.sessionRequestCounts.get(key) ?? { claude: 0, gemini: 0 };
+    current[family]++;
+    this.sessionRequestCounts.set(key, current);
+  }
+  /**
+   * Get a summary of the current session's request usage.
+   */
+  getSessionSummary() {
+    const durationMs = Date.now() - this.sessionStartTime;
+    const durationMinutes = Math.round(durationMs / 6e4);
+    const durationHours = durationMs / 36e5;
+    let totalClaude = 0;
+    let totalGemini = 0;
+    const perAccount = [];
+    for (const [key, counts] of this.sessionRequestCounts) {
+      const idx = Number(key);
+      const account = this.accounts[idx];
+      totalClaude += counts.claude;
+      totalGemini += counts.gemini;
+      if (counts.claude > 0 || counts.gemini > 0) {
+        perAccount.push({ index: idx, email: account?.email, claude: counts.claude, gemini: counts.gemini });
+      }
+    }
+    const totalRequests = totalClaude + totalGemini;
+    const requestsPerHour = durationHours > 0 ? Math.round(totalRequests / durationHours) : 0;
+    return {
+      durationMinutes,
+      totalClaude,
+      totalGemini,
+      requestsPerHour,
+      accountsUsed: perAccount.length,
+      perAccount: perAccount.sort((a, b) => b.claude + b.gemini - (a.claude + a.gemini))
+    };
+  }
   isAccountOverSoftQuota(account, family, thresholdPercent, cacheTtlMs, model) {
     return isOverSoftQuotaThreshold(account, family, thresholdPercent, cacheTtlMs, model);
   }
@@ -11972,6 +12068,16 @@ var createAntigravityPlugin = (providerId) => async ({ client, directory }) => {
   const eventHandler = async (input2) => {
     await updateChecker.event(input2);
     if (input2.event.type === "session.created") {
+      const prevSummary = activeAccountManager?.getSessionSummary();
+      if (prevSummary && (prevSummary.totalClaude > 0 || prevSummary.totalGemini > 0)) {
+        log10.debug("prev-session-quota-summary", {
+          durationMinutes: prevSummary.durationMinutes,
+          totalClaude: prevSummary.totalClaude,
+          totalGemini: prevSummary.totalGemini,
+          requestsPerHour: prevSummary.requestsPerHour,
+          accountsUsed: prevSummary.accountsUsed
+        });
+      }
       const props = input2.event.properties;
       if (props?.info?.parentID) {
         isChildSession = true;
@@ -12592,6 +12698,11 @@ var createAntigravityPlugin = (providerId) => async ({ client, directory }) => {
                     }
                     const response = await fetch(prepared.request, prepared.init);
                     apiRequestCount++;
+                    accountManager.recordRequest(account.index, family);
+                    const requestCounts = accountManager.getDailyRequestCounts(account.index);
+                    if (requestCounts) {
+                      pushDebug(`[Quota] account=${account.index} ${family}_today=${requestCounts[family]} total_${family}_today=${accountManager.getTotalDailyRequests(family)}`);
+                    }
                     pushDebug(`status=${response.status} ${response.statusText} (api_request #${apiRequestCount})`);
                     if (response.status === 429 || response.status === 503 || response.status === 529) {
                       if (tokenConsumed) {
@@ -12877,6 +12988,29 @@ Alternatively, you can:
                     if (apiRequestCount > 1) {
                       pushDebug(`[Quota] Total API requests for this user message: ${apiRequestCount} (${apiRequestCount - 1} retries)`);
                     }
+                    const dailyCounts = accountManager.getDailyRequestCounts(account.index);
+                    if (dailyCounts) {
+                      pushDebug(`[Quota] Account ${account.index} (${account.email ?? "unknown"}) today: claude=${dailyCounts.claude} gemini=${dailyCounts.gemini}`);
+                    }
+                    const totalToday = accountManager.getTotalDailyRequests(family);
+                    pushDebug(`[Quota] Total ${family} requests today (all accounts): ${totalToday}`);
+                    const cachedQuota = account.cachedQuota;
+                    if (cachedQuota) {
+                      const quotaFamily = family === "claude" ? "claude" : "gemini-flash";
+                      const groupQuota = cachedQuota[quotaFamily];
+                      if (groupQuota?.remainingFraction != null) {
+                        const pct = Math.round(groupQuota.remainingFraction * 100);
+                        pushDebug(`[Quota] Account ${account.index} cached ${quotaFamily} remaining: ${pct}%${groupQuota.resetTime ? ` (resets ${groupQuota.resetTime})` : ""}`);
+                      }
+                    }
+                    const sessionSummary = accountManager.getSessionSummary();
+                    if (sessionSummary.durationMinutes >= 1) {
+                      const familyTotal = family === "claude" ? sessionSummary.totalClaude : sessionSummary.totalGemini;
+                      if (familyTotal > 0) {
+                        const ratePerHour = sessionSummary.requestsPerHour;
+                        pushDebug(`[Quota] Session: ${sessionSummary.durationMinutes}min, ${familyTotal} ${family} reqs, ~${ratePerHour} reqs/hr, ${sessionSummary.accountsUsed} accounts used`);
+                      }
+                    }
                     return transformedResponse;
                   } catch (error) {
                     if (tokenConsumed) {