@opentiny/webmcp-cli 0.0.1-alpha.5 → 0.0.1-alpha.6

package/dist/bin.cjs

@@ -26,25 +26,197 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 // src/bin.ts
 var import_commander = require("commander");
 var import_picocolors2 = __toESM(require("picocolors"), 1);
-var import_fs2 = require("fs");
-var import_path2 = require("path");
+var import_fs3 = require("fs");
+var import_path3 = require("path");
+
+// src/expand-file-refs.ts
+var import_fs = require("fs");
+var import_path = require("path");
+var import_url = require("url");
+var JSON_STRING_LITERAL = /"(?:[^"\\]|\\.)*"/g;
+var UNQUOTED_FILE_REF = /(?<![\\"])@(base64)?file:([^\s"',}\]]+)/g;
+var FILE_REF_PREFIX = /^@(base64)?file:(.+)$/s;
+function resolveFilePath(filePath, baseDir = process.cwd()) {
+  let normalizedPath = filePath.trim();
+  if (normalizedPath.startsWith('"') && normalizedPath.endsWith('"') || normalizedPath.startsWith("'") && normalizedPath.endsWith("'")) {
+    normalizedPath = normalizedPath.slice(1, -1).trim();
+  }
+  if (normalizedPath.startsWith("file://")) {
+    normalizedPath = (0, import_url.fileURLToPath)(normalizedPath);
+  }
+  if ((0, import_path.isAbsolute)(normalizedPath)) {
+    return (0, import_path.normalize)(normalizedPath);
+  }
+  return (0, import_path.resolve)(baseDir, normalizedPath);
+}
+function decodeFilePath(rawPath) {
+  return rawPath.replace(/\\\\/g, "\\");
+}
+function readFileRefContent(filePath, baseDir, base64) {
+  const absPath = resolveFilePath(filePath, baseDir);
+  let content;
+  try {
+    content = (0, import_fs.readFileSync)(absPath, "utf-8");
+  } catch (e) {
+    throw new Error(`\u65E0\u6CD5\u8BFB\u53D6\u6587\u4EF6\u5F15\u7528 "${filePath}" (${absPath}): ${e.message}`);
+  }
+  return base64 ? Buffer.from(content, "utf-8").toString("base64") : content;
+}
+function expandFileRefValue(rawInner, baseDir) {
+  const match = rawInner.match(FILE_REF_PREFIX);
+  if (!match) {
+    return null;
+  }
+  const [, base64Flag, rawPath] = match;
+  const filePath = decodeFilePath(rawPath);
+  const value = readFileRefContent(filePath, baseDir, Boolean(base64Flag));
+  return JSON.stringify(value);
+}
+function expandQuotedFileRefs(argsJson, baseDir) {
+  return argsJson.replace(JSON_STRING_LITERAL, (jsonStringLiteral) => {
+    const rawInner = jsonStringLiteral.slice(1, -1);
+    const replacement = expandFileRefValue(rawInner, baseDir);
+    return replacement ?? jsonStringLiteral;
+  });
+}
+function expandUnquotedFileRefs(argsJson, baseDir) {
+  return argsJson.replace(UNQUOTED_FILE_REF, (match, base64Flag, filePath) => {
+    const value = readFileRefContent(filePath, baseDir, Boolean(base64Flag));
+    return JSON.stringify(value);
+  });
+}
+function expandFileRefs(argsJson, baseDir = process.cwd()) {
+  const withQuotedExpanded = expandQuotedFileRefs(argsJson, baseDir);
+  return expandUnquotedFileRefs(withQuotedExpanded, baseDir);
+}
+function getFileBaseDir(fileOption) {
+  if (!fileOption) {
+    return process.cwd();
+  }
+  return (0, import_path.dirname)((0, import_path.resolve)(process.cwd(), fileOption));
+}
+
+// src/parse-run-args.ts
+var import_json5 = __toESM(require("json5"), 1);
+function cleanJsonString(str) {
+  let cleaned = str.trim();
+  if (cleaned.startsWith("'") && cleaned.endsWith("'")) {
+    cleaned = cleaned.slice(1, -1).trim();
+  }
+  return cleaned;
+}
+function looksLikeObjectBody(raw) {
+  return /^("?[\w-]+"?\s*:)/.test(raw);
+}
+function joinShellArgs(args) {
+  const cleaned = args.map(cleanJsonString);
+  const first = cleaned[0] ?? "";
+  if (first.startsWith("{")) {
+    return cleaned.join(" ");
+  }
+  if (looksLikeObjectBody(first)) {
+    let joined = cleaned.join(" ");
+    if (!joined.startsWith("{")) {
+      joined = `{${joined}}`;
+    }
+    if (!joined.endsWith("}")) {
+      joined = `${joined}}`;
+    }
+    return joined;
+  }
+  return cleaned.join(" ");
+}
+function insertMissingCommas(raw) {
+  return raw.replace(/"(\s+)("?[\w-]+"?\s*:)/g, '",$2');
+}
+function quoteUnquotedStringValues(raw) {
+  return raw.replace(
+    /([{,]\s*(?:"([^"]+)"|([\w-]+))\s*:\s*)(?!"@(?:base64)?file:|"|true|false|null|-?\d|\{|\[)([^",}\]]+?)(?=\s*[,}])/g,
+    (_match, prefix, _quotedKey, _unquotedKey, value) => {
+      const trimmed = value.trim();
+      if (/^-?\d+(\.\d+)?([eE][+-]?\d+)?$/.test(trimmed)) {
+        return prefix + trimmed;
+      }
+      return prefix + JSON.stringify(trimmed);
+    }
+  );
+}
+function normalizeShellDamagedJson(raw) {
+  let normalized = raw.trim();
+  normalized = insertMissingCommas(normalized);
+  normalized = quoteUnquotedStringValues(normalized);
+  return normalized;
+}
+function buildArgsFromKeyValuePairs(args) {
+  const obj = {};
+  for (const pair of args) {
+    const index = pair.indexOf("=");
+    if (index === -1) {
+      throw new Error(`\u65E0\u6548\u7684\u53C2\u6570\u683C\u5F0F "${pair}"\u3002\u8BF7\u4F7F\u7528 key=value\uFF0C\u6216 JSON / -f \u6587\u4EF6\u3002`);
+    }
+    const key = pair.substring(0, index).trim();
+    let val = cleanJsonString(pair.substring(index + 1).trim());
+    try {
+      obj[key] = JSON.parse(val);
+    } catch {
+      obj[key] = val;
+    }
+  }
+  return JSON.stringify(obj);
+}
+function parseArgsJson(raw) {
+  const trimmed = cleanJsonString(raw);
+  try {
+    return JSON.stringify(import_json5.default.parse(trimmed));
+  } catch (firstError) {
+    const repaired = normalizeShellDamagedJson(trimmed);
+    try {
+      return JSON.stringify(import_json5.default.parse(repaired));
+    } catch {
+      const hint = `\u63D0\u793A\uFF1Abash \u4E2D JSON \u5FC5\u987B\u7528\u5355\u5F15\u53F7\u5305\u88F9\uFF0C\u4F8B\u5982 '{"title":"\u6807\u9898","content":"@base64file:./article.md"}'\uFF1B\u6216\u4F7F\u7528 key=value\uFF1Atitle=\u6807\u9898 content=@base64file:./article.md\uFF1B\u6216\u4F7F\u7528 -f args.json\u3002`;
+      const message = firstError instanceof Error ? firstError.message : String(firstError);
+      throw new Error(`\u53C2\u6570\u4E0D\u662F\u6709\u6548\u7684 JSON \u6216 JS \u5BF9\u8C61: ${message}
+${hint}`);
+    }
+  }
+}
+function prepareRunArgsJson(args, fileContent, baseDir) {
+  let raw = "";
+  if (fileContent !== void 0) {
+    raw = fileContent.trim();
+  } else if (args.length > 0) {
+    const joined = joinShellArgs(args);
+    if (joined.startsWith("{") || looksLikeObjectBody(joined)) {
+      raw = joined;
+    } else if (args.every((arg) => arg.includes("="))) {
+      raw = buildArgsFromKeyValuePairs(args);
+    } else {
+      raw = joined;
+    }
+  }
+  if (!raw) {
+    throw new Error("\u5FC5\u987B\u63D0\u4F9B\u53C2\u6570\u6216\u901A\u8FC7 -f/--file \u6307\u5B9A\u53C2\u6570\u6587\u4EF6");
+  }
+  const expanded = expandFileRefs(raw, baseDir);
+  return parseArgsJson(expanded);
+}
 
 // src/browser.ts
 var import_puppeteer_core = __toESM(require("puppeteer-core"), 1);
 var import_picocolors = __toESM(require("picocolors"), 1);
-var import_fs = __toESM(require("fs"), 1);
+var import_fs2 = __toESM(require("fs"), 1);
 var import_os = __toESM(require("os"), 1);
-var import_path = __toESM(require("path"), 1);
+var import_path2 = __toESM(require("path"), 1);
 var import_http = __toESM(require("http"), 1);
 var import_child_process = require("child_process");
-var import_url = require("url");
+var import_url2 = require("url");
 var import_meta = {};
-var __filename = (0, import_url.fileURLToPath)(import_meta.url);
-var __dirname = import_path.default.dirname(__filename);
+var __filename = (0, import_url2.fileURLToPath)(import_meta.url);
+var __dirname = import_path2.default.dirname(__filename);
 var CDP_PORT = 9222;
 var CDP_URL = `http://localhost:${CDP_PORT}`;
 function httpGet(url, timeoutMs = 1500) {
-  return new Promise((resolve2, reject) => {
+  return new Promise((resolve3, reject) => {
     const req = import_http.default.get(url, { timeout: timeoutMs }, (res) => {
       let data = "";
       res.on("data", (chunk) => {
@@ -52,7 +224,7 @@ function httpGet(url, timeoutMs = 1500) {
       });
       res.on("end", () => {
         if (res.statusCode && res.statusCode >= 200 && res.statusCode < 300) {
-          resolve2(data);
+          resolve3(data);
         } else {
           reject(new Error(`HTTP ${res.statusCode}`));
         }
@@ -93,9 +265,9 @@ function getDefaultBrowserPath() {
   const platform = import_os.default.platform();
   if (platform === "darwin") {
     const chromePath = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome";
-    if (import_fs.default.existsSync(chromePath)) return { path: chromePath, name: "Chrome" };
+    if (import_fs2.default.existsSync(chromePath)) return { path: chromePath, name: "Chrome" };
     const edgePath = "/Applications/Microsoft Edge.app/Contents/MacOS/Microsoft Edge";
-    if (import_fs.default.existsSync(edgePath)) return { path: edgePath, name: "Edge" };
+    if (import_fs2.default.existsSync(edgePath)) return { path: edgePath, name: "Edge" };
     return null;
   } else if (platform === "win32") {
     const chromePaths = [
@@ -103,7 +275,7 @@ function getDefaultBrowserPath() {
       process.env.PROGRAMFILES + "\\Google\\Chrome\\Application\\chrome.exe",
       process.env["PROGRAMFILES(X86)"] + "\\Google\\Chrome\\Application\\chrome.exe"
     ];
-    const foundChrome = chromePaths.find((p) => import_fs.default.existsSync(p));
+    const foundChrome = chromePaths.find((p) => import_fs2.default.existsSync(p));
     if (foundChrome) return { path: foundChrome, name: "Chrome" };
     const edgePaths = [
       process.env.LOCALAPPDATA + "\\Microsoft\\Edge\\Application\\msedge.exe",
@@ -112,7 +284,7 @@ function getDefaultBrowserPath() {
       "C:\\Program Files (x86)\\Microsoft\\Edge\\Application\\msedge.exe",
       "C:\\Program Files\\Microsoft\\Edge\\Application\\msedge.exe"
     ];
-    const foundEdge = edgePaths.find((p) => import_fs.default.existsSync(p));
+    const foundEdge = edgePaths.find((p) => import_fs2.default.existsSync(p));
     if (foundEdge) return { path: foundEdge, name: "Edge" };
     return null;
   } else {
@@ -124,17 +296,17 @@ function getDefaultBrowserPath() {
       { p: "/usr/bin/microsoft-edge", name: "Edge" },
       { p: "/usr/bin/microsoft-edge-stable", name: "Edge" }
     ];
-    const found = paths.find(({ p }) => import_fs.default.existsSync(p));
+    const found = paths.find(({ p }) => import_fs2.default.existsSync(p));
     return found ? { path: found.p, name: found.name } : null;
   }
 }
 async function startBrowserInBackground() {
   const browserInfo = getDefaultBrowserPath();
-  if (!browserInfo || !import_fs.default.existsSync(browserInfo.path)) {
+  if (!browserInfo || !import_fs2.default.existsSync(browserInfo.path)) {
     throw new Error("\u65E0\u6CD5\u5728\u7CFB\u7EDF\u4E2D\u627E\u5230 Chrome \u6216 Edge \u6D4F\u89C8\u5668\u7684\u9ED8\u8BA4\u5B89\u88C5\u8DEF\u5F84\u3002");
   }
   console.log(import_picocolors.default.yellow(`\u6B63\u5728\u542F\u52A8\u540E\u53F0 ${browserInfo.name} \u5B9E\u4F8B (\u7AEF\u53E3: ${CDP_PORT})...`));
-  const userDataDir = process.env.WEBMCP_WORKSPACE || import_path.default.join(import_os.default.homedir(), ".webmcp_chrome_profile");
+  const userDataDir = process.env.WEBMCP_WORKSPACE || import_path2.default.join(import_os.default.homedir(), ".webmcp_chrome_profile");
   const child = (0, import_child_process.spawn)(
     browserInfo.path,
     [
@@ -162,7 +334,7 @@ async function startBrowserInBackground() {
       }
     } catch (e) {
     }
-    await new Promise((resolve2) => setTimeout(resolve2, 500));
+    await new Promise((resolve3) => setTimeout(resolve3, 500));
   }
   throw new Error(`${browserInfo.name} \u542F\u52A8\u8D85\u65F6\uFF0C\u65E0\u6CD5\u8FDE\u63A5\u5230 CDP \u7AEF\u53E3\u3002`);
 }
@@ -345,18 +517,18 @@ async function injectWebMCPPolyfillAndTools(page2, force = false) {
   }).catch(() => false);
   if (!polyfillReady) {
     console.log(import_picocolors.default.cyan("\u5F53\u524D\u9875\u9762\u5C1A\u672A\u6CE8\u5165 WebMCP \u73AF\u5883\uFF0C\u6B63\u5728\u6267\u884C\u81EA\u52A8\u6CE8\u5165..."));
-    const injectScriptPath = import_path.default.resolve(__dirname, "inject-bundle.js");
-    if (!import_fs.default.existsSync(injectScriptPath)) {
+    const injectScriptPath = import_path2.default.resolve(__dirname, "inject-bundle.js");
+    if (!import_fs2.default.existsSync(injectScriptPath)) {
       throw new Error(`Cannot find inject-bundle.js at ${injectScriptPath}. Please ensure you run 'pnpm build:inject' first.`);
     }
-    const scriptContent = import_fs.default.readFileSync(injectScriptPath, "utf-8");
+    const scriptContent = import_fs2.default.readFileSync(injectScriptPath, "utf-8");
     try {
       await page2.evaluate(scriptContent);
     } catch (err) {
       const msg = err instanceof Error ? err.message : String(err);
       throw new Error("\u81EA\u52A8\u6CE8\u5165\u811A\u672C\u6267\u884C\u5931\u8D25: " + msg);
     }
-    await new Promise((resolve2) => setTimeout(resolve2, 300));
+    await new Promise((resolve3) => setTimeout(resolve3, 300));
   }
   await injectDomainTools(page2);
 }
@@ -368,12 +540,12 @@ async function injectDomainTools(page2) {
   } catch {
     return;
   }
-  const toolBundlePath = import_path.default.resolve(__dirname, "webmcp-tools", `${hostname}.js`);
-  if (!import_fs.default.existsSync(toolBundlePath)) {
+  const toolBundlePath = import_path2.default.resolve(__dirname, "webmcp-tools", `${hostname}.js`);
+  if (!import_fs2.default.existsSync(toolBundlePath)) {
     return;
   }
   console.log(import_picocolors.default.cyan(`\u68C0\u6D4B\u5230\u57DF\u540D ${hostname} \u6709\u9884\u7F6E\u5DE5\u5177\uFF0C\u6B63\u5728\u6CE8\u5165...`));
-  const toolScript = import_fs.default.readFileSync(toolBundlePath, "utf-8");
+  const toolScript = import_fs2.default.readFileSync(toolBundlePath, "utf-8");
   try {
     await page2.evaluate(toolScript);
     console.log(import_picocolors.default.green(`\u5DF2\u4E3A ${hostname} \u6CE8\u5165\u9884\u7F6E\u5DE5\u5177`));
@@ -444,7 +616,7 @@ async function stateCommand({ tabid: tabid2 }) {
         if (pUrl.startsWith("devtools://")) return null;
         const pTitle = await Promise.race([
           p.title().catch(() => "Unknown"),
-          new Promise((resolve2) => setTimeout(() => resolve2("Unknown"), 500))
+          new Promise((resolve3) => setTimeout(() => resolve3("Unknown"), 500))
         ]);
         return {
           // 使用真实的 Chrome target ID，而非数组下标
@@ -464,7 +636,6 @@ async function stateCommand({ tabid: tabid2 }) {
 }
 
 // src/commands/run.ts
-var import_json5 = __toESM(require("json5"), 1);
 async function runCommand({
   toolName,
   argsJson,
@@ -473,16 +644,7 @@ async function runCommand({
   const browser2 = await connectBrowser();
   try {
     const page2 = await getTargetPage(browser2, tabid2);
-    let cleanedArgs = argsJson.trim();
-    if (cleanedArgs.startsWith("'") && cleanedArgs.endsWith("'")) {
-      cleanedArgs = cleanedArgs.slice(1, -1).trim();
-    }
-    try {
-      const obj = import_json5.default.parse(cleanedArgs);
-      cleanedArgs = JSON.stringify(obj);
-    } catch (e) {
-      throw new Error(`\u53C2\u6570\u4E0D\u662F\u6709\u6548\u7684 JSON \u6216 JS \u5BF9\u8C61: ${e.message}`);
-    }
+    const cleanedArgs = argsJson.trim();
     const result2 = await page2.evaluate(async (name, inputString) => {
       const mcp = navigator.modelContextTesting || navigator.modelContext;
       if (!mcp || typeof mcp.executeTool !== "function") {
@@ -492,9 +654,17 @@ async function runCommand({
       if (typeof res === "string") {
         try {
           res = JSON.parse(res);
-        } catch (e) {
+        } catch {
         }
       }
+      if (res === void 0 || res === null) {
+        throw new Error("\u5DE5\u5177 execute \u672A\u8FD4\u56DE\u7ED3\u679C\uFF0C\u8BF7\u68C0\u67E5\u5DE5\u5177\u5B9E\u73B0\u662F\u5426\u7F3A\u5C11 return");
+      }
+      if (typeof res === "object" && res.success === false) {
+        const failed = res;
+        throw new Error(failed.error || failed.message || "\u5DE5\u5177\u6267\u884C\u5931\u8D25");
+      }
+      return res;
     }, toolName, cleanedArgs);
     return result2;
   } finally {
@@ -555,7 +725,7 @@ async function tabsOpenCommand(url) {
     url = normalizeUrl(url);
     await page2.goto(url, { waitUntil: "domcontentloaded" });
     await injectIntoPage(page2);
-    await new Promise((resolve2) => setTimeout(resolve2, 500));
+    await new Promise((resolve3) => setTimeout(resolve3, 500));
     return {
       success: true,
       tabid: await getPageTargetId(page2),
@@ -656,7 +826,7 @@ async function tabsForwardCommand(tabid2) {
 // package.json
 var package_default = {
   name: "@opentiny/webmcp-cli",
-  version: "0.0.1-alpha.5",
+  version: "0.0.1-alpha.6",
   type: "module",
   description: "WebMCP CLI for AI Agents to interact with browser via Puppeteer",
   main: "dist/index.js",
@@ -710,13 +880,6 @@ var package_default = {
 
 // src/bin.ts
 var program = new import_commander.Command();
-function cleanJsonString(str) {
-  let cleaned = str.trim();
-  if (cleaned.startsWith("'") && cleaned.endsWith("'")) {
-    cleaned = cleaned.slice(1, -1).trim();
-  }
-  return cleaned;
-}
 function parseTabId(id) {
   if (!id) return void 0;
   return id;
@@ -726,19 +889,6 @@ function handleCommandError(error, commandName) {
   console.error(import_picocolors2.default.red(`Error executing ${commandName} command: ${msg}`));
   process.exit(1);
 }
-function expandFileRefs(argsJson) {
-  return argsJson.replace(/"@(base64)?file:([^"]+)"/g, (_match, base64Flag, filePath) => {
-    const absPath = (0, import_path2.resolve)(process.cwd(), filePath);
-    let content;
-    try {
-      content = (0, import_fs2.readFileSync)(absPath, "utf-8");
-    } catch (e) {
-      throw new Error(`\u65E0\u6CD5\u8BFB\u53D6\u6587\u4EF6\u5F15\u7528 "${filePath}": ${e.message}`);
-    }
-    const value = base64Flag ? Buffer.from(content, "utf-8").toString("base64") : content;
-    return JSON.stringify(value);
-  });
-}
 program.name("webmcp-cli").description("WebMCP CLI for interacting with browser via CDP").version(package_default.version).option("-w, --workspace <path>", "\u6307\u5B9A\u81EA\u5B9A\u4E49\u7684\u6D4F\u89C8\u5668\u5DE5\u4F5C\u7A7A\u95F4\uFF08\u7528\u6237\u914D\u7F6E\u76EE\u5F55\uFF09\u8DEF\u5F84").hook("preAction", (thisCommand) => {
   const opts = thisCommand.opts();
   if (opts.workspace) {
@@ -762,42 +912,18 @@ program.command("run <toolName> [args...]").description("\u5411\u6307\u5B9A\u987
 \u793A\u4F8B\uFF1Awebmcp-cli run mytool '{"content":"@base64file:./doc.md"}'`
 ).action(async (toolName, args, options) => {
   try {
-    let finalArgsJson = "";
+    let fileContent;
+    let fileBaseDir = process.cwd();
     if (options.file) {
-      const filePath = (0, import_path2.resolve)(process.cwd(), options.file);
+      const filePath = (0, import_path3.resolve)(process.cwd(), options.file);
+      fileBaseDir = getFileBaseDir(options.file);
       try {
-        finalArgsJson = (0, import_fs2.readFileSync)(filePath, "utf-8").trim();
+        fileContent = (0, import_fs3.readFileSync)(filePath, "utf-8");
       } catch (e) {
         throw new Error(`\u65E0\u6CD5\u8BFB\u53D6\u6587\u4EF6 "${filePath}": ${e.message}`);
       }
-    } else if (args && args.length > 0) {
-      const firstArg = args[0].trim();
-      const cleanedFirstArg = cleanJsonString(firstArg);
-      if (cleanedFirstArg.startsWith("{") && cleanedFirstArg.endsWith("}")) {
-        finalArgsJson = args.map(cleanJsonString).join(" ");
-      } else {
-        const obj = {};
-        for (const pair of args) {
-          const index = pair.indexOf("=");
-          if (index === -1) {
-            throw new Error(`\u65E0\u6548\u7684\u53C2\u6570\u683C\u5F0F "${pair}"\u3002\u5FC5\u987B\u662F JSON \u5B57\u7B26\u4E32\u6216 key=value \u683C\u5F0F\u3002`);
-          }
-          const key = pair.substring(0, index).trim();
-          let val = pair.substring(index + 1).trim();
-          val = cleanJsonString(val);
-          try {
-            obj[key] = JSON.parse(val);
-          } catch {
-            obj[key] = val;
-          }
-        }
-        finalArgsJson = JSON.stringify(obj);
-      }
-    }
-    if (!finalArgsJson) {
-      throw new Error("\u5FC5\u987B\u63D0\u4F9B\u53C2\u6570\u6216\u901A\u8FC7 -f/--file \u6307\u5B9A\u53C2\u6570\u6587\u4EF6");
     }
-    finalArgsJson = expandFileRefs(finalArgsJson);
+    const finalArgsJson = prepareRunArgsJson(args ?? [], fileContent, fileBaseDir);
     const result2 = await runCommand({
       toolName,
       argsJson: finalArgsJson,