z-mcp-codesign 0.1.2 → 0.1.4

package/README.md

@@ -21,46 +21,40 @@
 }
 ```
 
-
-
 # 提供的工具
 
-| 工具名 | 说明 |
-| ------ | ---- |
-| `codesign_open_login` | 有界面打开 CoDesign 登录页，用于在本机 Edge 用户数据目录中完成登录 |
-| `codesign_screen_inspect` | 打开指定 CoDesign 页面，截取最大的可见 `.screen-inspect` 区域，以 PNG 的 base64 返回 |
-| `codesign_screen_inspect_save` | 打开指定 CoDesign 页面，截取最大的可见 `.screen-inspect` 区域，保存为本地 PNG 文件并返回保存路径（默认保存到项目根目录 `.codesign_screen_imgs/`） |
-| `codesign_screen_list` | 获取 CoDesign 设计页面左侧菜单中的所有页面名称和访问地址 |
+| 工具名                                 | 说明                                                                                                                                    |
+| -------------------------------------- | --------------------------------------------------------------------------------------------------------------------------------------- |
+| `codesign_open_login`                  | 有界面打开 CoDesign 登录页，用于在本机 Edge 用户数据目录中完成登录                                                                      |
+| `codesign_screen_inspect_save`         | 打开指定 CoDesign 页面，截取最大的可见 `.screen-inspect` 区域，保存为本地 PNG 文件并返回保存路径（默认保存到项目根目录 `_z-mcp-imgs/`） |
+| `codesign_screen_viewer_download_save` | 打开 CoDesign 页面，点击 `.screen-viewer__action` 选择“下载图片”，捕获下载并保存到项目根目录 `_z-mcp-imgs/`                             |
+| `codesign_screen_list`                 | 获取 CoDesign 设计页面左侧菜单中的所有页面名称和访问地址                                                                                |
 
 # 使用示例
 
 **打开登录页（用于后续抓图）**
 
 ```bash
-请在本机用可见浏览器打开 CoDesign 登录页
+调用MCP 打开CoDesign登录页
 ```
 
 **列出当前项目的页面列表**
 
 ```bash
-请打开这个 CoDesign inspect 链接，并把左侧菜单里的页面列表整理出来（包含每个页面的名称和对应的 inspect 访问地址）：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+调用MCP 获取CoDesign所有页面列表使用表格展示。
+URL为：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
 ```
 
-**抓图并返回图片（base64）**
+**通过页面“下载图片”保存到本地**
 
 ```bash
-请打开这个 CoDesign inspect 链接 把截图以 base64 返回给我：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
-```
-
-**抓图并保存到本地**
-
-```bash
-请打开这个 CoDesign inspect 链接，获取图片，把截图保存到本地，并告诉我保存路径：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+调用MCP 打开CoDesign页面，下载图片保存到本地，并告诉我保存路径。
+传入参数url为：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
 ```
 
 **生成项目**
 
 ```bash
-请打开这个 CoDesign inspect 链接，获取图片，把截图保存到本地，并识别图片内容生成Vue3+Vite项目：https://codesign.qq.com/app/design/<projectId>/<pageId>/inspect
+调用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;
@@ -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
@@ -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);
       }
     });
   }
@@ -21008,8 +21008,7 @@ import { resolve as resolve4 } from "node:path";
 import { join } from "node:path";
 var ENV_EDGE_USER_DATA = "Z_MCP_CODESIGN_EDGE_USER_DATA";
 var DEFAULT_CODESIGN_LOGIN_URL = "https://codesign.qq.com/app/login";
-var SCREEN_INSPECT_SELECTOR = ".screen-inspect:visible";
-var DEFAULT_MCP_EDGE_PROFILE = join(process.cwd(), ".z-mcp-codesign-edge");
+var DEFAULT_MCP_EDGE_PROFILE = join(process.cwd(), "_z-mcp-edge");
 
 // src/lib/login-window.ts
 import { resolve as resolve2 } from "node:path";
@@ -21205,9 +21204,8 @@ var toolError = (text) => ({
 // src/tools/open-login.ts
 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) {
@@ -21226,7 +21224,7 @@ function registerOpenLoginTool(mcpServer2, deps2) {
           return toolError("loginUrl \u4E0D\u662F\u5408\u6CD5 URL\u3002");
         }
       }
-      const persistentDir = resolveEdgeUserDataDir(input.userDataDir);
+      const persistentDir = resolveEdgeUserDataDir(void 0);
       const ok = await openOrNavigateCodesignLogin(
         deps2,
         loginUrl,
@@ -21243,77 +21241,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"
-          },
-          null,
-          2
-        )
-      );
-    }
-  );
-}
-
-// src/tools/save-image.ts
-import { resolve as resolve5, dirname, extname } from "node:path";
-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;
-      try {
-        buf = Buffer.from(parsed.base64, "base64");
-      } 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
+            hint: "\u5728\u5F39\u51FA\u7684\u6D4F\u89C8\u5668\u4E2D\u5B8C\u6210\u767B\u5F55\u540E\u5173\u95ED\u7A97\u53E3\uFF0C\u518D\u8C03\u7528\u6293\u56FE/\u5217\u8868/\u4E0B\u8F7D\u7B49\u5DE5\u5177\u3002"
           },
           null,
           2
@@ -21337,9 +21265,9 @@ async function withCodesignPage(deps2, input, fn, opts = {}) {
     ...opts.defaults ?? {}
   };
   const url = input.url;
-  const headless = input.headless ?? merged.headless;
+  const headless = opts.allowHeaded ? merged.headless : true;
   const timeoutMs = input.timeoutMs ?? merged.timeoutMs;
-  const persistentDir = resolveEdgeUserDataDir(input.userDataDir);
+  const persistentDir = resolveEdgeUserDataDir(void 0);
   const context = await chromium2.launchPersistentContext(
     persistentDir,
     persistentBrowserContextOptions({
@@ -21378,49 +21306,62 @@ async function withCodesignPage(deps2, input, fn, opts = {}) {
   }
 }
 
-// src/lib/screen-inspect-png.ts
-async function screenshotLargestScreenInspectPng(page) {
-  const handles = await page.$$(SCREEN_INSPECT_SELECTOR);
-  let best = null;
-  for (const h of handles) {
-    const box = await h.boundingBox();
-    if (!box || box.width < 1 || box.height < 1) continue;
-    const area = box.width * box.height;
-    const bestArea = best ? best.box.width * best.box.height : 0;
-    if (!best || area > bestArea) best = { handle: h, box };
-  }
-  if (!best) return null;
-  const buffer = Buffer.from(await best.handle.screenshot({ type: "png" }));
-  return { buffer, boundingBox: best.box };
-}
-
-// src/tools/screen-inspect.ts
-var screenInspectInput = 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"
+// src/tools/screen-list.ts
+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"
   ),
+  timeoutMs: external_exports.number().int().positive().optional().describe("\u6574\u4F53\u8D85\u65F6\u6BEB\u79D2\u6570\uFF0C\u9ED8\u8BA4 60000"),
   profileDirectory: external_exports.string().optional().describe("User Data \u4E0B\u914D\u7F6E\u540D\uFF0C\u5982 Default\u3002")
 });
-async function runScreenInspect(deps2, input) {
+async function runScreenList(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 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(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(
           {
-            url: input.url,
-            title: await page.title(),
-            mimeType: "image/png",
-            imageBase64: shot.buffer.toString("base64"),
-            boundingBox: shot.boundingBox
+            projectId,
+            totalCount: resultData.length,
+            pages: resultData
           },
           null,
           2
@@ -21428,24 +21369,24 @@ async function runScreenInspect(deps2, input) {
       );
     },
     {
-      waitForSelector: { selector: ".screen-inspect", state: "visible" },
+      waitForSelector: { selector: ".screen-list__item", state: "attached" },
       defaults: { afterLoadWaitMs: 250 }
     }
   );
 }
-function registerScreenInspectTool(mcpServer2, deps2) {
+function registerScreenListTool(mcpServer2, deps2) {
   mcpServer2.registerTool(
-    "codesign_screen_inspect",
+    "codesign_screen_list",
     {
-      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\u3002",
-      inputSchema: screenInspectInput
+      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) => runScreenInspect(deps2, input)
+    (input) => runScreenList(deps2, input)
   );
 }
 
-// src/tools/screen-inspect-save.ts
-import { mkdir as mkdir3, writeFile as writeFile2 } from "node:fs/promises";
+// src/tools/screen-viewer-download-save.ts
+import { mkdir as mkdir2 } from "node:fs/promises";
 import path from "node:path";
 function safeSegment(s) {
   return s.replace(/[^a-zA-Z0-9._-]+/g, "-").replace(/-+/g, "-").replace(/^-|-$/g, "");
@@ -21465,124 +21406,78 @@ function deriveCodesignIds(url) {
     return {};
   }
 }
-var screenInspectSaveInput = external_exports.object({
+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 runScreenInspectSave(deps2, input) {
+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 shot = await screenshotLargestScreenInspectPng(page);
-      if (!shot) {
-        return toolError("\u672A\u627E\u5230\u53EF\u89C1\u7684 .screen-inspect \u533A\u57DF\u3002");
+      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"
+        );
       }
-      const outDir = path.resolve(process.cwd(), ".codesign_screen_imgs");
+      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 = path.extname(suggested) || ".png";
+      const outDir = path.resolve(process.cwd(), "_z-mcp-imgs");
       const { projectId, pageId } = deriveCodesignIds(input.url);
       const filenameParts = [
         "codesign",
+        "viewer_download",
         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 mkdir3(path.dirname(finalPath), { recursive: true });
-      await writeFile2(finalPath, shot.buffer);
+      const finalPath = path.join(outDir, `${filenameParts.join("_")}${ext}`);
+      await mkdir2(path.dirname(finalPath), { recursive: true });
+      await download.saveAs(finalPath);
       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 }
-    }
-  );
-}
-function registerScreenInspectSaveTool(mcpServer2, deps2) {
-  mcpServer2.registerTool(
-    "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\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) => runScreenInspectSave(deps2, input)
-  );
-}
-
-// src/tools/screen-list.ts
-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"
-  ),
-  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"),
-  profileDirectory: external_exports.string().optional().describe("User Data \u4E0B\u914D\u7F6E\u540D\uFF0C\u5982 Default\u3002")
-});
-async function runScreenList(deps2, input) {
-  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;
-      });
-      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
+            suggestedFilename: suggested,
+            savedPath: finalPath
           },
           null,
           2
@@ -21590,35 +21485,33 @@ async function runScreenList(deps2, input) {
       );
     },
     {
-      waitForSelector: { selector: ".screen-list__item", state: "attached" },
-      defaults: { afterLoadWaitMs: 250 }
+      waitForSelector: { selector: ".screen-viewer__action", state: "visible" },
+      defaults: { afterLoadWaitMs: 600, networkIdleMaxWaitMs: 2e3 }
     }
   );
 }
-function registerScreenListTool(mcpServer2, deps2) {
+function registerScreenViewerDownloadSaveTool(mcpServer2, deps2) {
   mcpServer2.registerTool(
-    "codesign_screen_list",
+    "codesign_screen_viewer_download_save",
     {
-      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
+      description: "\u6253\u5F00 CoDesign \u9875\u9762\uFF0C\u70B9\u51FB\u201C\u4E0B\u8F7D\u56FE\u7247\u201D\uFF0C\u6355\u83B7\u6D4F\u89C8\u5668\u4E0B\u8F7D\u5E76\u4FDD\u5B58\u5230\u9879\u76EE\u6839\u76EE\u5F55 _z-mcp-imgs/\u3002\u4E0B\u8F7DCoDesign\u9875\u9762\u56FE\u7247\u5230\u672C\u5730",
+      inputSchema: screenViewerDownloadSaveInput
     },
-    (input) => runScreenList(deps2, input)
+    (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.2",
+  "version": "0.1.4",
   "type": "module",
   "description": "An MCP server to capture images from Tencent CoDesign pages via Edge",
   "main": "dist/index.js",