antigravity-claude-proxy 1.0.0

package/src/constants.js

@@ -0,0 +1,164 @@
+/**
+ * Constants for Antigravity Cloud Code API integration
+ * Based on: https://github.com/NoeFabris/opencode-antigravity-auth
+ */
+
+import { homedir, platform, arch } from 'os';
+import { join } from 'path';
+
+/**
+ * Get the Antigravity database path based on the current platform.
+ * - macOS: ~/Library/Application Support/Antigravity/...
+ * - Windows: ~/AppData/Roaming/Antigravity/...
+ * - Linux/other: ~/.config/Antigravity/...
+ * @returns {string} Full path to the Antigravity state database
+ */
+function getAntigravityDbPath() {
+    const home = homedir();
+    switch (platform()) {
+        case 'darwin':
+            return join(home, 'Library/Application Support/Antigravity/User/globalStorage/state.vscdb');
+        case 'win32':
+            return join(home, 'AppData/Roaming/Antigravity/User/globalStorage/state.vscdb');
+        default: // linux, freebsd, etc.
+            return join(home, '.config/Antigravity/User/globalStorage/state.vscdb');
+    }
+}
+
+/**
+ * Generate platform-specific User-Agent string.
+ * @returns {string} User-Agent in format "antigravity/version os/arch"
+ */
+function getPlatformUserAgent() {
+    const os = platform();
+    const architecture = arch();
+    return `antigravity/1.11.5 ${os}/${architecture}`;
+}
+
+// Cloud Code API endpoints (in fallback order)
+const ANTIGRAVITY_ENDPOINT_DAILY = 'https://daily-cloudcode-pa.sandbox.googleapis.com';
+const ANTIGRAVITY_ENDPOINT_PROD = 'https://cloudcode-pa.googleapis.com';
+
+// Endpoint fallback order (daily → prod)
+export const ANTIGRAVITY_ENDPOINT_FALLBACKS = [
+    ANTIGRAVITY_ENDPOINT_DAILY,
+    ANTIGRAVITY_ENDPOINT_PROD
+];
+
+// Required headers for Antigravity API requests
+export const ANTIGRAVITY_HEADERS = {
+    'User-Agent': getPlatformUserAgent(),
+    'X-Goog-Api-Client': 'google-cloud-sdk vscode_cloudshelleditor/0.1',
+    'Client-Metadata': JSON.stringify({
+        ideType: 'IDE_UNSPECIFIED',
+        platform: 'PLATFORM_UNSPECIFIED',
+        pluginType: 'GEMINI'
+    })
+};
+
+// Model name mappings: Anthropic format → Antigravity format
+export const MODEL_MAPPINGS = {
+    // Claude models
+    'claude-3-opus-20240229': 'claude-opus-4-5-thinking',
+    'claude-3-5-opus-20240229': 'claude-opus-4-5-thinking',
+    'claude-3-5-sonnet-20241022': 'claude-sonnet-4-5',
+    'claude-3-5-sonnet-20240620': 'claude-sonnet-4-5',
+    'claude-3-sonnet-20240229': 'claude-sonnet-4-5',
+    'claude-sonnet-4-5': 'claude-sonnet-4-5',
+    'claude-sonnet-4-5-thinking': 'claude-sonnet-4-5-thinking',
+    'claude-opus-4-5-thinking': 'claude-opus-4-5-thinking'
+};
+
+// Available models exposed by this proxy
+export const AVAILABLE_MODELS = [
+    {
+        id: 'claude-sonnet-4-5',
+        name: 'Claude Sonnet 4.5 (Antigravity)',
+        description: 'Claude Sonnet 4.5 via Antigravity Cloud Code',
+        context: 200000,
+        output: 64000
+    },
+    {
+        id: 'claude-sonnet-4-5-thinking',
+        name: 'Claude Sonnet 4.5 Thinking (Antigravity)',
+        description: 'Claude Sonnet 4.5 with extended thinking via Antigravity',
+        context: 200000,
+        output: 64000
+    },
+    {
+        id: 'claude-opus-4-5-thinking',
+        name: 'Claude Opus 4.5 Thinking (Antigravity)',
+        description: 'Claude Opus 4.5 with extended thinking via Antigravity',
+        context: 200000,
+        output: 64000
+    }
+];
+
+// Default project ID if none can be discovered
+export const DEFAULT_PROJECT_ID = 'rising-fact-p41fc';
+
+export const TOKEN_REFRESH_INTERVAL_MS = 5 * 60 * 1000; // 5 minutes
+export const REQUEST_BODY_LIMIT = '50mb';
+export const ANTIGRAVITY_AUTH_PORT = 9092;
+export const DEFAULT_PORT = 8080;
+
+// Multi-account configuration
+export const ACCOUNT_CONFIG_PATH = join(
+    homedir(),
+    '.config/antigravity-proxy/accounts.json'
+);
+
+// Antigravity app database path (for legacy single-account token extraction)
+// Uses platform-specific path detection
+export const ANTIGRAVITY_DB_PATH = getAntigravityDbPath();
+
+export const DEFAULT_COOLDOWN_MS = 60 * 1000; // 1 minute default cooldown
+export const MAX_RETRIES = 5; // Max retry attempts across accounts
+export const MAX_ACCOUNTS = 10; // Maximum number of accounts allowed
+
+// Rate limit wait thresholds
+export const MAX_WAIT_BEFORE_ERROR_MS = 120000; // 2 minutes - throw error if wait exceeds this
+
+// Thinking model constants
+export const CLAUDE_THINKING_MAX_OUTPUT_TOKENS = 64000; // Max output tokens for thinking models
+export const MIN_SIGNATURE_LENGTH = 50; // Minimum valid thinking signature length
+
+// Google OAuth configuration (from opencode-antigravity-auth)
+export const OAUTH_CONFIG = {
+    clientId: '1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com',
+    clientSecret: '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',
+    authUrl: 'https://accounts.google.com/o/oauth2/v2/auth',
+    tokenUrl: 'https://oauth2.googleapis.com/token',
+    userInfoUrl: 'https://www.googleapis.com/oauth2/v1/userinfo',
+    callbackPort: 51121,
+    scopes: [
+        'https://www.googleapis.com/auth/cloud-platform',
+        'https://www.googleapis.com/auth/userinfo.email',
+        'https://www.googleapis.com/auth/userinfo.profile',
+        'https://www.googleapis.com/auth/cclog',
+        'https://www.googleapis.com/auth/experimentsandconfigs'
+    ]
+};
+export const OAUTH_REDIRECT_URI = `http://localhost:${OAUTH_CONFIG.callbackPort}/oauth-callback`;
+
+export default {
+    ANTIGRAVITY_ENDPOINT_FALLBACKS,
+    ANTIGRAVITY_HEADERS,
+    MODEL_MAPPINGS,
+    AVAILABLE_MODELS,
+    DEFAULT_PROJECT_ID,
+    TOKEN_REFRESH_INTERVAL_MS,
+    REQUEST_BODY_LIMIT,
+    ANTIGRAVITY_AUTH_PORT,
+    DEFAULT_PORT,
+    ACCOUNT_CONFIG_PATH,
+    ANTIGRAVITY_DB_PATH,
+    DEFAULT_COOLDOWN_MS,
+    MAX_RETRIES,
+    MAX_ACCOUNTS,
+    MAX_WAIT_BEFORE_ERROR_MS,
+    CLAUDE_THINKING_MAX_OUTPUT_TOKENS,
+    MIN_SIGNATURE_LENGTH,
+    OAUTH_CONFIG,
+    OAUTH_REDIRECT_URI
+};

package/src/errors.js

@@ -0,0 +1,159 @@
+/**
+ * Custom Error Classes
+ *
+ * Provides structured error types for better error handling and classification.
+ * Replaces string-based error detection with proper error class checking.
+ */
+
+/**
+ * Base error class for Antigravity proxy errors
+ */
+export class AntigravityError extends Error {
+    /**
+     * @param {string} message - Error message
+     * @param {string} code - Error code for programmatic handling
+     * @param {boolean} retryable - Whether the error is retryable
+     * @param {Object} metadata - Additional error metadata
+     */
+    constructor(message, code, retryable = false, metadata = {}) {
+        super(message);
+        this.name = 'AntigravityError';
+        this.code = code;
+        this.retryable = retryable;
+        this.metadata = metadata;
+    }
+
+    /**
+     * Convert to JSON for API responses
+     */
+    toJSON() {
+        return {
+            name: this.name,
+            code: this.code,
+            message: this.message,
+            retryable: this.retryable,
+            ...this.metadata
+        };
+    }
+}
+
+/**
+ * Rate limit error (429 / RESOURCE_EXHAUSTED)
+ */
+export class RateLimitError extends AntigravityError {
+    /**
+     * @param {string} message - Error message
+     * @param {number|null} resetMs - Time in ms until rate limit resets
+     * @param {string} accountEmail - Email of the rate-limited account
+     */
+    constructor(message, resetMs = null, accountEmail = null) {
+        super(message, 'RATE_LIMITED', true, { resetMs, accountEmail });
+        this.name = 'RateLimitError';
+        this.resetMs = resetMs;
+        this.accountEmail = accountEmail;
+    }
+}
+
+/**
+ * Authentication error (invalid credentials, token expired, etc.)
+ */
+export class AuthError extends AntigravityError {
+    /**
+     * @param {string} message - Error message
+     * @param {string} accountEmail - Email of the account with auth issues
+     * @param {string} reason - Specific reason for auth failure
+     */
+    constructor(message, accountEmail = null, reason = null) {
+        super(message, 'AUTH_INVALID', false, { accountEmail, reason });
+        this.name = 'AuthError';
+        this.accountEmail = accountEmail;
+        this.reason = reason;
+    }
+}
+
+/**
+ * No accounts available error
+ */
+export class NoAccountsError extends AntigravityError {
+    /**
+     * @param {string} message - Error message
+     * @param {boolean} allRateLimited - Whether all accounts are rate limited
+     */
+    constructor(message = 'No accounts available', allRateLimited = false) {
+        super(message, 'NO_ACCOUNTS', allRateLimited, { allRateLimited });
+        this.name = 'NoAccountsError';
+        this.allRateLimited = allRateLimited;
+    }
+}
+
+/**
+ * Max retries exceeded error
+ */
+export class MaxRetriesError extends AntigravityError {
+    /**
+     * @param {string} message - Error message
+     * @param {number} attempts - Number of attempts made
+     */
+    constructor(message = 'Max retries exceeded', attempts = 0) {
+        super(message, 'MAX_RETRIES', false, { attempts });
+        this.name = 'MaxRetriesError';
+        this.attempts = attempts;
+    }
+}
+
+/**
+ * API error from upstream service
+ */
+export class ApiError extends AntigravityError {
+    /**
+     * @param {string} message - Error message
+     * @param {number} statusCode - HTTP status code
+     * @param {string} errorType - Type of API error
+     */
+    constructor(message, statusCode = 500, errorType = 'api_error') {
+        super(message, errorType.toUpperCase(), statusCode >= 500, { statusCode, errorType });
+        this.name = 'ApiError';
+        this.statusCode = statusCode;
+        this.errorType = errorType;
+    }
+}
+
+/**
+ * Check if an error is a rate limit error
+ * Works with both custom error classes and legacy string-based errors
+ * @param {Error} error - Error to check
+ * @returns {boolean}
+ */
+export function isRateLimitError(error) {
+    if (error instanceof RateLimitError) return true;
+    const msg = (error.message || '').toLowerCase();
+    return msg.includes('429') ||
+        msg.includes('resource_exhausted') ||
+        msg.includes('quota_exhausted') ||
+        msg.includes('rate limit');
+}
+
+/**
+ * Check if an error is an authentication error
+ * Works with both custom error classes and legacy string-based errors
+ * @param {Error} error - Error to check
+ * @returns {boolean}
+ */
+export function isAuthError(error) {
+    if (error instanceof AuthError) return true;
+    const msg = (error.message || '').toUpperCase();
+    return msg.includes('AUTH_INVALID') ||
+        msg.includes('INVALID_GRANT') ||
+        msg.includes('TOKEN REFRESH FAILED');
+}
+
+export default {
+    AntigravityError,
+    RateLimitError,
+    AuthError,
+    NoAccountsError,
+    MaxRetriesError,
+    ApiError,
+    isRateLimitError,
+    isAuthError
+};