@xiaou66/vite-plugin-vue-mcp-next 1.1.0 → 1.1.1

package/dist/index.cjs

@@ -94,6 +94,11 @@ var RESOLVED_VIRTUAL_SNAPDOM_LOADER_ID = `\0${VIRTUAL_SNAPDOM_LOADER_ID}`;
 var DEFAULT_MCP_CLIENT_SERVER_NAME = "vite-mcp-next";
 var LEGACY_MCP_CLIENT_SERVER_NAMES = ["vue-mcp-next"];
 var RUNTIME_PAGE_RECONNECTED_EVENT = "vite-plugin-vue-mcp-next:page-reconnected";
+var RUNTIME_PAGE_CONNECTED_EVENT = "vite-plugin-vue-mcp-next:page-connected";
+var RUNTIME_PAGE_DISCONNECTED_EVENT = "vite-plugin-vue-mcp-next:page-disconnected";
+var RUNTIME_PAGE_HEARTBEAT_EVENT = "vite-plugin-vue-mcp-next:heartbeat";
+var DEFAULT_RUNTIME_PAGE_HEARTBEAT_TIMEOUT_MS = 45e3;
+var DEFAULT_RUNTIME_PAGE_HEARTBEAT_SCAN_INTERVAL_MS = 45e3;
 var DEFAULT_OPTIONS = {
   mcpPath: DEFAULT_MCP_PATH,
   host: "localhost",
@@ -2949,32 +2954,53 @@ function vueMcpNext(userOptions = {}) {
         () => createMcpServer(ctx, server),
         server
       );
-      server.ws.on(
-        "vite-plugin-vue-mcp-next:page-connected",
-        (payload) => {
-          if (isRuntimePageTarget(payload)) {
-            ctx.pages.upsert(payload);
-            void ctx.hooks.callHook(RUNTIME_PAGE_RECONNECTED_EVENT, payload);
-            void cdpLifecycle.connectPage(payload);
+      const lastSeenAt = /* @__PURE__ */ new Map();
+      const heartbeatTimer = setInterval(() => {
+        const now = Date.now();
+        for (const [pageId, seenAt] of lastSeenAt) {
+          const target = ctx.pages.get(pageId);
+          if (!target || target.source !== "runtime" || !target.connected) {
+            lastSeenAt.delete(pageId);
+            continue;
           }
-        }
-      );
-      server.ws.on(
-        "vite-plugin-vue-mcp-next:console-record",
-        (payload) => {
-          if (isConsoleRecord(payload)) {
-            ctx.consoleRecords.push(payload);
+          if (now - seenAt >= DEFAULT_RUNTIME_PAGE_HEARTBEAT_TIMEOUT_MS) {
+            ctx.pages.disconnect(pageId, now);
+            lastSeenAt.delete(pageId);
           }
         }
-      );
-      server.ws.on(
-        "vite-plugin-vue-mcp-next:network-record",
-        (payload) => {
-          if (isNetworkRecord(payload)) {
-            ctx.networkRecords.push(payload);
+      }, DEFAULT_RUNTIME_PAGE_HEARTBEAT_SCAN_INTERVAL_MS);
+      server.ws.on(RUNTIME_PAGE_CONNECTED_EVENT, (payload) => {
+        if (isRuntimePageTarget(payload)) {
+          ctx.pages.upsert(payload);
+          lastSeenAt.set(payload.pageId, Date.now());
+          void ctx.hooks.callHook(RUNTIME_PAGE_RECONNECTED_EVENT, payload);
+          void cdpLifecycle.connectPage(payload);
+        }
+      });
+      server.ws.on(RUNTIME_PAGE_HEARTBEAT_EVENT, (payload) => {
+        if (isRuntimeHeartbeatTarget(payload)) {
+          const target = ctx.pages.get(payload.pageId);
+          if (target?.source === "runtime" && target.connected) {
+            lastSeenAt.set(payload.pageId, payload.timestamp);
           }
         }
-      );
+      });
+      server.ws.on(RUNTIME_PAGE_DISCONNECTED_EVENT, (payload) => {
+        if (isRuntimeDisconnectTarget(payload)) {
+          ctx.pages.disconnect(payload.pageId);
+          lastSeenAt.delete(payload.pageId);
+        }
+      });
+      server.ws.on("vite-plugin-vue-mcp-next:console-record", (payload) => {
+        if (isConsoleRecord(payload)) {
+          ctx.consoleRecords.push(payload);
+        }
+      });
+      server.ws.on("vite-plugin-vue-mcp-next:network-record", (payload) => {
+        if (isNetworkRecord(payload)) {
+          ctx.networkRecords.push(payload);
+        }
+      });
       server.ws.on(
         "vite-plugin-vue-mcp-next:performance-record",
         (payload) => {
@@ -3004,6 +3030,7 @@ function vueMcpNext(userOptions = {}) {
         }, 300);
       }
       server.httpServer?.once("close", () => {
+        clearInterval(heartbeatTimer);
         void cdpLifecycle.closeAll();
       });
     },
@@ -3028,6 +3055,20 @@ function isRuntimePageTarget(payload) {
   const target = payload;
   return target.source === "runtime" && typeof target.pageId === "string" && typeof target.url === "string" && typeof target.pathname === "string" && typeof target.connected === "boolean" && (target.runtimeClientId === void 0 || typeof target.runtimeClientId === "string");
 }
+function isRuntimeHeartbeatTarget(payload) {
+  if (!payload || typeof payload !== "object") {
+    return false;
+  }
+  const target = payload;
+  return typeof target.pageId === "string" && typeof target.timestamp === "number";
+}
+function isRuntimeDisconnectTarget(payload) {
+  if (!payload || typeof payload !== "object") {
+    return false;
+  }
+  const target = payload;
+  return typeof target.pageId === "string";
+}
 function isConsoleRecord(payload) {
   if (!payload || typeof payload !== "object") {
     return false;