antigravity-auth 1.6.0 → 1.7.1

package/dist/plugin/config/updater.js

@@ -1,205 +0,0 @@
-/**
- * OpenCode configuration file updater.
- *
- * Updates ~/.config/opencode/opencode.json(c) with plugin models.
- */
-import { existsSync, readFileSync, writeFileSync, mkdirSync } from "node:fs";
-import { join, dirname, resolve } from "node:path";
-import { homedir } from "node:os";
-import { OPENCODE_MODEL_DEFINITIONS } from "./models";
-function isRecord(value) {
-    return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-function definitionToPlainObject(def) {
-    return JSON.parse(JSON.stringify(def));
-}
-// =============================================================================
-// =============================================================================
-const PLUGIN_NAME = "opencode-antigravity-auth@latest";
-const SCHEMA_URL = "https://opencode.ai/config.json";
-const OPENCODE_JSON_FILENAME = "opencode.json";
-const OPENCODE_JSONC_FILENAME = "opencode.jsonc";
-export function stripJsonCommentsAndTrailingCommas(json) {
-    return json
-        .replace(/\\"|"(?:\\"|[^"])*"|(\/\/.*|\/\*[\s\S]*?\*\/)/g, (match, group) => (group ? "" : match))
-        .replace(/,(\s*[}\]])/g, "$1");
-}
-/**
- * Get the opencode config directory path.
- */
-export function getOpencodeConfigDir() {
-    const xdgConfig = process.env.XDG_CONFIG_HOME || join(homedir(), ".config");
-    return join(xdgConfig, "opencode");
-}
-/**
- * Get the opencode config file path.
- *
- * Prefers opencode.jsonc when present so we update the active config file
- * instead of creating a new opencode.json.
- */
-export function getOpencodeConfigPath() {
-    const configDir = getOpencodeConfigDir();
-    const jsoncPath = join(configDir, OPENCODE_JSONC_FILENAME);
-    const jsonPath = join(configDir, OPENCODE_JSON_FILENAME);
-    if (existsSync(jsoncPath)) {
-        return jsoncPath;
-    }
-    if (existsSync(jsonPath)) {
-        return jsonPath;
-    }
-    return jsonPath;
-}
-/**
- * Resolves the active OpenCode config file: explicit `configPath`, then `OPENCODE_CONFIG`,
- * then default XDG layout (jsonc preferred).
- */
-export function resolveActiveOpencodeConfigPath(options = {}) {
-    if (options.configPath) {
-        return resolve(options.configPath);
-    }
-    const envSource = options.env ?? process.env;
-    const override = envSource.OPENCODE_CONFIG?.trim();
-    if (override) {
-        return resolve(override);
-    }
-    return getOpencodeConfigPath();
-}
-// =============================================================================
-// =============================================================================
-/**
- * Updates the opencode configuration file with plugin models.
- *
- * This function:
- * 1. Reads existing opencode.json/opencode.jsonc (or creates default structure)
- * 2. Replaces `provider.google.models` with plugin models
- * 3. Writes back to disk with proper formatting
- *
- * Preserves:
- * - $schema and other top-level config keys
- * - Non-google provider sections
- * - Other settings within google provider (except models)
- *
- * @param options - Optional configuration (e.g., custom configPath for testing)
- * @returns UpdateConfigResult with success status and path
- */
-export async function updateOpencodeConfig(options = {}) {
-    const configPath = resolveActiveOpencodeConfigPath(options);
-    try {
-        let config;
-        if (existsSync(configPath)) {
-            const content = readFileSync(configPath, "utf-8");
-            config = JSON.parse(stripJsonCommentsAndTrailingCommas(content));
-        }
-        else {
-            config = {
-                $schema: SCHEMA_URL,
-                plugin: [],
-                provider: {},
-            };
-        }
-        if (!config.$schema) {
-            config.$schema = SCHEMA_URL;
-        }
-        if (!config.provider) {
-            config.provider = {};
-        }
-        if (!config.provider.google) {
-            config.provider.google = {};
-        }
-        config.provider.google.models = { ...OPENCODE_MODEL_DEFINITIONS };
-        const configDir = dirname(configPath);
-        if (!existsSync(configDir)) {
-            mkdirSync(configDir, { recursive: true });
-        }
-        writeFileSync(configPath, JSON.stringify(config, null, 2), "utf-8");
-        return {
-            success: true,
-            configPath,
-        };
-    }
-    catch (error) {
-        return {
-            success: false,
-            configPath,
-            error: error instanceof Error ? error.message : String(error),
-        };
-    }
-}
-/**
- * Merges built-in Antigravity/Google model definitions into `provider.google.models`
- * without removing user-defined model ids.
- *
- * For each model id in {@link OPENCODE_MODEL_DEFINITIONS}, plugin fields overwrite
- * the same keys on the existing entry so names/limits/variants stay current.
- * Custom models (ids not in the plugin map) are left unchanged.
- *
- * Use on plugin startup; use {@link updateOpencodeConfig} when a full replace is intended.
- */
-export async function mergeAntigravityGoogleModelsIntoOpencodeConfig(options = {}) {
-    const configPath = resolveActiveOpencodeConfigPath(options);
-    try {
-        let config;
-        if (existsSync(configPath)) {
-            const content = readFileSync(configPath, "utf-8");
-            config = JSON.parse(stripJsonCommentsAndTrailingCommas(content));
-        }
-        else {
-            config = {
-                $schema: SCHEMA_URL,
-                plugin: [],
-                provider: {},
-            };
-        }
-        let needsWrite = false;
-        if (!config.$schema) {
-            config.$schema = SCHEMA_URL;
-            needsWrite = true;
-        }
-        if (!config.provider) {
-            config.provider = {};
-            needsWrite = true;
-        }
-        if (!config.provider.google) {
-            config.provider.google = {};
-            needsWrite = true;
-        }
-        const google = config.provider.google;
-        const existingModels = isRecord(google.models) ? { ...google.models } : {};
-        let modelsTouched = false;
-        for (const [id, def] of Object.entries(OPENCODE_MODEL_DEFINITIONS)) {
-            const pluginPlain = definitionToPlainObject(def);
-            const prev = isRecord(existingModels[id])
-                ? { ...existingModels[id] }
-                : {};
-            const next = { ...prev, ...pluginPlain };
-            const prevNorm = isRecord(existingModels[id]) ? existingModels[id] : {};
-            if (JSON.stringify(next) !== JSON.stringify(prevNorm)) {
-                modelsTouched = true;
-            }
-            existingModels[id] = next;
-        }
-        if (modelsTouched) {
-            google.models = existingModels;
-            needsWrite = true;
-        }
-        if (!needsWrite) {
-            return { success: true, configPath };
-        }
-        const configDir = dirname(configPath);
-        if (!existsSync(configDir)) {
-            mkdirSync(configDir, { recursive: true });
-        }
-        writeFileSync(configPath, JSON.stringify(config, null, 2), "utf-8");
-        return {
-            success: true,
-            configPath,
-        };
-    }
-    catch (error) {
-        return {
-            success: false,
-            configPath,
-            error: error instanceof Error ? error.message : String(error),
-        };
-    }
-}

package/dist/plugin/core/streaming/index.d.ts

@@ -1,2 +0,0 @@
-export * from './types';
-export * from './transformer';

package/dist/plugin/core/streaming/index.js

@@ -1,2 +0,0 @@
-export * from './types';
-export * from './transformer';

package/dist/plugin/core/streaming/transformer.d.ts

@@ -1,9 +0,0 @@
-import type { SignatureStore, StreamingCallbacks, StreamingOptions, ThoughtBuffer } from './types';
-export declare function createThoughtBuffer(): ThoughtBuffer;
-export declare function transformStreamingPayload(payload: string, transformThinkingParts?: (response: unknown) => unknown): string;
-export declare function deduplicateThinkingText(response: unknown, sentBuffer: ThoughtBuffer, displayedThinkingHashes?: Set<string>): unknown;
-export declare function transformSseLine(line: string, signatureStore: SignatureStore, thoughtBuffer: ThoughtBuffer, sentThinkingBuffer: ThoughtBuffer, callbacks: StreamingCallbacks, options: StreamingOptions, debugState: {
-    injected: boolean;
-}): string;
-export declare function cacheThinkingSignaturesFromResponse(response: unknown, signatureSessionKey: string, signatureStore: SignatureStore, thoughtBuffer: ThoughtBuffer, onCacheSignature?: (sessionKey: string, text: string, signature: string) => void): void;
-export declare function createStreamingTransformer(signatureStore: SignatureStore, callbacks: StreamingCallbacks, options?: StreamingOptions): TransformStream<Uint8Array, Uint8Array>;

package/dist/plugin/core/streaming/transformer.js

@@ -1,301 +0,0 @@
-import { processImageData } from '../../image-saver';
-/**
- * Simple string hash for thinking deduplication.
- * Uses DJB2-like algorithm.
- */
-function hashString(str) {
-    let hash = 5381;
-    for (let i = 0; i < str.length; i++) {
-        hash = ((hash << 5) + hash) + str.charCodeAt(i); /* hash * 33 + c */
-    }
-    return (hash >>> 0).toString(16);
-}
-export function createThoughtBuffer() {
-    const buffer = new Map();
-    return {
-        get: (index) => buffer.get(index),
-        set: (index, text) => buffer.set(index, text),
-        clear: () => buffer.clear(),
-    };
-}
-export function transformStreamingPayload(payload, transformThinkingParts) {
-    return payload
-        .split('\n')
-        .map((line) => {
-        if (!line.startsWith('data:')) {
-            return line;
-        }
-        const json = line.slice(5).trim();
-        if (!json) {
-            return line;
-        }
-        try {
-            const parsed = JSON.parse(json);
-            if (parsed.response !== undefined) {
-                const transformed = transformThinkingParts
-                    ? transformThinkingParts(parsed.response)
-                    : parsed.response;
-                return `data: ${JSON.stringify(transformed)}`;
-            }
-        }
-        catch (_) { }
-        return line;
-    })
-        .join('\n');
-}
-export function deduplicateThinkingText(response, sentBuffer, displayedThinkingHashes) {
-    if (!response || typeof response !== 'object')
-        return response;
-    const resp = response;
-    if (Array.isArray(resp.candidates)) {
-        const newCandidates = resp.candidates.map((candidate, index) => {
-            const cand = candidate;
-            if (!cand?.content)
-                return candidate;
-            const content = cand.content;
-            if (!Array.isArray(content.parts))
-                return candidate;
-            const newParts = content.parts.map((part) => {
-                const p = part;
-                if (p.inlineData) {
-                    const inlineData = p.inlineData;
-                    const result = processImageData({
-                        mimeType: inlineData.mimeType,
-                        data: inlineData.data,
-                    });
-                    if (result) {
-                        return { text: result };
-                    }
-                }
-                if (p.thought === true || p.type === 'thinking') {
-                    const fullText = (p.text || p.thinking || '');
-                    if (displayedThinkingHashes) {
-                        const hash = hashString(fullText);
-                        if (displayedThinkingHashes.has(hash)) {
-                            sentBuffer.set(index, fullText);
-                            return null;
-                        }
-                        displayedThinkingHashes.add(hash);
-                    }
-                    const sentText = sentBuffer.get(index) ?? '';
-                    if (fullText.startsWith(sentText)) {
-                        const delta = fullText.slice(sentText.length);
-                        sentBuffer.set(index, fullText);
-                        if (delta) {
-                            return { ...p, text: delta, thinking: delta };
-                        }
-                        return null;
-                    }
-                    sentBuffer.set(index, fullText);
-                    return part;
-                }
-                return part;
-            });
-            const filteredParts = newParts.filter((p) => p !== null);
-            return {
-                ...cand,
-                content: { ...content, parts: filteredParts },
-            };
-        });
-        return { ...resp, candidates: newCandidates };
-    }
-    if (Array.isArray(resp.content)) {
-        let thinkingIndex = 0;
-        const newContent = resp.content.map((block) => {
-            const b = block;
-            if (b?.type === 'thinking') {
-                const fullText = (b.thinking || b.text || '');
-                if (displayedThinkingHashes) {
-                    const hash = hashString(fullText);
-                    if (displayedThinkingHashes.has(hash)) {
-                        sentBuffer.set(thinkingIndex, fullText);
-                        thinkingIndex++;
-                        return null;
-                    }
-                    displayedThinkingHashes.add(hash);
-                }
-                const sentText = sentBuffer.get(thinkingIndex) ?? '';
-                if (fullText.startsWith(sentText)) {
-                    const delta = fullText.slice(sentText.length);
-                    sentBuffer.set(thinkingIndex, fullText);
-                    thinkingIndex++;
-                    if (delta) {
-                        return { ...b, thinking: delta, text: delta };
-                    }
-                    return null;
-                }
-                sentBuffer.set(thinkingIndex, fullText);
-                thinkingIndex++;
-                return block;
-            }
-            return block;
-        });
-        const filteredContent = newContent.filter((b) => b !== null);
-        return { ...resp, content: filteredContent };
-    }
-    return response;
-}
-export function transformSseLine(line, signatureStore, thoughtBuffer, sentThinkingBuffer, callbacks, options, debugState) {
-    if (!line.startsWith('data:')) {
-        return line;
-    }
-    const json = line.slice(5).trim();
-    if (!json) {
-        return line;
-    }
-    try {
-        const parsed = JSON.parse(json);
-        if (parsed.response !== undefined) {
-            if (options.cacheSignatures && options.signatureSessionKey) {
-                cacheThinkingSignaturesFromResponse(parsed.response, options.signatureSessionKey, signatureStore, thoughtBuffer, callbacks.onCacheSignature);
-            }
-            let response = deduplicateThinkingText(parsed.response, sentThinkingBuffer, options.displayedThinkingHashes);
-            if (options.debugText && callbacks.onInjectDebug && !debugState.injected) {
-                response = callbacks.onInjectDebug(response, options.debugText);
-                debugState.injected = true;
-            }
-            // Note: onInjectSyntheticThinking removed - keep_thinking now uses debugText path
-            const transformed = callbacks.transformThinkingParts
-                ? callbacks.transformThinkingParts(response)
-                : response;
-            return `data: ${JSON.stringify(transformed)}`;
-        }
-    }
-    catch (_) { }
-    return line;
-}
-export function cacheThinkingSignaturesFromResponse(response, signatureSessionKey, signatureStore, thoughtBuffer, onCacheSignature) {
-    if (!response || typeof response !== 'object')
-        return;
-    const resp = response;
-    if (Array.isArray(resp.candidates)) {
-        resp.candidates.forEach((candidate, index) => {
-            const cand = candidate;
-            if (!cand?.content)
-                return;
-            const content = cand.content;
-            if (!Array.isArray(content.parts))
-                return;
-            content.parts.forEach((part) => {
-                const p = part;
-                if (p.thought === true || p.type === 'thinking') {
-                    const text = (p.text || p.thinking || '');
-                    if (text) {
-                        const current = thoughtBuffer.get(index) ?? '';
-                        thoughtBuffer.set(index, current + text);
-                    }
-                }
-                if (p.thoughtSignature) {
-                    const fullText = thoughtBuffer.get(index) ?? '';
-                    if (fullText) {
-                        const signature = p.thoughtSignature;
-                        onCacheSignature?.(signatureSessionKey, fullText, signature);
-                        signatureStore.set(signatureSessionKey, { text: fullText, signature });
-                    }
-                }
-            });
-        });
-    }
-    if (Array.isArray(resp.content)) {
-        const CLAUDE_BUFFER_KEY = 0; // Use index 0 for Claude's single-stream content
-        resp.content.forEach((block) => {
-            const b = block;
-            if (b?.type === 'thinking') {
-                const text = (b.thinking || b.text || '');
-                if (text) {
-                    const current = thoughtBuffer.get(CLAUDE_BUFFER_KEY) ?? '';
-                    thoughtBuffer.set(CLAUDE_BUFFER_KEY, current + text);
-                }
-            }
-            if (b?.signature) {
-                const fullText = thoughtBuffer.get(CLAUDE_BUFFER_KEY) ?? '';
-                if (fullText) {
-                    const signature = b.signature;
-                    onCacheSignature?.(signatureSessionKey, fullText, signature);
-                    signatureStore.set(signatureSessionKey, { text: fullText, signature });
-                }
-            }
-        });
-    }
-}
-export function createStreamingTransformer(signatureStore, callbacks, options = {}) {
-    const decoder = new TextDecoder();
-    const encoder = new TextEncoder();
-    let buffer = '';
-    const thoughtBuffer = createThoughtBuffer();
-    const sentThinkingBuffer = createThoughtBuffer();
-    const debugState = { injected: false };
-    let hasSeenUsageMetadata = false;
-    let watchdogTimer = null;
-    let isDone = false;
-    let controllerRef = null;
-    const resetWatchdog = () => {
-        if (isDone || !controllerRef)
-            return;
-        if (watchdogTimer)
-            clearTimeout(watchdogTimer);
-        watchdogTimer = setTimeout(() => {
-            if (isDone)
-                return;
-            isDone = true;
-            const finish = () => {
-                try {
-                    controllerRef?.terminate();
-                }
-                catch (e) { }
-                options.onComplete?.();
-            };
-            if (options.onWatchdogTimeout) {
-                Promise.resolve(options.onWatchdogTimeout()).finally(finish);
-            }
-            else {
-                finish();
-            }
-        }, 45000); // 45 seconds of silence terminates the stream cleanly
-    };
-    return new TransformStream({
-        start(controller) {
-            controllerRef = controller;
-            resetWatchdog();
-        },
-        transform(chunk, controller) {
-            resetWatchdog();
-            buffer += decoder.decode(chunk, { stream: true });
-            const lines = buffer.split('\n');
-            buffer = lines.pop() || '';
-            for (const line of lines) {
-                if (line.includes('usageMetadata')) {
-                    hasSeenUsageMetadata = true;
-                }
-                const transformedLine = transformSseLine(line, signatureStore, thoughtBuffer, sentThinkingBuffer, callbacks, options, debugState);
-                controller.enqueue(encoder.encode(transformedLine + '\n'));
-            }
-        },
-        flush(controller) {
-            isDone = true;
-            if (watchdogTimer)
-                clearTimeout(watchdogTimer);
-            buffer += decoder.decode();
-            if (buffer) {
-                if (buffer.includes('usageMetadata')) {
-                    hasSeenUsageMetadata = true;
-                }
-                const transformedLine = transformSseLine(buffer, signatureStore, thoughtBuffer, sentThinkingBuffer, callbacks, options, debugState);
-                controller.enqueue(encoder.encode(transformedLine));
-            }
-            if (!hasSeenUsageMetadata) {
-                const syntheticUsage = {
-                    response: {
-                        usageMetadata: {
-                            promptTokenCount: 0,
-                            candidatesTokenCount: 0,
-                            totalTokenCount: 0,
-                        }
-                    }
-                };
-                controller.enqueue(encoder.encode(`\ndata: ${JSON.stringify(syntheticUsage)}\n\n`));
-            }
-            options.onComplete?.();
-        },
-    });
-}

package/dist/plugin/core/streaming/types.d.ts

@@ -1,28 +0,0 @@
-export interface SignedThinking {
-    text: string;
-    signature: string;
-}
-export interface SignatureStore {
-    get(sessionKey: string): SignedThinking | undefined;
-    set(sessionKey: string, value: SignedThinking): void;
-    has(sessionKey: string): boolean;
-    delete(sessionKey: string): void;
-}
-export interface StreamingCallbacks {
-    onCacheSignature?: (sessionKey: string, text: string, signature: string) => void;
-    onInjectDebug?: (response: unknown, debugText: string) => unknown;
-    transformThinkingParts?: (parts: unknown) => unknown;
-}
-export interface StreamingOptions {
-    signatureSessionKey?: string;
-    debugText?: string;
-    cacheSignatures?: boolean;
-    displayedThinkingHashes?: Set<string>;
-    onComplete?: () => void;
-    onWatchdogTimeout?: () => Promise<void> | void;
-}
-export interface ThoughtBuffer {
-    get(index: number): string | undefined;
-    set(index: number, text: string): void;
-    clear(): void;
-}

package/dist/plugin/core/streaming/types.js

@@ -1 +0,0 @@
-export {};

package/dist/plugin/debug.d.ts

@@ -1,93 +0,0 @@
-import type { AntigravityConfig } from "./config";
-export declare const DEBUG_MESSAGE_PREFIX = "[opencode-antigravity-auth debug]";
-/**
- * Initialize or reinitialize debug state with the given config.
- * Call this once at plugin startup after loading config.
- */
-export declare function initializeDebug(config: AntigravityConfig): void;
-export declare function isDebugEnabled(): boolean;
-export declare function isDebugTuiEnabled(): boolean;
-export declare function getLogFilePath(): string | undefined;
-export interface AntigravityDebugContext {
-    id: string;
-    streaming: boolean;
-    startedAt: number;
-}
-interface AntigravityDebugRequestMeta {
-    originalUrl: string;
-    resolvedUrl: string;
-    method?: string;
-    headers?: HeadersInit;
-    body?: BodyInit | null;
-    streaming: boolean;
-    projectId?: string;
-}
-interface AntigravityDebugResponseMeta {
-    body?: string;
-    note?: string;
-    error?: unknown;
-    headersOverride?: HeadersInit;
-}
-/**
- * Begins a debug trace for an Antigravity request.
- */
-export declare function startAntigravityDebugRequest(meta: AntigravityDebugRequestMeta): AntigravityDebugContext | null;
-/**
- * Logs response details for a previously started debug trace.
- */
-export declare function logAntigravityDebugResponse(context: AntigravityDebugContext | null | undefined, response: Response, meta?: AntigravityDebugResponseMeta): void;
-export interface AccountDebugInfo {
-    index: number;
-    email?: string;
-    family: string;
-    totalAccounts: number;
-    rateLimitState?: {
-        claude?: number;
-        gemini?: number;
-    };
-}
-export declare function logAccountContext(label: string, info: AccountDebugInfo): void;
-export declare function logRateLimitEvent(accountIndex: number, email: string | undefined, family: string, status: number, retryAfterMs: number, bodyInfo: {
-    message?: string;
-    quotaResetTime?: string;
-    retryDelayMs?: number | null;
-    reason?: string;
-}): void;
-export declare function logRateLimitSnapshot(family: string, accounts: Array<{
-    index: number;
-    email?: string;
-    rateLimitResetTimes?: {
-        claude?: number;
-        gemini?: number;
-    };
-}>): void;
-export declare function logResponseBody(context: AntigravityDebugContext | null | undefined, response: Response, status: number): Promise<string | undefined>;
-export declare function logModelFamily(url: string, extractedModel: string | null, family: string): void;
-export declare function debugLogToFile(message: string): void;
-/**
- * Logs a toast message to the debug file.
- * This helps correlate what the user saw with debug events.
- */
-export declare function logToast(message: string, variant: "info" | "warning" | "success" | "error"): void;
-/**
- * Logs retry attempt information.
- * @param maxAttempts - Use -1 for unlimited retries
- */
-export declare function logRetryAttempt(attempt: number, maxAttempts: number, reason: string, delayMs?: number): void;
-/**
- * Logs cache hit/miss information from response usage metadata.
- */
-export declare function logCacheStats(model: string, cacheReadTokens: number, cacheWriteTokens: number, totalInputTokens: number): void;
-/**
- * Logs quota status for an account.
- */
-export declare function logQuotaStatus(accountEmail: string | undefined, accountIndex: number, quotaPercent: number, family?: string): void;
-/**
- * Logs background quota fetch events.
- */
-export declare function logQuotaFetch(event: "start" | "complete" | "error", accountCount?: number, details?: string): void;
-/**
- * Logs which model is being used for a request.
- */
-export declare function logModelUsed(requestedModel: string, actualModel: string, accountEmail?: string): void;
-export {};