nstantpage-agent 0.5.23 → 0.5.24

package/dist/tunnel.d.ts

@@ -42,7 +42,7 @@ export declare class TunnelClient {
     private requestsForwarded;
     private connectedAt;
     private lastPongReceived;
-    /** Active WebSocket channels (terminal WS relay through tunnel) */
+    /** Active WebSocket channels (terminal WS + HMR WS relay through tunnel) */
     private wsChannels;
     /** Listeners for status changes (used by tray app) */
     private statusListeners;
@@ -87,14 +87,20 @@ export declare class TunnelClient {
      */
     private handleWsOpen;
     /**
-     * Handle ws-data: frontend sent a message through the terminal WebSocket.
-     * Parse it and route to the terminal session.
+     * Handle ws-data: frontend sent a message through a WebSocket channel.
+     * Routes to either terminal session or dev-server WebSocket.
      */
     private handleWsData;
     /**
-     * Handle ws-close: frontend disconnected the terminal WebSocket.
+     * Handle ws-close: frontend disconnected the terminal WebSocket or HMR WebSocket.
      */
     private handleWsClose;
+    /**
+     * Handle ws-open for dev-server target: open a WebSocket to the local
+     * Vite dev server and relay data bidirectionally through the tunnel.
+     * This enables Vite HMR to work through the cloud preview URL.
+     */
+    private handleWsOpenDevServer;
     /**
      * Handle start-project: gateway wants this agent to start serving a new project.
      * Called when user clicks "Connect" in the web editor's Cloud panel.

package/dist/tunnel.js

@@ -32,7 +32,7 @@ export class TunnelClient {
     requestsForwarded = 0;
     connectedAt = 0;
     lastPongReceived = 0;
-    /** Active WebSocket channels (terminal WS relay through tunnel) */
+    /** Active WebSocket channels (terminal WS + HMR WS relay through tunnel) */
     wsChannels = new Map();
     /** Listeners for status changes (used by tray app) */
     statusListeners = new Set();
@@ -267,13 +267,31 @@ export class TunnelClient {
         });
     }
     /**
-     * Handle ws-data: frontend sent a message through the terminal WebSocket.
-     * Parse it and route to the terminal session.
+     * Handle ws-data: frontend sent a message through a WebSocket channel.
+     * Routes to either terminal session or dev-server WebSocket.
      */
-    handleWsData(wsId, data) {
+    handleWsData(wsId, data, binary) {
         const channel = this.wsChannels.get(wsId);
         if (!channel)
             return;
+        // If this channel has a dev-server WebSocket, forward data directly
+        if (channel.devServerWs) {
+            try {
+                if (channel.devServerWs.readyState === WebSocket.OPEN) {
+                    if (binary) {
+                        channel.devServerWs.send(Buffer.from(data, 'base64'));
+                    }
+                    else {
+                        channel.devServerWs.send(data);
+                    }
+                }
+            }
+            catch (err) {
+                console.warn(`  [Tunnel] Failed to forward HMR data to dev server:`, err.message);
+            }
+            return;
+        }
+        // Terminal session handling
         try {
             const msg = JSON.parse(data);
             const session = getTerminalSession(channel.sessionId);
@@ -298,16 +316,79 @@ export class TunnelClient {
         }
     }
     /**
-     * Handle ws-close: frontend disconnected the terminal WebSocket.
+     * Handle ws-close: frontend disconnected the terminal WebSocket or HMR WebSocket.
      */
     handleWsClose(wsId) {
         const channel = this.wsChannels.get(wsId);
         if (channel) {
             if (channel.cleanup)
                 channel.cleanup();
+            // Close dev-server WebSocket if this was an HMR relay
+            if (channel.devServerWs) {
+                try {
+                    channel.devServerWs.close();
+                }
+                catch { }
+            }
             this.wsChannels.delete(wsId);
         }
     }
+    // ─── Dev Server WebSocket Relay (Vite HMR through tunnel) ──
+    /**
+     * Handle ws-open for dev-server target: open a WebSocket to the local
+     * Vite dev server and relay data bidirectionally through the tunnel.
+     * This enables Vite HMR to work through the cloud preview URL.
+     */
+    handleWsOpenDevServer(wsId, url, protocol) {
+        const devPort = this.options.devPort;
+        const wsUrl = `ws://127.0.0.1:${devPort}${url}`;
+        console.log(`  [Tunnel] Opening HMR WS relay: ${wsId} → ${wsUrl}`);
+        const wsOptions = {};
+        const protocols = protocol ? protocol.split(',').map(p => p.trim()) : undefined;
+        const devWs = new WebSocket(wsUrl, protocols, wsOptions);
+        devWs.on('open', () => {
+            console.log(`  [Tunnel] HMR WS connected: ${wsId} → ${wsUrl}`);
+        });
+        devWs.on('message', (data, isBinary) => {
+            // Relay dev server messages back to browser through tunnel
+            if (isBinary) {
+                const buf = data instanceof Buffer ? data : Buffer.from(data);
+                this.send({
+                    type: 'ws-data',
+                    wsId,
+                    data: buf.toString('base64'),
+                    binary: true,
+                });
+            }
+            else {
+                this.send({
+                    type: 'ws-data',
+                    wsId,
+                    data: data.toString(),
+                });
+            }
+        });
+        devWs.on('close', () => {
+            console.log(`  [Tunnel] HMR WS closed: ${wsId}`);
+            this.send({ type: 'ws-close', wsId });
+            this.wsChannels.delete(wsId);
+        });
+        devWs.on('error', (err) => {
+            console.warn(`  [Tunnel] HMR WS error: ${wsId}:`, err.message);
+            this.send({ type: 'ws-close', wsId });
+            this.wsChannels.delete(wsId);
+        });
+        this.wsChannels.set(wsId, {
+            sessionId: `hmr-${wsId}`,
+            cleanup: () => {
+                try {
+                    devWs.close();
+                }
+                catch { }
+            },
+            devServerWs: devWs,
+        });
+    }
     /**
      * Handle start-project: gateway wants this agent to start serving a new project.
      * Called when user clicks "Connect" in the web editor's Cloud panel.
@@ -369,10 +450,15 @@ export class TunnelClient {
                 this.handleHttpRequest(msg);
                 break;
             case 'ws-open':
-                this.handleWsOpen(msg.wsId, msg.sessionId, msg.projectId);
+                if (msg.target === 'dev-server') {
+                    this.handleWsOpenDevServer(msg.wsId, msg.url || '/', msg.protocol);
+                }
+                else {
+                    this.handleWsOpen(msg.wsId, msg.sessionId, msg.projectId);
+                }
                 break;
             case 'ws-data':
-                this.handleWsData(msg.wsId, msg.data);
+                this.handleWsData(msg.wsId, msg.data, msg.binary);
                 break;
             case 'ws-close':
                 this.handleWsClose(msg.wsId);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nstantpage-agent",
-  "version": "0.5.23",
+  "version": "0.5.24",
   "description": "Local development agent for nstantpage.com — run your projects locally, preview in the cloud. Replaces cloud containers for faster builds.",
   "type": "module",
   "bin": {