viyv-browser-mcp 0.5.4 → 0.5.5

package/dist/index.js

@@ -683,9 +683,10 @@ function startBridge(options) {
 
 // src/server.ts
 import { randomUUID as randomUUID3 } from "crypto";
-import { existsSync, statSync } from "fs";
+import { existsSync, mkdirSync, statSync, writeFileSync } from "fs";
 import http from "http";
 import { createConnection as createConnection2 } from "net";
+import { dirname } from "path";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
@@ -2301,7 +2302,8 @@ var smExportTool = {
   name: "sm_export",
   description: SM_EXPORT_DESCRIPTION,
   inputSchema: z.object({
-    domain: z.string().optional().describe("Export only pages matching this domain (subdomain-aware). Omit for all.")
+    domain: z.string().optional().describe("Export only pages matching this domain (subdomain-aware). Omit for all."),
+    file_path: z.string().optional().describe("Save to this path and return metadata only (keeps data out of context)")
   })
 };
 var smImportTool = {
@@ -2462,7 +2464,8 @@ var smReportExportTool = {
   description: SM_REPORT_EXPORT_DESCRIPTION,
   inputSchema: z.object({
     report_id: z.string().describe("Report ID to export"),
-    format: z.enum(["json", "csv"]).optional().describe("Export format (default: json)")
+    format: z.enum(["json", "csv"]).optional().describe("Export format (default: json)"),
+    file_path: z.string().optional().describe("Save to this path and return metadata only (keeps data out of context)")
   })
 };
 var jobEntrySchema = z.object({
@@ -2532,7 +2535,8 @@ var smJobReportExportTool = {
   description: SM_JOB_REPORT_EXPORT_DESCRIPTION,
   inputSchema: z.object({
     job_report_id: z.string().describe("Job report ID to export"),
-    format: z.enum(["json", "csv"]).optional().describe("Export format (default: json)")
+    format: z.enum(["json", "csv"]).optional().describe("Export format (default: json)"),
+    file_path: z.string().optional().describe("Save to this path and return metadata only (keeps data out of context)")
   })
 };
 var smJobHistoryTool = {
@@ -2766,6 +2770,32 @@ function coerceShape(shape) {
   }
   return result;
 }
+var FILE_EXPORT_TOOLS = /* @__PURE__ */ new Set(["sm_report_export", "sm_job_report_export", "sm_export"]);
+function handleFileExport(filePath, result) {
+  mkdirSync(dirname(filePath), { recursive: true });
+  let content;
+  let metadata;
+  if (typeof result.data === "string") {
+    const { data, ...rest } = result;
+    content = data;
+    metadata = { ...rest, file_path: filePath };
+    if (result.format === "csv") {
+      const lines = content.split("\n").filter((l) => l.trim());
+      const headerLine = lines[0]?.replace(/^\uFEFF/, "") ?? "";
+      const columns = headerLine.split(",").map((h) => h.replace(/^"|"$/g, "").trim());
+      metadata.row_count = lines.length - 1;
+      metadata.column_count = columns.length;
+      metadata.columns = columns;
+    }
+  } else {
+    content = JSON.stringify(result, null, 2);
+    const pages = Array.isArray(result.pages) ? result.pages : [];
+    metadata = { ok: true, file_path: filePath, page_count: pages.length };
+  }
+  writeFileSync(filePath, content, "utf-8");
+  metadata.file_size = Buffer.byteLength(content, "utf-8");
+  return metadata;
+}
 function createConfiguredMcpServer() {
   const server = new McpServer({
     name: MCP_SERVER.NAME,
@@ -3428,9 +3458,34 @@ async function callExtensionTool(tool, input) {
         ]
       });
     }, toolTimeout);
+    const { chromeProfile, file_path: filePath, ...cleanInput } = input;
     pendingRequests.set(requestId, {
       resolve: (result) => {
         removeErrorListener();
+        if (FILE_EXPORT_TOOLS.has(tool) && typeof filePath === "string" && ("data" in result || "pages" in result)) {
+          try {
+            const metadata = handleFileExport(filePath, result);
+            resolve2({
+              content: [{ type: "text", text: JSON.stringify(metadata) }]
+            });
+          } catch (e) {
+            const msg = e instanceof Error ? e.message : String(e);
+            resolve2({
+              content: [
+                {
+                  type: "text",
+                  text: JSON.stringify({
+                    error: {
+                      code: "FILE_WRITE_ERROR",
+                      message: `Failed to write ${filePath}: ${msg}`
+                    }
+                  })
+                }
+              ]
+            });
+          }
+          return;
+        }
         if (tool === "screenshot" && typeof result.data === "string") {
           const { data, ...metadata } = result;
           const mimeType = metadata.format === "png" ? "image/png" : "image/jpeg";
@@ -3461,7 +3516,6 @@ async function callExtensionTool(tool, input) {
       },
       timer
     });
-    const { chromeProfile, ...cleanInput } = input;
     const request = {
       id: requestId,
       type: "tool_call",
@@ -3540,9 +3594,9 @@ async function handleSwitchBrowser() {
 
 // src/setup.ts
 import { execSync } from "child_process";
-import { chmodSync, existsSync as existsSync2, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { chmodSync, existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync, writeFileSync as writeFileSync2 } from "fs";
 import { homedir, platform } from "os";
-import { dirname, resolve } from "path";
+import { dirname as dirname2, resolve } from "path";
 function runSetup(options = {}) {
   const os = platform();
   const binaryPath = getBinaryPath();
@@ -3569,11 +3623,11 @@ function runSetup(options = {}) {
     allowed_origins: allowedOrigins
   };
   const manifestPath = getManifestPath(os);
-  const manifestDir = dirname(manifestPath);
+  const manifestDir = dirname2(manifestPath);
   console.log(`Wrapper: ${wrapperPath}`);
   console.log(`Manifest path: ${manifestPath}`);
-  mkdirSync(manifestDir, { recursive: true });
-  writeFileSync(manifestPath, JSON.stringify(manifest, null, 2));
+  mkdirSync2(manifestDir, { recursive: true });
+  writeFileSync2(manifestPath, JSON.stringify(manifest, null, 2));
   chmodSync(manifestPath, 420);
   console.log("\nNative Messaging Host registered successfully!");
   console.log("\nNext steps:");
@@ -3595,18 +3649,18 @@ function getBinaryPath() {
   return resolve(currentScript);
 }
 function createNativeHostWrapper(os, binaryPath) {
-  const manifestDir = dirname(getManifestPath(os));
-  mkdirSync(manifestDir, { recursive: true });
+  const manifestDir = dirname2(getManifestPath(os));
+  mkdirSync2(manifestDir, { recursive: true });
   const nodePath = getNodePath();
   if (os === "win32") {
     const wrapperPath2 = resolve(manifestDir, `${NATIVE_HOST_NAME}.bat`);
-    writeFileSync(wrapperPath2, `@echo off\r
+    writeFileSync2(wrapperPath2, `@echo off\r
 "${nodePath}" "${binaryPath}" --native-host\r
 `);
     return wrapperPath2;
   }
   const wrapperPath = resolve(manifestDir, `${NATIVE_HOST_NAME}.sh`);
-  writeFileSync(wrapperPath, `#!/bin/bash
+  writeFileSync2(wrapperPath, `#!/bin/bash
 exec "${nodePath}" "${binaryPath}" --native-host
 `);
   chmodSync(wrapperPath, 493);
@@ -3664,7 +3718,7 @@ function setupClaudeDesktopConfig() {
   if (!config.mcpServers || typeof config.mcpServers !== "object") {
     config.mcpServers = {};
   }
-  const nodeBinDir = dirname(getNodePath());
+  const nodeBinDir = dirname2(getNodePath());
   const defaultPath = "/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin";
   const envPath = nodeBinDir ? `${nodeBinDir}:${defaultPath}` : defaultPath;
   config.mcpServers["viyv-browser"] = {
@@ -3672,9 +3726,9 @@ function setupClaudeDesktopConfig() {
     args: ["-y", "viyv-browser-mcp@latest"],
     env: { PATH: envPath }
   };
-  const configDir = dirname(configPath);
-  mkdirSync(configDir, { recursive: true });
-  writeFileSync(configPath, JSON.stringify(config, null, 2));
+  const configDir = dirname2(configPath);
+  mkdirSync2(configDir, { recursive: true });
+  writeFileSync2(configPath, JSON.stringify(config, null, 2));
   console.log(`npx path: ${npxPath}`);
   console.log("\nClaude Desktop config updated successfully!");
   console.log("Please restart Claude Desktop for changes to take effect.");