polydev-ai 1.8.90 → 1.8.92

package/mcp/stdio-wrapper.js

@@ -410,18 +410,16 @@ class StdioMCPWrapper {
    */
   async handleLoginTool(params, id) {
     const http = require('http');
-    const { spawn } = require('child_process');
     
-    return new Promise((resolve) => {
-      // Check if already authenticated
-      if (this.isAuthenticated && this.userToken) {
-        resolve({
-          jsonrpc: '2.0',
-          id,
-          result: {
-            content: [{
-              type: 'text',
-              text: `╭─────────────────────────────────────────╮
+    // Check if already authenticated
+    if (this.isAuthenticated && this.userToken) {
+      return {
+        jsonrpc: '2.0',
+        id,
+        result: {
+          content: [{
+            type: 'text',
+            text: `╭─────────────────────────────────────────╮
 │        ALREADY AUTHENTICATED ✓          │
 ╰─────────────────────────────────────────╯
 
@@ -433,128 +431,120 @@ class StdioMCPWrapper {
   • get_perspectives - Direct tool call
 
 To re-login: npx polydev-ai`
-            }]
-          }
+          }]
+        }
+      };
+    }
+    
+    // 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;
       }
-      
-      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 (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; // Flag for opening models page after CLI detection
+        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] Login successful, token saved');
+          console.error('[Polydev] ✓ You can now use Polydev tools!');
 
-            // Wait 7 seconds before closing server (gives time for 5s auto-close countdown + buffer)
-            setTimeout(() => {
-              server.close();
-              resolve({
-                jsonrpc: '2.0',
-                id,
-                result: {
-                  content: [{
-                    type: 'text',
-                    text: `╭─────────────────────────────────────────╮
-│        LOGIN SUCCESSFUL! ✓              │
-╰─────────────────────────────────────────╯
+          // 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');
+      }
+    });
 
-🔐 Token saved to:
-  • ~/.polydev.env
-  • ~/.zshrc
+    // 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`;
 
-⚠️  IMPORTANT: Restart your IDE to activate.
+      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);
+      });
+    });
 
-🤖 After restart, you can:
-  • Use /polydev:ask to query multiple AI models
-  • Use /polydev:auth to check status & credits
-  • Use get_perspectives tool directly
+    // 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);
 
-📊 Dashboard: https://polydev.ai/dashboard`
-                  }]
-                }
-              });
-            }, 7000);
-          } else {
-            res.writeHead(400, { 'Content-Type': 'text/plain' });
-            res.end('Invalid or missing token');
-          }
-        } else {
-          res.writeHead(404);
-          res.end('Not found');
-        }
-      });
+    this._loginServer.on('error', (err) => {
+      console.error('[Polydev] Login server error:', err.message);
+      this._loginServer = null;
+    });
 
-      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`;
+    // Return immediately - server stays alive at this._loginServer
+    return {
+      jsonrpc: '2.0',
+      id,
+      result: {
+        content: [{
+          type: 'text',
+          text: `╭─────────────────────────────────────────╮
+│     BROWSER OPENED FOR LOGIN 🌐        │
+╰─────────────────────────────────────────╯
 
-        console.error(`[Polydev] Opening browser for authentication: ${authUrl}`);
-        
-        // Best-in-class browser opening using 'open' package (cross-platform)
-        // Falls back to platform-specific commands if package fails
-        this.openBrowser(authUrl).catch(() => {
-          console.error('[Polydev] All browser open methods failed');
-          console.error('[Polydev] Please open this URL manually:', authUrl);
-        });
+📱 Complete authentication in your browser.
 
-        // Timeout after 5 minutes
-        setTimeout(() => {
-          server.close();
-          resolve({
-            jsonrpc: '2.0',
-            id,
-            result: {
-              content: [{
-                type: 'text',
-                text: `Login timed out.\n\nPlease try again or run in terminal: npx polydev-ai`
-              }],
-              isError: true
-            }
-          });
-        }, 5 * 60 * 1000);
-      });
+After login:
+  • Token will be saved automatically
+  • You'll see a success page with auto-close
+  • Return here and the tools will be ready!
 
-      server.on('error', (err) => {
-        console.error('[Polydev] Login server error:', err.message);
-        resolve({
-          jsonrpc: '2.0',
-          id,
-          result: {
-            content: [{
-              type: 'text',
-              text: `Login failed: ${err.message}\n\nPlease run in terminal: npx polydev-ai`
-            }],
-            isError: true
-          }
-        });
-      });
-    });
+⏳ Waiting for authentication...
+   (Server listening for up to 5 minutes)`
+        }]
+      }
+    };
   }
 
   /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "polydev-ai",
-  "version": "1.8.90",
+  "version": "1.8.92",
   "engines": {
     "node": ">=20.x <=22.x"
   },
@@ -74,7 +74,7 @@
     "marked": "^16.2.1",
     "next": "^15.5.7",
     "open": "^11.0.0",
-    "polydev-ai": "^1.8.42",
+    "polydev-ai": "^1.8.90",
     "posthog-js": "^1.157.2",
     "prismjs": "^1.30.0",
     "react": "^18.3.1",