weixin-devtools-mcp 0.0.1

package/build/tools/page.js

@@ -0,0 +1,150 @@
+/**
+ * 页面查询和等待工具
+ * 提供类似浏览器的$选择器和waitFor等待功能
+ */
+import { z } from 'zod';
+import { defineTool } from './ToolDefinition.js';
+import { queryElements, waitForCondition } from '../tools.js';
+/**
+ * $ 选择器工具 - 通过CSS选择器查找页面元素
+ */
+export const querySelectorTool = defineTool({
+    name: '$',
+    description: '通过CSS选择器查找页面元素，返回匹配元素的详细信息',
+    schema: z.object({
+        selector: z.string().min(1, '选择器不能为空').describe('CSS选择器，如：view.container、#myId、.myClass、text=按钮'),
+    }),
+    annotations: {
+        audience: ['developers'],
+    },
+    handler: async (request, response, context) => {
+        const { selector } = request.params;
+        // 验证选择器
+        if (!selector || typeof selector !== 'string' || selector.trim() === '') {
+            throw new Error('选择器不能为空');
+        }
+        if (!context.currentPage) {
+            throw new Error('请先获取当前页面');
+        }
+        try {
+            const options = { selector };
+            const results = await queryElements(context.currentPage, context.elementMap, options);
+            if (results.length === 0) {
+                response.appendResponseLine(`未找到匹配选择器 "${selector}" 的元素`);
+                return;
+            }
+            response.appendResponseLine(`找到 ${results.length} 个匹配元素:`);
+            response.appendResponseLine('');
+            for (let i = 0; i < results.length; i++) {
+                const element = results[i];
+                response.appendResponseLine(`[${i + 1}] ${element.tagName} (uid: ${element.uid})`);
+                if (element.text) {
+                    response.appendResponseLine(`    文本: ${element.text}`);
+                }
+                if (element.attributes) {
+                    const attrs = Object.entries(element.attributes)
+                        .map(([key, value]) => `${key}="${value}"`)
+                        .join(' ');
+                    response.appendResponseLine(`    属性: ${attrs}`);
+                }
+                if (element.position) {
+                    const { left, top, width, height } = element.position;
+                    response.appendResponseLine(`    位置: (${left}, ${top}) 大小: ${width}x${height}`);
+                }
+                response.appendResponseLine('');
+            }
+            // 查询可能会发现新元素，包含快照信息
+            response.setIncludeSnapshot(true);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            response.appendResponseLine(`查询元素失败: ${errorMessage}`);
+            throw error;
+        }
+    },
+});
+/**
+ * waitFor 等待工具 - 等待条件满足
+ */
+export const waitForTool = defineTool({
+    name: 'waitFor',
+    description: '等待条件满足，支持等待元素出现、消失、文本匹配等',
+    schema: z.object({
+        // 支持三种模式:
+        // 1. 时间等待: { delay: 1000 }
+        // 2. 选择器等待: { selector: ".button" }
+        // 3. 复杂条件: { selector: ".button", text: "提交", timeout: 5000 }
+        delay: z.number().optional().describe('等待指定毫秒数（时间等待模式）'),
+        selector: z.string().optional().describe('等待元素选择器（选择器等待模式）'),
+        timeout: z.number().optional().default(5000).describe('超时时间(毫秒)，默认5000ms'),
+        text: z.string().optional().describe('等待元素包含指定文本'),
+        visible: z.boolean().optional().describe('等待元素可见状态，true为可见，false为隐藏'),
+        disappear: z.boolean().optional().default(false).describe('等待元素消失，默认false'),
+    }).refine((data) => data.delay !== undefined || data.selector !== undefined, { message: '必须提供 delay（时间等待）或 selector（选择器等待）' }),
+    annotations: {
+        audience: ['developers'],
+    },
+    handler: async (request, response, context) => {
+        const options = request.params;
+        if (!context.currentPage) {
+            throw new Error('请先获取当前页面');
+        }
+        try {
+            const startTime = Date.now();
+            // 构建等待描述信息和实际等待参数
+            let waitDescription = '';
+            let waitParam;
+            if (options.delay !== undefined) {
+                // 时间等待模式
+                waitDescription = `等待 ${options.delay}ms`;
+                waitParam = options.delay;
+            }
+            else if (options.selector) {
+                // 选择器等待模式
+                const parts = [];
+                parts.push(`选择器 "${options.selector}"`);
+                if (options.disappear)
+                    parts.push('消失');
+                else
+                    parts.push('出现');
+                if (options.text)
+                    parts.push(`包含文本 "${options.text}"`);
+                if (options.visible !== undefined) {
+                    parts.push(options.visible ? '可见' : '隐藏');
+                }
+                waitDescription = `等待 ${parts.join(' 且 ')}`;
+                if (options.timeout) {
+                    waitDescription += ` (超时: ${options.timeout}ms)`;
+                }
+                // 构建 WaitForOptions 参数
+                waitParam = {
+                    selector: options.selector,
+                    timeout: options.timeout,
+                    ...(options.text && { text: options.text }),
+                    ...(options.visible !== undefined && { visible: options.visible }),
+                    ...(options.disappear !== undefined && { disappear: options.disappear }),
+                };
+            }
+            else {
+                throw new Error('必须提供 delay 或 selector 参数');
+            }
+            response.appendResponseLine(`开始 ${waitDescription}...`);
+            const result = await waitForCondition(context.currentPage, waitParam);
+            const endTime = Date.now();
+            const duration = endTime - startTime;
+            if (result) {
+                response.appendResponseLine(`等待成功，耗时 ${duration}ms`);
+                // 等待完成后，页面可能发生变化
+                response.setIncludeSnapshot(true);
+            }
+            else {
+                response.appendResponseLine(`等待失败，耗时 ${duration}ms`);
+            }
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            response.appendResponseLine(`等待失败: ${errorMessage}`);
+            throw error;
+        }
+    },
+});

package/build/tools/screenshot.js

@@ -0,0 +1,47 @@
+/**
+ * 截图工具
+ * 负责页面截图功能，支持保存到文件或返回base64数据
+ */
+import { z } from 'zod';
+import { defineTool } from './ToolDefinition.js';
+import { takeScreenshot } from '../tools.js';
+/**
+ * 页面截图
+ */
+export const screenshotTool = defineTool({
+    name: 'screenshot',
+    description: '对当前页面截图，支持返回base64数据或保存到文件',
+    schema: z.object({
+        path: z.string().optional().describe('图片保存路径（可选），如果不提供则返回base64编码的图片数据'),
+    }),
+    annotations: {
+        audience: ['developers'],
+    },
+    handler: async (request, response, context) => {
+        if (!context.miniProgram) {
+            throw new Error('请先连接到微信开发者工具');
+        }
+        const { path } = request.params;
+        try {
+            const options = {};
+            if (path)
+                options.path = path;
+            const result = await takeScreenshot(context.miniProgram, options);
+            if (path) {
+                response.appendResponseLine(`截图已保存到: ${path}`);
+            }
+            else if (result) {
+                response.appendResponseLine(`截图获取成功`);
+                response.appendResponseLine(`Base64数据长度: ${result.length} 字符`);
+                response.appendResponseLine(`格式: ${result.startsWith('data:image') ? 'data URL' : 'base64'}`);
+                // 附加图片到响应
+                response.attachImage(result, 'image/png');
+            }
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            response.appendResponseLine(`截图失败: ${errorMessage}`);
+            throw error;
+        }
+    },
+});

package/build/tools/snapshot.js

@@ -0,0 +1,45 @@
+/**
+ * 页面快照工具
+ * 负责获取页面元素快照和UID映射
+ */
+import { z } from 'zod';
+import { defineTool } from './ToolDefinition.js';
+import { getPageSnapshot } from '../tools.js';
+/**
+ * 获取页面快照
+ */
+export const getPageSnapshotTool = defineTool({
+    name: 'get_page_snapshot',
+    description: '获取当前页面的元素快照，包含所有元素的uid信息',
+    schema: z.object({}),
+    annotations: {
+        audience: ['developers'],
+    },
+    handler: async (request, response, context) => {
+        if (!context.currentPage) {
+            throw new Error('请先获取当前页面');
+        }
+        try {
+            // 清空之前的元素映射
+            context.elementMap.clear();
+            // 获取页面快照
+            const { snapshot, elementMap } = await getPageSnapshot(context.currentPage);
+            // 更新上下文中的元素映射
+            elementMap.forEach((value, key) => {
+                context.elementMap.set(key, value);
+            });
+            response.appendResponseLine(`页面快照获取成功`);
+            response.appendResponseLine(`页面路径: ${snapshot.path}`);
+            response.appendResponseLine(`元素数量: ${snapshot.elements.length}`);
+            response.appendResponseLine('');
+            response.appendResponseLine(JSON.stringify(snapshot, null, 2));
+            // 设置包含快照信息
+            response.setIncludeSnapshot(true);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            response.appendResponseLine(`获取页面快照失败: ${errorMessage}`);
+            throw error;
+        }
+    },
+});