@cortexkit/opencode-antigravity-auth 1.1.0 → 2.0.1

package/dist/src/plugin/quota.d.ts

@@ -1,44 +1,141 @@
-import type { PluginClient } from "./types";
-import type { AccountMetadataV3 } from "./storage";
-export type QuotaGroup = "claude" | "gemini-pro" | "gemini-flash" | "gpt-oss";
-export interface QuotaGroupSummary {
-    remainingFraction?: number;
-    resetTime?: string;
-    modelCount: number;
+/**
+ * OpenCode adapter for the harness-agnostic quota manager.
+ *
+ * Re-exports the core `QuotaManager` types and helpers so call sites in
+ * `plugin.ts` and other modules don't need to switch imports. Also wires up
+ * the host-specific fetch callback that handles:
+ *   1. Token refresh via the existing `refreshAccessToken` path.
+ *   2. Persisting rotated refresh tokens via `client.auth.set` (matching
+ *      legacy behavior).
+ *   3. Resolving project context via `ensureProjectContext`.
+ *
+ * The legacy `checkAccountsQuota(accounts, client, providerId)` export is
+ * retained as a compatibility wrapper that creates a short-lived manager
+ * with `force: true` — manual quota screens must always refresh, even if
+ * the background manager has backed off.
+ */
+import { type AccountMetadataV3, type AccountQuotaResult, type FetchAccountQuota, type FetchAvailableModelsOptions, type QuotaManager } from '@cortexkit/antigravity-auth-core';
+import type { PluginClient } from './types';
+type QuotaFetch = NonNullable<FetchAvailableModelsOptions['fetchVia']>;
+export type { AccountQuotaResult, AccountQuotaStatus, GeminiCliQuotaModel, GeminiCliQuotaSummary, PerModelQuotaEntry, QuotaGroup, QuotaGroupSummary, QuotaManager, QuotaManagerOptions, QuotaSummary, } from '@cortexkit/antigravity-auth-core';
+export { classifyQuotaGroup, createQuotaManager, defaultKeyOf, } from '@cortexkit/antigravity-auth-core';
+export interface CreateOpenCodeQuotaManagerOptions {
+    /** Override the default key derivation (email → refresh-token hash). */
+    keyOf?: (account: AccountMetadataV3) => string;
+    baseBackoffMs?: number;
+    maxBackoffMs?: number;
+    fetchTimeoutMs?: number;
 }
-export interface PerModelQuotaEntry {
-    modelId: string;
-    displayName?: string;
-    group: QuotaGroup | null;
-    remainingFraction: number;
-    resetTime?: string;
-}
-export interface QuotaSummary {
-    groups: Partial<Record<QuotaGroup, QuotaGroupSummary>>;
-    perModel?: PerModelQuotaEntry[];
-    modelCount: number;
-    error?: string;
-}
-export interface GeminiCliQuotaModel {
-    modelId: string;
-    remainingFraction: number;
-    resetTime?: string;
-}
-export interface GeminiCliQuotaSummary {
-    models: GeminiCliQuotaModel[];
-    error?: string;
-}
-export type AccountQuotaStatus = "ok" | "disabled" | "error";
-export interface AccountQuotaResult {
-    index: number;
-    email?: string;
-    status: AccountQuotaStatus;
-    error?: string;
-    disabled?: boolean;
-    quota?: QuotaSummary;
-    geminiCliQuota?: GeminiCliQuotaSummary;
-    updatedAccount?: AccountMetadataV3;
-}
-export declare function classifyQuotaGroup(modelName: string, displayName?: string): QuotaGroup | null;
+/**
+ * Build an OpenCode-wired quota manager.
+ *
+ * The returned manager owns its cache, in-flight dedupe, and backoff state.
+ * Register its `dispose()` with `PluginLifecycle` so refreshes abort on plugin
+ * shutdown.
+ *
+ * The wrapper observes `refreshAccount` / `refreshAccounts` and pushes a
+ * redacted sidebar snapshot after every refresh (success or backoff) so
+ * the TUI's next poll renders the freshest cached quota. The snapshot is
+ * sourced from the live AccountManager view (`getAccountsForSidebar`) so
+ * it carries the just-updated percentages; before bootstrapping it is a
+ * no-op.
+ */
+export declare function createOpenCodeQuotaManager(client: PluginClient, providerId?: string, options?: CreateOpenCodeQuotaManagerOptions & {
+    /**
+     * Optional account-snapshot provider. Wired by the plugin entry to
+     * the live `AccountManager.getAccounts()` so each refresh can build
+     * a sidebar snapshot from the actual cached quota + cooldown. When
+     * omitted, the wrapper falls back to a no-op snapshot push.
+     */
+    getAccountsForSidebar?: () => Array<{
+        index: number;
+        label?: string;
+        enabled?: boolean;
+        coolingDownUntil?: number;
+        cachedQuota?: AccountMetadataV3['cachedQuota'];
+        cachedQuotaAccountId?: string;
+        currentQuotaAccountId?: string;
+    }> | null;
+    /**
+     * Optional provider for the active-account indexes per model family.
+     * Wired by the plugin entry so every quota-refresh sidebar snapshot
+     * carries the real `current` flag — not a hardcoded `false`.
+     */
+    getActiveIndexByFamily?: () => {
+        claude: number;
+        gemini: number;
+    } | null;
+    /**
+     * Optional transport adapter used for both `fetchAvailableModels`
+     * and the project-context lookup. When omitted, the production
+     * `fetchWithAgyCliTransport` runs and binds to the real
+     * Antigravity endpoints; the e2e harness injects a mock here so
+     * quota refresh + project discovery stay on the loopback server.
+     */
+    fetchVia?: QuotaFetch;
+}): QuotaManager;
+/**
+ * Compatibility wrapper used by code paths that want a one-shot check across
+ * the full account pool with no shared cache.
+ *
+ * Equivalent to spinning up a short-lived manager with `force: true` so
+ * manual quota dialogs always reflect the latest data even if the background
+ * manager has backed off.
+ */
+export declare function checkAccountsQuotaWith(accounts: AccountMetadataV3[], fetchAccountQuota: FetchAccountQuota): Promise<AccountQuotaResult[]>;
+export declare function checkAccountsQuotaStandalone(accounts: AccountMetadataV3[], options: {
+    refresh: boolean;
+}): Promise<AccountQuotaResult[]>;
 export declare function checkAccountsQuota(accounts: AccountMetadataV3[], client: PluginClient, providerId?: string): Promise<AccountQuotaResult[]>;
+/**
+ * Push a quota refresh into the sidebar. Called by every quota refresh
+ * call site (manual `/antigravity-quota`, the `check` menu action, and the
+ * background refresh in `fetch-interceptor`) AFTER the results have been
+ * folded back into the AccountManager's cached quota. The function reads
+ * the live account snapshot through `getAccounts` so the redacted entry
+ * carries the just-refreshed percentages — not the previous tick's stale
+ * numbers and not `undefined`.
+ *
+ * The mapping is deliberately tolerant: if `getAccounts` returns `null`
+ * (e.g. before the plugin has finished bootstrapping) the call is a no-op.
+ * On lock contention the error is logged-and-swallowed so a quota dialog
+ * never fails just because the sidebar file is busy.
+ */
+export declare function pushSidebarQuotaSnapshot(getAccounts: () => Array<{
+    index: number;
+    label?: string;
+    enabled?: boolean;
+    coolingDownUntil?: number;
+    cachedQuota?: AccountMetadataV3['cachedQuota'];
+    cachedQuotaAccountId?: string;
+    currentQuotaAccountId?: string;
+    /** Captured plan tier to surface in the sidebar state file. */
+    tier?: {
+        id: string;
+        paidId?: string;
+        capturedAt: number;
+    };
+}> | null, backoffUntil?: number, getActiveIndexByFamily?: () => {
+    claude: number;
+    gemini: number;
+} | null): Promise<void>;
+/**
+ * Build a per-account tier-loader callback for the background poller.
+ *
+ * Calls `loadManagedProject` (loadCodeAssist) directly, bypassing the
+ * `ensureProjectContext` cache that fast-paths on `managedProjectId` and
+ * never returns a tier for existing accounts. One call per account per 24 h.
+ *
+ * Uses the same token-refresh infrastructure as `makeFetchAccountQuota` so
+ * an expired access token does not silently fail the tier lookup.
+ *
+ * `loadManagedProject` uses the production TLS transport (`fetchWithAgyCliTransport`)
+ * and is not interceptable via `fetchVia` -- the same design constraint applies to
+ * `ensureProjectContext`. Tier lookup is best-effort; any failure resolves `null`.
+ */
+export declare function makeTierLoader(client: PluginClient | undefined, providerId: string): (account: AccountMetadataV3) => Promise<{
+    id: string;
+    paidId?: string;
+    capturedAt: number;
+} | null>;
 //# sourceMappingURL=quota.d.ts.map

package/dist/src/plugin/quota.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"quota.d.ts","sourceRoot":"","sources":["../../../src/plugin/quota.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,SAAS,CAAC;AAC9D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAInD,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,YAAY,GAAG,cAAc,GAAG,SAAS,CAAC;AAE9E,MAAM,WAAW,iBAAiB;IAChC,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,UAAU,GAAG,IAAI,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC,CAAC;IACvD,QAAQ,CAAC,EAAE,kBAAkB,EAAE,CAAC;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,mBAAmB,EAAE,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAYD,MAAM,MAAM,kBAAkB,GAAG,IAAI,GAAG,UAAU,GAAG,OAAO,CAAC;AAE7D,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,kBAAkB,CAAC;IAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,cAAc,CAAC,EAAE,qBAAqB,CAAC;IACvC,cAAc,CAAC,EAAE,iBAAiB,CAAC;CACpC;AA+CD,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI,CAgB7F;AAwPD,wBAAsB,kBAAkB,CACtC,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,MAAM,EAAE,YAAY,EACpB,UAAU,SAA0B,GACnC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAiF/B"}
+{"version":3,"file":"quota.d.ts","sourceRoot":"","sources":["../../../src/plugin/quota.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EAMvB,KAAK,iBAAiB,EACtB,KAAK,2BAA2B,EAMhC,KAAK,YAAY,EAElB,MAAM,kCAAkC,CAAA;AAsBzC,OAAO,KAAK,EAAoB,YAAY,EAAE,MAAM,SAAS,CAAA;AAE7D,KAAK,UAAU,GAAG,WAAW,CAAC,2BAA2B,CAAC,UAAU,CAAC,CAAC,CAAA;AAKtE,YAAY,EACV,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACnB,qBAAqB,EACrB,kBAAkB,EAClB,UAAU,EACV,iBAAiB,EACjB,YAAY,EACZ,mBAAmB,EACnB,YAAY,GACb,MAAM,kCAAkC,CAAA;AACzC,OAAO,EACL,kBAAkB,EAClB,kBAAkB,EAClB,YAAY,GACb,MAAM,kCAAkC,CAAA;AAEzC,MAAM,WAAW,iCAAiC;IAChD,wEAAwE;IACxE,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,iBAAiB,KAAK,MAAM,CAAA;IAC9C,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,cAAc,CAAC,EAAE,MAAM,CAAA;CACxB;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,YAAY,EACpB,UAAU,GAAE,MAAgC,EAC5C,OAAO,GAAE,iCAAiC,GAAG;IAC3C;;;;;OAKG;IACH,qBAAqB,CAAC,EAAE,MAAM,KAAK,CAAC;QAClC,KAAK,EAAE,MAAM,CAAA;QACb,KAAK,CAAC,EAAE,MAAM,CAAA;QACd,OAAO,CAAC,EAAE,OAAO,CAAA;QACjB,gBAAgB,CAAC,EAAE,MAAM,CAAA;QACzB,WAAW,CAAC,EAAE,iBAAiB,CAAC,aAAa,CAAC,CAAA;QAC9C,oBAAoB,CAAC,EAAE,MAAM,CAAA;QAC7B,qBAAqB,CAAC,EAAE,MAAM,CAAA;KAC/B,CAAC,GAAG,IAAI,CAAA;IACT;;;;OAIG;IACH,sBAAsB,CAAC,EAAE,MAAM;QAC7B,MAAM,EAAE,MAAM,CAAA;QACd,MAAM,EAAE,MAAM,CAAA;KACf,GAAG,IAAI,CAAA;IACR;;;;;;OAMG;IACH,QAAQ,CAAC,EAAE,UAAU,CAAA;CACjB,GACL,YAAY,CAiFd;AAED;;;;;;;GAOG;AACH,wBAAsB,sBAAsB,CAC1C,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,iBAAiB,EAAE,iBAAiB,GACnC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAa/B;AAED,wBAAsB,4BAA4B,CAChD,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,OAAO,EAAE;IAAE,OAAO,EAAE,OAAO,CAAA;CAAE,GAC5B,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAiB/B;AAED,wBAAsB,kBAAkB,CACtC,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,MAAM,EAAE,YAAY,EACpB,UAAU,GAAE,MAAgC,GAC3C,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAK/B;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAsB,wBAAwB,CAC5C,WAAW,EAAE,MAAM,KAAK,CAAC;IACvB,KAAK,EAAE,MAAM,CAAA;IACb,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,OAAO,CAAA;IACjB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,WAAW,CAAC,EAAE,iBAAiB,CAAC,aAAa,CAAC,CAAA;IAC9C,oBAAoB,CAAC,EAAE,MAAM,CAAA;IAC7B,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B,+DAA+D;IAC/D,IAAI,CAAC,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAA;CAC3D,CAAC,GAAG,IAAI,EACT,YAAY,GAAE,MAAU,EACxB,sBAAsB,CAAC,EAAE,MAAM;IAC7B,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,EAAE,MAAM,CAAA;CACf,GAAG,IAAI,GACP,OAAO,CAAC,IAAI,CAAC,CA8Bf;AAsSD;;;;;;;;;;;;;GAaG;AACH,wBAAgB,cAAc,CAC5B,MAAM,EAAE,YAAY,GAAG,SAAS,EAChC,UAAU,EAAE,MAAM,GACjB,CACD,OAAO,EAAE,iBAAiB,KACvB,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAAC,CAiCvE"}