hedgequantx 2.6.109 → 2.6.111

package/package.json

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

package/src/menus/ai-agent.js

@@ -1029,53 +1029,51 @@ const setupProxyOAuth = async (provider, config) => {
   }
   console.log();
   
-  // Step 4: Wait for OAuth callback (with manual fallback for VPS/remote)
-  const waitSpinner = ora({ text: 'Waiting for authorization...', color: 'cyan' }).start();
+  // Step 4: Wait for OAuth callback
+  // Detect if we're on a remote server (SSH) - callback won't work automatically
+  const isRemote = process.env.SSH_CONNECTION || process.env.SSH_CLIENT || 
+                   (process.env.DISPLAY === undefined && process.platform === 'linux');
   
   let authSuccess = false;
   
-  // First try: wait 3 seconds for automatic callback (instant for local, very fast fallback for VPS)
-  try {
-    await proxyManager.waitForAuth(authState, 3000, (status) => {
-      waitSpinner.text = status;
-    });
-    authSuccess = true;
-    waitSpinner.succeed('Authorization successful!');
-  } catch (error) {
-    // Timeout - offer manual callback entry
-    if (error.message === 'Authentication timeout') {
-      waitSpinner.stop();
-      console.log();
-      console.log(chalk.yellow('  Paste the callback URL from your browser:'));
-      console.log(chalk.gray('  (The page showing http://localhost:54545/callback?code=...)'));
-      console.log();
-      
-      const callbackUrl = await prompts.input('  Callback URL: ');
-      
-      if (callbackUrl && callbackUrl.trim()) {
-        const submitSpinner = ora({ text: 'Submitting callback...', color: 'cyan' }).start();
-        try {
-          await proxyManager.submitCallback(callbackUrl.trim());
-          submitSpinner.text = 'Verifying authorization...';
-          
-          // Wait a bit for the proxy to process the callback
-          await new Promise(resolve => setTimeout(resolve, 2000));
-          
-          // Verify auth status
-          await proxyManager.waitForAuth(authState, 10000, () => {});
-          authSuccess = true;
-          submitSpinner.succeed('Authorization successful!');
-        } catch (submitError) {
-          submitSpinner.fail(`Failed: ${submitError.message}`);
-          await prompts.waitForEnter();
-          return await selectProviderOption(provider);
-        }
-      } else {
-        console.log(chalk.gray('  Cancelled.'));
+  if (isRemote) {
+    // Remote/VPS: Ask for callback URL directly (no point waiting)
+    console.log(chalk.yellow('  After authorizing, paste the callback URL from your browser:'));
+    console.log(chalk.gray('  (The page showing http://localhost:54545/callback?code=...)'));
+    console.log();
+    
+    const callbackUrl = await prompts.textInput('  Callback URL: ');
+    
+    if (callbackUrl && callbackUrl.trim()) {
+      const submitSpinner = ora({ text: 'Submitting callback...', color: 'cyan' }).start();
+      try {
+        await proxyManager.submitCallback(callbackUrl.trim());
+        submitSpinner.text = 'Verifying authorization...';
+        await new Promise(resolve => setTimeout(resolve, 1000));
+        await proxyManager.waitForAuth(authState, 10000, () => {});
+        authSuccess = true;
+        submitSpinner.succeed('Authorization successful!');
+      } catch (submitError) {
+        submitSpinner.fail(`Failed: ${submitError.message}`);
         await prompts.waitForEnter();
         return await selectProviderOption(provider);
       }
     } else {
+      console.log(chalk.gray('  Cancelled.'));
+      await prompts.waitForEnter();
+      return await selectProviderOption(provider);
+    }
+  } else {
+    // Local: Wait for automatic callback
+    const waitSpinner = ora({ text: 'Waiting for authorization...', color: 'cyan' }).start();
+    
+    try {
+      await proxyManager.waitForAuth(authState, 300000, (status) => {
+        waitSpinner.text = status;
+      });
+      authSuccess = true;
+      waitSpinner.succeed('Authorization successful!');
+    } catch (error) {
       waitSpinner.fail(`Authorization failed: ${error.message}`);
       await prompts.waitForEnter();
       return await selectProviderOption(provider);