@omnicross/daemon 0.4.0 → 0.4.1

package/dist/index.d.cts

@@ -1636,6 +1636,56 @@ declare class CopilotAllowanceCollector {
     private unsupportedSnapshot;
 }
 
+/**
+ * GeminiAllowanceCollector — gemini subscription quota via the Cloud Code
+ * Assist internal user-quota RPC.
+ *
+ * Probes `POST ${cloudcode-pa}/v1internal:retrieveUserQuota` per gemini
+ * account (Bearer access token + the GeminiCLI masquerade pair the inference
+ * path also carries), parsing `buckets[]` into one per-model window:
+ * `remainingFraction` (0–1) → used percent, `resetTime` → resetsAt. The
+ * Code Assist project id comes from the SHARED `GeminiCodeAssistProjectResolver`
+ * (the same instance the dispatch seam uses, so the handshake runs at most
+ * once per token across inference + quota); a handshake hard failure degrades
+ * to a project-less probe (the valid free-tier envelope), mirroring the
+ * reference implementation's tolerant behavior.
+ *
+ * Mirrors the sibling collectors' cache contract: 5-minute cache, per-account
+ * in-flight merging, one 401/403→refresh→retry. Display-only — the scheduling
+ * whitelist deliberately excludes gemini (per-model fraction buckets do not
+ * fit the worst-window pause/demote rule).
+ *
+ * @module @omnicross/daemon/allowance/GeminiAllowanceCollector
+ */
+
+interface GeminiAllowanceCredentialReader {
+    getAccessTokenForAccount(providerId: 'gemini', accountId: string): Promise<string | null>;
+    refreshAccountToken(providerId: 'gemini', accountId: string): Promise<boolean>;
+}
+type GeminiAllowanceFetch = (url: string, init: RequestInit, accountId: string) => Promise<Response>;
+interface GeminiAllowanceCollectOptions {
+    force?: boolean;
+    refreshAheadMs?: number;
+}
+declare class GeminiAllowanceCollector {
+    private readonly credentials;
+    private readonly store;
+    private readonly fetchImpl;
+    private readonly now;
+    private readonly projectResolver;
+    private readonly inFlight;
+    constructor(credentials: GeminiAllowanceCredentialReader, store?: AccountAllowanceStore, fetchImpl?: GeminiAllowanceFetch, now?: () => number, projectResolver?: {
+        resolveProject(accessToken: string): Promise<string | undefined>;
+    });
+    collectMany(accounts: readonly SubscriptionAccountEntry<GeminiTokenConfig>[], options?: GeminiAllowanceCollectOptions): Promise<AccountAllowanceSnapshot[]>;
+    collect(account: SubscriptionAccountEntry<GeminiTokenConfig>, options?: GeminiAllowanceCollectOptions): Promise<AccountAllowanceSnapshot>;
+    private isCacheValid;
+    private fetchAccount;
+    private request;
+    private failureSnapshot;
+    private unsupportedSnapshot;
+}
+
 /**
  * OpenCodeGo usage collector.
  *
@@ -1683,8 +1733,8 @@ declare class OpenCodeGoAllowanceCollector {
  * literals — extending them all would make the overloads conflict.
  */
 interface AccountAllowanceCredentialReader {
-    getAccessTokenForAccount(providerId: 'claude' | 'codex' | 'kimi' | 'opencodego' | 'grok' | 'copilot', accountId: string): Promise<string | null>;
-    refreshAccountToken(providerId: 'claude' | 'codex' | 'kimi' | 'grok' | 'copilot', accountId: string): Promise<boolean>;
+    getAccessTokenForAccount(providerId: 'claude' | 'codex' | 'kimi' | 'opencodego' | 'grok' | 'copilot' | 'gemini', accountId: string): Promise<string | null>;
+    refreshAccountToken(providerId: 'claude' | 'codex' | 'kimi' | 'grok' | 'copilot' | 'gemini', accountId: string): Promise<boolean>;
     getFullConfig(): Promise<AccountTokensConfig>;
 }
 interface AccountAllowanceFilter {
@@ -1705,7 +1755,8 @@ declare class AccountAllowanceService {
     readonly grokCollector: GrokAllowanceCollector;
     readonly copilotCollector: CopilotAllowanceCollector;
     readonly opencodegoCollector: OpenCodeGoAllowanceCollector;
-    constructor(credentials: AccountAllowanceCredentialReader, store?: AccountAllowanceStore, collector?: ClaudeAllowanceCollector, codexCollector?: CodexAllowanceCollector, kimiCollector?: KimiAllowanceCollector, opencodegoCollector?: OpenCodeGoAllowanceCollector, grokCollector?: GrokAllowanceCollector, copilotCollector?: CopilotAllowanceCollector, now?: () => number);
+    readonly geminiCollector: GeminiAllowanceCollector;
+    constructor(credentials: AccountAllowanceCredentialReader, store?: AccountAllowanceStore, collector?: ClaudeAllowanceCollector, codexCollector?: CodexAllowanceCollector, kimiCollector?: KimiAllowanceCollector, opencodegoCollector?: OpenCodeGoAllowanceCollector, grokCollector?: GrokAllowanceCollector, copilotCollector?: CopilotAllowanceCollector, geminiCollector?: GeminiAllowanceCollector, now?: () => number);
     /**
      * Read all/filtered snapshots. Claude's and Codex's five-minute caches are
      * refreshed lazily on read (Codex polls `/backend-api/wham/usage`; the
@@ -1729,6 +1780,8 @@ declare class AccountAllowanceService {
     refreshCopilot(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     /** Force-refresh Grok usage (CLI billing proxy) for one/all accounts. */
     refreshGrok(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
+    /** Force-refresh Gemini usage (Code Assist retrieveUserQuota) for one/all accounts. */
+    refreshGemini(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     /**
      * Keep Claude + Codex + Kimi snapshots warm for allowance-aware routing. All
      * collectors preserve their cache + per-account in-flight coalescing; a tick
@@ -2939,6 +2992,8 @@ interface AccountAllowanceAdminReader {
     refreshGrok?(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     /** Optional: Copilot user-quota refresh (absent on older daemons). */
     refreshCopilot?(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
+    /** Optional: Gemini Code-Assist quota refresh (absent on older daemons). */
+    refreshGemini?(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     removeAccountSnapshot?(providerId: SubscriptionProviderId, accountId: string): void;
     removeProviderSnapshots?(providerId: SubscriptionProviderId): void;
     getSchedulingStatus?(): AccountAllowanceSchedulingStatus;

package/dist/index.d.ts

@@ -1636,6 +1636,56 @@ declare class CopilotAllowanceCollector {
     private unsupportedSnapshot;
 }
 
+/**
+ * GeminiAllowanceCollector — gemini subscription quota via the Cloud Code
+ * Assist internal user-quota RPC.
+ *
+ * Probes `POST ${cloudcode-pa}/v1internal:retrieveUserQuota` per gemini
+ * account (Bearer access token + the GeminiCLI masquerade pair the inference
+ * path also carries), parsing `buckets[]` into one per-model window:
+ * `remainingFraction` (0–1) → used percent, `resetTime` → resetsAt. The
+ * Code Assist project id comes from the SHARED `GeminiCodeAssistProjectResolver`
+ * (the same instance the dispatch seam uses, so the handshake runs at most
+ * once per token across inference + quota); a handshake hard failure degrades
+ * to a project-less probe (the valid free-tier envelope), mirroring the
+ * reference implementation's tolerant behavior.
+ *
+ * Mirrors the sibling collectors' cache contract: 5-minute cache, per-account
+ * in-flight merging, one 401/403→refresh→retry. Display-only — the scheduling
+ * whitelist deliberately excludes gemini (per-model fraction buckets do not
+ * fit the worst-window pause/demote rule).
+ *
+ * @module @omnicross/daemon/allowance/GeminiAllowanceCollector
+ */
+
+interface GeminiAllowanceCredentialReader {
+    getAccessTokenForAccount(providerId: 'gemini', accountId: string): Promise<string | null>;
+    refreshAccountToken(providerId: 'gemini', accountId: string): Promise<boolean>;
+}
+type GeminiAllowanceFetch = (url: string, init: RequestInit, accountId: string) => Promise<Response>;
+interface GeminiAllowanceCollectOptions {
+    force?: boolean;
+    refreshAheadMs?: number;
+}
+declare class GeminiAllowanceCollector {
+    private readonly credentials;
+    private readonly store;
+    private readonly fetchImpl;
+    private readonly now;
+    private readonly projectResolver;
+    private readonly inFlight;
+    constructor(credentials: GeminiAllowanceCredentialReader, store?: AccountAllowanceStore, fetchImpl?: GeminiAllowanceFetch, now?: () => number, projectResolver?: {
+        resolveProject(accessToken: string): Promise<string | undefined>;
+    });
+    collectMany(accounts: readonly SubscriptionAccountEntry<GeminiTokenConfig>[], options?: GeminiAllowanceCollectOptions): Promise<AccountAllowanceSnapshot[]>;
+    collect(account: SubscriptionAccountEntry<GeminiTokenConfig>, options?: GeminiAllowanceCollectOptions): Promise<AccountAllowanceSnapshot>;
+    private isCacheValid;
+    private fetchAccount;
+    private request;
+    private failureSnapshot;
+    private unsupportedSnapshot;
+}
+
 /**
  * OpenCodeGo usage collector.
  *
@@ -1683,8 +1733,8 @@ declare class OpenCodeGoAllowanceCollector {
  * literals — extending them all would make the overloads conflict.
  */
 interface AccountAllowanceCredentialReader {
-    getAccessTokenForAccount(providerId: 'claude' | 'codex' | 'kimi' | 'opencodego' | 'grok' | 'copilot', accountId: string): Promise<string | null>;
-    refreshAccountToken(providerId: 'claude' | 'codex' | 'kimi' | 'grok' | 'copilot', accountId: string): Promise<boolean>;
+    getAccessTokenForAccount(providerId: 'claude' | 'codex' | 'kimi' | 'opencodego' | 'grok' | 'copilot' | 'gemini', accountId: string): Promise<string | null>;
+    refreshAccountToken(providerId: 'claude' | 'codex' | 'kimi' | 'grok' | 'copilot' | 'gemini', accountId: string): Promise<boolean>;
     getFullConfig(): Promise<AccountTokensConfig>;
 }
 interface AccountAllowanceFilter {
@@ -1705,7 +1755,8 @@ declare class AccountAllowanceService {
     readonly grokCollector: GrokAllowanceCollector;
     readonly copilotCollector: CopilotAllowanceCollector;
     readonly opencodegoCollector: OpenCodeGoAllowanceCollector;
-    constructor(credentials: AccountAllowanceCredentialReader, store?: AccountAllowanceStore, collector?: ClaudeAllowanceCollector, codexCollector?: CodexAllowanceCollector, kimiCollector?: KimiAllowanceCollector, opencodegoCollector?: OpenCodeGoAllowanceCollector, grokCollector?: GrokAllowanceCollector, copilotCollector?: CopilotAllowanceCollector, now?: () => number);
+    readonly geminiCollector: GeminiAllowanceCollector;
+    constructor(credentials: AccountAllowanceCredentialReader, store?: AccountAllowanceStore, collector?: ClaudeAllowanceCollector, codexCollector?: CodexAllowanceCollector, kimiCollector?: KimiAllowanceCollector, opencodegoCollector?: OpenCodeGoAllowanceCollector, grokCollector?: GrokAllowanceCollector, copilotCollector?: CopilotAllowanceCollector, geminiCollector?: GeminiAllowanceCollector, now?: () => number);
     /**
      * Read all/filtered snapshots. Claude's and Codex's five-minute caches are
      * refreshed lazily on read (Codex polls `/backend-api/wham/usage`; the
@@ -1729,6 +1780,8 @@ declare class AccountAllowanceService {
     refreshCopilot(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     /** Force-refresh Grok usage (CLI billing proxy) for one/all accounts. */
     refreshGrok(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
+    /** Force-refresh Gemini usage (Code Assist retrieveUserQuota) for one/all accounts. */
+    refreshGemini(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     /**
      * Keep Claude + Codex + Kimi snapshots warm for allowance-aware routing. All
      * collectors preserve their cache + per-account in-flight coalescing; a tick
@@ -2939,6 +2992,8 @@ interface AccountAllowanceAdminReader {
     refreshGrok?(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     /** Optional: Copilot user-quota refresh (absent on older daemons). */
     refreshCopilot?(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
+    /** Optional: Gemini Code-Assist quota refresh (absent on older daemons). */
+    refreshGemini?(accountId?: string): Promise<AccountAllowanceSnapshot[]>;
     removeAccountSnapshot?(providerId: SubscriptionProviderId, accountId: string): void;
     removeProviderSnapshots?(providerId: SubscriptionProviderId): void;
     getSchedulingStatus?(): AccountAllowanceSchedulingStatus;