z-mcp-codesign 0.1.1 → 0.1.3

package/README.md

@@ -1,132 +1,70 @@
 # 前言
 
-
-
 这是一个基于 `@modelcontextprotocol/sdk` 实现的 **MCP stdio 服务**，用于在本机通过 Edge 打开腾讯 CoDesign 页面并抓取当前画板图片。
 
-
-
 # 配置说明
 
-
-
 在 Cursor 工作区根目录创建或编辑 `.cursor/mcp.json`。保存后在 Cursor 的 MCP 相关设置中启用该服务，必要时重新加载窗口或刷新 MCP 列表。
 
-
-
 **前提**：本机需已安装 **Microsoft Edge**；服务通过 Playwright 使用系统 Edge（`channel: "msedge"`）。
 
-
-
-## 使用已发布的 npm 包（推荐）
-
-
-
-在任意项目中均可使用，无需克隆本仓库：
-
-
+在任意项目中均可使用：
 
 ```json
-
 {
-
   "mcpServers": {
-
     "z-mcp-codesign": {
-
       "command": "npx",
-
       "args": ["-y", "z-mcp-codesign", "--stdio"]
-
-    }
-
-  }
-
-}
-
-```
-
-
-
-## 使用本地源码（开发与调试）
-
-
-
-在克隆后的仓库根目录配置 `.cursor/mcp.json`，并先执行 `npm install` 与 `npm run build`（生成 `dist/`）。`args` 中的 `"."` 表示使用当前目录作为 npm 包：
-
-
-
-```json
-
-{
-
-  "mcpServers": {
-
-    "z-mcp-codesign": {
-
-      "command": "npx",
-
-      "args": ["-y", ".", "--stdio"]
-
     }
-
   }
-
 }
-
 ```
 
 
 
-若 MCP 启动时工作目录不是仓库根目录，可在该条目中增加 `"cwd"`，值为本仓库的绝对路径（Windows 下 JSON 字符串里反斜杠需写成 `\\`）。
-
-
-
-## 环境变量与用户数据目录
-
-
-
-调用 **`codesign_open_login`** 成功后，进程会设置环境变量 **`Z_MCP_CODESIGN_EDGE_USER_DATA`**，指向本次使用的 Edge 用户数据根目录；随后在同一 MCP 会话里调用 **`codesign_screen_inspect`** 且不显式传 `userDataDir` 时，会自动复用该目录下的登录态。
-
-
-
-也可在两条工具调用里显式传入相同的 **`userDataDir`**（绝对路径）；若使用非 `Default` 的 Edge profile，请在登录与抓图两次调用中传入相同的 **`profileDirectory`**。
-
-
-
-# 使用 MCP 登录 CoDesign
-
-
-
-1. 按上文在 Cursor 中启用本 MCP 服务。
-
-2. 在对话中让助手调用工具 **`codesign_open_login`**（参数可全部省略；默认打开 `https://codesign.qq.com/app/login`）。
-
-3. 在弹出的 **Edge** 窗口中完成腾讯 CoDesign 登录（扫码或账号密码等）。
-
-4. **关闭该 Edge 窗口**（释放用户数据目录锁），否则后续抓图可能启动失败。
-
-5. 再调用 **`codesign_screen_inspect`**，传入目标 CoDesign 页面 **`url`**（例如设计稿 inspect 链接）；返回的 JSON 中含 **`imageBase64`**（PNG），供客户端展示或落盘。
-
-
-
 # 提供的工具
 
+| 工具名 | 说明 |
+| ------ | ---- |
+| `codesign_open_login` | 有界面打开 CoDesign 登录页，用于在本机 Edge 用户数据目录中完成登录 |
+| `codesign_screen_inspect_save` | 打开指定 CoDesign 页面，截取最大的可见 `.screen-inspect` 区域，保存为本地 PNG 文件并返回保存路径（默认保存到项目根目录 `.codesign_screen_imgs/`） |
+| `codesign_screen_viewer_download_save` | 打开 CoDesign 页面，点击 `.screen-viewer__action` 选择“下载图片”，捕获下载并保存到项目根目录 `.codesign_screen_imgs/` |
+| `codesign_screen_list` | 获取 CoDesign 设计页面左侧菜单中的所有页面名称和访问地址 |
 
+# 使用示例
 
-| 工具名 | 说明 |
+**打开登录页（用于后续抓图）**
 
-| ------ | ---- |
+```bash
+使用MCP 打开 CoDesign 登录页
+```
 
-| `codesign_open_login` | 有界面打开 CoDesign 登录页，用于在本机 Edge 用户数据目录中完成登录 |
+**列出当前项目的页面列表**
 
-| `codesign_screen_inspect` | 打开指定 CoDesign 页面，截取最大的可见 `.screen-inspect` 区域，以 PNG 的 base64 返回 |
+```bash
+使用MCP 获取CoDesign所有页面列表使用表格展示。
+URL为：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+```
 
+**抓图并保存到本地**
 
+```bash
+使用MCP 获取CoDesign页面图片，把截图保存到本地，并告诉我保存路径。
+URL为：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+```
 
-# 使用示例
+**通过页面“下载图片”保存到本地**
 
+```bash
+使用MCP 打开CoDesign页面，下载图片保存到本地，并告诉我保存路径。
+URL为：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+```
 
+**生成项目**
 
-在已登录（步骤见上一节）的前提下，调用 **`codesign_screen_inspect`**，传入 `url`（CoDesign 页面地址）。工具成功时返回的文本内容为 JSON，字段包括 `imageBase64`、`title`、`boundingBox` 等；将 `imageBase64` 解码即可得到 PNG 二进制数据。
+```bash
+使用MCP 打开CoDesign页面，下载图片保存到本地，并识别图片内容生成Vue3+Vite项目。
+URL为：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+```
 

package/dist/index.js

@@ -2980,7 +2980,7 @@ var require_compile = __commonJS({
       const schOrFunc = root.refs[ref];
       if (schOrFunc)
         return schOrFunc;
-      let _sch = resolve6.call(this, root, ref);
+      let _sch = resolve5.call(this, root, ref);
       if (_sch === void 0) {
         const schema = (_a = root.localRefs) === null || _a === void 0 ? void 0 : _a[ref];
         const { schemaId } = this.opts;
@@ -3007,7 +3007,7 @@ var require_compile = __commonJS({
     function sameSchemaEnv(s1, s2) {
       return s1.schema === s2.schema && s1.root === s2.root && s1.baseId === s2.baseId;
     }
-    function resolve6(root, ref) {
+    function resolve5(root, ref) {
       let sch;
       while (typeof (sch = this.refs[ref]) == "string")
         ref = sch;
@@ -3222,8 +3222,8 @@ var require_utils = __commonJS({
       }
       return ind;
     }
-    function removeDotSegments(path) {
-      let input = path;
+    function removeDotSegments(path3) {
+      let input = path3;
       const output = [];
       let nextSlash = -1;
       let len = 0;
@@ -3422,8 +3422,8 @@ var require_schemes = __commonJS({
         wsComponent.secure = void 0;
       }
       if (wsComponent.resourceName) {
-        const [path, query] = wsComponent.resourceName.split("?");
-        wsComponent.path = path && path !== "/" ? path : void 0;
+        const [path3, query] = wsComponent.resourceName.split("?");
+        wsComponent.path = path3 && path3 !== "/" ? path3 : void 0;
         wsComponent.query = query;
         wsComponent.resourceName = void 0;
       }
@@ -3582,7 +3582,7 @@ var require_fast_uri = __commonJS({
       }
       return uri;
     }
-    function resolve6(baseURI, relativeURI, options) {
+    function resolve5(baseURI, relativeURI, options) {
       const schemelessOptions = options ? Object.assign({ scheme: "null" }, options) : { scheme: "null" };
       const resolved = resolveComponent(parse3(baseURI, schemelessOptions), parse3(relativeURI, schemelessOptions), schemelessOptions, true);
       schemelessOptions.skipEscape = true;
@@ -3809,7 +3809,7 @@ var require_fast_uri = __commonJS({
     var fastUri = {
       SCHEMES,
       normalize,
-      resolve: resolve6,
+      resolve: resolve5,
       resolveComponent,
       equal,
       serialize,
@@ -6800,7 +6800,7 @@ var require_dist = __commonJS({
 
 // src/index.ts
 import { createRequire } from "node:module";
-import { dirname as dirname2, join as join4 } from "node:path";
+import { dirname, join as join4 } from "node:path";
 import { fileURLToPath } from "node:url";
 
 // node_modules/zod/v3/external.js
@@ -7281,8 +7281,8 @@ function getErrorMap() {
 
 // node_modules/zod/v3/helpers/parseUtil.js
 var makeIssue = (params) => {
-  const { data, path, errorMaps, issueData } = params;
-  const fullPath = [...path, ...issueData.path || []];
+  const { data, path: path3, errorMaps, issueData } = params;
+  const fullPath = [...path3, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -7398,11 +7398,11 @@ var errorUtil;
 
 // node_modules/zod/v3/types.js
 var ParseInputLazyPath = class {
-  constructor(parent, value, path, key) {
+  constructor(parent, value, path3, key) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path;
+    this._path = path3;
     this._key = key;
   }
   get path() {
@@ -11039,10 +11039,10 @@ function assignProp(target, prop, value) {
     configurable: true
   });
 }
-function getElementAtPath(obj, path) {
-  if (!path)
+function getElementAtPath(obj, path3) {
+  if (!path3)
     return obj;
-  return path.reduce((acc, key) => acc?.[key], obj);
+  return path3.reduce((acc, key) => acc?.[key], obj);
 }
 function promiseAllObject(promisesObj) {
   const keys = Object.keys(promisesObj);
@@ -11362,11 +11362,11 @@ function aborted(x, startIndex = 0) {
   }
   return false;
 }
-function prefixIssues(path, issues) {
+function prefixIssues(path3, issues) {
   return issues.map((iss) => {
     var _a;
     (_a = iss).path ?? (_a.path = []);
-    iss.path.unshift(path);
+    iss.path.unshift(path3);
     return iss;
   });
 }
@@ -18875,7 +18875,7 @@ var Protocol = class {
           return;
         }
         const pollInterval = task2.pollInterval ?? this._options?.defaultTaskPollInterval ?? 1e3;
-        await new Promise((resolve6) => setTimeout(resolve6, pollInterval));
+        await new Promise((resolve5) => setTimeout(resolve5, pollInterval));
         options?.signal?.throwIfAborted();
       }
     } catch (error2) {
@@ -18892,7 +18892,7 @@ var Protocol = class {
    */
   request(request, resultSchema, options) {
     const { relatedRequestId, resumptionToken, onresumptiontoken, task, relatedTask } = options ?? {};
-    return new Promise((resolve6, reject) => {
+    return new Promise((resolve5, reject) => {
       const earlyReject = (error2) => {
         reject(error2);
       };
@@ -18970,7 +18970,7 @@ var Protocol = class {
           if (!parseResult.success) {
             reject(parseResult.error);
           } else {
-            resolve6(parseResult.data);
+            resolve5(parseResult.data);
           }
         } catch (error2) {
           reject(error2);
@@ -19231,12 +19231,12 @@ var Protocol = class {
       }
     } catch {
     }
-    return new Promise((resolve6, reject) => {
+    return new Promise((resolve5, reject) => {
       if (signal.aborted) {
         reject(new McpError(ErrorCode.InvalidRequest, "Request cancelled"));
         return;
       }
-      const timeoutId = setTimeout(resolve6, interval);
+      const timeoutId = setTimeout(resolve5, interval);
       signal.addEventListener("abort", () => {
         clearTimeout(timeoutId);
         reject(new McpError(ErrorCode.InvalidRequest, "Request cancelled"));
@@ -20336,7 +20336,7 @@ var McpServer = class {
     let task = createTaskResult.task;
     const pollInterval = task.pollInterval ?? 5e3;
     while (task.status !== "completed" && task.status !== "failed" && task.status !== "cancelled") {
-      await new Promise((resolve6) => setTimeout(resolve6, pollInterval));
+      await new Promise((resolve5) => setTimeout(resolve5, pollInterval));
       const updatedTask = await extra.taskStore.getTask(taskId);
       if (!updatedTask) {
         throw new McpError(ErrorCode.InternalError, `Task ${taskId} not found during polling`);
@@ -20979,12 +20979,12 @@ var StdioServerTransport = class {
     this.onclose?.();
   }
   send(message) {
-    return new Promise((resolve6) => {
+    return new Promise((resolve5) => {
       const json = serializeMessage(message);
       if (this._stdout.write(json)) {
-        resolve6();
+        resolve5();
       } else {
-        this._stdout.once("drain", resolve6);
+        this._stdout.once("drain", resolve5);
       }
     });
   }
@@ -21207,7 +21207,7 @@ var codesignLoginInput = external_exports.object({
   loginUrl: external_exports.string().url().optional().describe(`\u53EF\u9009\uFF1B\u9ED8\u8BA4 ${DEFAULT_CODESIGN_LOGIN_URL}`),
   userDataDir: external_exports.string().optional().describe("\u6D4F\u89C8\u5668\u7528\u6237\u6570\u636E\u6839\u76EE\u5F55\uFF1B\u672A\u4F20\u65F6\u540C\u6293\u56FE\u5DE5\u5177\uFF08\u73AF\u5883\u53D8\u91CF\u6216\u4E34\u65F6\u76EE\u5F55\uFF09\u3002"),
   profileDirectory: external_exports.string().optional().describe(
-    "User Data \u4E0B Edge profile \u76EE\u5F55\u540D\uFF08\u5982 Default\uFF09\uFF1B\u4E0E\u6293\u56FE\u5DE5\u5177 codesign_screen_inspect \u4F7F\u7528\u540C\u4E00\u76EE\u5F55\u65F6\u987B\u4F20\u76F8\u540C\u503C\u3002"
+    "User Data \u4E0B Edge profile \u76EE\u5F55\u540D\uFF08\u5982 Default\uFF09\uFF1B\u4E0E\u6293\u56FE\u5DE5\u5177\u4F7F\u7528\u540C\u4E00\u76EE\u5F55\u65F6\u987B\u4F20\u76F8\u540C\u503C\u3002"
   )
 });
 function registerOpenLoginTool(mcpServer2, deps2) {
@@ -21243,7 +21243,7 @@ function registerOpenLoginTool(mcpServer2, deps2) {
           {
             loginUrl,
             userDataDir: resolve4(persistentDir),
-            hint: "\u5728\u5F39\u51FA\u7684\u6D4F\u89C8\u5668\u4E2D\u5B8C\u6210\u767B\u5F55\u540E\u5173\u95ED\u7A97\u53E3\uFF0C\u518D\u8C03\u7528 codesign_screen_inspect\u3002"
+            hint: "\u5728\u5F39\u51FA\u7684\u6D4F\u89C8\u5668\u4E2D\u5B8C\u6210\u767B\u5F55\u540E\u5173\u95ED\u7A97\u53E3\uFF0C\u518D\u8C03\u7528\u6293\u56FE\u5DE5\u5177\uFF08\u5982 codesign_screen_inspect_save\uFF09\u3002"
           },
           null,
           2
@@ -21253,79 +21253,65 @@ function registerOpenLoginTool(mcpServer2, deps2) {
   );
 }
 
-// src/tools/save-image.ts
-import { resolve as resolve5, dirname, extname } from "node:path";
+// src/tools/screen-inspect-save.ts
 import { mkdir as mkdir2, writeFile } from "node:fs/promises";
-var saveImageInput = external_exports.object({
-  imageBase64: external_exports.string().min(1).describe(
-    "\u56FE\u7247 base64\uFF1B\u652F\u6301\u7EAF base64\uFF0C\u6216 data URL\uFF08\u5982 data:image/png;base64,AAAA...\uFF09\u3002"
-  ),
-  filePath: external_exports.string().min(1).describe("\u4FDD\u5B58\u8DEF\u5F84\uFF08\u76F8\u5BF9\u6216\u7EDD\u5BF9\uFF09\uFF0C\u5982 ./out/shot.png \u6216 D:/tmp/shot.png"),
-  mimeType: external_exports.string().optional().describe("\u53EF\u9009\uFF1B\u7528\u4E8E\u63A8\u65AD\u6269\u5C55\u540D\uFF08\u5F53 filePath \u65E0\u6269\u5C55\u540D\u65F6\uFF09\u3002\u5982 image/png")
-});
-function extFromMime(mimeType) {
-  const m = (mimeType || "").trim().toLowerCase();
-  if (!m) return null;
-  if (m === "image/png") return ".png";
-  if (m === "image/jpeg" || m === "image/jpg") return ".jpg";
-  if (m === "image/webp") return ".webp";
-  if (m === "image/gif") return ".gif";
-  return null;
-}
-function parseBase64(input) {
-  const s = input.trim();
-  const m = s.match(/^data:([^;]+);base64,(.*)$/i);
-  if (m) {
-    const mimeType = m[1]?.trim();
-    const base642 = (m[2] || "").trim();
-    if (!base642) return null;
-    return { mimeType, base64: base642 };
-  }
-  return { base64: s };
-}
-function registerSaveImageTool(mcpServer2) {
-  mcpServer2.registerTool(
-    "codesign_save_image",
-    {
-      description: "\u5C06 base64 \u56FE\u7247\u4FDD\u5B58\u4E3A\u672C\u5730\u6587\u4EF6\u8DEF\u5F84\u3002",
-      inputSchema: saveImageInput
-    },
-    async (input) => {
-      const parsed = parseBase64(input.imageBase64);
-      if (!parsed) return toolError("imageBase64 \u4E3A\u7A7A\u6216\u4E0D\u662F\u5408\u6CD5\u7684 base64/dataURL\u3002");
-      let filePath = input.filePath.trim();
-      if (!filePath) return toolError("filePath \u4E0D\u80FD\u4E3A\u7A7A\u3002");
-      const inferredExt = extFromMime(parsed.mimeType) || extFromMime(input.mimeType);
-      if (!extname(filePath) && inferredExt) {
-        filePath += inferredExt;
-      }
-      const absPath = resolve5(filePath);
-      await mkdir2(dirname(absPath), { recursive: true });
-      let buf;
+import path from "node:path";
+
+// src/lib/codesign-page.ts
+import { chromium as chromium2 } from "playwright";
+var DEFAULT_CODESIGN_PAGE_DEFAULTS = {
+  headless: true,
+  timeoutMs: 6e4,
+  afterLoadWaitMs: 1e3,
+  networkIdleMaxWaitMs: 3e4
+};
+async function withCodesignPage(deps2, input, fn, opts = {}) {
+  const merged = {
+    ...DEFAULT_CODESIGN_PAGE_DEFAULTS,
+    ...opts.defaults ?? {}
+  };
+  const url = input.url;
+  const headless = input.headless ?? merged.headless;
+  const timeoutMs = input.timeoutMs ?? merged.timeoutMs;
+  const persistentDir = resolveEdgeUserDataDir(input.userDataDir);
+  const context = await chromium2.launchPersistentContext(
+    persistentDir,
+    persistentBrowserContextOptions({
+      headless,
+      profileDirectory: input.profileDirectory,
+      channel: deps2.browserChannel,
+      edgeExecutablePath: deps2.edgeExecutableArg
+    })
+  );
+  try {
+    const page = await context.newPage();
+    page.setDefaultTimeout(timeoutMs);
+    await page.goto(url, { waitUntil: "domcontentloaded" });
+    try {
+      await page.waitForLoadState("networkidle", {
+        timeout: Math.max(5e3, Math.min(timeoutMs, merged.networkIdleMaxWaitMs))
+      });
+    } catch {
+    }
+    if (opts.waitForSelector?.selector) {
+      const waitTimeout = opts.waitForSelector.timeoutMs ?? Math.max(5e3, Math.floor(timeoutMs * 2 / 3));
       try {
-        buf = Buffer.from(parsed.base64, "base64");
+        await page.waitForSelector(opts.waitForSelector.selector, {
+          state: opts.waitForSelector.state ?? "attached",
+          timeout: Math.min(timeoutMs, waitTimeout)
+        });
       } catch {
-        return toolError("base64 \u89E3\u7801\u5931\u8D25\uFF08\u8BF7\u786E\u8BA4\u4E0D\u662F\u88AB\u622A\u65AD/\u4E0D\u662F base64\uFF09\u3002");
       }
-      if (buf.length === 0) return toolError("base64 \u89E3\u7801\u540E\u4E3A\u7A7A\uFF08\u8BF7\u786E\u8BA4\u8F93\u5165\u6B63\u786E\uFF09\u3002");
-      await writeFile(absPath, buf);
-      return toolResult(
-        JSON.stringify(
-          {
-            savedPath: absPath,
-            bytes: buf.length
-          },
-          null,
-          2
-        )
-      );
     }
-  );
+    if (merged.afterLoadWaitMs > 0) {
+      await page.waitForTimeout(merged.afterLoadWaitMs);
+    }
+    return await fn(page, { persistentDir, headless, timeoutMs });
+  } finally {
+    await context.close();
+  }
 }
 
-// src/tools/screen-inspect.ts
-import { chromium as chromium2 } from "playwright";
-
 // src/lib/screen-inspect-png.ts
 async function screenshotLargestScreenInspectPng(page) {
   const handles = await page.$$(SCREEN_INSPECT_SELECTOR);
@@ -21342,8 +21328,26 @@ async function screenshotLargestScreenInspectPng(page) {
   return { buffer, boundingBox: best.box };
 }
 
-// src/tools/screen-inspect.ts
-var screenInspectInput = external_exports.object({
+// src/tools/screen-inspect-save.ts
+function safeSegment(s) {
+  return s.replace(/[^a-zA-Z0-9._-]+/g, "-").replace(/-+/g, "-").replace(/^-|-$/g, "");
+}
+function timestampCompact(d) {
+  const pad = (n) => String(n).padStart(2, "0");
+  return `${d.getFullYear()}${pad(d.getMonth() + 1)}${pad(d.getDate())}_${pad(d.getHours())}${pad(d.getMinutes())}${pad(d.getSeconds())}`;
+}
+function deriveCodesignIds(url) {
+  try {
+    const u = new URL(url);
+    const parts = u.pathname.split("/").filter(Boolean);
+    const projectId = parts.find((p) => /^\d{10,}$/.test(p));
+    const pageId = parts.find((p) => /^\d+$/.test(p));
+    return { projectId, pageId };
+  } catch {
+    return {};
+  }
+}
+var screenInspectSaveInput = external_exports.object({
   url: external_exports.string().url().describe("CoDesign \u9875\u9762\u5730\u5740"),
   headless: external_exports.boolean().optional().describe("\u65E0\u5934\u6A21\u5F0F\uFF0C\u9ED8\u8BA4 true"),
   timeoutMs: external_exports.number().int().positive().optional().describe("\u6574\u4F53\u8D85\u65F6\u6BEB\u79D2\u6570\uFF0C\u9ED8\u8BA4 60000"),
@@ -21352,171 +21356,270 @@ var screenInspectInput = external_exports.object({
   ),
   profileDirectory: external_exports.string().optional().describe("User Data \u4E0B\u914D\u7F6E\u540D\uFF0C\u5982 Default\u3002")
 });
-async function runScreenInspect(deps2, input) {
-  const url = input.url;
-  const headless = input.headless ?? true;
-  const timeoutMs = input.timeoutMs ?? 6e4;
-  const persistentDir = resolveEdgeUserDataDir(input.userDataDir);
-  const context = await chromium2.launchPersistentContext(
-    persistentDir,
-    persistentBrowserContextOptions({
-      headless,
-      profileDirectory: input.profileDirectory,
-      channel: deps2.browserChannel,
-      edgeExecutablePath: deps2.edgeExecutableArg
-    })
-  );
-  try {
-    const page = await context.newPage();
-    page.setDefaultTimeout(timeoutMs);
-    await page.goto(url, { waitUntil: "domcontentloaded" });
-    try {
-      await page.waitForLoadState("networkidle", {
-        timeout: Math.max(5e3, Math.min(timeoutMs, 3e4))
-      });
-    } catch {
-    }
-    await page.waitForTimeout(1e3);
-    const shot = await screenshotLargestScreenInspectPng(page);
-    if (!shot) {
-      return toolError("\u672A\u627E\u5230\u53EF\u89C1\u7684 .screen-inspect \u533A\u57DF\u3002");
+async function runScreenInspectSave(deps2, input) {
+  return withCodesignPage(
+    deps2,
+    input,
+    async (page) => {
+      const shot = await screenshotLargestScreenInspectPng(page);
+      if (!shot) {
+        return toolError("\u672A\u627E\u5230\u53EF\u89C1\u7684 .screen-inspect \u533A\u57DF\u3002");
+      }
+      const outDir = path.resolve(process.cwd(), ".codesign_screen_imgs");
+      const { projectId, pageId } = deriveCodesignIds(input.url);
+      const filenameParts = [
+        "codesign",
+        projectId ? `p${safeSegment(projectId)}` : void 0,
+        pageId ? `s${safeSegment(pageId)}` : void 0,
+        timestampCompact(/* @__PURE__ */ new Date())
+      ].filter(Boolean);
+      const finalPath = path.join(outDir, `${filenameParts.join("_")}.png`);
+      await mkdir2(path.dirname(finalPath), { recursive: true });
+      await writeFile(finalPath, shot.buffer);
+      return toolResult(
+        JSON.stringify(
+          {
+            url: input.url,
+            title: await page.title(),
+            mimeType: "image/png",
+            savedPath: finalPath,
+            boundingBox: shot.boundingBox
+          },
+          null,
+          2
+        )
+      );
+    },
+    {
+      waitForSelector: { selector: ".screen-inspect", state: "visible" },
+      defaults: { afterLoadWaitMs: 250 }
     }
-    return toolResult(
-      JSON.stringify(
-        {
-          url,
-          title: await page.title(),
-          mimeType: "image/png",
-          imageBase64: shot.buffer.toString("base64"),
-          boundingBox: shot.boundingBox
-        },
-        null,
-        2
-      )
-    );
-  } finally {
-    await context.close();
-  }
+  );
 }
-function registerScreenInspectTool(mcpServer2, deps2) {
+function registerScreenInspectSaveTool(mcpServer2, deps2) {
   mcpServer2.registerTool(
-    "codesign_screen_inspect",
+    "codesign_screen_inspect_save",
     {
-      description: "\u6253\u5F00 CoDesign \u9875\u9762\uFF0C\u4EC5\u622A\u53D6\u9762\u79EF\u6700\u5927\u7684\u53EF\u89C1 .screen-inspect \u533A\u57DF\uFF0C\u4EE5 PNG \u7684 base64 \u8FD4\u56DE\uFF0C\u4E0D\u5199\u78C1\u76D8\u3002",
-      inputSchema: screenInspectInput
+      description: "\u6253\u5F00 CoDesign \u9875\u9762\uFF0C\u4EC5\u622A\u53D6\u9762\u79EF\u6700\u5927\u7684\u53EF\u89C1 .screen-inspect \u533A\u57DF\uFF0C\u4FDD\u5B58\u4E3A\u672C\u5730 PNG \u6587\u4EF6\u5E76\u8FD4\u56DE\u8DEF\u5F84\uFF08\u9ED8\u8BA4\u4FDD\u5B58\u5230\u9879\u76EE\u6839\u76EE\u5F55 .codesign_screen_imgs\uFF09\u3002",
+      inputSchema: screenInspectSaveInput
     },
-    (input) => runScreenInspect(deps2, input)
+    (input) => runScreenInspectSave(deps2, input)
   );
 }
 
 // src/tools/screen-list.ts
-import { chromium as chromium3 } from "playwright";
 var screenListInput = external_exports.object({
-  url: external_exports.string().url().describe("CoDesign \u8BBE\u8BA1\u9875\u9762\u5730\u5740\uFF0C\u5982 https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect"),
+  url: external_exports.string().url().describe(
+    "CoDesign \u8BBE\u8BA1\u9875\u9762\u5730\u5740\uFF0C\u5982 https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect"
+  ),
   headless: external_exports.boolean().optional().describe("\u65E0\u5934\u6A21\u5F0F\uFF0C\u9ED8\u8BA4 true"),
   timeoutMs: external_exports.number().int().positive().optional().describe("\u6574\u4F53\u8D85\u65F6\u6BEB\u79D2\u6570\uFF0C\u9ED8\u8BA4 60000"),
-  userDataDir: external_exports.string().optional().describe(
-    "\u6D4F\u89C8\u5668\u7528\u6237\u6570\u636E\u6839\u76EE\u5F55\uFF1B\u672A\u4F20\u65F6\u540C\u6293\u56FE\u5DE5\u5177\uFF08\u73AF\u5883\u53D8\u91CF\u6216\u4E34\u65F6\u76EE\u5F55\uFF09\u3002"
-  ),
+  userDataDir: external_exports.string().optional().describe("\u6D4F\u89C8\u5668\u7528\u6237\u6570\u636E\u6839\u76EE\u5F55\uFF1B\u672A\u4F20\u65F6\u540C\u6293\u56FE\u5DE5\u5177\uFF08\u73AF\u5883\u53D8\u91CF\u6216\u4E34\u65F6\u76EE\u5F55\uFF09\u3002"),
   profileDirectory: external_exports.string().optional().describe("User Data \u4E0B\u914D\u7F6E\u540D\uFF0C\u5982 Default\u3002")
 });
 async function runScreenList(deps2, input) {
-  const url = input.url;
-  const headless = input.headless ?? true;
-  const timeoutMs = input.timeoutMs ?? 6e4;
-  const persistentDir = resolveEdgeUserDataDir(input.userDataDir);
-  const context = await chromium3.launchPersistentContext(
-    persistentDir,
-    persistentBrowserContextOptions({
-      headless,
-      profileDirectory: input.profileDirectory,
-      channel: deps2.browserChannel,
-      edgeExecutablePath: deps2.edgeExecutableArg
-    })
-  );
-  try {
-    const page = await context.newPage();
-    page.setDefaultTimeout(timeoutMs);
-    await page.goto(url, { waitUntil: "domcontentloaded" });
-    try {
-      await page.waitForLoadState("networkidle", {
-        timeout: Math.max(5e3, Math.min(timeoutMs, 3e4))
+  return withCodesignPage(
+    deps2,
+    input,
+    async (page) => {
+      const pages = await page.evaluate(() => {
+        const items = document.querySelectorAll(".screen-list__item");
+        const result = [];
+        for (const item of items) {
+          const id = item.getAttribute("id") || "";
+          const match = id.match(/^screen-(\d+)$/);
+          if (!match) continue;
+          const pageId = match[1];
+          let name = "";
+          const nameEl = item.querySelector(
+            "[class*='name'], [class*='title'], [class*='label']"
+          );
+          if (nameEl) {
+            name = nameEl.textContent?.trim() || "";
+          }
+          if (!name) {
+            name = item.textContent?.trim().replace(/\s+/g, " ") || "";
+          }
+          result.push({
+            name,
+            pageId
+          });
+        }
+        return result;
       });
-    } catch {
+      if (pages.length === 0) {
+        return toolError("\u672A\u627E\u5230\u5DE6\u4FA7\u83DC\u5355\u4E2D\u7684\u9875\u9762\u5217\u8868\u3002");
+      }
+      const currentUrl = new URL(input.url);
+      const pathParts = currentUrl.pathname.split("/");
+      const projectId = pathParts.find((p) => /^\d{10,}$/.test(p)) || "";
+      const currentPageId = pathParts.find((p) => /^\d+$/.test(p)) || "";
+      const resultData = pages.map((p) => ({
+        name: p.name,
+        pageId: p.pageId,
+        url: projectId ? `https://codesign.qq.com/app/design/${projectId}/${p.pageId}/inspect` : currentPageId ? input.url.replace(`/${currentPageId}/`, `/${p.pageId}/`) : input.url
+      }));
+      return toolResult(
+        JSON.stringify(
+          {
+            projectId,
+            totalCount: resultData.length,
+            pages: resultData
+          },
+          null,
+          2
+        )
+      );
+    },
+    {
+      waitForSelector: { selector: ".screen-list__item", state: "attached" },
+      defaults: { afterLoadWaitMs: 250 }
     }
-    await page.waitForTimeout(2e3);
-    const pages = await page.evaluate(() => {
-      const items = document.querySelectorAll(".screen-list__item");
-      const result = [];
-      for (const item of items) {
-        const id = item.getAttribute("id") || "";
-        const match = id.match(/^screen-(\d+)$/);
-        if (!match) continue;
-        const pageId = match[1];
-        let name = "";
-        const nameEl = item.querySelector("[class*='name'], [class*='title'], [class*='label']");
-        if (nameEl) {
-          name = nameEl.textContent?.trim() || "";
-        }
-        if (!name) {
-          name = item.textContent?.trim().replace(/\s+/g, " ") || "";
-        }
-        result.push({
-          name,
-          pageId
-        });
-      }
-      return result;
-    });
-    if (pages.length === 0) {
-      return toolError("\u672A\u627E\u5230\u5DE6\u4FA7\u83DC\u5355\u4E2D\u7684\u9875\u9762\u5217\u8868\u3002");
-    }
-    const currentUrl = new URL(url);
-    const pathParts = currentUrl.pathname.split("/");
-    const projectId = pathParts.find((p) => /^\d{10,}$/.test(p)) || "";
-    const resultData = pages.map((p) => ({
-      name: p.name,
-      pageId: p.pageId,
-      url: projectId ? `https://codesign.qq.com/app/design/${projectId}/${p.pageId}/inspect` : p.url.replace("PROJECT_ID", projectId)
-    }));
-    return toolResult(
-      JSON.stringify(
-        {
-          projectId,
-          totalCount: resultData.length,
-          pages: resultData
-        },
-        null,
-        2
-      )
-    );
-  } finally {
-    await context.close();
-  }
+  );
 }
 function registerScreenListTool(mcpServer2, deps2) {
   mcpServer2.registerTool(
     "codesign_screen_list",
     {
-      description: "\u83B7\u53D6 CoDesign \u8BBE\u8BA1\u9875\u9762\u5DE6\u4FA7\u83DC\u5355\u4E2D\u7684\u6240\u6709\u9875\u9762\u540D\u79F0\u548C\u8BBF\u95EE\u5730\u5740\u3002",
+      description: "\u6839\u636ECoDesign\u9875\u9762\u5730\u5740\u83B7\u53D6\u9879\u76EE\u6240\u6709\u9875\u9762\u5217\u8868\uFF0C\u8FD4\u56DE\u9879\u76EEID\u3001\u603B\u9875\u9762\u6570\u3001\u9875\u9762\u5217\u8868\u3002",
       inputSchema: screenListInput
     },
     (input) => runScreenList(deps2, input)
   );
 }
 
+// src/tools/screen-viewer-download-save.ts
+import { mkdir as mkdir3 } from "node:fs/promises";
+import path2 from "node:path";
+function safeSegment2(s) {
+  return s.replace(/[^a-zA-Z0-9._-]+/g, "-").replace(/-+/g, "-").replace(/^-|-$/g, "");
+}
+function timestampCompact2(d) {
+  const pad = (n) => String(n).padStart(2, "0");
+  return `${d.getFullYear()}${pad(d.getMonth() + 1)}${pad(d.getDate())}_${pad(d.getHours())}${pad(d.getMinutes())}${pad(d.getSeconds())}`;
+}
+function deriveCodesignIds2(url) {
+  try {
+    const u = new URL(url);
+    const parts = u.pathname.split("/").filter(Boolean);
+    const projectId = parts.find((p) => /^\d{10,}$/.test(p));
+    const pageId = parts.find((p) => /^\d+$/.test(p));
+    return { projectId, pageId };
+  } catch {
+    return {};
+  }
+}
+var screenViewerDownloadSaveInput = external_exports.object({
+  url: external_exports.string().url().describe("CoDesign \u9875\u9762\u5730\u5740"),
+  headless: external_exports.boolean().optional().describe("\u65E0\u5934\u6A21\u5F0F\uFF0C\u9ED8\u8BA4 true"),
+  timeoutMs: external_exports.number().int().positive().optional().describe("\u6574\u4F53\u8D85\u65F6\u6BEB\u79D2\u6570\uFF0C\u9ED8\u8BA4 60000"),
+  userDataDir: external_exports.string().optional().describe(
+    "\u6D4F\u89C8\u5668\u7528\u6237\u6570\u636E\u6839\u76EE\u5F55\uFF08\u987B\u5148\u9000\u51FA\u5360\u7528\u8BE5\u76EE\u5F55\u7684\u6D4F\u89C8\u5668\uFF09\u3002\u672A\u4F20\u65F6\u4F7F\u7528\u73AF\u5883\u53D8\u91CF Z_MCP_CODESIGN_EDGE_USER_DATA \u6216\u7CFB\u7EDF\u4E34\u65F6\u76EE\u5F55\u4E0B\u9ED8\u8BA4\u914D\u7F6E\u3002"
+  ),
+  profileDirectory: external_exports.string().optional().describe("User Data \u4E0B\u914D\u7F6E\u540D\uFF0C\u5982 Default\u3002")
+});
+async function clickFirstVisibleText(page, candidates) {
+  for (const text of candidates) {
+    const loc = page.getByText(text, { exact: true }).first();
+    try {
+      if (await loc.isVisible()) {
+        await loc.click();
+        return true;
+      }
+    } catch {
+    }
+  }
+  return false;
+}
+async function runScreenViewerDownloadSave(deps2, input) {
+  return withCodesignPage(
+    deps2,
+    input,
+    async (page) => {
+      const action = page.locator(".screen-viewer__action").first();
+      try {
+        await action.waitFor({ state: "visible", timeout: 5e3 });
+      } catch {
+        return toolError(
+          "\u672A\u627E\u5230\u53EF\u89C1\u7684 .screen-viewer__action\uFF08\u53EF\u80FD\u672A\u8FDB\u5165\u53EF\u4E0B\u8F7D\u7684\u753B\u677F\u89C6\u56FE\uFF09\u3002"
+        );
+      }
+      const downloadPromise = page.waitForEvent("download", {
+        timeout: Math.max(1e4, Math.min(45e3, input.timeoutMs ?? 6e4))
+      });
+      await action.click();
+      const clicked = await clickFirstVisibleText(page, [
+        "\u4E0B\u8F7D\u56FE\u7247",
+        "\u4E0B\u8F7D\u539F\u56FE",
+        "\u4E0B\u8F7D"
+      ]);
+      if (!clicked) {
+        return toolError(
+          "\u672A\u627E\u5230\u201C\u4E0B\u8F7D\u56FE\u7247\u201D\u6309\u94AE/\u83DC\u5355\u9879\uFF08\u53EF\u80FD\u9875\u9762\u7ED3\u6784\u53D8\u66F4\u6216\u672A\u767B\u5F55\uFF09\u3002"
+        );
+      }
+      let download;
+      try {
+        download = await downloadPromise;
+      } catch {
+        return toolError("\u672A\u6355\u83B7\u5230\u4E0B\u8F7D\u4E8B\u4EF6\uFF08\u70B9\u51FB\u540E\u672A\u89E6\u53D1\u4E0B\u8F7D\uFF09\u3002");
+      }
+      const suggested = download.suggestedFilename() || "codesign_download.png";
+      const ext = path2.extname(suggested) || ".png";
+      const outDir = path2.resolve(process.cwd(), ".codesign_screen_imgs");
+      const { projectId, pageId } = deriveCodesignIds2(input.url);
+      const filenameParts = [
+        "codesign",
+        "viewer_download",
+        projectId ? `p${safeSegment2(projectId)}` : void 0,
+        pageId ? `s${safeSegment2(pageId)}` : void 0,
+        timestampCompact2(/* @__PURE__ */ new Date())
+      ].filter(Boolean);
+      const finalPath = path2.join(outDir, `${filenameParts.join("_")}${ext}`);
+      await mkdir3(path2.dirname(finalPath), { recursive: true });
+      await download.saveAs(finalPath);
+      return toolResult(
+        JSON.stringify(
+          {
+            url: input.url,
+            title: await page.title(),
+            suggestedFilename: suggested,
+            savedPath: finalPath
+          },
+          null,
+          2
+        )
+      );
+    },
+    {
+      waitForSelector: { selector: ".screen-viewer__action", state: "visible" },
+      defaults: { afterLoadWaitMs: 600, networkIdleMaxWaitMs: 2e3 }
+    }
+  );
+}
+function registerScreenViewerDownloadSaveTool(mcpServer2, deps2) {
+  mcpServer2.registerTool(
+    "codesign_screen_viewer_download_save",
+    {
+      description: "\u6253\u5F00 CoDesign \u9875\u9762\uFF0C\u70B9\u51FB .screen-viewer__action \u540E\u9009\u62E9\u201C\u4E0B\u8F7D\u56FE\u7247\u201D\uFF0C\u6355\u83B7\u6D4F\u89C8\u5668\u4E0B\u8F7D\u5E76\u4FDD\u5B58\u5230\u9879\u76EE\u6839\u76EE\u5F55 .codesign_screen_imgs/\u3002",
+      inputSchema: screenViewerDownloadSaveInput
+    },
+    (input) => runScreenViewerDownloadSave(deps2, input)
+  );
+}
+
 // src/tools/register-all.ts
 function registerAllCodesignTools(mcpServer2, deps2) {
-  registerScreenInspectTool(mcpServer2, deps2);
+  registerScreenInspectSaveTool(mcpServer2, deps2);
+  registerScreenViewerDownloadSaveTool(mcpServer2, deps2);
   registerScreenListTool(mcpServer2, deps2);
   registerOpenLoginTool(mcpServer2, deps2);
-  registerSaveImageTool(mcpServer2);
 }
 
 // src/index.ts
 var SERVER_NAME = "z-mcp-codesign";
 var _require = createRequire(import.meta.url);
-var _dirname = dirname2(fileURLToPath(import.meta.url));
+var _dirname = dirname(fileURLToPath(import.meta.url));
 var SERVER_VERSION = _require(join4(_dirname, "..", "package.json")).version ?? "1.0.0";
 var USAGE = "npx -y z-mcp-codesign --stdio";
 var args = process.argv.slice(2);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "z-mcp-codesign",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "type": "module",
   "description": "An MCP server to capture images from Tencent CoDesign pages via Edge",
   "main": "dist/index.js",