@opentiny/webmcp-cli 0.0.4 → 0.0.6

package/dist/bin.js

@@ -3,7 +3,7 @@
 // src/bin.ts
 import fs2 from "fs";
 import path2 from "path";
-import os2 from "os";
+import os3 from "os";
 import { Command } from "commander";
 import pc2 from "picocolors";
 import { readFileSync as readFileSync2 } from "fs";
@@ -625,10 +625,10 @@ async function stateCommand({ tabid }) {
     const state = await page.evaluate(async () => {
       const url = document.URL;
       const title = document.title;
-      const mcp = navigator.modelContextTesting || navigator.modelContext;
+      const mcp = document.modelContext;
       let webmcpTools = [];
       if (mcp && typeof mcp.listTools === "function") {
-        const toolsResult = await mcp.listTools();
+        const toolsResult = await mcp.getTools();
         webmcpTools = toolsResult?.tools || toolsResult || [];
       }
       return { url, title, webmcpTools };
@@ -661,6 +661,155 @@ async function stateCommand({ tabid }) {
   }
 }
 
+// src/zhihu/create-article.ts
+import os2 from "os";
+
+// src/zhihu/markdown.ts
+import { Marked } from "marked";
+function stripLeadingH1(markdown) {
+  return markdown.replace(/^#\s+[^\n]*\n+/, "");
+}
+function markdownToHtml(markdown) {
+  const marked = new Marked({
+    gfm: true,
+    breaks: false,
+    renderer: {
+      code({ text, lang }) {
+        const langClass = lang ? ` class="language-${lang}"` : "";
+        const escaped = text.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+        return `<pre><code${langClass}>${escaped}</code></pre>`;
+      },
+      codespan({ text }) {
+        const escaped = text.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+        return `<code>${escaped}</code>`;
+      }
+    }
+  });
+  return marked.parse(markdown);
+}
+function decodeBase64Content(content) {
+  return Buffer.from(content, "base64").toString("utf-8");
+}
+var ZHIHU_WRITE_PAGE_RE = /^https:\/\/zhuanlan\.zhihu\.com\/(write|p\/\d+\/edit)/;
+function isZhihuWriteUrl(url) {
+  return ZHIHU_WRITE_PAGE_RE.test(url);
+}
+
+// src/zhihu/create-article.ts
+function delay(ms) {
+  return new Promise((resolve3) => setTimeout(resolve3, ms));
+}
+function getPasteModifier() {
+  return os2.platform() === "darwin" ? "Meta" : "Control";
+}
+async function fillTitle(page, title) {
+  await page.evaluate((titleText) => {
+    const input = document.querySelector('textarea[placeholder*="\u6807\u9898"]') || document.querySelector(".WriteIndex-titleInput textarea") || document.querySelector(".WriteIndex-titleInput input");
+    if (!input) throw new Error("\u672A\u627E\u5230\u6807\u9898\u8F93\u5165\u6846\uFF0C\u8BF7\u786E\u8BA4\u7F16\u8F91\u5668\u9875\u9762\u5DF2\u5B8C\u5168\u52A0\u8F7D");
+    const setter = Object.getOwnPropertyDescriptor(
+      input instanceof HTMLTextAreaElement ? HTMLTextAreaElement.prototype : HTMLInputElement.prototype,
+      "value"
+    )?.set;
+    if (setter) {
+      setter.call(input, titleText);
+    } else {
+      input.value = titleText;
+    }
+    input.dispatchEvent(new InputEvent("input", { bubbles: true, cancelable: true, data: titleText }));
+    input.dispatchEvent(new Event("change", { bubbles: true }));
+    if (input.value.trim() !== titleText.trim()) {
+      throw new Error("\u6807\u9898\u586B\u5199\u5931\u8D25\uFF0C\u8BF7\u5237\u65B0\u9875\u9762\u540E\u91CD\u8BD5");
+    }
+  }, title);
+}
+async function pasteHtmlToEditor(page, html) {
+  const browser = page.browser();
+  const tempPage = await browser.newPage();
+  const modifier = getPasteModifier();
+  try {
+    const fullHtml = `<!DOCTYPE html><html><head><meta charset="utf-8"></head><body>${html}</body></html>`;
+    await tempPage.setContent(fullHtml, { waitUntil: "domcontentloaded" });
+    await delay(300);
+    await tempPage.click("body");
+    await tempPage.keyboard.down(modifier);
+    await tempPage.keyboard.press("KeyA");
+    await tempPage.keyboard.up(modifier);
+    await delay(100);
+    await tempPage.keyboard.down(modifier);
+    await tempPage.keyboard.press("KeyC");
+    await tempPage.keyboard.up(modifier);
+    await delay(300);
+  } finally {
+    await tempPage.close();
+  }
+  await page.bringToFront();
+  await page.waitForSelector(".public-DraftEditor-content", { timeout: 1e4 });
+  await page.click(".public-DraftEditor-content");
+  await delay(200);
+  await page.keyboard.down(modifier);
+  await page.keyboard.press("KeyA");
+  await page.keyboard.up(modifier);
+  await delay(100);
+  await page.keyboard.press("Backspace");
+  await delay(200);
+  await page.keyboard.down(modifier);
+  await page.keyboard.press("KeyV");
+  await page.keyboard.up(modifier);
+  await delay(1500);
+  const probe = html.replace(/<[^>]+>/g, "").trim().slice(0, 15);
+  const verifyResult = await page.evaluate((probeText) => {
+    const editor = document.querySelector(".public-DraftEditor-content");
+    if (!editor) return { ok: false, reason: "no editor" };
+    const text = (editor.innerText || "").trim();
+    if (!text) return { ok: false, reason: "empty" };
+    const inner = editor.innerHTML;
+    const hasRichFormat = inner.includes("data-text") || /font-weight:\s*bold|font-style:\s*italic|code-block|blockquote/i.test(inner);
+    const normalizedText = text.replace(/\s+/g, "");
+    const normalizedProbe = probeText.replace(/\s+/g, "");
+    const hasProbe = normalizedText.includes(normalizedProbe);
+    return {
+      ok: text.length > 20 && hasProbe && (hasRichFormat || text.length > 50),
+      reason: !hasProbe ? "probe mismatch" : !hasRichFormat && text.length <= 50 ? "no rich format" : void 0
+    };
+  }, probe);
+  if (!verifyResult.ok) {
+    throw new Error(
+      `\u6B63\u6587\u586B\u5199\u5931\u8D25\uFF08Markdown \u8F6C HTML \u7C98\u8D34\u672A\u751F\u6548\uFF09${verifyResult.reason ? `: ${verifyResult.reason}` : ""}`
+    );
+  }
+}
+async function zhihuCreateArticle(page, args) {
+  const { title, content } = args;
+  if (!title?.trim()) throw new Error("\u53C2\u6570 title \u4E0D\u80FD\u4E3A\u7A7A");
+  if (!content?.trim()) throw new Error("\u53C2\u6570 content \u4E0D\u80FD\u4E3A\u7A7A");
+  const url = page.url();
+  if (!isZhihuWriteUrl(url)) {
+    throw new Error("\u5F53\u524D\u9875\u9762\u4E0D\u662F\u77E5\u4E4E\u4E13\u680F\u7F16\u8F91\u5668\uFF0C\u8BF7\u5148\u6253\u5F00 https://zhuanlan.zhihu.com/write");
+  }
+  let markdown;
+  try {
+    markdown = stripLeadingH1(decodeBase64Content(content));
+  } catch {
+    throw new Error("content \u4E0D\u662F\u6709\u6548\u7684 Base64 \u7F16\u7801\uFF0C\u8BF7\u68C0\u67E5\u53C2\u6570\u6216\u4F7F\u7528 @base64file: \u5F15\u7528\u6587\u4EF6");
+  }
+  if (!markdown.trim()) {
+    throw new Error("\u6B63\u6587\u5185\u5BB9\u4E3A\u7A7A\uFF0C\u8BF7\u68C0\u67E5 Markdown \u6587\u4EF6");
+  }
+  const html = markdownToHtml(markdown);
+  await fillTitle(page, title.trim());
+  await delay(500);
+  await pasteHtmlToEditor(page, html);
+  const result = {
+    success: true,
+    message: "\u6587\u7AE0\u6807\u9898\u548C\u6B63\u6587\u5DF2\u6210\u529F\u586B\u5199\u5230\u77E5\u4E4E\u4E13\u680F\u7F16\u8F91\u5668\uFF08Markdown \u5DF2\u8F6C\u4E3A\u5BCC\u6587\u672C\uFF09\uFF0C\u8349\u7A3F\u5C06\u81EA\u52A8\u4FDD\u5B58",
+    title: title.trim(),
+    contentLength: markdown.length
+  };
+  return {
+    content: [{ type: "text", text: JSON.stringify(result) }]
+  };
+}
+
 // src/commands/run.ts
 async function runCommand({
   toolName,
@@ -672,14 +821,28 @@ async function runCommand({
     const page = await getTargetPage(browser, tabid);
     const urlBefore = page.url();
     const cleanedArgs = argsJson.trim();
+    if (toolName === "create_article") {
+      let parsedArgs;
+      try {
+        parsedArgs = JSON.parse(cleanedArgs);
+      } catch {
+        throw new Error("\u53C2\u6570\u4E0D\u662F\u6709\u6548\u7684 JSON");
+      }
+      if (isZhihuWriteUrl(page.url())) {
+        return await zhihuCreateArticle(page, parsedArgs);
+      }
+    }
     let result;
     try {
       result = await page.evaluate(async (name, inputString) => {
-        const mcp = navigator.modelContextTesting || navigator.modelContext;
+        const mcp = document.modelContext || document.modelContext || document.modelContext;
         if (!mcp || typeof mcp.executeTool !== "function") {
-          throw new Error("\u5F53\u524D\u9875\u9762\u6CA1\u6709\u6CE8\u5165 WebMCP \u73AF\u5883 (navigator.modelContextTesting.executeTool \u672A\u627E\u5230)");
+          throw new Error("\u5F53\u524D\u9875\u9762\u6CA1\u6709\u6CE8\u5165 WebMCP \u73AF\u5883 (document.modelContext \u672A\u627E\u5230)");
         }
-        let res = await mcp.executeTool(name, inputString);
+        const tools = await mcp.getTools();
+        const toolObj = tools.find((t) => t.name === name);
+        if (!toolObj) throw new Error(`Tool ${name} not found`);
+        let res = await mcp.executeTool(toolObj, inputString);
         if (typeof res === "string") {
           try {
             res = JSON.parse(res);
@@ -856,7 +1019,7 @@ async function tabsForwardCommand(tabid) {
 // package.json
 var package_default = {
   name: "@opentiny/webmcp-cli",
-  version: "0.0.4",
+  version: "0.0.6",
   type: "module",
   description: "WebMCP CLI for AI Agents to interact with browser via Puppeteer",
   main: "dist/index.js",
@@ -896,6 +1059,7 @@ var package_default = {
     clipboardy: "^5.3.1",
     commander: "^11.1.0",
     json5: "^2.2.3",
+    marked: "^15.0.7",
     picocolors: "^1.0.0",
     "puppeteer-core": "^22.0.0"
   },
@@ -956,9 +1120,66 @@ function cleanOldLogs(baseDir, logDir) {
   } catch (e) {
   }
 }
+function formatLogResult(result) {
+  if (result === void 0 || result === null) {
+    return "null";
+  }
+  try {
+    if (typeof result !== "object") {
+      return String(result);
+    }
+    const clone = JSON.parse(JSON.stringify(result));
+    let a11yTreeStr = "";
+    let searchResultStr = "";
+    if (clone && Array.isArray(clone.content) && clone.content.length > 0 && clone.content[0] && typeof clone.content[0].text === "string") {
+      const text = clone.content[0].text;
+      if (text.startsWith("\u6D4F\u89C8\u5668\u72B6\u6001: ")) {
+        let jsonStr = text.substring("\u6D4F\u89C8\u5668\u72B6\u6001: ".length);
+        let extraAlerts = "";
+        const firstNewline = jsonStr.indexOf("\n");
+        if (firstNewline !== -1) {
+          extraAlerts = jsonStr.substring(firstNewline);
+          jsonStr = jsonStr.substring(0, firstNewline);
+        }
+        try {
+          const parsed = JSON.parse(jsonStr);
+          if (parsed && typeof parsed.content === "string") {
+            a11yTreeStr = parsed.content;
+            parsed.content = "[Formatted A11y Tree - See details below]";
+            clone.content[0].text = `\u6D4F\u89C8\u5668\u72B6\u6001: ${JSON.stringify(parsed, null, 2)}${extraAlerts}`;
+          }
+        } catch {
+        }
+      } else if (text.includes("A11y Tree \u641C\u7D22\u7ED3\u679C") || text.includes("\u65E0\u969C\u788D\u6811\u641C\u7D22\u7ED3\u679C") || text.includes("\u65E0\u969C\u788D\u6811\u641C\u7D22") || text.includes("\u547D\u4E2D\u884C") || text.includes("context:")) {
+        searchResultStr = text;
+        clone.content[0].text = "[Search A11y Tree Result - See details below]";
+      }
+    }
+    let out = JSON.stringify(clone, null, 2);
+    if (a11yTreeStr) {
+      out += `
+
+[FORMATTED A11Y TREE]:
+${a11yTreeStr}`;
+    }
+    if (searchResultStr) {
+      out += `
+
+[FORMATTED SEARCH RESULT]:
+${searchResultStr}`;
+    }
+    return out;
+  } catch (e) {
+    try {
+      return JSON.stringify(result, null, 2);
+    } catch {
+      return String(result);
+    }
+  }
+}
 function writeLog(commandName, args, result, error) {
   try {
-    const baseDir = process.env.WEBMCP_WORKSPACE || path2.join(os2.homedir(), ".webmcp_chrome_profile");
+    const baseDir = process.env.WEBMCP_WORKSPACE || path2.join(os3.homedir(), ".webmcp_chrome_profile");
     const logDir = path2.join(baseDir, "logs");
     if (!fs2.existsSync(logDir)) {
       fs2.mkdirSync(logDir, { recursive: true });
@@ -981,15 +1202,8 @@ ${error instanceof Error ? error.stack : String(error)}
     } else if (result) {
       logText += `RESULT:
 `;
-      try {
-        if (result && typeof result === "object") {
-          logText += JSON.stringify(result, null, 2).split("\n").map((line) => `  ${line}`).join("\n") + "\n";
-        } else {
-          logText += String(result).split("\n").map((line) => `  ${line}`).join("\n") + "\n";
-        }
-      } catch {
-        logText += JSON.stringify(result, null, 2).split("\n").map((line) => `  ${line}`).join("\n") + "\n";
-      }
+      const formattedResult = formatLogResult(result);
+      logText += formattedResult.split("\n").map((line) => `  ${line}`).join("\n") + "\n";
     } else {
       logText += `RESULT: null
 `;
@@ -1019,7 +1233,7 @@ program.command("state").description("\u83B7\u53D6\u6D4F\u89C8\u5668\u5F53\u524D
   try {
     const result = await stateCommand(args);
     writeLog("state", args, result);
-    console.log(JSON.stringify(result, null, 2));
+    console.log(formatLogResult(result));
   } catch (error) {
     handleCommandError(error, "state", args);
   }
@@ -1053,7 +1267,7 @@ program.command("run <toolName> [args...]").description("\u5411\u6307\u5B9A\u987
       tabid: parseTabId(options.tabid)
     });
     writeLog("run", { ...rawArgs, finalArgsJson }, result);
-    console.log(JSON.stringify(result, null, 2));
+    console.log(formatLogResult(result));
   } catch (error) {
     handleCommandError(error, "run", { ...rawArgs, finalArgsJson });
   }