antigravity-auth 1.6.0 → 1.7.1

package/dist/src/plugin/quota.js

@@ -1,261 +0,0 @@
-import { ANTIGRAVITY_ENDPOINT_PROD, getAntigravityHeaders, ANTIGRAVITY_PROVIDER_ID, } from "../constants";
-import { accessTokenExpired, formatRefreshParts, parseRefreshParts } from "./auth";
-import { logQuotaFetch, logQuotaStatus } from "./debug";
-import { ensureProjectContext } from "./project";
-import { refreshAccessToken } from "./token";
-import { getModelFamily } from "./transform/model-resolver";
-const FETCH_TIMEOUT_MS = 10000;
-function buildAuthFromAccount(account) {
-    return {
-        type: "oauth",
-        refresh: formatRefreshParts({
-            refreshToken: account.refreshToken,
-            projectId: account.projectId,
-            managedProjectId: account.managedProjectId,
-        }),
-        access: undefined,
-        expires: undefined,
-    };
-}
-function normalizeRemainingFraction(value) {
-    if (typeof value !== "number" || !Number.isFinite(value)) {
-        return 0;
-    }
-    if (value < 0)
-        return 0;
-    if (value > 1)
-        return 1;
-    return value;
-}
-function parseResetTime(resetTime) {
-    if (!resetTime)
-        return null;
-    const timestamp = Date.parse(resetTime);
-    if (!Number.isFinite(timestamp)) {
-        return null;
-    }
-    return timestamp;
-}
-function classifyQuotaGroup(modelName, displayName) {
-    const combined = `${modelName} ${displayName ?? ""}`.toLowerCase();
-    if (combined.includes("claude")) {
-        return "claude";
-    }
-    const isGemini3 = combined.includes("gemini-3") || combined.includes("gemini 3");
-    if (!isGemini3) {
-        return null;
-    }
-    const family = getModelFamily(modelName);
-    return family === "gemini-flash" ? "gemini-flash" : "gemini-pro";
-}
-function aggregateQuota(models) {
-    const groups = {};
-    if (!models) {
-        return { groups, modelCount: 0 };
-    }
-    let totalCount = 0;
-    for (const [modelName, entry] of Object.entries(models)) {
-        const group = classifyQuotaGroup(modelName, entry.displayName ?? entry.modelName);
-        if (!group) {
-            continue;
-        }
-        const quotaInfo = entry.quotaInfo;
-        const remainingFraction = quotaInfo
-            ? normalizeRemainingFraction(quotaInfo.remainingFraction)
-            : undefined;
-        const resetTime = quotaInfo?.resetTime;
-        const resetTimestamp = parseResetTime(resetTime);
-        totalCount += 1;
-        const existing = groups[group];
-        const nextCount = (existing?.modelCount ?? 0) + 1;
-        const nextRemaining = remainingFraction === undefined
-            ? existing?.remainingFraction
-            : existing?.remainingFraction === undefined
-                ? remainingFraction
-                : Math.min(existing.remainingFraction, remainingFraction);
-        let nextResetTime = existing?.resetTime;
-        if (resetTimestamp !== null) {
-            if (!existing?.resetTime) {
-                nextResetTime = resetTime;
-            }
-            else {
-                const existingTimestamp = parseResetTime(existing.resetTime);
-                if (existingTimestamp === null || resetTimestamp < existingTimestamp) {
-                    nextResetTime = resetTime;
-                }
-            }
-        }
-        groups[group] = {
-            remainingFraction: nextRemaining,
-            resetTime: nextResetTime,
-            modelCount: nextCount,
-        };
-    }
-    return { groups, modelCount: totalCount };
-}
-async function fetchWithTimeout(url, options, timeoutMs = FETCH_TIMEOUT_MS) {
-    const controller = new AbortController();
-    const timeout = setTimeout(() => controller.abort(), timeoutMs);
-    try {
-        return await fetch(url, { ...options, signal: controller.signal });
-    }
-    finally {
-        clearTimeout(timeout);
-    }
-}
-async function fetchAvailableModels(accessToken, projectId) {
-    const endpoint = ANTIGRAVITY_ENDPOINT_PROD;
-    const quotaUserAgent = getAntigravityHeaders()["User-Agent"] || "antigravity/windows/amd64";
-    const errors = [];
-    const body = projectId ? { project: projectId } : {};
-    const response = await fetchWithTimeout(`${endpoint}/v1internal:fetchAvailableModels`, {
-        method: "POST",
-        headers: {
-            Authorization: `Bearer ${accessToken}`,
-            "Content-Type": "application/json",
-            "User-Agent": quotaUserAgent,
-        },
-        body: JSON.stringify(body),
-    });
-    if (response.ok) {
-        return (await response.json());
-    }
-    const message = await response.text().catch(() => "");
-    const snippet = message.trim().slice(0, 200);
-    errors.push(`fetchAvailableModels ${response.status} at ${endpoint}${snippet ? `: ${snippet}` : ""}`);
-    throw new Error(errors.join("; ") || "fetchAvailableModels failed");
-}
-async function fetchGeminiCliQuota(accessToken, projectId) {
-    const endpoint = ANTIGRAVITY_ENDPOINT_PROD;
-    const platform = process.platform || "darwin";
-    const arch = process.arch || "arm64";
-    const geminiCliUserAgent = `GeminiCLI/1.0.0/gemini-2.5-pro (${platform}; ${arch})`;
-    const body = projectId ? { project: projectId } : {};
-    try {
-        const response = await fetchWithTimeout(`${endpoint}/v1internal:retrieveUserQuota`, {
-            method: "POST",
-            headers: {
-                Authorization: `Bearer ${accessToken}`,
-                "Content-Type": "application/json",
-                "User-Agent": geminiCliUserAgent,
-            },
-            body: JSON.stringify(body),
-        });
-        if (response.ok) {
-            const data = (await response.json());
-            return data;
-        }
-        return { buckets: [] };
-    }
-    catch {
-        return { buckets: [] };
-    }
-}
-function aggregateGeminiCliQuota(response) {
-    const models = [];
-    if (!response.buckets || response.buckets.length === 0) {
-        return { models };
-    }
-    for (const bucket of response.buckets) {
-        if (!bucket.modelId) {
-            continue;
-        }
-        const modelId = bucket.modelId;
-        const isRelevantModel = modelId.startsWith("gemini-3-") ||
-            modelId === "gemini-2.5-pro";
-        if (!isRelevantModel) {
-            continue;
-        }
-        models.push({
-            modelId: bucket.modelId,
-            remainingFraction: normalizeRemainingFraction(bucket.remainingFraction),
-            resetTime: bucket.resetTime,
-        });
-    }
-    models.sort((a, b) => a.modelId.localeCompare(b.modelId));
-    return { models };
-}
-function applyAccountUpdates(account, auth) {
-    const parts = parseRefreshParts(auth.refresh);
-    if (!parts.refreshToken) {
-        return undefined;
-    }
-    const updated = {
-        ...account,
-        refreshToken: parts.refreshToken,
-        projectId: parts.projectId ?? account.projectId,
-        managedProjectId: parts.managedProjectId ?? account.managedProjectId,
-    };
-    const changed = updated.refreshToken !== account.refreshToken ||
-        updated.projectId !== account.projectId ||
-        updated.managedProjectId !== account.managedProjectId;
-    return changed ? updated : undefined;
-}
-export async function checkAccountsQuota(accounts, client, providerId = ANTIGRAVITY_PROVIDER_ID) {
-    const results = [];
-    logQuotaFetch("start", accounts.length);
-    for (const [index, account] of accounts.entries()) {
-        const disabled = account.enabled === false;
-        let auth = buildAuthFromAccount(account);
-        try {
-            if (accessTokenExpired(auth)) {
-                const refreshed = await refreshAccessToken(auth, client, providerId);
-                if (!refreshed) {
-                    throw new Error("Token refresh failed");
-                }
-                auth = refreshed;
-            }
-            const projectContext = await ensureProjectContext(auth);
-            auth = projectContext.auth;
-            const updatedAccount = applyAccountUpdates(account, auth);
-            let quotaResult;
-            let geminiCliQuotaResult;
-            const [antigravityResponse, geminiCliResponse] = await Promise.all([
-                fetchAvailableModels(auth.access ?? "", projectContext.effectiveProjectId)
-                    .catch((error) => ({ models: undefined })),
-                fetchGeminiCliQuota(auth.access ?? "", projectContext.effectiveProjectId),
-            ]);
-            if (antigravityResponse.models === undefined) {
-                quotaResult = {
-                    groups: {},
-                    modelCount: 0,
-                    error: "Failed to fetch Antigravity quota",
-                };
-            }
-            else {
-                quotaResult = aggregateQuota(antigravityResponse.models);
-            }
-            geminiCliQuotaResult = aggregateGeminiCliQuota(geminiCliResponse);
-            if (geminiCliResponse.buckets === undefined || geminiCliResponse.buckets.length === 0) {
-                geminiCliQuotaResult.error = geminiCliQuotaResult.models.length === 0
-                    ? "No Gemini CLI quota available"
-                    : undefined;
-            }
-            results.push({
-                index,
-                email: account.email,
-                status: "ok",
-                disabled,
-                quota: quotaResult,
-                geminiCliQuota: geminiCliQuotaResult,
-                updatedAccount,
-            });
-            for (const [family, groupQuota] of Object.entries(quotaResult.groups)) {
-                const remainingPercent = (groupQuota.remainingFraction ?? 0) * 100;
-                logQuotaStatus(account.email, index, remainingPercent, family);
-            }
-        }
-        catch (error) {
-            results.push({
-                index,
-                email: account.email,
-                status: "error",
-                disabled,
-                error: error instanceof Error ? error.message : String(error),
-            });
-            logQuotaFetch("error", undefined, `account=${account.email ?? index} error=${error instanceof Error ? error.message : String(error)}`);
-        }
-    }
-    logQuotaFetch("complete", accounts.length, `ok=${results.filter(r => r.status === "ok").length} errors=${results.filter(r => r.status === "error").length}`);
-    return results;
-}

package/dist/src/plugin/recovery/constants.d.ts

@@ -1,21 +0,0 @@
-/**
- * Constants for session recovery storage paths.
- *
- * Based on oh-my-opencode/src/hooks/session-recovery/constants.ts
- */
-/**
- * Get the XDG config directory for Antigravity config.
- * Falls back to ~/.config on Linux/Mac, or APPDATA on Windows.
- */
-export declare function getXdgConfig(): string;
-/**
- * Get the Antigravity config directory.
- * Default: ~/.config/opencode/antigravity.json
- */
-export declare function getAntigravityConfigDir(): string;
-export declare const OPENCODE_STORAGE: string;
-export declare const MESSAGE_STORAGE: string;
-export declare const PART_STORAGE: string;
-export declare const THINKING_TYPES: Set<string>;
-export declare const META_TYPES: Set<string>;
-export declare const CONTENT_TYPES: Set<string>;

package/dist/src/plugin/recovery/constants.js

@@ -1,42 +0,0 @@
-/**
- * Constants for session recovery storage paths.
- *
- * Based on oh-my-opencode/src/hooks/session-recovery/constants.ts
- */
-import { join } from "node:path";
-import { homedir } from "node:os";
-/**
- * Get the XDG data directory for OpenCode storage.
- * Falls back to ~/.local/share on Linux/Mac, or APPDATA on Windows.
- */
-function getXdgData() {
-    const platform = process.platform;
-    if (platform === "win32") {
-        return process.env.APPDATA || join(homedir(), "AppData", "Roaming");
-    }
-    return process.env.XDG_DATA_HOME || join(homedir(), ".local", "share");
-}
-/**
- * Get the XDG config directory for Antigravity config.
- * Falls back to ~/.config on Linux/Mac, or APPDATA on Windows.
- */
-export function getXdgConfig() {
-    const platform = process.platform;
-    if (platform === "win32") {
-        return process.env.APPDATA || join(homedir(), "AppData", "Roaming");
-    }
-    return process.env.XDG_CONFIG_HOME || join(homedir(), ".config");
-}
-/**
- * Get the Antigravity config directory.
- * Default: ~/.config/opencode/antigravity.json
- */
-export function getAntigravityConfigDir() {
-    return join(getXdgConfig(), "opencode");
-}
-export const OPENCODE_STORAGE = join(getXdgData(), "opencode", "storage");
-export const MESSAGE_STORAGE = join(OPENCODE_STORAGE, "message");
-export const PART_STORAGE = join(OPENCODE_STORAGE, "part");
-export const THINKING_TYPES = new Set(["thinking", "redacted_thinking", "reasoning"]);
-export const META_TYPES = new Set(["step-start", "step-finish"]);
-export const CONTENT_TYPES = new Set(["text", "tool", "tool_use", "tool_result"]);

package/dist/src/plugin/recovery/index.d.ts

@@ -1,11 +0,0 @@
-/**
- * Session recovery module for opencode-antigravity-auth.
- *
- * Provides recovery from:
- * - tool_result_missing: Interrupted tool executions
- * - thinking_block_order: Corrupted thinking blocks
- * - thinking_disabled_violation: Thinking in non-thinking model
- */
-export * from "./types";
-export * from "./constants";
-export * from "./storage";

package/dist/src/plugin/recovery/index.js

@@ -1,11 +0,0 @@
-/**
- * Session recovery module for opencode-antigravity-auth.
- *
- * Provides recovery from:
- * - tool_result_missing: Interrupted tool executions
- * - thinking_block_order: Corrupted thinking blocks
- * - thinking_disabled_violation: Thinking in non-thinking model
- */
-export * from "./types";
-export * from "./constants";
-export * from "./storage";

package/dist/src/plugin/recovery/storage.d.ts

@@ -1,23 +0,0 @@
-/**
- * Storage utilities for reading OpenCode's session data.
- *
- * Based on oh-my-opencode/src/hooks/session-recovery/storage.ts
- */
-import type { StoredMessageMeta, StoredPart } from "./types";
-export declare function generatePartId(): string;
-export declare function getMessageDir(sessionID: string): string;
-export declare function readMessages(sessionID: string): StoredMessageMeta[];
-export declare function readParts(messageID: string): StoredPart[];
-export declare function hasContent(part: StoredPart): boolean;
-export declare function messageHasContent(messageID: string): boolean;
-export declare function injectTextPart(sessionID: string, messageID: string, text: string): boolean;
-export declare function findMessagesWithThinkingBlocks(sessionID: string): string[];
-export declare function findMessagesWithThinkingOnly(sessionID: string): string[];
-export declare function findMessagesWithOrphanThinking(sessionID: string): string[];
-export declare function prependThinkingPart(sessionID: string, messageID: string): boolean;
-export declare function stripThinkingParts(messageID: string): boolean;
-export declare function findEmptyMessages(sessionID: string): string[];
-export declare function findEmptyMessageByIndex(sessionID: string, targetIndex: number): string | null;
-export declare function findMessageByIndexNeedingThinking(sessionID: string, targetIndex: number): string | null;
-export declare function replaceEmptyTextParts(messageID: string, replacementText: string): boolean;
-export declare function findMessagesWithEmptyTextParts(sessionID: string): string[];