hedgequantx 2.6.160 → 2.6.162

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

@@ -0,0 +1,161 @@
+/**
+ * @fileoverview Remote OAuth for CLIProxyAPI
+ * 
+ * For VPS/Server users without browser access
+ * Uses cli.hedgequantx.com as OAuth relay
+ */
+
+const https = require('https');
+
+const REMOTE_OAUTH_URL = 'https://cli.hedgequantx.com';
+
+/**
+ * Make HTTPS request
+ */
+const httpsRequest = (url, options, body = null) => {
+  return new Promise((resolve, reject) => {
+    const parsedUrl = new URL(url);
+    const req = https.request({
+      hostname: parsedUrl.hostname,
+      port: 443,
+      path: parsedUrl.pathname + parsedUrl.search,
+      method: options.method || 'GET',
+      headers: options.headers || {}
+    }, (res) => {
+      let data = '';
+      res.on('data', chunk => data += chunk);
+      res.on('end', () => {
+        try {
+          resolve(JSON.parse(data));
+        } catch (e) {
+          resolve(data);
+        }
+      });
+    });
+    
+    req.on('error', reject);
+    req.on('timeout', () => {
+      req.destroy();
+      reject(new Error('Request timeout'));
+    });
+    
+    if (body) {
+      req.write(typeof body === 'string' ? body : JSON.stringify(body));
+    }
+    req.end();
+  });
+};
+
+/**
+ * Create Remote OAuth session
+ * @param {string} provider - Provider ID (anthropic, openai, gemini)
+ * @returns {Promise<{sessionId: string, authUrl: string}>}
+ */
+const createRemoteSession = async (provider) => {
+  const response = await httpsRequest(
+    `${REMOTE_OAUTH_URL}/oauth/session/create`,
+    {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' }
+    },
+    JSON.stringify({ provider })
+  );
+  
+  if (response.error) {
+    throw new Error(response.error);
+  }
+  
+  return {
+    sessionId: response.sessionId,
+    authUrl: response.authUrl
+  };
+};
+
+/**
+ * Poll Remote OAuth session status
+ * @param {string} sessionId - Session ID from createRemoteSession
+ * @returns {Promise<{status: string, error?: string}>}
+ */
+const pollRemoteSession = async (sessionId) => {
+  const response = await httpsRequest(
+    `${REMOTE_OAUTH_URL}/oauth/session/${sessionId}/status`,
+    { method: 'GET' }
+  );
+  return response;
+};
+
+/**
+ * Get tokens from Remote OAuth session
+ * @param {string} sessionId - Session ID
+ * @returns {Promise<{provider: string, tokens: Object}>}
+ */
+const getRemoteTokens = async (sessionId) => {
+  const response = await httpsRequest(
+    `${REMOTE_OAUTH_URL}/oauth/session/${sessionId}/tokens`,
+    { method: 'GET' }
+  );
+  
+  if (response.error) {
+    throw new Error(response.error);
+  }
+  
+  return response;
+};
+
+/**
+ * Wait for Remote OAuth to complete
+ * @param {string} sessionId - Session ID
+ * @param {number} timeoutMs - Timeout in milliseconds
+ * @param {Function} onStatus - Status callback
+ * @returns {Promise<{provider: string, tokens: Object}>}
+ */
+const waitForRemoteAuth = async (sessionId, timeoutMs = 300000, onStatus = () => {}) => {
+  const startTime = Date.now();
+  
+  while (Date.now() - startTime < timeoutMs) {
+    const status = await pollRemoteSession(sessionId);
+    
+    if (status.status === 'success') {
+      return await getRemoteTokens(sessionId);
+    } else if (status.status === 'error') {
+      throw new Error(status.error || 'Authentication failed');
+    }
+    
+    onStatus('Waiting for authorization...');
+    await new Promise(resolve => setTimeout(resolve, 2000));
+  }
+  
+  throw new Error('Authentication timeout');
+};
+
+/**
+ * Detect if we're running on a server (no display/browser)
+ * @returns {boolean}
+ */
+const isServerEnvironment = () => {
+  if (process.env.SSH_CONNECTION || process.env.SSH_CLIENT) return true;
+  if (process.env.DISPLAY === undefined && process.platform === 'linux') return true;
+  if (process.env.TERM === 'dumb') return true;
+  if (process.env.HQX_REMOTE_OAUTH === '1') return true;
+  return false;
+};
+
+/**
+ * Detect if browser can be opened
+ * @returns {boolean}
+ */
+const canOpenBrowser = () => {
+  if (process.platform === 'darwin' || process.platform === 'win32') return true;
+  if (process.env.DISPLAY) return true;
+  return false;
+};
+
+module.exports = {
+  createRemoteSession,
+  pollRemoteSession,
+  getRemoteTokens,
+  waitForRemoteAuth,
+  isServerEnvironment,
+  canOpenBrowser,
+  REMOTE_OAUTH_URL
+};