hedgequantx 2.6.97 → 2.6.99

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hedgequantx",
-  "version": "2.6.97",
+  "version": "2.6.99",
   "description": "HedgeQuantX - Prop Futures Trading CLI",
   "main": "src/app.js",
   "bin": {

package/src/menus/ai-agent.js

@@ -782,9 +782,7 @@ const getOAuthConfig = (providerId) => {
 
 /**
  * Setup OAuth connection for any provider with OAuth support
- * Automatically detects environment and uses the appropriate method:
- * - Local (PC/Mac): Uses CLIProxyAPI for automatic token management
- * - Remote (VPS/Server): Uses cli.hedgequantx.com as OAuth relay
+ * Unified flow: copy code from URL (works on local and VPS)
  */
 const setupOAuthConnection = async (provider) => {
   const config = getOAuthConfig(provider.id);
@@ -794,52 +792,13 @@ const setupOAuthConnection = async (provider) => {
     return await selectProviderOption(provider);
   }
   
-  // Check if we can open a browser locally
-  const canUseBrowser = proxyManager.canOpenBrowser();
-  const isServer = proxyManager.isServerEnvironment();
-  
-  // If on server or can't open browser, use remote OAuth
-  if (isServer || !canUseBrowser) {
-    // Only anthropic is supported for remote OAuth currently
-    if (provider.id === 'anthropic') {
-      return await setupRemoteOAuth(provider, config);
-    } else {
-      // For other providers on VPS, show message
-      const boxWidth = getLogoWidth();
-      const W = boxWidth - 2;
-      const makeLine = (content) => {
-        const plainLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
-        const padding = W - plainLen;
-        return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
-      };
-      
-      console.clear();
-      displayBanner();
-      drawBoxHeaderContinue('SERVER DETECTED', boxWidth);
-      
-      console.log(makeLine(chalk.yellow('VPS/SERVER ENVIRONMENT DETECTED')));
-      console.log(makeLine(''));
-      console.log(makeLine(chalk.white('OAuth for this provider requires a browser.')));
-      console.log(makeLine(''));
-      console.log(makeLine(chalk.white('OPTIONS:')));
-      console.log(makeLine(chalk.cyan('1. Use Claude (supports remote OAuth)')));
-      console.log(makeLine(chalk.cyan('2. Use API Key instead of OAuth')));
-      console.log(makeLine(chalk.cyan('3. Run this on a local machine first')));
-      console.log(makeLine(''));
-      
-      drawBoxFooter(boxWidth);
-      await prompts.waitForEnter();
-      return await selectProviderOption(provider);
-    }
-  }
-  
-  // Local machine - use CLIProxyAPI
-  return await setupProxyOAuth(provider, config);
+  // Use unified manual flow for all providers and environments
+  return await setupRemoteOAuth(provider, config);
 };
 
 /**
- * Setup OAuth via Manual Code Entry (for VPS/Server users)
- * User copies the authorization code from the browser
+ * Setup OAuth via Manual Code Entry (unified flow for local and VPS)
+ * User copies the authorization code from the URL or page
  */
 const setupRemoteOAuth = async (provider, config) => {
   const boxWidth = getLogoWidth();
@@ -851,31 +810,46 @@ const setupRemoteOAuth = async (provider, config) => {
     return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
   };
   
-  // Generate OAuth URL using the existing oauth module
-  const authResult = oauthAnthropic.authorize('max');
+  // Get the right OAuth module for this provider
+  const oauthModules = {
+    anthropic: oauthAnthropic,
+    openai: oauthOpenai,
+    gemini: oauthGemini,
+    qwen: oauthQwen,
+    iflow: oauthIflow
+  };
+  
+  const oauthModule = oauthModules[provider.id];
+  if (!oauthModule) {
+    console.log(chalk.red(`OAuth not supported for ${provider.id}`));
+    await prompts.waitForEnter();
+    return await selectProviderOption(provider);
+  }
+  
+  // Generate OAuth URL using the provider's oauth module
+  const authResult = oauthModule.authorize(config.optionId || 'max');
   const url = authResult.url;
   const verifier = authResult.verifier;
   
   // Show instructions
   console.clear();
   displayBanner();
-  drawBoxHeaderContinue(`CONNECT ${config.name} (VPS MODE)`, boxWidth);
+  drawBoxHeaderContinue(`CONNECT ${config.name}`, boxWidth);
   
-  console.log(makeLine(chalk.yellow('MANUAL AUTHORIZATION')));
+  console.log(makeLine(chalk.yellow('CONNECT YOUR ACCOUNT')));
   console.log(makeLine(''));
-  console.log(makeLine(chalk.white('1. OPEN THE LINK BELOW IN ANY BROWSER')));
-  console.log(makeLine(chalk.white('   (Phone, laptop, any device)')));
+  console.log(makeLine(chalk.white('1. OPEN THE LINK BELOW IN YOUR BROWSER')));
   console.log(makeLine(''));
   console.log(makeLine(chalk.white(`2. LOGIN WITH YOUR ${config.accountName.toUpperCase()} ACCOUNT`)));
   console.log(makeLine(''));
   console.log(makeLine(chalk.white('3. CLICK "AUTHORIZE"')));
   console.log(makeLine(''));
-  console.log(makeLine(chalk.green('4. COPY THE CODE SHOWN ON SCREEN')));
-  console.log(makeLine(chalk.white('   (Format: abc123...#xyz789...)')));
+  console.log(makeLine(chalk.green('4. COPY THE CODE FROM THE URL BAR')));
+  console.log(makeLine(chalk.white('   Look for: code=XXXXXX in the URL')));
+  console.log(makeLine(chalk.white('   Copy everything after code= until &')));
   console.log(makeLine(''));
   console.log(makeLine(chalk.white('5. PASTE THE CODE BELOW')));
   console.log(makeLine(''));
-  console.log(makeLine(chalk.white('TYPE < TO CANCEL')));
   
   drawBoxFooter(boxWidth);
   
@@ -896,7 +870,7 @@ const setupRemoteOAuth = async (provider, config) => {
   // Exchange code for tokens
   const spinner = ora({ text: 'Exchanging code for tokens...', color: 'cyan' }).start();
   
-  const result = await oauthAnthropic.exchange(code.trim(), verifier);
+  const result = await oauthModule.exchange(code.trim(), verifier);
   
   if (result.type === 'failed') {
     spinner.fail(`Authentication failed: ${result.error || 'Invalid code'}`);
@@ -923,8 +897,8 @@ const setupRemoteOAuth = async (provider, config) => {
   
   let models = [];
   try {
-    const { fetchAnthropicModelsOAuth } = require('../services/ai/client');
-    models = await fetchAnthropicModelsOAuth(result.access);
+    const { fetchModelsWithOAuth } = require('../services/ai/client');
+    models = await fetchModelsWithOAuth(provider.id, result.access);
   } catch (e) {
     // Ignore - will use defaults
   }
@@ -988,177 +962,12 @@ const getDefaultModelsForProvider = (providerId) => {
 };
 
 /**
- * Setup OAuth via CLIProxyAPI (automatic local proxy)
- * This is the simplified flow for users with subscription plans on local machines
+ * Setup OAuth via Manual Code Entry (unified flow for local and VPS)
+ * User copies the authorization code from the URL or page
  */
 const setupProxyOAuth = async (provider, config) => {
-  const boxWidth = getLogoWidth();
-  const W = boxWidth - 2;
-  
-  const makeLine = (content) => {
-    const plainLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
-    const padding = W - plainLen;
-    return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
-  };
-  
-  console.clear();
-  displayBanner();
-  drawBoxHeaderContinue(`CONNECT ${config.name}`, boxWidth);
-  
-  console.log(makeLine(chalk.yellow('AUTOMATIC SETUP')));
-  console.log(makeLine(''));
-  console.log(makeLine(chalk.white('PREPARING CONNECTION...')));
-  
-  drawBoxFooter(boxWidth);
-  
-  // Ensure proxy is running (will install if needed)
-  const spinner = ora({ text: 'Setting up AI connection...', color: 'cyan' }).start();
-  
-  try {
-    await proxyManager.ensureRunning((msg) => {
-      spinner.text = msg;
-    });
-  } catch (error) {
-    spinner.fail(`Setup failed: ${error.message}`);
-    await prompts.waitForEnter();
-    return await selectProviderOption(provider);
-  }
-  
-  // Get OAuth URL from proxy
-  spinner.text = 'Generating authorization link...';
-  
-  let authData;
-  try {
-    authData = await proxyManager.getAuthUrl(provider.id);
-  } catch (error) {
-    spinner.fail(`Failed to get auth URL: ${error.message}`);
-    await prompts.waitForEnter();
-    return await selectProviderOption(provider);
-  }
-  
-  spinner.stop();
-  
-  // Show URL to user
-  console.clear();
-  displayBanner();
-  drawBoxHeaderContinue(`CONNECT ${config.name}`, boxWidth);
-  
-  console.log(makeLine(chalk.yellow('LOGIN TO YOUR ACCOUNT')));
-  console.log(makeLine(''));
-  console.log(makeLine(chalk.white('1. OPEN THE LINK BELOW IN YOUR BROWSER')));
-  console.log(makeLine(chalk.white(`2. LOGIN WITH YOUR ${config.accountName.toUpperCase()} ACCOUNT`)));
-  console.log(makeLine(chalk.white('3. AUTHORIZE THE APPLICATION')));
-  console.log(makeLine(''));
-  console.log(makeLine(chalk.green('THE CONNECTION WILL BE AUTOMATIC!')));
-  console.log(makeLine(chalk.white('NO CODE TO COPY - JUST LOGIN AND AUTHORIZE')));
-  console.log(makeLine(''));
-  console.log(makeLine(chalk.white('PRESS ENTER AFTER YOU AUTHORIZED...')));
-  
-  drawBoxFooter(boxWidth);
-  
-  // Display URL outside the box for easy copy
-  console.log();
-  console.log(chalk.yellow('  OPEN THIS URL IN YOUR BROWSER:'));
-  console.log();
-  console.log(chalk.cyan(`  ${authData.url}`));
-  console.log();
-  
-  // Try to open browser automatically
-  const browserOpened = await openBrowser(authData.url);
-  if (browserOpened) {
-    console.log(chalk.green('  Browser opened automatically!'));
-    console.log();
-  }
-  
-  // Wait for user to press enter
-  await prompts.waitForEnter();
-  
-  // Check if auth was successful
-  const authSpinner = ora({ text: 'Checking authorization...', color: 'cyan' }).start();
-  
-  try {
-    // Poll for auth status (with timeout)
-    const startTime = Date.now();
-    const timeout = 60000; // 1 minute
-    
-    while (Date.now() - startTime < timeout) {
-      const status = await proxyManager.pollAuthStatus(authData.state);
-      
-      if (status.status === 'ok') {
-        authSpinner.succeed('Authorization successful!');
-        break;
-      } else if (status.status === 'error') {
-        authSpinner.fail(`Authorization failed: ${status.error}`);
-        await prompts.waitForEnter();
-        return await selectProviderOption(provider);
-      }
-      
-      // Still waiting
-      await new Promise(resolve => setTimeout(resolve, 2000));
-    }
-  } catch (error) {
-    // If polling fails, try to get models anyway (user might have authorized)
-    authSpinner.text = 'Verifying connection...';
-  }
-  
-  // Fetch available models from proxy
-  authSpinner.text = 'Fetching available models...';
-  
-  let models = [];
-  try {
-    models = await proxyManager.getModels();
-  } catch (e) {
-    // Try again
-    await new Promise(resolve => setTimeout(resolve, 2000));
-    models = await proxyManager.getModels();
-  }
-  
-  if (!models || models.length === 0) {
-    authSpinner.fail('No models available - authorization may have failed');
-    console.log(chalk.red('\n  Please try again and make sure to complete the login.'));
-    await prompts.waitForEnter();
-    return await selectProviderOption(provider);
-  }
-  
-  // Filter models for this provider
-  const providerModels = models.filter(m => {
-    const modelLower = m.toLowerCase();
-    if (provider.id === 'anthropic') return modelLower.includes('claude');
-    if (provider.id === 'openai') return modelLower.includes('gpt') || modelLower.includes('o1') || modelLower.includes('o3');
-    if (provider.id === 'gemini') return modelLower.includes('gemini');
-    if (provider.id === 'qwen') return modelLower.includes('qwen');
-    if (provider.id === 'iflow') return true; // iFlow has various models
-    return true;
-  });
-  
-  const finalModels = providerModels.length > 0 ? providerModels : models;
-  
-  authSpinner.succeed(`Found ${finalModels.length} models`);
-  
-  // Let user select model
-  const selectedModel = await selectModelFromList(finalModels, config.name);
-  if (!selectedModel) {
-    return await selectProviderOption(provider);
-  }
-  
-  // Save agent configuration (using proxy)
-  const credentials = {
-    useProxy: true,
-    proxyPort: proxyManager.PROXY_PORT
-  };
-  
-  try {
-    await aiService.addAgent(provider.id, config.optionId, credentials, selectedModel, config.agentName);
-    
-    console.log(chalk.green(`\n  CONNECTED TO ${config.name}`));
-    console.log(chalk.white(`  MODEL: ${selectedModel}`));
-    console.log(chalk.white('  UNLIMITED USAGE WITH YOUR SUBSCRIPTION'));
-  } catch (error) {
-    console.log(chalk.red(`\n  FAILED TO SAVE: ${error.message}`));
-  }
-  
-  await prompts.waitForEnter();
-  return await aiAgentMenu();
+  // Use the same flow as VPS - it works everywhere
+  return await setupRemoteOAuth(provider, config);
 };
 
 /**