polydev-ai 1.9.38 → 1.9.40

package/lib/tunnelClient.js

@@ -21,11 +21,60 @@ class TunnelClient {
     this.pollInterval = null;
     this._processing = new Set(); // track in-flight request IDs
     this._started = false;
+    this._consecutive401s = 0; // track auth failures for token reload
 
     // Configurable intervals
     this.HEARTBEAT_INTERVAL_MS = 30_000;  // 30s
     this.POLL_INTERVAL_MS = 3_000;        // 3s
     this.CLI_TIMEOUT_MS = 120_000;        // 2 min per request
+
+    // Try to load freshest token from file on construction
+    // (env var may be stale if process was started long ago)
+    this._reloadTokenFromFile();
+  }
+
+  /**
+   * Reload token from ~/.polydev.env file.
+   * The env var POLYDEV_USER_TOKEN may be stale (set when the IDE started the process).
+   * The file is always updated by the latest login.
+   */
+  _reloadTokenFromFile() {
+    try {
+      const fs = require('fs');
+      const path = require('path');
+      const os = require('os');
+      const envFile = path.join(os.homedir(), '.polydev.env');
+
+      if (!fs.existsSync(envFile)) return false;
+
+      const content = fs.readFileSync(envFile, 'utf8');
+      const match = content.match(/POLYDEV_USER_TOKEN[=\s]["']?([^"'\n]+)["']?/);
+      if (match && match[1] && (match[1].startsWith('pd_') || match[1].startsWith('polydev_'))) {
+        if (match[1] !== this.authToken) {
+          console.error(`[Tunnel] Token reloaded from ${envFile} (was stale)`);
+          this.authToken = match[1];
+          this._consecutive401s = 0;
+          return true;
+        }
+      }
+    } catch {
+      // ignore file read errors
+    }
+    return false;
+  }
+
+  /**
+   * Handle a 401 response — try reloading token from file
+   */
+  _handle401() {
+    this._consecutive401s++;
+    // Try reload every 5 consecutive 401s (every ~15s at 3s poll interval)
+    if (this._consecutive401s % 5 === 1) {
+      const reloaded = this._reloadTokenFromFile();
+      if (reloaded) {
+        console.error('[Tunnel] Token refreshed after 401 — retrying');
+      }
+    }
   }
 
   /**
@@ -36,6 +85,7 @@ class TunnelClient {
     this._started = true;
 
     console.error('[Tunnel] Starting CLI-as-API tunnel client');
+    console.error(`[Tunnel] Auth token prefix: ${this.authToken ? this.authToken.substring(0, 8) + '...' : 'NONE'}`);
 
     // Send initial heartbeat immediately
     try {
@@ -102,9 +152,14 @@ class TunnelClient {
     });
 
     if (!res.ok) {
+      if (res.status === 401) {
+        this._handle401();
+        return;
+      }
       const text = await res.text().catch(() => '');
       throw new Error(`Heartbeat failed (${res.status}): ${text}`);
     }
+    this._consecutive401s = 0; // reset on success
   }
 
   /**
@@ -120,11 +175,14 @@ class TunnelClient {
     });
 
     if (!res.ok) {
-      // 401 is expected if token expired — don't spam logs
-      if (res.status === 401) return;
+      if (res.status === 401) {
+        this._handle401();
+        return;
+      }
       const text = await res.text().catch(() => '');
       throw new Error(`Poll failed (${res.status}): ${text}`);
     }
+    this._consecutive401s = 0; // reset on success
 
     const data = await res.json();
     const requests = data.requests || [];

package/mcp/stdio-wrapper.js

@@ -2770,7 +2770,7 @@ To re-login: /polydev:login`
       const statusFile = path.join(polydevevDir, 'cli-status.json');
 
       // Ensure directory exists
-      if (!fs.existsSync(polydeveevDir)) {
+      if (!fs.existsSync(polydevevDir)) {
         fs.mkdirSync(polydeveevDir, { recursive: true });
       }
 
@@ -2962,6 +2962,10 @@ To re-login: /polydev:login`
       return; // No auth, skip tunnel
     }
 
+    // Reload token from file to get the freshest one
+    // (env var may be stale if IDE started this process long ago)
+    this.reloadTokenFromFiles();
+
     try {
       this.tunnelClient = new TunnelClient(
         this.serverUrl, // https://www.polydev.ai/api/mcp

package/package.json

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