@kamel-ahmed/proxy-claude 1.0.0

package/src/cli/refresh.js

@@ -0,0 +1,201 @@
+#!/usr/bin/env node
+
+/**
+ * Token Refresh CLI
+ *
+ * Checks and refreshes OAuth tokens for all configured accounts.
+ * Can be run manually or scheduled as a cron job.
+ */
+
+import { existsSync, readFileSync, writeFileSync } from 'fs';
+import { ACCOUNT_CONFIG_PATH } from '../constants.js';
+import { refreshAccessToken } from '../auth/oauth.js';
+import { logger } from '../utils/logger.js';
+
+const COLORS = {
+    RESET: '\x1b[0m',
+    GREEN: '\x1b[32m',
+    RED: '\x1b[31m',
+    YELLOW: '\x1b[33m',
+    BLUE: '\x1b[34m',
+    CYAN: '\x1b[36m'
+};
+
+/**
+ * Load accounts from config file
+ */
+function loadAccounts() {
+    if (!existsSync(ACCOUNT_CONFIG_PATH)) {
+        return null;
+    }
+    try {
+        const data = readFileSync(ACCOUNT_CONFIG_PATH, 'utf-8');
+        return JSON.parse(data);
+    } catch (err) {
+        return null;
+    }
+}
+
+/**
+ * Save accounts to config file
+ */
+function saveAccounts(config) {
+    writeFileSync(ACCOUNT_CONFIG_PATH, JSON.stringify(config, null, 2));
+}
+
+/**
+ * Check if a token needs refresh (expires within threshold)
+ * @param {object} account - Account object
+ * @param {number} thresholdMs - Refresh if expiring within this time (default: 5 minutes)
+ */
+function needsRefresh(account, thresholdMs = 5 * 60 * 1000) {
+    if (!account.tokenExpiresAt) {
+        // No expiry info, assume it needs refresh
+        return true;
+    }
+    const expiresAt = new Date(account.tokenExpiresAt).getTime();
+    const now = Date.now();
+    return (expiresAt - now) < thresholdMs;
+}
+
+/**
+ * Refresh a single account's token
+ */
+async function refreshAccount(account) {
+    if (!account.refreshToken) {
+        return {
+            success: false,
+            error: 'No refresh token available'
+        };
+    }
+
+    try {
+        const result = await refreshAccessToken(account.refreshToken);
+        return {
+            success: true,
+            accessToken: result.accessToken,
+            expiresIn: result.expiresIn
+        };
+    } catch (err) {
+        return {
+            success: false,
+            error: err.message
+        };
+    }
+}
+
+/**
+ * Main refresh function
+ */
+export async function runRefresh(options = {}) {
+    const { force = false, quiet = false } = options;
+
+    const log = quiet ? () => {} : console.log;
+
+    log(`${COLORS.BLUE}╔════════════════════════════════════════╗${COLORS.RESET}`);
+    log(`${COLORS.BLUE}║   Antigravity Proxy Token Refresh      ║${COLORS.RESET}`);
+    log(`${COLORS.BLUE}╚════════════════════════════════════════╝${COLORS.RESET}`);
+    log('');
+
+    const config = loadAccounts();
+    
+    if (!config || !config.accounts || config.accounts.length === 0) {
+        log(`${COLORS.YELLOW}No accounts configured.${COLORS.RESET}`);
+        log(`Run 'proxy-claude accounts add' to add an account.`);
+        return { refreshed: 0, failed: 0, skipped: 0 };
+    }
+
+    const accounts = config.accounts;
+    let refreshed = 0;
+    let failed = 0;
+    let skipped = 0;
+    const invalidAccounts = [];
+
+    log(`Found ${accounts.length} account(s). Checking tokens...\n`);
+
+    for (let i = 0; i < accounts.length; i++) {
+        const account = accounts[i];
+        const email = account.email || `Account ${i + 1}`;
+
+        process.stdout.write(`  ${email}: `);
+
+        // Check if refresh is needed
+        if (!force && !needsRefresh(account)) {
+            const expiresIn = account.tokenExpiresAt 
+                ? Math.round((new Date(account.tokenExpiresAt).getTime() - Date.now()) / 60000)
+                : '?';
+            log(`${COLORS.GREEN}✓ Valid (expires in ${expiresIn} min)${COLORS.RESET}`);
+            skipped++;
+            continue;
+        }
+
+        // Attempt refresh
+        const result = await refreshAccount(account);
+
+        if (result.success) {
+            // Update account with new token
+            accounts[i].accessToken = result.accessToken;
+            accounts[i].tokenExpiresAt = new Date(Date.now() + result.expiresIn * 1000).toISOString();
+            accounts[i].isInvalid = false;
+            accounts[i].invalidReason = null;
+            
+            log(`${COLORS.GREEN}✓ Refreshed (expires in ${Math.round(result.expiresIn / 60)} min)${COLORS.RESET}`);
+            refreshed++;
+        } else {
+            log(`${COLORS.RED}✗ Failed: ${result.error}${COLORS.RESET}`);
+            
+            // Mark as invalid if refresh token is bad
+            if (result.error.includes('invalid_grant') || result.error.includes('Bad Request')) {
+                accounts[i].isInvalid = true;
+                accounts[i].invalidReason = 'Refresh token expired or revoked';
+                invalidAccounts.push(email);
+            }
+            failed++;
+        }
+    }
+
+    // Save updated config
+    config.accounts = accounts;
+    saveAccounts(config);
+
+    log('');
+    log(`${COLORS.CYAN}Summary:${COLORS.RESET}`);
+    log(`  Refreshed: ${COLORS.GREEN}${refreshed}${COLORS.RESET}`);
+    log(`  Skipped:   ${COLORS.BLUE}${skipped}${COLORS.RESET} (still valid)`);
+    log(`  Failed:    ${COLORS.RED}${failed}${COLORS.RESET}`);
+
+    if (invalidAccounts.length > 0) {
+        log('');
+        log(`${COLORS.RED}⚠ The following accounts need to be re-added:${COLORS.RESET}`);
+        invalidAccounts.forEach(email => {
+            log(`  - ${email}`);
+        });
+        log('');
+        log(`Run 'proxy-claude accounts add' to re-authenticate.`);
+    }
+
+    return { refreshed, failed, skipped, invalidAccounts };
+}
+
+// Run if called directly
+const args = process.argv.slice(2);
+if (args.includes('--help') || args.includes('-h')) {
+    console.log(`
+Usage: proxy-claude refresh [options]
+
+Options:
+  --force, -f    Force refresh all tokens, even if not expired
+  --quiet, -q    Suppress output (for cron jobs)
+  --help, -h     Show this help
+`);
+    process.exit(0);
+}
+
+const force = args.includes('--force') || args.includes('-f');
+const quiet = args.includes('--quiet') || args.includes('-q');
+
+runRefresh({ force, quiet }).then(result => {
+    if (result.failed > 0) {
+        process.exit(1);
+    }
+});

package/src/cli/setup.js

@@ -0,0 +1,338 @@
+import { execSync } from 'child_process';
+import fs from 'fs';
+import os from 'os';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { logger } from '../utils/logger.js';
+import { DEFAULT_PORT } from '../constants.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const PROXY_ROOT = path.resolve(__dirname, '../..');
+
+export async function runSetup() {
+    logger.info('Starting Antigravity Claude Proxy setup...');
+
+    // 1. Install Claude Code CLI
+    await installClaudeCode();
+
+    // 2. Install this package globally (so proxy-claude uses local code)
+    await installProxyGlobally();
+
+    // 3. Configure Environment/Script
+    await createProxyLauncher();
+
+    logger.success('Setup complete!');
+    console.log('');
+    console.log('You can now run "proxy-claude" from anywhere to start the proxy and Claude Code seamlessly.');
+    console.log('');
+}
+
+async function installClaudeCode() {
+    try {
+        // Check if claude is installed
+        try {
+            execSync('claude --version', { stdio: 'ignore' });
+            logger.info('Claude Code CLI is already installed.');
+        } catch (e) {
+            // If checking version fails, it might not be installed or return non-zero
+            throw new Error('Not found');
+        }
+    } catch (e) {
+        logger.info('Installing Claude Code CLI...');
+        try {
+            execSync('npm install -g @anthropic-ai/claude-code', { stdio: 'inherit' });
+            logger.success('Claude Code CLI installed successfully.');
+        } catch (err) {
+            logger.error('Failed to install Claude Code CLI. Please install it manually: npm install -g @anthropic-ai/claude-code');
+            // We don't exit here, maybe they want the script anyway
+        }
+    }
+}
+
+async function installProxyGlobally() {
+    // Skip global install - too complex with build steps
+    // Instead, proxy-claude will use npx or check for local project
+    logger.info('Skipping global install (use npm link if needed).');
+}
+
+async function createProxyLauncher() {
+    // Generate the script content.
+    // IMPORTANT: We escape ${} for Bash variables as \${} so JS doesn't parse them.
+    // We escape backslashes for colors as \\033 so they write correctly to the file.
+    
+    // Embed the local project path so it can be used for development
+    const localProjectPath = PROXY_ROOT;
+    
+    const scriptContent = `#!/bin/bash
+# proxy-claude - Full CLI for Antigravity Claude Proxy
+# This script provides complete control over the proxy and Claude Code.
+
+VERSION="1.0.0"
+PORT=\${PORT:-${DEFAULT_PORT || 8080}}
+PROXY_PID=""
+LOCAL_PROJECT="${localProjectPath}"
+
+# Colors for output
+RED='\\033[0;31m'
+GREEN='\\033[0;32m'
+YELLOW='\\033[1;33m'
+BLUE='\\033[0;34m'
+NC='\\033[0m' # No Color
+
+# Function to cleanup background process on exit
+cleanup() {
+  if [ -n "$PROXY_PID" ]; then
+    echo ""
+    echo -e "\${YELLOW}Stopping Antigravity Claude Proxy...\${NC}"
+    kill $PROXY_PID 2>/dev/null
+    wait $PROXY_PID 2>/dev/null
+    echo -e "\${GREEN}✓ Proxy stopped\${NC}"
+  fi
+}
+
+# Function to start proxy and wait for it
+start_proxy() {
+  echo -e "\${BLUE}Starting Antigravity Claude Proxy on port $PORT...\${NC}"
+  
+  if command -v antigravity-claude-proxy &> /dev/null; then
+    antigravity-claude-proxy start > /tmp/antigravity-proxy.log 2>&1 &
+  else
+    npx antigravity-claude-proxy@latest start > /tmp/antigravity-proxy.log 2>&1 &
+  fi
+  PROXY_PID=$!
+  
+  echo "Waiting for proxy to be ready..."
+  MAX_RETRIES=60
+  count=0
+  while ! nc -z localhost $PORT >/dev/null 2>&1; do
+    sleep 0.5
+    count=$((count+1))
+    if [ $count -ge $MAX_RETRIES ]; then
+      echo -e "\${RED}Error: Proxy failed to start within 30 seconds.\${NC}"
+      echo "Check logs: /tmp/antigravity-proxy.log"
+      tail -n 20 /tmp/antigravity-proxy.log
+      exit 1
+    fi
+  done
+  echo -e "\${GREEN}✓ Proxy is ready on port $PORT!\${NC}"
+}
+
+# Function to run the proxy command
+run_proxy_cmd() {
+  # Prefer local project if it exists (for development)
+  if [ -d "$LOCAL_PROJECT" ] && [ -f "$LOCAL_PROJECT/src/index.js" ]; then
+    node "$LOCAL_PROJECT/bin/cli.js" "$@"
+  elif command -v proxy-claude &> /dev/null; then
+    proxy-claude "$@"
+  else
+    npx @kamel-ahmed/proxy-claude@latest "$@"
+  fi
+}
+
+show_help() {
+  echo ""
+  echo -e "\${BLUE}proxy-claude\${NC} v$VERSION - Antigravity Claude Proxy CLI"
+  echo ""
+  echo "USAGE:"
+  echo "  proxy-claude [command] [options]"
+  echo ""
+  echo "COMMANDS:"
+  echo "  (default)           Start proxy + launch Claude Code (interactive)"
+  echo "  web                 Start proxy with web dashboard only"
+  echo "  start               Start proxy server only (foreground)"
+  echo "  accounts [action]   Manage Google accounts (add/list/remove/verify/clear)"
+  echo "  refresh             Check and refresh account tokens"
+  echo "  status              Check if proxy is running"
+  echo "  stop                Stop running proxy server"
+  echo "  logs                View proxy logs"
+  echo "  uninstall           Remove proxy-claude from system"
+  echo "  help                Show this help message"
+  echo ""
+  echo "OPTIONS:"
+  echo "  --port, -p <port>   Set custom port (default: 8080)"
+  echo "  --version, -v       Show version"
+  echo ""
+  echo "EXAMPLES:"
+  echo "  proxy-claude                    # Start proxy + Claude"
+  echo "  proxy-claude web                # Start web dashboard"
+  echo "  proxy-claude accounts add       # Add Google account"
+  echo "  proxy-claude refresh            # Check and refresh tokens"
+  echo "  proxy-claude refresh --force    # Force refresh all tokens"
+  echo "  PORT=3000 proxy-claude          # Use custom port"
+  echo "  proxy-claude --port 3000 web    # Web on custom port"
+  echo ""
+  echo "WEB DASHBOARD:"
+  echo "  http://localhost:$PORT"
+  echo ""
+}
+
+# Parse global options
+while [[ "$1" == --* ]] || [[ "$1" == -* ]]; do
+  case "$1" in
+    --port|-p)
+      PORT="$2"
+      shift 2
+      ;;
+    --version|-v)
+      echo "proxy-claude v$VERSION"
+      exit 0
+      ;;
+    --help|-h)
+      show_help
+      exit 0
+      ;;
+    *)
+      break
+      ;;
+  esac
+done
+
+COMMAND="\${1:-}"
+shift 2>/dev/null || true
+
+case "$COMMAND" in
+  ""|"claude")
+    # Default: Start proxy + Claude
+    trap cleanup EXIT INT TERM
+    start_proxy
+    echo ""
+    if command -v claude &> /dev/null; then
+      export ANTHROPIC_BASE_URL="http://localhost:$PORT"
+      export ANTHROPIC_API_KEY="dummy"
+      claude "$@"
+    else
+      echo -e "\${RED}Error: 'claude' command not found.\${NC}"
+      echo "Install it with: npm install -g @anthropic-ai/claude-code"
+      exit 1
+    fi
+    ;;
+    
+  "web")
+    # Start proxy with web dashboard only (foreground)
+    echo -e "\${BLUE}Starting Antigravity Claude Proxy with Web Dashboard...\${NC}"
+    echo -e "Web UI: \${GREEN}http://localhost:$PORT\${NC}"
+    echo ""
+    echo "Press Ctrl+C to stop"
+    echo ""
+    run_proxy_cmd start
+    ;;
+    
+  "start")
+    # Start proxy only (foreground)
+    echo -e "\${BLUE}Starting proxy server...\${NC}"
+    run_proxy_cmd start
+    ;;
+    
+  "accounts")
+    # Account management
+    run_proxy_cmd accounts "$@"
+    ;;
+    
+  "refresh")
+    # Token refresh - run inline since npm package may not have this yet
+    CONFIG_FILE="$HOME/.config/antigravity-proxy/accounts.json"
+    
+    if [ ! -f "$CONFIG_FILE" ]; then
+      echo -e "\${YELLOW}No accounts configured.\${NC}"
+      echo "Run 'proxy-claude accounts add' to add an account."
+      exit 0
+    fi
+    
+    # Try to run via the package first, fall back to verify which does similar
+    if run_proxy_cmd refresh "$@" 2>/dev/null; then
+      exit 0
+    else
+      echo "Running token verification..."
+      run_proxy_cmd accounts verify
+    fi
+    ;;
+    
+  "status")
+    # Check if proxy is running
+    if nc -z localhost $PORT >/dev/null 2>&1; then
+      echo -e "\${GREEN}✓ Proxy is running on port $PORT\${NC}"
+      # Try to get health info
+      curl -s "http://localhost:$PORT/health" 2>/dev/null || true
+    else
+      echo -e "\${YELLOW}✗ Proxy is not running on port $PORT\${NC}"
+    fi
+    ;;
+    
+  "stop")
+    # Stop running proxy
+    echo "Stopping proxy on port $PORT..."
+    # Find and kill process listening on the port
+    PID=$(lsof -ti tcp:$PORT 2>/dev/null)
+    if [ -n "$PID" ]; then
+      kill $PID 2>/dev/null
+      echo -e "\${GREEN}✓ Proxy stopped (PID: $PID)\${NC}"
+    else
+      echo -e "\${YELLOW}No proxy found running on port $PORT\${NC}"
+    fi
+    ;;
+    
+  "logs")
+    # View logs
+    if [ -f /tmp/antigravity-proxy.log ]; then
+      echo -e "\${BLUE}=== Proxy Logs ===\${NC}"
+      tail -f /tmp/antigravity-proxy.log
+    else
+      echo "No log file found at /tmp/antigravity-proxy.log"
+    fi
+    ;;
+    
+  "uninstall")
+    # Uninstall proxy-claude
+    echo -e "\${YELLOW}Uninstalling proxy-claude...\${NC}"
+    if [ -f /usr/local/bin/proxy-claude ]; then
+      sudo rm /usr/local/bin/proxy-claude
+      echo -e "\${GREEN}✓ Removed /usr/local/bin/proxy-claude\${NC}"
+    else
+      echo "proxy-claude not found in /usr/local/bin"
+    fi
+    echo ""
+    echo "To also uninstall the npm package (if installed globally):"
+    echo "  npm uninstall -g antigravity-claude-proxy"
+    ;;
+    
+  "help"|"--help"|"-h")
+    show_help
+    ;;
+    
+  *)
+    echo -e "\${RED}Unknown command: $COMMAND\${NC}"
+    echo "Run 'proxy-claude help' for usage"
+    exit 1
+    ;;
+esac
+`;
+
+    // Try to install it to /usr/local/bin directly if possible
+    const targetPath = '/usr/local/bin/proxy-claude';
+    
+    try {
+        // Try writing directly (might fail if not sudo)
+        fs.writeFileSync(targetPath, scriptContent);
+        fs.chmodSync(targetPath, '755');
+        logger.success(`Installed global command: ${targetPath}`);
+    } catch (err) {
+        // Fallback: Write to local and try to move with sudo, or just tell user what to do
+        if (err.code === 'EACCES') {
+             logger.warn('Could not write directly to /usr/local/bin. Trying with sudo...');
+             const tempPath = path.join(os.tmpdir(), 'proxy-claude');
+             fs.writeFileSync(tempPath, scriptContent);
+             fs.chmodSync(tempPath, '755');
+             
+             try {
+                execSync(`sudo mv "${tempPath}" "${targetPath}"`, { stdio: 'inherit' });
+                logger.success(`Installed global command: ${targetPath}`);
+             } catch (sudoErr) {
+                 logger.error('Failed to install global command. Please run the setup with sudo or move the file manually.');
+                 console.log(`Command to run: sudo mv "${tempPath}" "${targetPath}"`);
+             }
+        } else {
+            logger.error(`Failed to create launcher: ${err.message}`);
+        }
+    }
+}

package/src/cloudcode/index.js

@@ -0,0 +1,29 @@
+/**
+ * Cloud Code Client for Antigravity
+ *
+ * Communicates with Google's Cloud Code internal API using the
+ * v1internal:streamGenerateContent endpoint with proper request wrapping.
+ *
+ * Supports multi-account load balancing with automatic failover.
+ *
+ * Based on: https://github.com/NoeFabris/opencode-antigravity-auth
+ */
+
+// Re-export public API
+export { sendMessage } from './message-handler.js';
+export { sendMessageStream } from './streaming-handler.js';
+export { listModels, fetchAvailableModels, getModelQuotas, getSubscriptionTier } from './model-api.js';
+
+// Default export for backwards compatibility
+import { sendMessage } from './message-handler.js';
+import { sendMessageStream } from './streaming-handler.js';
+import { listModels, fetchAvailableModels, getModelQuotas, getSubscriptionTier } from './model-api.js';
+
+export default {
+    sendMessage,
+    sendMessageStream,
+    listModels,
+    fetchAvailableModels,
+    getModelQuotas,
+    getSubscriptionTier
+};