@weapp-vite/mcp 1.1.2 → 1.2.1

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

@@ -1,16 +1,16 @@
 import process from 'node:process';
 import { fileURLToPath } from 'node:url';
-import fs$1 from 'node:fs/promises';
-import { createRequire } from 'node:module';
-import path from 'node:path';
-import fs from 'node:fs';
-import { spawn } from 'node:child_process';
 import { Buffer } from 'node:buffer';
 import http from 'node:http';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+import fs$1 from 'node:fs/promises';
 import { McpServer, ResourceTemplate } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { z } from 'zod';
+import { createRequire } from 'node:module';
+import path from 'node:path';
+import fs from 'node:fs';
+import { spawn } from 'node:child_process';
 
 const MCP_SERVER_NAME = "@weapp-vite/mcp";
 const MCP_SERVER_VERSION = "2.0.0";
@@ -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);
+      }
+      if (outputPath) {
+        args.push("--output", outputPath);
+      }
+      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);
       }
-      const finalArgs = [cliPath, subCommand];
-      if (projectPath) {
-        finalArgs.push(resolveSubPath(workspaceRoot, projectPath));
+      if (currentOutputPath) {
+        args.push("--current-output", currentOutputPath);
       }
-      if (platform) {
-        finalArgs.push("--platform", platform);
+      if (diffOutputPath) {
+        args.push("--diff-output", diffOutputPath);
       }
-      if (Array.isArray(args) && args.length > 0) {
-        finalArgs.push(...args);
+      if (threshold != null) {
+        args.push("--threshold", String(threshold));
       }
-      const result = await runCommand(workspaceRoot, "node", finalArgs, {
-        timeoutMs: timeoutMs ?? DEFAULT_TIMEOUT_MS
+      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);
     }
@@ -776,7 +874,7 @@ function writeJson(res, statusCode, payload) {
   res.setHeader("content-type", "application/json");
   res.end(JSON.stringify(payload));
 }
-async function startStdioServer$1(options) {
+async function startStdioServer(options) {
   const previousCwd = process.cwd();
   if (options?.workspaceRoot) {
     process.chdir(options.workspaceRoot);
@@ -881,7 +979,7 @@ async function startWeappViteMcpServer(options) {
   if (transport === "streamable-http") {
     return startStreamableHttpServer(options ?? {});
   }
-  await startStdioServer$1(options);
+  await startStdioServer(options);
   return {
     transport: "stdio"
   };
@@ -904,4 +1002,4 @@ ${message}
   });
 }
 
-export { DEFAULT_MAX_FILE_CHARS, DEFAULT_MAX_OUTPUT_CHARS, DEFAULT_MAX_RESULTS, DEFAULT_MCP_ENDPOINT, DEFAULT_MCP_HOST, DEFAULT_MCP_PORT, DEFAULT_TIMEOUT_MS, EXPOSED_PACKAGES, MCP_SERVER_NAME, MCP_SERVER_VERSION, SKIPPED_DIR_NAMES, assertInsideRoot, createWeappViteMcpServer, formatJson, listFilesInDirectory, loadExposedCatalog, loadPackageSummary, normalizeErrorMessage, readFileContent, resolveSubPath, resolveWorkspaceRoot, runCommand, searchTextInDirectory, startStdioServer$1 as startStdioServer, startWeappViteMcpServer, toToolError, toToolResult };
+export { DEFAULT_MAX_FILE_CHARS, DEFAULT_MAX_OUTPUT_CHARS, DEFAULT_MAX_RESULTS, DEFAULT_MCP_ENDPOINT, DEFAULT_MCP_HOST, DEFAULT_MCP_PORT, DEFAULT_TIMEOUT_MS, EXPOSED_PACKAGES, MCP_SERVER_NAME, MCP_SERVER_VERSION, SKIPPED_DIR_NAMES, assertInsideRoot, createWeappViteMcpServer, formatJson, listFilesInDirectory, loadExposedCatalog, loadPackageSummary, normalizeErrorMessage, readFileContent, resolveSubPath, resolveWorkspaceRoot, runCommand, searchTextInDirectory, startStdioServer, startWeappViteMcpServer, toToolError, toToolResult };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@weapp-vite/mcp",
   "type": "module",
-  "version": "1.1.2",
+  "version": "1.2.1",
   "description": "mcp",
   "author": "ice breaker <1324318532@qq.com>",
   "license": "ISC",
@@ -40,6 +40,8 @@
     "build": "unbuild",
     "test": "vitest run",
     "test:dev": "vitest",
+    "pretest:types": "pnpm build",
+    "test:types": "tsd",
     "typecheck": "tsc --noEmit",
     "release": "pnpm publish",
     "lint": "eslint .",