polydev-ai 1.8.92 → 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,100 +437,56 @@ To re-login: npx polydev-ai`
       };
     }
     
-    // Close any existing login server
-    if (this._loginServer) {
-      try { this._loginServer.close(); } catch (e) {}
-      this._loginServer = null;
-    }
+    // Generate unique session ID for polling-based auth
+    const sessionId = crypto.randomBytes(32).toString('hex');
     
-    // Create and STORE the server at class level so it stays alive
-    this._loginServer = 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 (let browser render success page)
-          setTimeout(() => {
-            if (this._loginServer) {
-              this._loginServer.close();
-              this._loginServer = null;
-            }
-          }, 7000);
-        } else {
-          res.writeHead(400, { 'Content-Type': 'text/plain' });
-          res.end('Invalid or missing token');
-        }
-      } else {
-        res.writeHead(404);
-        res.end('Not found');
+    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' }
+      });
+      
+      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
+          }
+        };
       }
-    });
-
-    // Start listening
-    this._loginServer.listen(0, 'localhost', () => {
-      const port = this._loginServer.address().port;
-      const callbackUrl = `http://localhost:${port}/callback`;
-      const authUrl = `https://polydev.ai/auth?callback=${encodeURIComponent(callbackUrl)}&redirect=ide-plugin&auto=true`;
-
+      
+      // 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] Callback server listening on port ${port}`);
+      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);
       });
-    });
-
-    // Timeout after 5 minutes
-    this._loginServerTimeout = setTimeout(() => {
-      if (this._loginServer) {
-        console.error('[Polydev] Login timeout - closing server');
-        this._loginServer.close();
-        this._loginServer = null;
-      }
-    }, 5 * 60 * 1000);
-
-    this._loginServer.on('error', (err) => {
-      console.error('[Polydev] Login server error:', err.message);
-      this._loginServer = null;
-    });
-
-    // Return immediately - server stays alive at this._loginServer
-    return {
-      jsonrpc: '2.0',
-      id,
-      result: {
-        content: [{
-          type: 'text',
-          text: `╭─────────────────────────────────────────╮
+      
+      // 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 🌐        │
 ╰─────────────────────────────────────────╯
 
@@ -537,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.92",
+  "version": "1.8.93",
   "engines": {
     "node": ">=20.x <=22.x"
   },