viyv-browser-mcp 0.7.2 → 0.7.3

package/dist/index.js

@@ -88,6 +88,57 @@ var MCP_SERVER = {
   SOCKET_PATH_TEMPLATE: "/tmp/viyv-browser-{pid}.sock"
 };
 
+// src/native-host-updater.ts
+import { chmodSync, copyFileSync, existsSync, mkdirSync, renameSync, unlinkSync } from "fs";
+import { homedir } from "os";
+import { resolve } from "path";
+import { fileURLToPath } from "url";
+var NATIVE_HOST_DIR = resolve(homedir(), ".viyv-browser", "native-host");
+function getSourceBinaryPath() {
+  return fileURLToPath(import.meta.url);
+}
+function getNativeHostBinaryPath() {
+  return resolve(NATIVE_HOST_DIR, "index.js");
+}
+function isNativeHostDeployed() {
+  return existsSync(getNativeHostBinaryPath());
+}
+function syncNativeHostBinary() {
+  const source = getSourceBinaryPath();
+  if (!existsSync(source)) {
+    throw new Error(`Source binary not found: ${source}`);
+  }
+  mkdirSync(NATIVE_HOST_DIR, { recursive: true });
+  const target = getNativeHostBinaryPath();
+  const tmp = resolve(NATIVE_HOST_DIR, `index.js.tmp.${process.pid}`);
+  try {
+    copyFileSync(source, tmp);
+    chmodSync(tmp, 493);
+    renameSync(tmp, target);
+  } catch (err) {
+    try {
+      unlinkSync(tmp);
+    } catch {
+    }
+    throw err;
+  }
+}
+function shouldUpdateBridge(bridgeVersion) {
+  return compareSemver("0.7.3", bridgeVersion) > 0;
+}
+function getPackageVersion() {
+  return "0.7.3";
+}
+function compareSemver(a, b) {
+  const pa = a.split(".").map(Number);
+  const pb = b.split(".").map(Number);
+  for (let i = 0; i < 3; i++) {
+    const diff = (pa[i] ?? 0) - (pb[i] ?? 0);
+    if (diff !== 0) return diff;
+  }
+  return 0;
+}
+
 // src/native-host/bridge-relay.ts
 import { createConnection } from "net";
 
@@ -382,6 +433,7 @@ function startBridge(options) {
     const msg = {
       type: "bridge_status",
       connected: true,
+      bridgeVersion: getPackageVersion(),
       chromeProfiles: getChromeProfilesList(),
       timestamp: Date.now()
     };
@@ -460,6 +512,28 @@ function startBridge(options) {
     } else if (type === "bridge_status" || type === "bridge_closing") {
       process.stderr.write(`${LOG_PREFIX2} Ignoring ${type} from MCP Server ${connId}
 `);
+    } else if (type === "bridge_shutdown_request") {
+      process.stderr.write(`${LOG_PREFIX2} Shutdown requested by MCP Server ${connId}
+`);
+      const DRAIN_TIMEOUT = 5e3;
+      const drainAndExit = () => {
+        if (requestOrigin.size === 0) {
+          gracefulExit("Bridge shutdown requested for version update");
+          return;
+        }
+        process.stderr.write(
+          `${LOG_PREFIX2} Waiting for ${requestOrigin.size} pending request(s) to drain...
+`
+        );
+        const start = Date.now();
+        const check = setInterval(() => {
+          if (requestOrigin.size === 0 || Date.now() - start > DRAIN_TIMEOUT) {
+            clearInterval(check);
+            gracefulExit("Bridge shutdown requested for version update");
+          }
+        }, 200);
+      };
+      drainAndExit();
     } else {
       sendToChrome(agentId ? getChromeForAgent(agentId) : primaryChromeId, message);
     }
@@ -684,6 +758,7 @@ function startBridge(options) {
     sendToMcp(connId, {
       type: "bridge_status",
       connected: true,
+      bridgeVersion: getPackageVersion(),
       chromeProfiles: getChromeProfilesList(),
       timestamp: Date.now()
     });
@@ -770,7 +845,7 @@ function startBridge(options) {
 
 // src/server.ts
 import { randomUUID as randomUUID3 } from "crypto";
-import { existsSync, mkdirSync, readFileSync, statSync, unlinkSync, writeFileSync } from "fs";
+import { existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync, statSync, unlinkSync as unlinkSync2, writeFileSync } from "fs";
 import http from "http";
 import { createConnection as createConnection2 } from "net";
 import { dirname } from "path";
@@ -4846,6 +4921,7 @@ function computeToolTimeout(tool, input) {
 // src/server.ts
 var pendingRequests = /* @__PURE__ */ new Map();
 var extensionSocket = null;
+var bridgeUpdateInfo = null;
 var configuredChromeProfile;
 function coerceValue(v) {
   if (typeof v !== "string") return v;
@@ -4867,7 +4943,7 @@ function coerceShape(shape) {
   return result;
 }
 function handleFileExport(filePath, result) {
-  mkdirSync(dirname(filePath), { recursive: true });
+  mkdirSync2(dirname(filePath), { recursive: true });
   let content;
   let metadata;
   if (typeof result.data === "string") {
@@ -4931,6 +5007,14 @@ function createConfiguredMcpServer() {
         } catch {
         }
       }
+      if (bridgeUpdateInfo) {
+        const info = bridgeUpdateInfo;
+        bridgeUpdateInfo = null;
+        result.content.unshift({
+          type: "text",
+          text: `[INFO] Bridge was restarted for version update (${info.from} \u2192 ${info.to}). Connection re-established.`
+        });
+      }
       return result;
     });
   }
@@ -5221,7 +5305,7 @@ async function handleStreamableHttpRequest(req, res, sessions2) {
 }
 var MAX_BODY_SIZE = 4 * 1024 * 1024;
 function parseJsonBody(req) {
-  return new Promise((resolve2, reject) => {
+  return new Promise((resolve3, reject) => {
     const chunks = [];
     let size = 0;
     let destroyed = false;
@@ -5239,7 +5323,7 @@ function parseJsonBody(req) {
     req.on("end", () => {
       if (destroyed) return;
       try {
-        resolve2(JSON.parse(Buffer.concat(chunks).toString("utf-8")));
+        resolve3(JSON.parse(Buffer.concat(chunks).toString("utf-8")));
       } catch (error) {
         reject(new Error(`Invalid JSON: ${error.message}`));
       }
@@ -5447,6 +5531,21 @@ function handleExtensionMessage(message) {
         }
       }
     }
+    const bridgeVersion = typeof msg.bridgeVersion === "string" ? msg.bridgeVersion : null;
+    if (bridgeVersion && shouldUpdateBridge(bridgeVersion)) {
+      const serverVersion = getPackageVersion();
+      process.stderr.write(
+        `[viyv-browser:mcp] Bridge version mismatch (bridge=${bridgeVersion}, server=${serverVersion}), requesting restart
+`
+      );
+      bridgeUpdateInfo = { from: bridgeVersion, to: serverVersion };
+      if (extensionSocket && !extensionSocket.destroyed) {
+        extensionSocket.write(
+          `${JSON.stringify({ type: "bridge_shutdown_request", timestamp: Date.now() })}
+`
+        );
+      }
+    }
   } else if (type === "browser_event") {
     process.stderr.write(`[viyv-browser:mcp] Browser event: ${String(msg.eventType)}
 `);
@@ -5584,7 +5683,7 @@ async function callExtensionTool(tool, input) {
     if (paths) {
       for (const p of paths) {
         try {
-          if (!existsSync(p) || !statSync(p).isFile()) {
+          if (!existsSync2(p) || !statSync(p).isFile()) {
             return {
               content: [
                 {
@@ -5616,7 +5715,7 @@ async function callExtensionTool(tool, input) {
     let csvData;
     if (typeof input.file_path === "string") {
       const fp = input.file_path;
-      if (!existsSync(fp) || !statSync(fp).isFile()) {
+      if (!existsSync2(fp) || !statSync(fp).isFile()) {
         return {
           content: [
             {
@@ -5674,13 +5773,13 @@ async function callExtensionTool(tool, input) {
   touchSession(agentId);
   const sock = extensionSocket;
   const toolTimeout = computeToolTimeout(tool, input);
-  return new Promise((resolve2) => {
+  return new Promise((resolve3) => {
     const onError = () => {
       const pending = pendingRequests.get(requestId);
       if (pending) {
         clearTimeout(pending.timer);
         pendingRequests.delete(requestId);
-        resolve2({
+        resolve3({
           content: [
             {
               type: "text",
@@ -5701,7 +5800,7 @@ async function callExtensionTool(tool, input) {
     const timer = setTimeout(() => {
       pendingRequests.delete(requestId);
       removeErrorListener();
-      resolve2({
+      resolve3({
         content: [
           {
             type: "text",
@@ -5726,12 +5825,12 @@ async function callExtensionTool(tool, input) {
         if (FILE_EXPORT_TOOLS.has(tool) && typeof filePath === "string" && ("data" in result || "pages" in result)) {
           try {
             const metadata = handleFileExport(filePath, result);
-            resolve2({
+            resolve3({
               content: [{ type: "text", text: JSON.stringify(metadata) }]
             });
           } catch (e) {
             const msg = e instanceof Error ? e.message : String(e);
-            resolve2({
+            resolve3({
               content: [
                 {
                   type: "text",
@@ -5750,9 +5849,9 @@ async function callExtensionTool(tool, input) {
         if (RESULT_EXPORT_TOOLS.has(tool) && typeof filePath === "string") {
           try {
             const content = JSON.stringify(result, null, 2);
-            mkdirSync(dirname(filePath), { recursive: true });
+            mkdirSync2(dirname(filePath), { recursive: true });
             writeFileSync(filePath, content, "utf-8");
-            resolve2({
+            resolve3({
               content: [
                 {
                   type: "text",
@@ -5765,7 +5864,7 @@ async function callExtensionTool(tool, input) {
             });
           } catch (e) {
             const msg = e instanceof Error ? e.message : String(e);
-            resolve2({
+            resolve3({
               content: [
                 {
                   type: "text",
@@ -5792,21 +5891,21 @@ async function callExtensionTool(tool, input) {
         }
         if (tempImportFile) {
           try {
-            unlinkSync(tempImportFile);
+            unlinkSync2(tempImportFile);
           } catch {
           }
         }
-        resolve2({ content: buildResponseContent(tool, finalResult) });
+        resolve3({ content: buildResponseContent(tool, finalResult) });
       },
       reject: (error) => {
         removeErrorListener();
         if (tempImportFile) {
           try {
-            unlinkSync(tempImportFile);
+            unlinkSync2(tempImportFile);
           } catch {
           }
         }
-        resolve2({
+        resolve3({
           content: [
             {
               type: "text",
@@ -5858,12 +5957,12 @@ async function handleSwitchBrowser() {
     setExtensionConnected(false);
   }
   process.stderr.write("[viyv-browser:mcp] switch_browser: waiting for bridge reconnection...\n");
-  return new Promise((resolve2) => {
+  return new Promise((resolve3) => {
     const checkInterval = setInterval(() => {
       if (extensionSocket && !extensionSocket.destroyed && isExtensionConnected()) {
         clearInterval(checkInterval);
         clearTimeout(timer);
-        resolve2({
+        resolve3({
           content: [
             {
               type: "text",
@@ -5878,7 +5977,7 @@ async function handleSwitchBrowser() {
     }, 500);
     const timer = setTimeout(() => {
       clearInterval(checkInterval);
-      resolve2({
+      resolve3({
         content: [
           {
             type: "text",
@@ -5897,27 +5996,24 @@ async function handleSwitchBrowser() {
 
 // src/setup.ts
 import { execSync } from "child_process";
-import { chmodSync, existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync as readFileSync2, writeFileSync as writeFileSync2 } from "fs";
-import { homedir, platform } from "os";
-import { dirname as dirname2, resolve } from "path";
+import { chmodSync as chmodSync2, existsSync as existsSync3, mkdirSync as mkdirSync3, readFileSync as readFileSync2, writeFileSync as writeFileSync2 } from "fs";
+import { homedir as homedir2, platform } from "os";
+import { dirname as dirname2, resolve as resolve2 } from "path";
 function runSetup(options = {}) {
   const os = platform();
-  const binaryPath = getBinaryPath();
   console.log("Viyv Browser MCP - Native Messaging Host Setup");
   console.log("================================================");
   console.log(`Platform: ${os}`);
-  console.log(`Binary: ${binaryPath}`);
-  if (!existsSync2(binaryPath)) {
-    console.error(`WARNING: Binary not found at ${binaryPath}`);
-    console.error("The Native Messaging Host may not work until the binary is available.");
-  }
   const allowedOrigins = options.extensionId ? [`chrome-extension://${options.extensionId}/`] : ["chrome-extension://*/"];
   if (!options.extensionId) {
     process.stderr.write(
       "WARNING: Using wildcard allowed_origins (chrome-extension://*/). This allows any Chrome extension to connect. For production, specify --extension-id to restrict access.\n"
     );
   }
-  const wrapperPath = createNativeHostWrapper(os, binaryPath);
+  syncNativeHostBinary();
+  const nativeHostPath = getNativeHostBinaryPath();
+  console.log(`Native host binary: ${nativeHostPath}`);
+  const wrapperPath = createNativeHostWrapper(os, nativeHostPath);
   const manifest = {
     name: NATIVE_HOST_NAME,
     description: "Viyv Browser MCP Native Messaging Host",
@@ -5929,9 +6025,9 @@ function runSetup(options = {}) {
   const manifestDir = dirname2(manifestPath);
   console.log(`Wrapper: ${wrapperPath}`);
   console.log(`Manifest path: ${manifestPath}`);
-  mkdirSync2(manifestDir, { recursive: true });
+  mkdirSync3(manifestDir, { recursive: true });
   writeFileSync2(manifestPath, JSON.stringify(manifest, null, 2));
-  chmodSync(manifestPath, 420);
+  chmodSync2(manifestPath, 420);
   console.log("\nNative Messaging Host registered successfully!");
   console.log("\nNext steps:");
   console.log("1. Start the MCP Server: node <path>/dist/index.js");
@@ -5941,32 +6037,22 @@ function runSetup(options = {}) {
     setupClaudeDesktopConfig();
   }
 }
-function getBinaryPath() {
-  const whichCmd = process.platform === "win32" ? "where" : "which";
-  try {
-    const found = execSync(`${whichCmd} viyv-browser-mcp`, { encoding: "utf-8" }).trim();
-    if (found) return found;
-  } catch {
-  }
-  const currentScript = process.argv[1];
-  return resolve(currentScript);
-}
 function createNativeHostWrapper(os, binaryPath) {
   const manifestDir = dirname2(getManifestPath(os));
-  mkdirSync2(manifestDir, { recursive: true });
+  mkdirSync3(manifestDir, { recursive: true });
   const nodePath = getNodePath();
   if (os === "win32") {
-    const wrapperPath2 = resolve(manifestDir, `${NATIVE_HOST_NAME}.bat`);
+    const wrapperPath2 = resolve2(manifestDir, `${NATIVE_HOST_NAME}.bat`);
     writeFileSync2(wrapperPath2, `@echo off\r
 "${nodePath}" "${binaryPath}" --native-host\r
 `);
     return wrapperPath2;
   }
-  const wrapperPath = resolve(manifestDir, `${NATIVE_HOST_NAME}.sh`);
+  const wrapperPath = resolve2(manifestDir, `${NATIVE_HOST_NAME}.sh`);
   writeFileSync2(wrapperPath, `#!/bin/bash
 exec "${nodePath}" "${binaryPath}" --native-host
 `);
-  chmodSync(wrapperPath, 493);
+  chmodSync2(wrapperPath, 493);
   return wrapperPath;
 }
 function getNodePath() {
@@ -5977,18 +6063,18 @@ function getNodePath() {
   }
 }
 function getManifestPath(os) {
-  const home = homedir();
+  const home = homedir2();
   switch (os) {
     case "darwin":
-      return resolve(
+      return resolve2(
         home,
         "Library/Application Support/Google/Chrome/NativeMessagingHosts",
         `${NATIVE_HOST_NAME}.json`
       );
     case "linux":
-      return resolve(home, ".config/google-chrome/NativeMessagingHosts", `${NATIVE_HOST_NAME}.json`);
+      return resolve2(home, ".config/google-chrome/NativeMessagingHosts", `${NATIVE_HOST_NAME}.json`);
     case "win32":
-      return resolve(
+      return resolve2(
         home,
         "AppData/Local/Google/Chrome/User Data/NativeMessagingHosts",
         `${NATIVE_HOST_NAME}.json`
@@ -6008,7 +6094,7 @@ function setupClaudeDesktopConfig() {
   const configPath = getClaudeDesktopConfigPath();
   console.log(`Config path: ${configPath}`);
   let config = {};
-  if (existsSync2(configPath)) {
+  if (existsSync3(configPath)) {
     try {
       config = JSON.parse(readFileSync2(configPath, "utf-8"));
     } catch {
@@ -6030,7 +6116,7 @@ function setupClaudeDesktopConfig() {
     env: { PATH: envPath }
   };
   const configDir = dirname2(configPath);
-  mkdirSync2(configDir, { recursive: true });
+  mkdirSync3(configDir, { recursive: true });
   writeFileSync2(configPath, JSON.stringify(config, null, 2));
   console.log(`npx path: ${npxPath}`);
   console.log("\nClaude Desktop config updated successfully!");
@@ -6045,16 +6131,16 @@ function getNpxPath() {
   }
 }
 function getClaudeDesktopConfigPath() {
-  const home = homedir();
+  const home = homedir2();
   switch (process.platform) {
     case "darwin":
-      return resolve(home, "Library/Application Support/Claude/claude_desktop_config.json");
+      return resolve2(home, "Library/Application Support/Claude/claude_desktop_config.json");
     case "linux":
-      return resolve(home, ".config/Claude/claude_desktop_config.json");
+      return resolve2(home, ".config/Claude/claude_desktop_config.json");
     case "win32":
-      return resolve(home, "AppData/Roaming/Claude/claude_desktop_config.json");
+      return resolve2(home, "AppData/Roaming/Claude/claude_desktop_config.json");
     default:
-      return resolve(home, ".config/Claude/claude_desktop_config.json");
+      return resolve2(home, ".config/Claude/claude_desktop_config.json");
   }
 }
 
@@ -6089,6 +6175,16 @@ if (args.includes("setup")) {
   const port = portIdx >= 0 ? Number(args[portIdx + 1]) : void 0;
   const chromeProfileIdx = args.indexOf("--chrome-profile");
   const chromeProfile = chromeProfileIdx >= 0 ? args[chromeProfileIdx + 1] : process.env.VIYV_CHROME_PROFILE;
+  if (isNativeHostDeployed()) {
+    try {
+      syncNativeHostBinary();
+    } catch (err) {
+      process.stderr.write(
+        `[viyv-browser:mcp] WARNING: Failed to update native host binary: ${err.message}
+`
+      );
+    }
+  }
   startMcpServer(agentName, { transport: transportMode, port, chromeProfile });
 }
 //# sourceMappingURL=index.js.map