polydev-ai 1.8.91 → 1.8.92

package/mcp/stdio-wrapper.js

@@ -436,89 +436,93 @@ 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;
-        }
+    // Close any existing login server
+    if (this._loginServer) {
+      try { this._loginServer.close(); } catch (e) {}
+      this._loginServer = null;
+    }
+    
+    // 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 (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;
+        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());
+          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!');
+          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');
-          }
+          // 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(404);
-          res.end('Not found');
+          res.writeHead(400, { 'Content-Type': 'text/plain' });
+          res.end('Invalid or missing token');
         }
-      });
-
-      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);
-        });
-      });
+      } else {
+        res.writeHead(404);
+        res.end('Not found');
+      }
+    });
 
-      // Timeout after 5 minutes
-      setTimeout(() => {
-        server.close();
-        resolve(false);
-      }, 5 * 60 * 1000);
+    // 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`;
 
-      server.on('error', (err) => {
-        console.error('[Polydev] Login server error:', err.message);
-        resolve(false);
+      console.error(`[Polydev] Opening browser for authentication...`);
+      console.error(`[Polydev] Callback server listening on port ${port}`);
+      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);
       });
     });
 
-    // Don't await - let it run in background
-    loginPromise.then((success) => {
-      if (success) {
-        console.error('[Polydev] Authentication completed successfully!');
+    // 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 - don't wait for auth to complete
+    // Return immediately - server stays alive at this._loginServer
     return {
       jsonrpc: '2.0',
       id,

package/package.json

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