@cortexkit/opencode-antigravity-auth 1.0.1 → 2.0.0

package/dist/src/plugin/fetch/warmup.js

@@ -0,0 +1,76 @@
+/**
+ * Per-interceptor warmup attempt + success bookkeeping.
+ *
+ * The plugin used to keep two module-level sets: one for sessions that already
+ * attempted a thinking warmup and another for sessions that succeeded. They
+ * leaked across plugin lifetimes and across concurrent interceptors. This
+ * factory scopes the same sets to a single interceptor; `dispose()` evicts
+ * everything so the next interceptor starts clean.
+ */
+const MAX_WARMUP_SESSIONS = 1000;
+const MAX_WARMUP_RETRIES = 2;
+export function createWarmupState() {
+    const warmupAttemptedSessionIds = new Set();
+    const warmupSucceededSessionIds = new Set();
+    let disposed = false;
+    function trackAttempt(sessionId) {
+        if (disposed)
+            return false;
+        if (warmupSucceededSessionIds.has(sessionId)) {
+            return false;
+        }
+        if (warmupAttemptedSessionIds.size >= MAX_WARMUP_SESSIONS) {
+            const first = warmupAttemptedSessionIds.values().next().value;
+            if (first) {
+                warmupAttemptedSessionIds.delete(first);
+                warmupSucceededSessionIds.delete(first);
+            }
+        }
+        const attempts = getAttemptCount(sessionId);
+        if (attempts >= MAX_WARMUP_RETRIES) {
+            return false;
+        }
+        warmupAttemptedSessionIds.add(sessionId);
+        return true;
+    }
+    function getAttemptCount(sessionId) {
+        return warmupAttemptedSessionIds.has(sessionId) ? 1 : 0;
+    }
+    function markSuccess(sessionId) {
+        if (disposed)
+            return;
+        warmupSucceededSessionIds.add(sessionId);
+        if (warmupSucceededSessionIds.size >= MAX_WARMUP_SESSIONS) {
+            const first = warmupSucceededSessionIds.values().next().value;
+            if (first)
+                warmupSucceededSessionIds.delete(first);
+        }
+    }
+    function clearWarmupAttempt(sessionId) {
+        if (disposed)
+            return;
+        warmupAttemptedSessionIds.delete(sessionId);
+    }
+    function clear() {
+        warmupAttemptedSessionIds.clear();
+        warmupSucceededSessionIds.clear();
+    }
+    function dispose() {
+        if (disposed)
+            return;
+        clear();
+        disposed = true;
+    }
+    return {
+        get disposed() {
+            return disposed;
+        },
+        trackAttempt,
+        getAttemptCount,
+        markSuccess,
+        clearWarmupAttempt,
+        clear,
+        dispose,
+    };
+}
+//# sourceMappingURL=warmup.js.map

package/dist/src/plugin/fetch/warmup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"warmup.js","sourceRoot":"","sources":["../../../../src/plugin/fetch/warmup.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,MAAM,mBAAmB,GAAG,IAAI,CAAA;AAChC,MAAM,kBAAkB,GAAG,CAAC,CAAA;AAsB5B,MAAM,UAAU,iBAAiB;IAC/B,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAU,CAAA;IACnD,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAU,CAAA;IACnD,IAAI,QAAQ,GAAG,KAAK,CAAA;IAEpB,SAAS,YAAY,CAAC,SAAiB;QACrC,IAAI,QAAQ;YAAE,OAAO,KAAK,CAAA;QAC1B,IAAI,yBAAyB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,OAAO,KAAK,CAAA;QACd,CAAC;QACD,IAAI,yBAAyB,CAAC,IAAI,IAAI,mBAAmB,EAAE,CAAC;YAC1D,MAAM,KAAK,GAAG,yBAAyB,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAA;YAC7D,IAAI,KAAK,EAAE,CAAC;gBACV,yBAAyB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gBACvC,yBAAyB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YACzC,CAAC;QACH,CAAC;QACD,MAAM,QAAQ,GAAG,eAAe,CAAC,SAAS,CAAC,CAAA;QAC3C,IAAI,QAAQ,IAAI,kBAAkB,EAAE,CAAC;YACnC,OAAO,KAAK,CAAA;QACd,CAAC;QACD,yBAAyB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QACxC,OAAO,IAAI,CAAA;IACb,CAAC;IAED,SAAS,eAAe,CAAC,SAAiB;QACxC,OAAO,yBAAyB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IACzD,CAAC;IAED,SAAS,WAAW,CAAC,SAAiB;QACpC,IAAI,QAAQ;YAAE,OAAM;QACpB,yBAAyB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAA;QACxC,IAAI,yBAAyB,CAAC,IAAI,IAAI,mBAAmB,EAAE,CAAC;YAC1D,MAAM,KAAK,GAAG,yBAAyB,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAA;YAC7D,IAAI,KAAK;gBAAE,yBAAyB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QACpD,CAAC;IACH,CAAC;IAED,SAAS,kBAAkB,CAAC,SAAiB;QAC3C,IAAI,QAAQ;YAAE,OAAM;QACpB,yBAAyB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;IAC7C,CAAC;IAED,SAAS,KAAK;QACZ,yBAAyB,CAAC,KAAK,EAAE,CAAA;QACjC,yBAAyB,CAAC,KAAK,EAAE,CAAA;IACnC,CAAC;IAED,SAAS,OAAO;QACd,IAAI,QAAQ;YAAE,OAAM;QACpB,KAAK,EAAE,CAAA;QACP,QAAQ,GAAG,IAAI,CAAA;IACjB,CAAC;IAED,OAAO;QACL,IAAI,QAAQ;YACV,OAAO,QAAQ,CAAA;QACjB,CAAC;QACD,YAAY;QACZ,eAAe;QACf,WAAW;QACX,kBAAkB;QAClB,KAAK;QACL,OAAO;KACR,CAAA;AACH,CAAC"}

package/dist/src/plugin/fetch-interceptor.d.ts

@@ -0,0 +1,59 @@
+import type { AccountManager } from './accounts';
+import type { GetAuth } from './auth';
+import type { AntigravityConfig } from './config';
+import type { AgyTransport, FetchImpl } from './dependencies';
+import type { OperatorSettingsController } from './operator-settings';
+import type { QuotaManager } from './quota';
+import { type AgySessionRegistry } from './session-context';
+import type { PluginClient } from './types';
+/**
+ * Inputs the fetch interceptor needs from the plugin bootstrap. Everything
+ * the original closure captured from `plugin.ts` now flows through this
+ * record so a fresh interceptor can be built per plugin instance without
+ * sharing state with siblings.
+ */
+export interface FetchInterceptorContext {
+    readonly client: PluginClient;
+    readonly directory: string;
+    readonly providerId: string;
+    readonly config: AntigravityConfig;
+    readonly accountManager: AccountManager;
+    readonly quotaManager: QuotaManager;
+    readonly getAuth: GetAuth;
+    readonly agySessionRegistry: AgySessionRegistry;
+    /**
+     * Live operator settings controller. Optional for backward
+     * compatibility — when present, the interceptor reads routing
+     * overrides and killswitch thresholds per request.
+     */
+    readonly operatorSettings?: OperatorSettingsController;
+    /**
+     * Transport adapter used for Antigravity HTTPS requests. Defaults to
+     * the production `fetchWithAgyCliTransport` when omitted; tests inject
+     * a deterministic stub so the e2e workspace can route calls at a mock
+     * server bound to 127.0.0.1.
+     */
+    readonly agyTransport?: AgyTransport;
+    /**
+     * HTTP primitive used for non-Antigravity URLs. Defaults to
+     * `globalThis.fetch`; tests inject a guarded stub that refuses any
+     * non-loopback target so a regression cannot silently leak a real
+     * network call.
+     */
+    readonly fetchImpl?: FetchImpl;
+}
+/**
+ * Public surface exposed to the auth-loader plumbing. `fetch` mirrors the
+ * host signature so it can be slotted into `LoaderResult` unchanged.
+ */
+export interface FetchInterceptor {
+    fetch(input: RequestInfo | URL, init?: RequestInit): Promise<Response>;
+    dispose(): void;
+}
+/**
+ * Wraps the upstream fetch with the full account-rotation, rate-limit, and
+ * quota-fallback pipeline. Each interceptor owns its retry/warmup bookkeeping
+ * so disposing the plugin releases every counter.
+ */
+export declare function createFetchInterceptor(context: FetchInterceptorContext): FetchInterceptor;
+//# sourceMappingURL=fetch-interceptor.d.ts.map

package/dist/src/plugin/fetch-interceptor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fetch-interceptor.d.ts","sourceRoot":"","sources":["../../../src/plugin/fetch-interceptor.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAOhD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAA;AAErC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAA;AAUjD,OAAO,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAA;AAqB7D,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAA;AAErE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAe3C,OAAO,EACL,KAAK,kBAAkB,EAExB,MAAM,mBAAmB,CAAA;AAE1B,OAAO,KAAK,EAAE,YAAY,EAAwB,MAAM,SAAS,CAAA;AAyHjE;;;;;GAKG;AACH,MAAM,WAAW,uBAAuB;IACtC,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAA;IAC7B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAA;IAC1B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAA;IAClC,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAA;IACvC,QAAQ,CAAC,YAAY,EAAE,YAAY,CAAA;IACnC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAA;IACzB,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,CAAA;IAC/C;;;;OAIG;IACH,QAAQ,CAAC,gBAAgB,CAAC,EAAE,0BAA0B,CAAA;IACtD;;;;;OAKG;IACH,QAAQ,CAAC,YAAY,CAAC,EAAE,YAAY,CAAA;IACpC;;;;;OAKG;IACH,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,CAAA;CAC/B;AAED;;;GAGG;AACH,MAAM,WAAW,gBAAgB;IAC/B,KAAK,CAAC,KAAK,EAAE,WAAW,GAAG,GAAG,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAA;IACtE,OAAO,IAAI,IAAI,CAAA;CAChB;AAED;;;;GAIG;AACH,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,gBAAgB,CAy1DlB"}