polydev-ai 1.8.91 → 1.8.93

package/mcp/stdio-wrapper.js

@@ -407,9 +407,10 @@ class StdioMCPWrapper {
 
   /**
    * Handle login tool - opens browser for authentication
+   * Uses polling-based flow that survives MCP reconnections
    */
   async handleLoginTool(params, id) {
-    const http = require('http');
+    const crypto = require('crypto');
     
     // Check if already authenticated
     if (this.isAuthenticated && this.userToken) {
@@ -436,96 +437,56 @@ To re-login: npx polydev-ai`
       };
     }
     
-    // Start login server in background (don't block the tool response)
-    const loginPromise = new Promise((resolve) => {
-      const server = http.createServer((req, res) => {
-        const url = new URL(req.url, `http://localhost`);
-
-        if (req.method === 'OPTIONS') {
-          res.writeHead(204, {
-            'Access-Control-Allow-Origin': '*',
-            'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
-            'Access-Control-Allow-Headers': 'Content-Type'
-          });
-          res.end();
-          return;
-        }
-
-        if (url.pathname === '/callback') {
-          const token = url.searchParams.get('token');
-
-          if (token && token.startsWith('pd_')) {
-            // Save token
-            this.saveTokenToFiles(token);
-            this.userToken = token;
-            this.isAuthenticated = true;
-            this._freshLogin = true;
-
-            res.writeHead(200, {
-              'Content-Type': 'text/html; charset=utf-8',
-              'Access-Control-Allow-Origin': '*'
-            });
-            res.end(this.getLoginSuccessHTML());
-
-            console.error('[Polydev] Login successful, token saved');
-            console.error('[Polydev] ✓ You can now use Polydev tools!');
-
-            // Wait 7 seconds before closing server
-            setTimeout(() => {
-              server.close();
-              resolve(true);
-            }, 7000);
-          } else {
-            res.writeHead(400, { 'Content-Type': 'text/plain' });
-            res.end('Invalid or missing token');
-          }
-        } else {
-          res.writeHead(404);
-          res.end('Not found');
-        }
-      });
-
-      server.listen(0, 'localhost', () => {
-        const port = server.address().port;
-        const callbackUrl = `http://localhost:${port}/callback`;
-        const authUrl = `https://polydev.ai/auth?callback=${encodeURIComponent(callbackUrl)}&redirect=ide-plugin&auto=true`;
-
-        console.error(`[Polydev] Opening browser for authentication...`);
-        console.error(`[Polydev] Auth URL: ${authUrl}`);
-        
-        this.openBrowser(authUrl).catch(() => {
-          console.error('[Polydev] Could not open browser automatically');
-          console.error('[Polydev] Please open this URL manually:', authUrl);
-        });
-      });
-
-      // Timeout after 5 minutes
-      setTimeout(() => {
-        server.close();
-        resolve(false);
-      }, 5 * 60 * 1000);
-
-      server.on('error', (err) => {
-        console.error('[Polydev] Login server error:', err.message);
-        resolve(false);
+    // Generate unique session ID for polling-based auth
+    const sessionId = crypto.randomBytes(32).toString('hex');
+    
+    try {
+      // Create session on server
+      const createResponse = await fetch(`https://www.polydev.ai/api/auth/cli-session/${sessionId}`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' }
       });
-    });
-
-    // Don't await - let it run in background
-    loginPromise.then((success) => {
-      if (success) {
-        console.error('[Polydev] Authentication completed successfully!');
+      
+      if (!createResponse.ok) {
+        const error = await createResponse.text();
+        console.error('[Polydev] Failed to create login session:', error);
+        return {
+          jsonrpc: '2.0',
+          id,
+          result: {
+            content: [{
+              type: 'text',
+              text: `Login session creation failed. Please try again or run: npx polydev-ai`
+            }],
+            isError: true
+          }
+        };
       }
-    });
-
-    // Return immediately - don't wait for auth to complete
-    return {
-      jsonrpc: '2.0',
-      id,
-      result: {
-        content: [{
-          type: 'text',
-          text: `╭─────────────────────────────────────────╮
+      
+      // Open browser with session_id (no localhost callback needed)
+      const authUrl = `https://polydev.ai/auth?session_id=${sessionId}&redirect=ide-plugin&auto=true`;
+      
+      console.error(`[Polydev] Opening browser for authentication...`);
+      console.error(`[Polydev] Session ID: ${sessionId.slice(0, 8)}...`);
+      console.error(`[Polydev] Auth URL: ${authUrl}`);
+      
+      this.openBrowser(authUrl).catch(() => {
+        console.error('[Polydev] Could not open browser automatically');
+        console.error('[Polydev] Please open this URL manually:', authUrl);
+      });
+      
+      // Start polling in background (survives reconnections because
+      // if this process dies, the new one can resume polling with same session)
+      this.startLoginPolling(sessionId);
+      
+      // Return immediately
+      return {
+        jsonrpc: '2.0',
+        id,
+        result: {
+          content: [{
+            type: 'text',
+            text: `╭─────────────────────────────────────────╮
 │     BROWSER OPENED FOR LOGIN 🌐        │
 ╰─────────────────────────────────────────╯
 
@@ -533,14 +494,86 @@ To re-login: npx polydev-ai`
 
 After login:
   • Token will be saved automatically
-  • You'll see a success page with auto-close
-  • Return here and the tools will be ready!
+  • You can safely reconnect MCP or restart IDE
+  • Use /polydev:auth to check status
+
+⏳ Polling for authentication completion...
+   (Session expires in 10 minutes)`
+          }]
+        }
+      };
+      
+    } catch (error) {
+      console.error('[Polydev] Login error:', error);
+      return {
+        jsonrpc: '2.0',
+        id,
+        result: {
+          content: [{
+            type: 'text',
+            text: `Login failed: ${error.message}\n\nPlease try: npx polydev-ai`
+          }],
+          isError: true
+        }
+      };
+    }
+  }
 
-⏳ Waiting for authentication...
-   (Server listening for up to 5 minutes)`
-        }]
+  /**
+   * Poll for login session completion
+   */
+  async startLoginPolling(sessionId) {
+    const pollInterval = 2000; // 2 seconds
+    const maxPolls = 300; // 10 minutes max
+    let polls = 0;
+    
+    const poll = async () => {
+      polls++;
+      
+      try {
+        const response = await fetch(`https://www.polydev.ai/api/auth/cli-session/${sessionId}`);
+        const data = await response.json();
+        
+        if (data.status === 'completed' && data.token) {
+          // Success! Save the token
+          this.saveTokenToFiles(data.token);
+          this.userToken = data.token;
+          this.isAuthenticated = true;
+          this._freshLogin = true;
+          
+          console.error('[Polydev] ✓ Login successful! Token saved.');
+          console.error('[Polydev] You can now use Polydev tools.');
+          return; // Stop polling
+        }
+        
+        if (data.status === 'expired' || response.status === 410) {
+          console.error('[Polydev] Login session expired. Please try again.');
+          return; // Stop polling
+        }
+        
+        if (data.status === 'not_found' || response.status === 404) {
+          console.error('[Polydev] Login session not found.');
+          return; // Stop polling
+        }
+        
+        // Still pending, continue polling
+        if (polls < maxPolls) {
+          setTimeout(poll, pollInterval);
+        } else {
+          console.error('[Polydev] Login polling timed out.');
+        }
+        
+      } catch (error) {
+        console.error('[Polydev] Poll error:', error.message);
+        // Continue polling on network errors
+        if (polls < maxPolls) {
+          setTimeout(poll, pollInterval);
+        }
       }
     };
+    
+    // Start polling
+    setTimeout(poll, pollInterval);
   }
 
   /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "polydev-ai",
-  "version": "1.8.91",
+  "version": "1.8.93",
   "engines": {
     "node": ">=20.x <=22.x"
   },