polydev-ai 1.8.93 → 1.8.95

package/mcp/stdio-wrapper.js

@@ -253,6 +253,9 @@ class StdioMCPWrapper {
     // Server URL for API calls
     this.serverUrl = 'https://www.polydev.ai/api/mcp';
     
+    // Pending session file for surviving restarts
+    this.PENDING_SESSION_FILE = path.join(os.homedir(), '.polydev-pending-session');
+    
     // Initialize CLI Manager for local CLI functionality
     this.cliManager = new CLIManager();
     
@@ -264,6 +267,12 @@ class StdioMCPWrapper {
       this.reloadTokenFromFiles();
     }
     
+    // Check for pending login session from previous process
+    // This allows login to survive MCP reconnections
+    if (!this.isAuthenticated) {
+      this.resumePendingLoginSession();
+    }
+    
     // Log startup status
     if (this.userToken) {
       console.error('[Polydev] Starting with authentication token');
@@ -475,6 +484,9 @@ To re-login: npx polydev-ai`
         console.error('[Polydev] Please open this URL manually:', authUrl);
       });
       
+      // Save pending session to file (allows resuming after MCP reconnect)
+      this.savePendingSession(sessionId);
+      
       // Start polling in background (survives reconnections because
       // if this process dies, the new one can resume polling with same session)
       this.startLoginPolling(sessionId);
@@ -521,12 +533,18 @@ After login:
 
   /**
    * Poll for login session completion
+   * @param {string} sessionId - The session ID to poll
+   * @param {boolean} isResume - Whether this is resuming from a previous process
    */
-  async startLoginPolling(sessionId) {
+  async startLoginPolling(sessionId, isResume = false) {
     const pollInterval = 2000; // 2 seconds
     const maxPolls = 300; // 10 minutes max
     let polls = 0;
     
+    if (isResume) {
+      console.error('[Polydev] Resuming login polling from previous session...');
+    }
+    
     const poll = async () => {
       polls++;
       
@@ -535,7 +553,8 @@ After login:
         const data = await response.json();
         
         if (data.status === 'completed' && data.token) {
-          // Success! Save the token
+          // Success! Save the token and clear pending session
+          this.clearPendingSession();
           this.saveTokenToFiles(data.token);
           this.userToken = data.token;
           this.isAuthenticated = true;
@@ -543,15 +562,18 @@ After login:
           
           console.error('[Polydev] ✓ Login successful! Token saved.');
           console.error('[Polydev] You can now use Polydev tools.');
+          console.error('[Polydev] No restart required - token is active immediately.');
           return; // Stop polling
         }
         
         if (data.status === 'expired' || response.status === 410) {
+          this.clearPendingSession();
           console.error('[Polydev] Login session expired. Please try again.');
           return; // Stop polling
         }
         
         if (data.status === 'not_found' || response.status === 404) {
+          this.clearPendingSession();
           console.error('[Polydev] Login session not found.');
           return; // Stop polling
         }
@@ -560,6 +582,7 @@ After login:
         if (polls < maxPolls) {
           setTimeout(poll, pollInterval);
         } else {
+          this.clearPendingSession();
           console.error('[Polydev] Login polling timed out.');
         }
         
@@ -568,6 +591,8 @@ After login:
         // Continue polling on network errors
         if (polls < maxPolls) {
           setTimeout(poll, pollInterval);
+        } else {
+          this.clearPendingSession();
         }
       }
     };
@@ -827,6 +852,79 @@ Error: ${error.message}`
     return false;
   }
 
+  /**
+   * Save pending login session ID to file
+   * Allows login to survive MCP reconnections
+   */
+  savePendingSession(sessionId) {
+    try {
+      const data = {
+        sessionId,
+        createdAt: Date.now(),
+        expiresAt: Date.now() + (10 * 60 * 1000) // 10 minutes
+      };
+      fs.writeFileSync(this.PENDING_SESSION_FILE, JSON.stringify(data));
+      console.error(`[Polydev] Saved pending session: ${sessionId.slice(0, 8)}...`);
+    } catch (error) {
+      console.error('[Polydev] Failed to save pending session:', error.message);
+    }
+  }
+
+  /**
+   * Load pending login session from file
+   */
+  loadPendingSession() {
+    try {
+      if (!fs.existsSync(this.PENDING_SESSION_FILE)) {
+        return null;
+      }
+      const content = fs.readFileSync(this.PENDING_SESSION_FILE, 'utf8');
+      const data = JSON.parse(content);
+      
+      // Check if expired
+      if (data.expiresAt && Date.now() > data.expiresAt) {
+        this.clearPendingSession();
+        console.error('[Polydev] Pending session expired, cleared');
+        return null;
+      }
+      
+      return data;
+    } catch (error) {
+      console.error('[Polydev] Failed to load pending session:', error.message);
+      return null;
+    }
+  }
+
+  /**
+   * Clear pending login session file
+   */
+  clearPendingSession() {
+    try {
+      if (fs.existsSync(this.PENDING_SESSION_FILE)) {
+        fs.unlinkSync(this.PENDING_SESSION_FILE);
+        console.error('[Polydev] Cleared pending session file');
+      }
+    } catch (error) {
+      // Ignore errors
+    }
+  }
+
+  /**
+   * Resume polling for a pending login session (called on startup)
+   */
+  resumePendingLoginSession() {
+    const pending = this.loadPendingSession();
+    if (!pending || !pending.sessionId) {
+      return;
+    }
+    
+    console.error(`[Polydev] Found pending login session: ${pending.sessionId.slice(0, 8)}...`);
+    console.error('[Polydev] Resuming authentication polling...');
+    
+    // Resume polling (this will auto-clear on success/failure)
+    this.startLoginPolling(pending.sessionId, true /* isResume */);
+  }
+
   /**
    * Get login success HTML page
    */
@@ -1444,7 +1542,7 @@ Error: ${error.message}`
     try {
       // Wait for initial CLI detection to complete if it's still running
       if (this._cliDetectionReady && !this._cliDetectionComplete) {
-        console.error('[Stdio Wrapper] Waiting for initial CLI detection to complete...');
+        console.error('[Polydev] Waiting for initial CLI detection to complete...');
         await this._cliDetectionReady;
         console.error('[Polydev] CLI detection ready, proceeding with request');
       }

package/package.json

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