hedgequantx 2.6.120 → 2.6.121

package/package.json

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

package/src/services/ai/proxy-manager.js

@@ -495,13 +495,72 @@ const getAuthUrl = async (provider) => {
 };
 
 /**
- * Poll for OAuth authentication status
- * @param {string} state - OAuth state from getAuthUrl
- * @returns {Promise<{status: string, error?: string}>}
+ * @param {string} callbackUrl - Full callback URL (http://localhost:54545/callback?code=xxx&state=yyy)
+ * @param {string} provider - Provider ID (anthropic, openai, gemini, qwen, iflow)
+ * @returns {Promise<boolean>}
  */
-const pollAuthStatus = async (state) => {
-  const response = await managementRequest('GET', `/v0/management/get-auth-status?state=${state}`);
-  return response;
+const submitCallback = async (callbackUrl, provider = 'anthropic') => {
+  // Parse the callback URL
+  let url;
+  try {
+    url = new URL(callbackUrl);
+  } catch (e) {
+    throw new Error('Invalid callback URL format');
+  }
+  
+  const code = url.searchParams.get('code');
+  const state = url.searchParams.get('state');
+  
+  if (!code || !state) {
+    throw new Error('Missing code or state in callback URL');
+  }
+  
+  // Each provider has its own OAuth callback port and path in CLIProxyAPI
+  // We need to submit the callback to the correct port
+  const providerConfig = {
+    anthropic: { port: 54545, path: '/callback' },
+    openai: { port: 16168, path: '/callback' },
+    gemini: { port: 8085, path: '/oauth2callback' },
+    qwen: { port: 8087, path: '/oauth2callback' },
+    iflow: { port: 8088, path: '/callback' }
+  };
+  
+  const config = providerConfig[provider] || providerConfig.anthropic;
+  
+  // Submit to the provider's OAuth callback port directly
+  return new Promise((resolve, reject) => {
+    const callbackPath = `${config.path}?code=${encodeURIComponent(code)}&state=${encodeURIComponent(state)}`;
+    
+    const req = http.request({
+      hostname: '127.0.0.1',
+      port: config.port,
+      path: callbackPath,
+      method: 'GET',
+      timeout: 30000
+    }, (res) => {
+      let data = '';
+      res.on('data', chunk => data += chunk);
+      res.on('end', () => {
+        // Success if we get a redirect or success page
+        if (res.statusCode === 200 || res.statusCode === 302 || res.statusCode === 301) {
+          resolve(true);
+        } else {
+          reject(new Error(`Callback failed with status ${res.statusCode}`));
+        }
+      });
+    });
+    
+    req.on('error', (e) => {
+      reject(new Error(`Failed to submit callback: ${e.message}`));
+    });
+    
+    req.on('timeout', () => {
+      req.destroy();
+      reject(new Error('Callback request timeout'));
+    });
+    
+    req.end();
+  });
 };
 
 /**