opencode-agy-auth 2.0.0

package/dist/src/plugin/project.js

@@ -0,0 +1,234 @@
+import { getAntigravityHeaders, ANTIGRAVITY_ENDPOINT_FALLBACKS, ANTIGRAVITY_LOAD_ENDPOINTS, ANTIGRAVITY_DEFAULT_PROJECT_ID, } from "../constants";
+import { formatRefreshParts, parseRefreshParts } from "./auth";
+import { createLogger } from "./logger";
+const log = createLogger("project");
+const projectContextResultCache = new Map();
+const projectContextPendingCache = new Map();
+const CODE_ASSIST_METADATA = {
+    ideType: "ANTIGRAVITY",
+    platform: process.platform === "win32" ? "WINDOWS" : "MACOS",
+    pluginType: "GEMINI",
+};
+function buildMetadata(projectId) {
+    const metadata = {
+        ideType: CODE_ASSIST_METADATA.ideType,
+        platform: CODE_ASSIST_METADATA.platform,
+        pluginType: CODE_ASSIST_METADATA.pluginType,
+    };
+    if (projectId) {
+        metadata.duetProject = projectId;
+    }
+    return metadata;
+}
+/**
+ * Selects the default tier ID from the allowed tiers list.
+ */
+function getDefaultTierId(allowedTiers) {
+    if (!allowedTiers || allowedTiers.length === 0) {
+        return undefined;
+    }
+    for (const tier of allowedTiers) {
+        if (tier?.isDefault) {
+            return tier.id;
+        }
+    }
+    return allowedTiers[0]?.id;
+}
+/**
+ * Promise-based delay utility.
+ */
+function wait(ms) {
+    return new Promise(function (resolve) {
+        setTimeout(resolve, ms);
+    });
+}
+/**
+ * Extracts the cloudaicompanion project id from loadCodeAssist responses.
+ */
+function extractManagedProjectId(payload) {
+    if (!payload) {
+        return undefined;
+    }
+    if (typeof payload.cloudaicompanionProject === "string") {
+        return payload.cloudaicompanionProject;
+    }
+    if (payload.cloudaicompanionProject && typeof payload.cloudaicompanionProject.id === "string") {
+        return payload.cloudaicompanionProject.id;
+    }
+    return undefined;
+}
+/**
+ * Generates a cache key for project context based on refresh token.
+ */
+function getCacheKey(auth) {
+    const refresh = auth.refresh?.trim();
+    return refresh ? refresh : undefined;
+}
+/**
+ * Clears cached project context results and pending promises, globally or for a refresh key.
+ */
+export function invalidateProjectContextCache(refresh) {
+    if (!refresh) {
+        projectContextPendingCache.clear();
+        projectContextResultCache.clear();
+        return;
+    }
+    projectContextPendingCache.delete(refresh);
+    projectContextResultCache.delete(refresh);
+}
+/**
+ * Loads managed project information for the given access token and optional project.
+ */
+export async function loadManagedProject(accessToken, projectId) {
+    const metadata = buildMetadata(projectId);
+    const requestBody = { metadata };
+    const loadHeaders = {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${accessToken}`,
+        "User-Agent": "google-api-nodejs-client/9.15.1",
+        "X-Goog-Api-Client": "google-cloud-sdk vscode_cloudshelleditor/0.1",
+        "Client-Metadata": getAntigravityHeaders()["Client-Metadata"],
+    };
+    const loadEndpoints = Array.from(new Set([...ANTIGRAVITY_LOAD_ENDPOINTS, ...ANTIGRAVITY_ENDPOINT_FALLBACKS]));
+    for (const baseEndpoint of loadEndpoints) {
+        try {
+            const response = await fetch(`${baseEndpoint}/v1internal:loadCodeAssist`, {
+                method: "POST",
+                headers: loadHeaders,
+                body: JSON.stringify(requestBody),
+            });
+            if (!response.ok) {
+                continue;
+            }
+            return (await response.json());
+        }
+        catch (error) {
+            log.debug("Failed to load managed project", { endpoint: baseEndpoint, error: String(error) });
+            continue;
+        }
+    }
+    return null;
+}
+/**
+ * Onboards a managed project for the user, optionally retrying until completion.
+ */
+export async function onboardManagedProject(accessToken, tierId, projectId, attempts = 10, delayMs = 5000) {
+    const metadata = buildMetadata(projectId);
+    const requestBody = {
+        tierId,
+        metadata,
+    };
+    for (const baseEndpoint of ANTIGRAVITY_ENDPOINT_FALLBACKS) {
+        for (let attempt = 0; attempt < attempts; attempt += 1) {
+            try {
+                const response = await fetch(`${baseEndpoint}/v1internal:onboardUser`, {
+                    method: "POST",
+                    headers: {
+                        "Content-Type": "application/json",
+                        Authorization: `Bearer ${accessToken}`,
+                        ...getAntigravityHeaders(),
+                    },
+                    body: JSON.stringify(requestBody),
+                });
+                if (!response.ok) {
+                    break;
+                }
+                const payload = (await response.json());
+                const managedProjectId = payload.response?.cloudaicompanionProject?.id;
+                if (payload.done && managedProjectId) {
+                    return managedProjectId;
+                }
+                if (payload.done && projectId) {
+                    return projectId;
+                }
+            }
+            catch (error) {
+                log.debug("Failed to onboard managed project", { endpoint: baseEndpoint, error: String(error) });
+                break;
+            }
+            await wait(delayMs);
+        }
+    }
+    return undefined;
+}
+/**
+ * Resolves an effective project ID for the current auth state, caching results per refresh token.
+ */
+export async function ensureProjectContext(auth) {
+    const accessToken = auth.access;
+    if (!accessToken) {
+        return { auth, effectiveProjectId: "" };
+    }
+    const cacheKey = getCacheKey(auth);
+    if (cacheKey) {
+        const cached = projectContextResultCache.get(cacheKey);
+        if (cached) {
+            return cached;
+        }
+        const pending = projectContextPendingCache.get(cacheKey);
+        if (pending) {
+            return pending;
+        }
+    }
+    const resolveContext = async () => {
+        const parts = parseRefreshParts(auth.refresh);
+        if (parts.managedProjectId) {
+            return { auth, effectiveProjectId: parts.managedProjectId };
+        }
+        const fallbackProjectId = ANTIGRAVITY_DEFAULT_PROJECT_ID;
+        const persistManagedProject = async (managedProjectId) => {
+            const updatedAuth = {
+                ...auth,
+                refresh: formatRefreshParts({
+                    refreshToken: parts.refreshToken,
+                    projectId: parts.projectId,
+                    managedProjectId,
+                }),
+            };
+            return { auth: updatedAuth, effectiveProjectId: managedProjectId };
+        };
+        // Try to resolve a managed project from Antigravity if possible.
+        const loadPayload = await loadManagedProject(accessToken, parts.projectId ?? fallbackProjectId);
+        const resolvedManagedProjectId = extractManagedProjectId(loadPayload);
+        if (resolvedManagedProjectId) {
+            return persistManagedProject(resolvedManagedProjectId);
+        }
+        // No managed project found - try to auto-provision one via onboarding.
+        // This handles accounts that were added before managed project provisioning was required.
+        const tierId = getDefaultTierId(loadPayload?.allowedTiers) ?? "FREE";
+        log.debug("Auto-provisioning managed project", { tierId, projectId: parts.projectId });
+        const provisionedProjectId = await onboardManagedProject(accessToken, tierId, parts.projectId);
+        if (provisionedProjectId) {
+            log.debug("Successfully provisioned managed project", { provisionedProjectId });
+            return persistManagedProject(provisionedProjectId);
+        }
+        log.warn("Failed to provision managed project - account may not work correctly", {
+            hasProjectId: !!parts.projectId,
+        });
+        if (parts.projectId) {
+            return { auth, effectiveProjectId: parts.projectId };
+        }
+        // No project id present in auth; fall back to the hardcoded id for requests.
+        return { auth, effectiveProjectId: fallbackProjectId };
+    };
+    if (!cacheKey) {
+        return resolveContext();
+    }
+    const promise = resolveContext()
+        .then((result) => {
+        const nextKey = getCacheKey(result.auth) ?? cacheKey;
+        projectContextPendingCache.delete(cacheKey);
+        projectContextResultCache.set(nextKey, result);
+        if (nextKey !== cacheKey) {
+            projectContextResultCache.delete(cacheKey);
+        }
+        return result;
+    })
+        .catch((error) => {
+        projectContextPendingCache.delete(cacheKey);
+        throw error;
+    });
+    projectContextPendingCache.set(cacheKey, promise);
+    return promise;
+}
+//# sourceMappingURL=project.js.map

package/dist/src/plugin/project.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"project.js","sourceRoot":"","sources":["../../../src/plugin/project.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,8BAA8B,EAC9B,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAGxC,MAAM,GAAG,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;AAEpC,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAgC,CAAC;AAC1E,MAAM,0BAA0B,GAAG,IAAI,GAAG,EAAyC,CAAC;AAEpF,MAAM,oBAAoB,GAAG;IAC3B,OAAO,EAAE,aAAa;IACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;IAC5D,UAAU,EAAE,QAAQ;CACZ,CAAC;AAyBX,SAAS,aAAa,CAAC,SAAkB;IACvC,MAAM,QAAQ,GAA2B;QACvC,OAAO,EAAE,oBAAoB,CAAC,OAAO;QACrC,QAAQ,EAAE,oBAAoB,CAAC,QAAQ;QACvC,UAAU,EAAE,oBAAoB,CAAC,UAAU;KAC5C,CAAC;IACF,IAAI,SAAS,EAAE,CAAC;QACd,QAAQ,CAAC,WAAW,GAAG,SAAS,CAAC;IACnC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,YAAoC;IAC5D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,IAAI,IAAI,EAAE,SAAS,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,SAAS,IAAI,CAAC,EAAU;IACtB,OAAO,IAAI,OAAO,CAAC,UAAU,OAAO;QAClC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,OAAqC;IACpE,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,OAAO,OAAO,CAAC,uBAAuB,KAAK,QAAQ,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC,uBAAuB,CAAC;IACzC,CAAC;IACD,IAAI,OAAO,CAAC,uBAAuB,IAAI,OAAO,OAAO,CAAC,uBAAuB,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;QAC9F,OAAO,OAAO,CAAC,uBAAuB,CAAC,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,IAAsB;IACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;IACrC,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AACvC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,6BAA6B,CAAC,OAAgB;IAC5D,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,0BAA0B,CAAC,KAAK,EAAE,CAAC;QACnC,yBAAyB,CAAC,KAAK,EAAE,CAAC;QAClC,OAAO;IACT,CAAC;IACD,0BAA0B,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3C,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,WAAmB,EACnB,SAAkB;IAElB,MAAM,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC1C,MAAM,WAAW,GAA4B,EAAE,QAAQ,EAAE,CAAC;IAE1D,MAAM,WAAW,GAA2B;QAC1C,cAAc,EAAE,kBAAkB;QAClC,aAAa,EAAE,UAAU,WAAW,EAAE;QACtC,YAAY,EAAE,iCAAiC;QAC/C,mBAAmB,EAAE,8CAA8C;QACnE,iBAAiB,EAAE,qBAAqB,EAAE,CAAC,iBAAiB,CAAC;KAC9D,CAAC;IAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAC9B,IAAI,GAAG,CAAS,CAAC,GAAG,0BAA0B,EAAE,GAAG,8BAA8B,CAAC,CAAC,CACpF,CAAC;IAEF,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,YAAY,4BAA4B,EAC3C;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;aAClC,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA0B,CAAC;QAC1D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC9F,SAAS;QACX,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAGD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,WAAmB,EACnB,MAAc,EACd,SAAkB,EAClB,QAAQ,GAAG,EAAE,EACb,OAAO,GAAG,IAAI;IAEd,MAAM,QAAQ,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC1C,MAAM,WAAW,GAA4B;QAC3C,MAAM;QACN,QAAQ;KACT,CAAC;IAEF,KAAK,MAAM,YAAY,IAAI,8BAA8B,EAAE,CAAC;QAC1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,QAAQ,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,YAAY,yBAAyB,EACxC;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;wBAClC,aAAa,EAAE,UAAU,WAAW,EAAE;wBACtC,GAAG,qBAAqB,EAAE;qBAC3B;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;iBAClC,CACF,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM;gBACR,CAAC;gBAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuB,CAAC;gBAC9D,MAAM,gBAAgB,GAAG,OAAO,CAAC,QAAQ,EAAE,uBAAuB,EAAE,EAAE,CAAC;gBACvE,IAAI,OAAO,CAAC,IAAI,IAAI,gBAAgB,EAAE,CAAC;oBACrC,OAAO,gBAAgB,CAAC;gBAC1B,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,IAAI,SAAS,EAAE,CAAC;oBAC9B,OAAO,SAAS,CAAC;gBACnB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACjG,MAAM;YACR,CAAC;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,IAAsB;IAC/D,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC;IAChC,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,EAAE,EAAE,CAAC;IAC1C,CAAC;IAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,MAAM,GAAG,yBAAyB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,MAAM,OAAO,GAAG,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,KAAK,IAAmC,EAAE;QAC/D,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAC3B,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC9D,CAAC;QAED,MAAM,iBAAiB,GAAG,8BAA8B,CAAC;QACzD,MAAM,qBAAqB,GAAG,KAAK,EAAE,gBAAwB,EAAiC,EAAE;YAC9F,MAAM,WAAW,GAAqB;gBACpC,GAAG,IAAI;gBACP,OAAO,EAAE,kBAAkB,CAAC;oBAC1B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;oBAC1B,gBAAgB;iBACjB,CAAC;aACH,CAAC;YAEF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,CAAC;QACrE,CAAC,CAAC;QAEF,iEAAiE;QACjE,MAAM,WAAW,GAAG,MAAM,kBAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,SAAS,IAAI,iBAAiB,CAAC,CAAC;QAChG,MAAM,wBAAwB,GAAG,uBAAuB,CAAC,WAAW,CAAC,CAAC;QAEtE,IAAI,wBAAwB,EAAE,CAAC;YAC7B,OAAO,qBAAqB,CAAC,wBAAwB,CAAC,CAAC;QACzD,CAAC;QAED,uEAAuE;QACvE,0FAA0F;QAC1F,MAAM,MAAM,GAAG,gBAAgB,CAAC,WAAW,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC;QACrE,GAAG,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;QAEvF,MAAM,oBAAoB,GAAG,MAAM,qBAAqB,CACtD,WAAW,EACX,MAAM,EACN,KAAK,CAAC,SAAS,CAChB,CAAC;QAEF,IAAI,oBAAoB,EAAE,CAAC;YACzB,GAAG,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAChF,OAAO,qBAAqB,CAAC,oBAAoB,CAAC,CAAC;QACrD,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,sEAAsE,EAAE;YAC/E,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,SAAS;SAChC,CAAC,CAAC;QAEH,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;QACvD,CAAC;QAED,6EAA6E;QAC7E,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;IACzD,CAAC,CAAC;IAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,cAAc,EAAE,CAAC;IAC1B,CAAC;IAED,MAAM,OAAO,GAAG,cAAc,EAAE;SAC7B,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;QACf,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC;QACrD,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,yBAAyB,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC/C,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;YACzB,yBAAyB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACf,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,KAAK,CAAC;IACd,CAAC,CAAC,CAAC;IAEL,0BAA0B,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAClD,OAAO,OAAO,CAAC;AACjB,CAAC"}

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

@@ -0,0 +1,2 @@
+export declare function configureProxy(): void;
+//# sourceMappingURL=proxy.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"proxy.d.ts","sourceRoot":"","sources":["../../../src/plugin/proxy.ts"],"names":[],"mappings":"AAKA,wBAAgB,cAAc,IAAI,IAAI,CAgBrC"}

package/dist/src/plugin/proxy.js

@@ -0,0 +1,20 @@
+import { ProxyAgent, setGlobalDispatcher } from "undici";
+import { createLogger } from "./logger";
+const log = createLogger("proxy");
+export function configureProxy() {
+    const proxyUrl = process.env.HTTPS_PROXY ||
+        process.env.HTTP_PROXY ||
+        process.env.https_proxy ||
+        process.env.http_proxy;
+    if (proxyUrl) {
+        try {
+            const dispatcher = new ProxyAgent(proxyUrl);
+            setGlobalDispatcher(dispatcher);
+            log.info(`Proxy configured using ${proxyUrl}`);
+        }
+        catch (error) {
+            log.warn(`Failed to configure proxy: ${error}`);
+        }
+    }
+}
+//# sourceMappingURL=proxy.js.map

package/dist/src/plugin/proxy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy.js","sourceRoot":"","sources":["../../../src/plugin/proxy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,mBAAmB,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAExC,MAAM,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;AAElC,MAAM,UAAU,cAAc;IAC5B,MAAM,QAAQ,GACZ,OAAO,CAAC,GAAG,CAAC,WAAW;QACvB,OAAO,CAAC,GAAG,CAAC,UAAU;QACtB,OAAO,CAAC,GAAG,CAAC,WAAW;QACvB,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAEzB,IAAI,QAAQ,EAAE,CAAC;QACb,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC5C,mBAAmB,CAAC,UAAU,CAAC,CAAC;YAChC,GAAG,CAAC,IAAI,CAAC,0BAA0B,QAAQ,EAAE,CAAC,CAAC;QACjD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,IAAI,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;AACH,CAAC"}

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

@@ -0,0 +1,58 @@
+import type { ModelLimits } from "../constants";
+import type { PluginClient } from "./types";
+import type { AccountMetadataV3 } from "./storage";
+export type QuotaGroup = "claude" | "gemini-pro" | "gemini-flash";
+export interface QuotaModelEntry {
+    model: string;
+    displayName?: string;
+    remainingFraction: number;
+    resetTime?: string | number;
+}
+export interface QuotaSummary {
+    groups: Partial<Record<QuotaGroup, QuotaModelEntry[]>>;
+    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;
+}
+interface FetchAvailableModelEntry {
+    quotaInfo?: {
+        remainingFraction?: number;
+        resetTime?: string;
+    };
+    displayName?: string;
+    modelName?: string;
+    inputTokenLimit?: number;
+    thinkingBudgetMax?: number;
+}
+export declare function getGroupMinFraction(entries: QuotaModelEntry[]): number | undefined;
+export declare function getGroupEarliestReset(entries: QuotaModelEntry[]): string | undefined;
+export declare function checkAccountsQuota(accounts: AccountMetadataV3[], client: PluginClient, providerId?: string): Promise<AccountQuotaResult[]>;
+export declare function readModelLimits(configDir: string): Promise<ModelLimits>;
+export declare function discoverAndPersistModelLimits(configDir: string, quotaModels: Map<string, FetchAvailableModelEntry>): Promise<void>;
+/**
+ * Resolves the effective thinking budget to use for a request.
+ * Priority: variant config > tier from model suffix > server default
+ * Special: -1 means server-controlled (pass through, don't send budget)
+ */
+export declare function resolveEffectiveThinkingBudget(serverBudget: number | undefined, variantBudget: number | undefined, modelDefault: number): number;
+export {};
+//# sourceMappingURL=quota.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"quota.d.ts","sourceRoot":"","sources":["../../../src/plugin/quota.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAMhD,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,CAAC;AAElE,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;IACvD,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAGD,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;AAMD,UAAU,wBAAwB;IAChC,SAAS,CAAC,EAAE;QACV,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAmGD,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,eAAe,EAAE,GAAG,MAAM,GAAG,SAAS,CAKlF;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,eAAe,EAAE,GAAG,MAAM,GAAG,SAAS,CAkCpF;AAkLD,wBAAsB,kBAAkB,CACtC,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,MAAM,EAAE,YAAY,EACpB,UAAU,SAA0B,GACnC,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAiF/B;AAED,wBAAsB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CA0B7E;AAED,wBAAsB,6BAA6B,CACjD,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,GACjD,OAAO,CAAC,IAAI,CAAC,CAuDf;AAED;;;;GAIG;AACH,wBAAgB,8BAA8B,CAC5C,YAAY,EAAE,MAAM,GAAG,SAAS,EAChC,aAAa,EAAE,MAAM,GAAG,SAAS,EACjC,YAAY,EAAE,MAAM,GACnB,MAAM,CASR"}