@weapp-vite/mcp 1.1.2 → 1.2.0

package/README.md

@@ -43,6 +43,8 @@ await handle.close?.()
 - `search_source_code`: 在包源码中检索关键词
 - `run_package_script`: 在指定包目录执行 `pnpm run <script>`
 - `run_weapp_vite_cli`: 执行 `node packages/weapp-vite/bin/weapp-vite.js ...`
+- `take_weapp_screenshot`: 面向“小程序截图 / 页面快照”语义，执行 `weapp-vite screenshot --json`
+- `compare_weapp_screenshot`: 面向“截图对比 / diff / baseline / 视觉回归”语义，执行 `weapp-vite compare --json`
 - `run_repo_command`: 执行仓库级命令（`pnpm/node/git/rg`）
 
 ## 主要 Resources

package/dist/index.mjs

@@ -413,6 +413,25 @@ function toDocsUri(packageId, fileName) {
 async function readTextFile(filePath) {
   return fs$1.readFile(filePath, "utf8");
 }
+async function runWeappViteCliTool(workspaceRoot, input) {
+  const cliPath = (await resolveExposedPackage(workspaceRoot, "weapp-vite")).cliPath;
+  if (!cliPath) {
+    throw new Error("\u5F53\u524D\u5DE5\u4F5C\u533A\u4E2D\u7684 weapp-vite \u672A\u66B4\u9732 CLI \u5165\u53E3");
+  }
+  const finalArgs = [cliPath, input.subCommand];
+  if (input.projectPath) {
+    finalArgs.push(resolveSubPath(workspaceRoot, input.projectPath));
+  }
+  if (input.platform) {
+    finalArgs.push("--platform", input.platform);
+  }
+  if (Array.isArray(input.args) && input.args.length > 0) {
+    finalArgs.push(...input.args);
+  }
+  return runCommand(workspaceRoot, "node", finalArgs, {
+    timeoutMs: input.timeoutMs ?? DEFAULT_TIMEOUT_MS
+  });
+}
 async function createWeappViteMcpServer(options) {
   const workspaceRoot = resolveWorkspaceRoot(options?.workspaceRoot);
   const server = new McpServer({
@@ -555,24 +574,103 @@ async function createWeappViteMcpServer(options) {
     }
   }, async ({ subCommand, projectPath, platform, args, timeoutMs }) => {
     try {
-      const cliPath = (await resolveExposedPackage(workspaceRoot, "weapp-vite")).cliPath;
-      if (!cliPath) {
-        throw new Error("\u5F53\u524D\u5DE5\u4F5C\u533A\u4E2D\u7684 weapp-vite \u672A\u66B4\u9732 CLI \u5165\u53E3");
+      const result = await runWeappViteCliTool(workspaceRoot, {
+        subCommand,
+        projectPath,
+        platform,
+        args,
+        timeoutMs
+      });
+      return toToolResult(result);
+    } catch (error) {
+      return toToolError(error);
+    }
+  });
+  server.registerTool("take_weapp_screenshot", {
+    title: "Take Weapp Screenshot",
+    description: "\u5F53\u7528\u6237\u63D0\u5230\u622A\u56FE\u3001\u9875\u9762\u5FEB\u7167\u3001\u8FD0\u884C\u65F6\u622A\u56FE\u65F6\uFF0C\u4F18\u5148\u8C03\u7528\u6B64\u5DE5\u5177\u6267\u884C weapp-vite screenshot",
+    inputSchema: {
+      projectPath: z.string().describe("\u76F8\u5BF9 workspace \u6839\u8DEF\u5F84\uFF0C\u901A\u5E38\u662F dist/build/mp-weixin \u6216\u5177\u4F53\u5C0F\u7A0B\u5E8F\u9879\u76EE\u76EE\u5F55"),
+      page: z.string().optional().describe("\u622A\u56FE\u524D\u5148\u8DF3\u8F6C\u7684\u5C0F\u7A0B\u5E8F\u9875\u9762\u8DEF\u5F84"),
+      outputPath: z.string().optional().describe("\u622A\u56FE\u8F93\u51FA\u8DEF\u5F84\uFF0C\u5EFA\u8BAE\u5199\u5165 .tmp/ \u6216\u5DE5\u4F5C\u533A\u76F8\u5BF9\u8DEF\u5F84"),
+      timeoutMs: z.number().int().positive().max(9e5).optional()
+    }
+  }, async ({ projectPath, page, outputPath, timeoutMs }) => {
+    try {
+      const args = ["--json"];
+      if (page) {
+        args.push("--page", page);
       }
-      const finalArgs = [cliPath, subCommand];
-      if (projectPath) {
-        finalArgs.push(resolveSubPath(workspaceRoot, projectPath));
+      if (outputPath) {
+        args.push("--output", outputPath);
       }
-      if (platform) {
-        finalArgs.push("--platform", platform);
+      const result = await runWeappViteCliTool(workspaceRoot, {
+        subCommand: "screenshot",
+        projectPath,
+        args,
+        timeoutMs
+      });
+      return toToolResult({
+        ...result,
+        projectPath,
+        page: page ?? null,
+        outputPath: outputPath ?? null,
+        recommendedIntent: "screenshot"
+      });
+    } catch (error) {
+      return toToolError(error);
+    }
+  });
+  server.registerTool("compare_weapp_screenshot", {
+    title: "Compare Weapp Screenshot",
+    description: "\u5F53\u7528\u6237\u63D0\u5230\u622A\u56FE\u5BF9\u6BD4\u3001diff\u3001baseline\u3001\u89C6\u89C9\u56DE\u5F52\u3001\u50CF\u7D20\u5BF9\u6BD4\u65F6\uFF0C\u4F18\u5148\u8C03\u7528\u6B64\u5DE5\u5177\u6267\u884C weapp-vite compare",
+    inputSchema: {
+      projectPath: z.string().describe("\u76F8\u5BF9 workspace \u6839\u8DEF\u5F84\uFF0C\u901A\u5E38\u662F dist/build/mp-weixin \u6216\u5177\u4F53\u5C0F\u7A0B\u5E8F\u9879\u76EE\u76EE\u5F55"),
+      baselinePath: z.string().describe("\u76F8\u5BF9 workspace \u6839\u8DEF\u5F84\u7684 baseline \u56FE\u7247\u8DEF\u5F84"),
+      page: z.string().optional().describe("\u622A\u56FE\u5BF9\u6BD4\u524D\u5148\u8DF3\u8F6C\u7684\u5C0F\u7A0B\u5E8F\u9875\u9762\u8DEF\u5F84"),
+      currentOutputPath: z.string().optional().describe("\u5F53\u524D\u622A\u56FE\u8F93\u51FA\u8DEF\u5F84"),
+      diffOutputPath: z.string().optional().describe("diff \u56FE\u7247\u8F93\u51FA\u8DEF\u5F84"),
+      threshold: z.number().min(0).max(1).optional(),
+      maxDiffPixels: z.number().int().min(0).optional(),
+      maxDiffRatio: z.number().min(0).max(1).optional(),
+      timeoutMs: z.number().int().positive().max(9e5).optional()
+    }
+  }, async ({ projectPath, baselinePath, page, currentOutputPath, diffOutputPath, threshold, maxDiffPixels, maxDiffRatio, timeoutMs }) => {
+    try {
+      const args = ["--json", "--baseline", baselinePath];
+      if (page) {
+        args.push("--page", page);
       }
-      if (Array.isArray(args) && args.length > 0) {
-        finalArgs.push(...args);
+      if (currentOutputPath) {
+        args.push("--current-output", currentOutputPath);
       }
-      const result = await runCommand(workspaceRoot, "node", finalArgs, {
-        timeoutMs: timeoutMs ?? DEFAULT_TIMEOUT_MS
+      if (diffOutputPath) {
+        args.push("--diff-output", diffOutputPath);
+      }
+      if (threshold != null) {
+        args.push("--threshold", String(threshold));
+      }
+      if (maxDiffPixels != null) {
+        args.push("--max-diff-pixels", String(maxDiffPixels));
+      }
+      if (maxDiffRatio != null) {
+        args.push("--max-diff-ratio", String(maxDiffRatio));
+      }
+      const result = await runWeappViteCliTool(workspaceRoot, {
+        subCommand: "compare",
+        projectPath,
+        args,
+        timeoutMs
+      });
+      return toToolResult({
+        ...result,
+        projectPath,
+        baselinePath,
+        page: page ?? null,
+        currentOutputPath: currentOutputPath ?? null,
+        diffOutputPath: diffOutputPath ?? null,
+        recommendedIntent: "compare"
       });
-      return toToolResult(result);
     } catch (error) {
       return toToolError(error);
     }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@weapp-vite/mcp",
   "type": "module",
-  "version": "1.1.2",
+  "version": "1.2.0",
   "description": "mcp",
   "author": "ice breaker <1324318532@qq.com>",
   "license": "ISC",