@weapp-vite/mcp 1.4.4 → 1.4.5

package/dist/index.d.mts

@@ -113,7 +113,9 @@ interface MiniProgramLike {
     reLaunch: (url: string) => Promise<MiniProgramPage>;
     switchTab: (url: string) => Promise<MiniProgramPage>;
     navigateBack: () => Promise<MiniProgramPage>;
-    screenshot: () => Promise<string | Uint8Array>;
+    screenshot: (options?: {
+        timeout?: number;
+    }) => Promise<string | Uint8Array>;
     callWxMethod: (method: string, ...args: unknown[]) => Promise<unknown>;
 }
 interface RuntimeConnectionInput {

package/dist/index.d.ts

@@ -113,7 +113,9 @@ interface MiniProgramLike {
     reLaunch: (url: string) => Promise<MiniProgramPage>;
     switchTab: (url: string) => Promise<MiniProgramPage>;
     navigateBack: () => Promise<MiniProgramPage>;
-    screenshot: () => Promise<string | Uint8Array>;
+    screenshot: (options?: {
+        timeout?: number;
+    }) => Promise<string | Uint8Array>;
     callWxMethod: (method: string, ...args: unknown[]) => Promise<unknown>;
 }
 interface RuntimeConnectionInput {

package/dist/index.mjs

@@ -622,6 +622,8 @@ async function registerServerResources(server, options) {
   });
 }
 
+const DEFAULT_SCREENSHOT_TIMEOUT = 6e4;
+const SCREENSHOT_RETRY_DELAY = 500;
 z.object({
   projectPath: z.string().trim().min(1).describe("\u5C0F\u7A0B\u5E8F\u9879\u76EE\u8DEF\u5F84\uFF1B\u652F\u6301 workspaceRoot \u76F8\u5BF9\u8DEF\u5F84"),
   timeout: z.number().int().positive().optional(),
@@ -752,6 +754,29 @@ function buildUrl(pagePath, query) {
   }
   return `${normalizedPath}${normalizedPath.includes("?") ? "&" : "?"}${search}`;
 }
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function isCaptureScreenshotProtocolTimeout(error) {
+  return error instanceof Error && "code" in error && error.code === "DEVTOOLS_PROTOCOL_TIMEOUT" && "method" in error && error.method === "App.captureScreenshot";
+}
+async function captureMiniProgramScreenshot(miniProgram, timeout = DEFAULT_SCREENSHOT_TIMEOUT) {
+  let lastError;
+  for (let attempt = 1; attempt <= 2; attempt += 1) {
+    try {
+      const screenshot = await miniProgram.screenshot({ timeout });
+      return typeof screenshot === "string" ? Buffer.from(screenshot, "base64") : Buffer.from(screenshot);
+    } catch (error) {
+      lastError = error;
+      if (!isCaptureScreenshotProtocolTimeout(error) || attempt === 2) {
+        throw error;
+      }
+      await miniProgram.currentPage().catch(() => void 0);
+      await sleep(SCREENSHOT_RETRY_DELAY);
+    }
+  }
+  throw lastError;
+}
 function parseSelectorWithIndex(selector) {
   const match = selector.match(/^(.+?)\[index=(\d+)\]$/);
   if (!match?.[1] || !match[2]) {
@@ -1066,8 +1091,10 @@ function registerDevtoolsRuntimeTools(server, manager) {
   }, async ({ outputPath, ...connection }) => {
     try {
       const result = await manager.withMiniProgram(connection, async (miniProgram) => {
-        const screenshot = await miniProgram.screenshot();
-        const buffer = typeof screenshot === "string" ? Buffer.from(screenshot, "base64") : Buffer.from(screenshot);
+        const buffer = await captureMiniProgramScreenshot(
+          miniProgram,
+          connection.timeout ?? DEFAULT_SCREENSHOT_TIMEOUT
+        );
         if (outputPath) {
           const resolvedOutputPath = manager.resolveWorkspacePath(outputPath);
           await fs$1.mkdir(path.dirname(resolvedOutputPath), { recursive: true });
@@ -2181,8 +2208,10 @@ async function handlePageStack(req, res, context, manager) {
 async function handleCapture(req, res, manager) {
   const input = captureBodySchema.parse(await readJsonBody(req));
   const result = await manager.withMiniProgram(input, async (miniProgram) => {
-    const screenshot = await miniProgram.screenshot();
-    const buffer = typeof screenshot === "string" ? Buffer.from(screenshot, "base64") : Buffer.from(screenshot);
+    const buffer = await captureMiniProgramScreenshot(
+      miniProgram,
+      input.timeout ?? DEFAULT_SCREENSHOT_TIMEOUT
+    );
     if (input.outputPath) {
       const resolvedOutputPath = manager.resolveWorkspacePath(input.outputPath);
       await fs$1.mkdir(path.dirname(resolvedOutputPath), { recursive: true });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@weapp-vite/mcp",
   "type": "module",
-  "version": "1.4.4",
+  "version": "1.4.5",
   "description": "mcp",
   "author": "ice breaker <1324318532@qq.com>",
   "license": "ISC",
@@ -33,7 +33,7 @@
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.29.0",
     "zod": "^4.4.3",
-    "@weapp-vite/devtools-runtime": "0.4.4"
+    "@weapp-vite/devtools-runtime": "0.4.5"
   },
   "scripts": {
     "dev": "unbuild --stub",