mobile-mcp-ai 2.2.6__py3-none-any.whl → 2.5.3__py3-none-any.whl

mobile_mcp/mcp/mcp_server.py

@@ -1,1126 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-Mobile MCP Server - 统一版本（合并了基础工具和智能工具）
-
-架构说明：
-- 基础工具：不需要 AI 密钥，提供精确的元素操作（设备管理、应用管理、高级交互等）
-- 智能工具：需要 AI 密钥（可选），提供自然语言定位
-
-用户可以选择：
-1. 只用基础工具 → 不需要配置 AI
-2. 启用智能功能 → 需要配置 AI（创建 .env 文件）
-
-v2.2.0: 合并了两个 MCP Server，移除了 browser_mcp 依赖
-"""
-
-import asyncio
-import json
-import os
-import sys
-from pathlib import Path
-from typing import Optional
-
-# 添加项目路径
-mobile_mcp_dir = Path(__file__).parent.parent
-project_root = mobile_mcp_dir.parent.parent
-backend_dir = project_root / "backend"
-
-# 确保系统的 mcp 包优先导入（避免与 mobile_mcp.mcp 冲突）
-# 将 site-packages 路径插入到最前面
-import site
-for site_dir in site.getsitepackages():
-    if (Path(site_dir) / 'mcp').exists():
-        sys.path.insert(0, str(site_dir))
-        break
-
-sys.path.insert(0, str(project_root))
-sys.path.insert(0, str(backend_dir))
-
-# 检测运行模式：full(完整版) 或 simple(简化版)
-SERVER_MODE = os.getenv("MOBILE_MCP_MODE", "full").lower()
-
-# 导入系统的 mcp 包（现在应该能正确导入）
-from mcp.types import Tool, TextContent
-from mcp.server import Server
-from mcp.server.stdio import stdio_server
-
-from mobile_mcp.core.mobile_client import MobileClient
-from mobile_mcp.core.basic_tools import BasicMobileTools
-from mobile_mcp.core.smart_tools import SmartMobileTools
-from mobile_mcp.core.dynamic_config import DynamicConfig
-
-
-class MobileMCPServer:
-    """简化的 Mobile MCP Server"""
-    
-    def __init__(self):
-        """初始化 MCP Server"""
-        self.client: Optional[MobileClient] = None
-        self.basic_tools: Optional[BasicMobileTools] = None
-        self.smart_tools: Optional[SmartMobileTools] = None
-        self._initialized = False
-    
-    @staticmethod
-    def format_response(result) -> str:
-        """
-        统一格式化返回值为JSON字符串
-        
-        Args:
-            result: 可以是字典、列表或字符串
-            
-        Returns:
-            格式化后的字符串（字典和列表会转为JSON）
-        """
-        if isinstance(result, (dict, list)):
-            return json.dumps(result, ensure_ascii=False, indent=2)
-        return str(result)
-    
-    async def initialize(self):
-        """延迟初始化设备连接"""
-        if not self._initialized:
-            # 初始化移动客户端
-            self.client = MobileClient()
-            
-            # 初始化基础工具（总是可用）
-            self.basic_tools = BasicMobileTools(self.client)
-            
-            # 初始化智能工具（检查 AI 可用性）
-            self.smart_tools = SmartMobileTools(self.client)
-            
-            ai_status = self.smart_tools.get_ai_status()
-            print(f"\n{ai_status['message']}\n", file=sys.stderr)
-            
-            self._initialized = True
-    
-    def get_tools(self):
-        """注册 MCP 工具"""
-        tools = []
-        
-        # ==================== 基础工具（不需要 AI）====================
-        
-        tools.extend([
-            Tool(
-                name="mobile_list_elements",
-                description="📋 列出页面所有可交互元素（不需要 AI）。返回 resource_id, text, bounds 等信息，供后续精确操作使用。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_click_by_id",
-                description="👆 通过 resource-id 点击元素（不需要 AI）。精确可靠的点击方式。先用 mobile_list_elements 查找元素 ID。\n\n"
-                           "✅ 点击成功后会自动等待 0.3 秒，无需重复点击！\n"
-                           "💡 提示：如果已经用 mobile_click_by_text 点击成功了，就不需要再用 ID 点击同一个元素。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "resource_id": {
-                            "type": "string",
-                            "description": "元素的 resource-id，如 'com.app:id/search_btn'"
-                        }
-                    },
-                    "required": ["resource_id"]
-                }
-            ),
-            Tool(
-                name="mobile_click_by_text",
-                description="👆 通过文本内容点击元素（不需要 AI）。适合文本完全匹配的场景。\n\n"
-                           "✅ 点击成功后会自动等待 0.3 秒，无需重复点击！\n"
-                           "⚠️ 如果需要确认是否成功，可以用 mobile_list_elements 查看页面变化，但不要重复点击同一个按钮。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "text": {
-                            "type": "string",
-                            "description": "元素的文本内容（精确匹配），如 '登录'"
-                        }
-                    },
-                    "required": ["text"]
-                }
-            ),
-            Tool(
-                name="mobile_click_at_coords",
-                description="👆 点击指定坐标（不需要 AI）。可以从 mobile_list_elements 获取的 bounds 计算坐标。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "x": {
-                            "type": "number",
-                            "description": "X 坐标（像素）"
-                        },
-                        "y": {
-                            "type": "number",
-                            "description": "Y 坐标（像素）"
-                        }
-                    },
-                    "required": ["x", "y"]
-                }
-            ),
-            Tool(
-                name="mobile_input_text_by_id",
-                description="⌨️ 通过 resource-id 在输入框输入文本（不需要 AI）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "resource_id": {
-                            "type": "string",
-                            "description": "输入框的 resource-id"
-                        },
-                        "text": {
-                            "type": "string",
-                            "description": "要输入的文本"
-                        }
-                    },
-                    "required": ["resource_id", "text"]
-                }
-            ),
-            Tool(
-                name="mobile_find_elements_by_class",
-                description="🔍 按类名查找元素（不需要 AI）。如查找所有输入框: 'android.widget.EditText'",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "class_name": {
-                            "type": "string",
-                            "description": "类名，如 'android.widget.EditText'"
-                        }
-                    },
-                    "required": ["class_name"]
-                }
-            ),
-            Tool(
-                name="mobile_wait_for_element",
-                description="⏳ 等待元素出现（不需要 AI）。用于等待页面加载完成。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "resource_id": {
-                            "type": "string",
-                            "description": "元素的 resource-id"
-                        },
-                        "timeout": {
-                            "type": "number",
-                            "description": "超时时间（秒），默认 10秒",
-                            "default": 10
-                        }
-                    },
-                    "required": ["resource_id"]
-                }
-            ),
-        ])
-        
-        # ==================== 完整版独有工具 ====================
-        if SERVER_MODE == "full":
-            tools.append(
-                Tool(
-                    name="mobile_wait",
-                    description="⏰ 通用等待工具 - AI 可根据场景灵活控制等待（不需要 AI）。\n\n"
-                               "🔥 强烈建议在以下场景使用：\n"
-                               "1. App 启动后：mobile_launch_app() → mobile_wait(seconds=2-3)\n"
-                               "2. 等待广告：mobile_wait(seconds=3-5)\n"
-                               "3. 等待搜索结果：mobile_wait(wait_for_text='搜索结果')\n"
-                               "4. 等待页面加载：mobile_wait(wait_for_id='com.app:id/home')\n\n"
-                               "⚠️ 不要立即操作刚启动的 App，先等待加载完成！",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "seconds": {
-                                "type": "number",
-                                "description": "固定等待时间（秒）。适用于等待广告、动画等"
-                            },
-                            "wait_for_text": {
-                                "type": "string",
-                                "description": "等待指定文本出现。如 '首页'、'搜索结果'"
-                            },
-                            "wait_for_id": {
-                                "type": "string",
-                                "description": "等待指定元素ID出现。如 'com.app:id/home'"
-                            },
-                            "timeout": {
-                                "type": "number",
-                                "description": "等待元素的超时时间（秒），默认 10秒",
-                                "default": 10
-                            }
-                        },
-                        "required": []
-                    }
-                )
-            )
-        
-        tools.extend([
-            Tool(
-                name="mobile_take_screenshot",
-                description="📸 截取屏幕截图（不需要 AI）。用于 Cursor AI 视觉识别、调试或记录测试过程。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "description": {
-                            "type": "string",
-                            "description": "截图描述（可选），用于生成文件名"
-                        }
-                    },
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_take_screenshot_region",
-                description="📸 截取屏幕指定区域（不需要 AI）。用于局部截图和分析。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "x1": {
-                            "type": "number",
-                            "description": "左上角X坐标"
-                        },
-                        "y1": {
-                            "type": "number",
-                            "description": "左上角Y坐标"
-                        },
-                        "x2": {
-                            "type": "number",
-                            "description": "右下角X坐标"
-                        },
-                        "y2": {
-                            "type": "number",
-                            "description": "右下角Y坐标"
-                        },
-                        "description": {
-                            "type": "string",
-                            "description": "截图描述（可选）"
-                        }
-                    },
-                    "required": ["x1", "y1", "x2", "y2"]
-                }
-            ),
-            # ==================== 设备管理工具 ====================
-            Tool(
-                name="mobile_list_devices",
-                description="📱 列出所有已连接的Android设备（不需要 AI）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_get_screen_size",
-                description="📐 获取设备屏幕尺寸（不需要 AI）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_get_orientation",
-                description="🔄 获取当前屏幕方向（portrait/landscape）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_set_orientation",
-                description="🔄 设置屏幕方向。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "orientation": {
-                            "type": "string",
-                            "enum": ["portrait", "landscape"],
-                            "description": "屏幕方向：portrait(竖屏) 或 landscape(横屏)"
-                        }
-                    },
-                    "required": ["orientation"]
-                }
-            ),
-            Tool(
-                name="mobile_check_connection",
-                description="🔌 检查设备连接状态。返回设备信息和连接状态。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_reconnect_device",
-                description="🔄 重新连接设备。当设备连接断开时使用。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            # ==================== 应用管理工具 ====================
-            Tool(
-                name="mobile_list_apps",
-                description="📦 列出设备上已安装的应用（不需要 AI）。可按关键词过滤。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "filter": {
-                            "type": "string",
-                            "description": "过滤关键词（可选），如包名的一部分"
-                        }
-                    },
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_install_app",
-                description="📲 安装APK文件（不需要 AI）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "apk_path": {
-                            "type": "string",
-                            "description": "APK文件路径"
-                        }
-                    },
-                    "required": ["apk_path"]
-                }
-            ),
-            Tool(
-                name="mobile_uninstall_app",
-                description="🗑️ 卸载应用（不需要 AI）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "package_name": {
-                            "type": "string",
-                            "description": "应用包名，如 'com.example.app'"
-                        }
-                    },
-                    "required": ["package_name"]
-                }
-            ),
-            Tool(
-                name="mobile_terminate_app",
-                description="⏹️ 终止应用（强制停止）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "package_name": {
-                            "type": "string",
-                            "description": "应用包名，如 'com.example.app'"
-                        }
-                    },
-                    "required": ["package_name"]
-                }
-            ),
-            Tool(
-                name="mobile_get_current_package",
-                description="📍 获取当前前台应用的包名。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            # ==================== 高级交互工具 ====================
-            Tool(
-                name="mobile_double_click",
-                description="👆👆 双击屏幕上的元素（不需要 AI）。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "x": {
-                            "type": "number",
-                            "description": "X坐标"
-                        },
-                        "y": {
-                            "type": "number",
-                            "description": "Y坐标"
-                        }
-                    },
-                    "required": ["x", "y"]
-                }
-            ),
-            Tool(
-                name="mobile_long_press",
-                description="👆⏱️ 长按屏幕上的元素。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "x": {
-                            "type": "number",
-                            "description": "X坐标"
-                        },
-                        "y": {
-                            "type": "number",
-                            "description": "Y坐标"
-                        },
-                        "duration": {
-                            "type": "number",
-                            "default": 1.0,
-                            "description": "长按时长（秒），默认1秒"
-                        }
-                    },
-                    "required": ["x", "y"]
-                }
-            ),
-            Tool(
-                name="mobile_open_url",
-                description="🌐 在设备浏览器中打开URL。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "url": {
-                            "type": "string",
-                            "description": "要打开的URL，如 'https://example.com'"
-                        }
-                    },
-                    "required": ["url"]
-                }
-            ),
-            Tool(
-                name="mobile_assert_text",
-                description="✅ 断言页面中是否包含指定文本。用于验证操作结果。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "text": {
-                            "type": "string",
-                            "description": "要检查的文本内容"
-                        }
-                    },
-                    "required": ["text"]
-                }
-            ),
-        ])
-        
-        # ==================== 智能工具（需要 AI，可选）====================
-        
-        tools.extend([
-            Tool(
-                name="mobile_smart_click",
-                description="🤖 智能定位并点击（需要 AI 密钥，可选功能）。使用自然语言描述元素，如'右上角的设置按钮'。\n\n"
-                           "⚠️ 如未配置 AI，请使用基础工具：mobile_list_elements + mobile_click_by_id",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "description": {
-                            "type": "string",
-                            "description": "元素的自然语言描述，如 '顶部搜索框'、'登录按钮'"
-                        }
-                    },
-                    "required": ["description"]
-                }
-            ),
-            Tool(
-                name="mobile_smart_input",
-                description="🤖 智能定位输入框并输入（需要 AI 密钥，可选功能）。使用自然语言描述输入框。\n\n"
-                           "⚠️ 如未配置 AI，请使用：mobile_input_text_by_id",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "description": {
-                            "type": "string",
-                            "description": "输入框的自然语言描述，如 '用户名输入框'"
-                        },
-                        "text": {
-                            "type": "string",
-                            "description": "要输入的文本"
-                        }
-                    },
-                    "required": ["description", "text"]
-                }
-            ),
-            Tool(
-                name="mobile_analyze_screenshot",
-                description="🤖 使用 AI 分析截图并返回坐标（需要 AI 密钥，可选功能）。用于 Cursor AI 无法直接识别的复杂场景。\n\n"
-                           "使用流程：\n"
-                           "1. 先用 mobile_take_screenshot 截图\n"
-                           "2. 调用此工具分析截图\n"
-                           "3. 根据返回的坐标使用 mobile_click_at_coords 点击\n\n"
-                           "⚠️ 需要配置支持视觉识别的 AI（GPT-4V、Claude 3、Qwen-VL）",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "screenshot_path": {
-                            "type": "string",
-                            "description": "截图文件路径"
-                        },
-                        "description": {
-                            "type": "string",
-                            "description": "要查找的元素描述"
-                        }
-                    },
-                    "required": ["screenshot_path", "description"]
-                }
-            ),
-            Tool(
-                name="mobile_get_ai_status",
-                description="ℹ️ 获取 AI 功能状态。检查是否已配置 AI 密钥，智能工具是否可用。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-        ])
-        
-        # ==================== 完整版独有：智能测试工具 ====================
-        if SERVER_MODE == "full":
-            tools.extend([
-                Tool(
-                    name="mobile_execute_test_case",
-                    description="🤖 智能执行测试用例（需要 AI）。AI 会自动规划、执行、验证每一步操作，遇到问题自动分析解决。",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "test_description": {
-                                "type": "string",
-                                "description": "自然语言描述的测试用例，如：'打开 com.im30.mind\\n点击底部云文档\\n点击我的空间'"
-                            }
-                        },
-                        "required": ["test_description"]
-                    }
-                ),
-                Tool(
-                    name="mobile_generate_test_script",
-                    description="📝 基于操作历史生成 pytest 格式的测试脚本（不需要 AI）。\n\n"
-                               "🔥 重要功能：\n"
-                               "1. 自动记录所有 mobile_click、mobile_input 等操作\n"
-                               "2. 一键生成可执行的 pytest 测试脚本\n"
-                               "3. 支持 pytest 批量执行和 allure 报告\n\n"
-                               "使用场景：\n"
-                               "- 手动测试完成后，生成自动化脚本\n"
-                               "- 快速创建回归测试用例\n"
-                               "- 录制复杂的操作流程\n\n"
-                               "💡 提示：执行完一系列操作后，调用此工具即可生成脚本！",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {
-                            "test_name": {
-                                "type": "string",
-                                "description": "测试用例名称，如 '登录测试'"
-                            },
-                            "package_name": {
-                                "type": "string",
-                                "description": "App 包名，如 'com.im30.mind'"
-                            },
-                            "filename": {
-                                "type": "string",
-                                "description": "生成的脚本文件名（不含 .py 后缀），如 'test_login'"
-                            },
-                            "output_dir": {
-                                "type": "string",
-                                "description": "输出目录路径（可选），默认为 tests 子目录"
-                            }
-                        },
-                        "required": ["test_name", "package_name", "filename"]
-                    }
-                ),
-            ])
-        
-        # ==================== 通用工具 ====================
-        
-        tools.extend([
-            Tool(
-                name="mobile_snapshot",
-                description="📸 获取页面快照。查看当前页面结构和元素信息。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_launch_app",
-                description="🚀 启动应用\n\n"
-                           "⚠️ 重要提示：\n"
-                           "1. 启动后需要等待 App 加载完成\n"
-                           "2. 建议启动后主动调用 mobile_wait(seconds=2-3) 等待界面稳定\n"
-                           "3. 或使用 mobile_wait(wait_for_text='首页关键文本') 等待特定元素\n"
-                           "4. 如果有广告/弹窗，可能需要等待 3-5 秒\n\n"
-                           "示例：\n"
-                           "mobile_launch_app('com.example.app')\n"
-                           "mobile_wait(seconds=3)  # 等待 App 加载\n"
-                           "mobile_click_by_text('开始使用')",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "package_name": {
-                            "type": "string",
-                            "description": "应用包名"
-                        }
-                    },
-                    "required": ["package_name"]
-                }
-            ),
-            Tool(
-                name="mobile_press_key",
-                description="⌨️ 按键操作（home, back, enter 等）",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "key": {
-                            "type": "string",
-                            "description": "按键名称：home, back, enter, search"
-                        }
-                    },
-                    "required": ["key"]
-                }
-            ),
-            Tool(
-                name="mobile_swipe",
-                description="👆 滑动屏幕",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "direction": {
-                            "type": "string",
-                            "enum": ["up", "down", "left", "right"],
-                            "description": "滑动方向"
-                        }
-                    },
-                    "required": ["direction"]
-                }
-            ),
-        ])
-        
-        # ==================== 完整版独有：操作历史管理工具 ====================
-        if SERVER_MODE == "full":
-            tools.extend([
-                Tool(
-                    name="mobile_get_operation_history",
-                description="📜 获取操作历史记录。查看之前执行的所有操作。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "limit": {
-                            "type": "number",
-                            "description": "返回最近的N条记录，不指定则返回全部"
-                        }
-                    },
-                    "required": []
-                }
-            ),
-            Tool(
-                name="mobile_clear_operation_history",
-                description="🗑️ 清空操作历史记录。清空后将无法生成测试脚本。",
-                inputSchema={
-                    "type": "object",
-                    "properties": {},
-                    "required": []
-                }
-            ),
-            # ==================== 动态配置工具 ====================
-            Tool(
-                name="mobile_configure",
-                description="⚙️ 动态配置自动化行为 - AI 可根据 App 特性和测试场景优化参数（不需要 AI）。\n\n"
-                           "适用场景：\n"
-                           "1. 游戏App：增加等待时间、调整页面变化阈值、使用横屏\n"
-                           "2. 电商App：启用广告自动关闭、使用竖屏\n"
-                           "3. 回归测试：禁用验证、减少等待时间、不截图\n"
-                           "4. 慢速设备：增加所有超时时间\n\n"
-                           "💡 提示：不调用此工具则使用默认配置（与当前行为完全一致）",
-                inputSchema={
-                    "type": "object",
-                    "properties": {
-                        "wait_strategy": {
-                            "type": "object",
-                            "description": "等待时间策略",
-                            "properties": {
-                                "click_wait": {
-                                    "type": "number",
-                                    "description": "点击后等待时间（秒），默认0.3"
-                                },
-                                "input_wait": {
-                                    "type": "number",
-                                    "description": "输入后等待时间（秒），默认0.3"
-                                },
-                                "page_stable_wait": {
-                                    "type": "number",
-                                    "description": "页面稳定等待时间（秒），默认0.8"
-                                },
-                                "element_timeout": {
-                                    "type": "number",
-                                    "description": "元素等待超时（秒），默认10"
-                                },
-                                "page_change_timeout": {
-                                    "type": "number",
-                                    "description": "页面变化检测超时（秒），默认2"
-                                }
-                            }
-                        },
-                        "verify_strategy": {
-                            "type": "object",
-                            "description": "验证策略",
-                            "properties": {
-                                "verify_clicks": {
-                                    "type": "boolean",
-                                    "description": "是否验证点击操作，默认true"
-                                },
-                                "verify_inputs": {
-                                    "type": "boolean",
-                                    "description": "是否验证输入操作，默认false"
-                                },
-                                "verify_keys": {
-                                    "type": "boolean",
-                                    "description": "是否验证按键操作，默认true"
-                                }
-                            }
-                        },
-                        "page_change_threshold": {
-                            "type": "number",
-                            "description": "页面变化阈值（0-1），游戏App建议0.1-0.15，工具App建议0.05，默认0.05"
-                        },
-                        "screen_orientation": {
-                            "type": "string",
-                            "enum": ["portrait", "landscape", "auto"],
-                            "description": "屏幕方向：portrait=竖屏, landscape=横屏, auto=跟随App，默认portrait"
-                        },
-                        "ad_handling": {
-                            "type": "object",
-                            "description": "广告/弹窗处理策略",
-                            "properties": {
-                                "auto_close": {
-                                    "type": "boolean",
-                                    "description": "是否自动关闭广告，默认true"
-                                },
-                                "wait_before_close": {
-                                    "type": "number",
-                                    "description": "点击关闭前等待（秒），默认0.3"
-                                },
-                                "max_close_buttons": {
-                                    "type": "number",
-                                    "description": "最多点击几个关闭按钮，默认1"
-                                }
-                            }
-                        },
-                        "screenshot_strategy": {
-                            "type": "string",
-                            "enum": ["always", "on_failure", "never", "smart"],
-                            "description": "截图策略：always=总是, on_failure=失败时, never=从不, smart=智能，默认smart"
-                        },
-                        "retry_strategy": {
-                            "type": "object",
-                            "description": "重试策略",
-                            "properties": {
-                                "max_retries": {
-                                    "type": "number",
-                                    "description": "最大重试次数，默认3"
-                                },
-                                "retry_delay": {
-                                    "type": "number",
-                                    "description": "重试间隔（秒），默认1.0"
-                                }
-                            }
-                        },
-                        "reset": {
-                            "type": "boolean",
-                            "description": "是否重置为默认配置，默认false"
-                        }
-                    },
-                    "required": []
-                }
-            ),
-                Tool(
-                    name="mobile_get_config",
-                    description="📋 获取当前动态配置。查看当前所有配置值。",
-                    inputSchema={
-                        "type": "object",
-                        "properties": {},
-                        "required": []
-                    }
-                ),
-            ])
-        
-        return tools
-    
-    async def handle_tool_call(self, name: str, arguments: dict):
-        """处理工具调用"""
-        await self.initialize()
-        
-        try:
-            # ==================== 基础工具 ====================
-            if name == "mobile_list_elements":
-                result = self.basic_tools.list_elements()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_click_by_id":
-                result = self.basic_tools.click_by_id(arguments["resource_id"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_click_by_text":
-                result = self.basic_tools.click_by_text(arguments["text"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_click_at_coords":
-                result = self.basic_tools.click_at_coords(arguments["x"], arguments["y"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_input_text_by_id":
-                result = self.basic_tools.input_text_by_id(
-                    arguments["resource_id"],
-                    arguments["text"]
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_find_elements_by_class":
-                result = self.basic_tools.find_elements_by_class(arguments["class_name"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_wait_for_element":
-                timeout = arguments.get("timeout", 10)
-                result = self.basic_tools.wait_for_element(arguments["resource_id"], timeout)
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_take_screenshot":
-                description = arguments.get("description", "")
-                result = self.basic_tools.take_screenshot(description)
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_take_screenshot_region":
-                description = arguments.get("description", "")
-                result = self.basic_tools.take_screenshot_region(
-                    arguments["x1"], arguments["y1"],
-                    arguments["x2"], arguments["y2"],
-                    description
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            # ==================== 设备管理工具 ====================
-            elif name == "mobile_list_devices":
-                result = self.basic_tools.list_devices()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_get_screen_size":
-                result = self.basic_tools.get_screen_size()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_get_orientation":
-                result = self.basic_tools.get_orientation()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_set_orientation":
-                result = self.basic_tools.set_orientation(arguments["orientation"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_check_connection":
-                result = self.basic_tools.check_connection()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_reconnect_device":
-                result = self.basic_tools.reconnect_device()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            # ==================== 应用管理工具 ====================
-            elif name == "mobile_list_apps":
-                filter_keyword = arguments.get("filter", "")
-                result = self.basic_tools.list_apps(filter_keyword)
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_install_app":
-                result = self.basic_tools.install_app(arguments["apk_path"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_uninstall_app":
-                result = self.basic_tools.uninstall_app(arguments["package_name"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_terminate_app":
-                result = self.basic_tools.terminate_app(arguments["package_name"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_get_current_package":
-                result = self.basic_tools.get_current_package()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            # ==================== 高级交互工具 ====================
-            elif name == "mobile_double_click":
-                result = self.basic_tools.double_click_at_coords(
-                    int(arguments["x"]), int(arguments["y"])
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_long_press":
-                duration = arguments.get("duration", 1.0)
-                result = self.basic_tools.long_press_at_coords(
-                    int(arguments["x"]), int(arguments["y"]), duration
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_open_url":
-                result = self.basic_tools.open_url(arguments["url"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_assert_text":
-                result = self.basic_tools.assert_text(arguments["text"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            # ==================== 智能工具 ====================
-            elif name == "mobile_smart_click":
-                result = await self.smart_tools.smart_click(arguments["description"])
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_smart_input":
-                result = await self.smart_tools.smart_input(
-                    arguments["description"],
-                    arguments["text"]
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_analyze_screenshot":
-                result = await self.smart_tools.analyze_screenshot_with_ai(
-                    arguments["screenshot_path"],
-                    arguments["description"]
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_get_ai_status":
-                result = self.smart_tools.get_ai_status()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            # ==================== 通用工具 ====================
-            elif name == "mobile_snapshot":
-                snapshot = await self.client.snapshot()
-                return [TextContent(type="text", text=snapshot)]
-            
-            elif name == "mobile_launch_app":
-                await self.client.launch_app(arguments["package_name"])
-                return [TextContent(type="text", text=f"✅ 已启动: {arguments['package_name']}")]
-            
-            elif name == "mobile_press_key":
-                await self.client.press_key(arguments["key"])
-                return [TextContent(type="text", text=f"✅ 已按键: {arguments['key']}")]
-            
-            elif name == "mobile_swipe":
-                await self.client.swipe(arguments["direction"])
-                return [TextContent(type="text", text=f"✅ 已滑动: {arguments['direction']}")]
-            
-            # ==================== 完整版独有工具处理 ====================
-            elif name == "mobile_wait":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                seconds = arguments.get("seconds")
-                wait_for_text = arguments.get("wait_for_text")
-                wait_for_id = arguments.get("wait_for_id")
-                timeout = arguments.get("timeout", 10)
-                result = self.basic_tools.wait(
-                    seconds=seconds,
-                    wait_for_text=wait_for_text,
-                    wait_for_id=wait_for_id,
-                    timeout=timeout
-                )
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_get_operation_history":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                limit = arguments.get("limit")
-                result = self.basic_tools.get_operation_history(limit)
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_clear_operation_history":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                result = self.basic_tools.clear_operation_history()
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_configure":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                if arguments.get("reset", False):
-                    result = DynamicConfig.reset()
-                else:
-                    result = DynamicConfig.update(arguments)
-                return [TextContent(type="text", text=self.format_response(result))]
-            
-            elif name == "mobile_get_config":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                current_config = DynamicConfig.get_current()
-                config_str = json.dumps(current_config, indent=2, ensure_ascii=False)
-                return [TextContent(type="text", text=f"📋 当前配置：\n{config_str}")]
-            
-            elif name == "mobile_execute_test_case":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                try:
-                    from mobile_mcp.core.ai.smart_test_executor import SmartTestExecutor
-                    executor = SmartTestExecutor(self.client)
-                    result = await executor.execute_test_case(arguments["test_description"])
-                    return [TextContent(type="text", text=self.format_response(result))]
-                except ImportError:
-                    return [TextContent(type="text", text="❌ 智能测试执行器模块未安装")]
-                except Exception as e:
-                    return [TextContent(type="text", text=f"❌ 测试执行失败: {str(e)}")]
-            
-            elif name == "mobile_generate_test_script":
-                if SERVER_MODE != "full":
-                    return [TextContent(type="text", text=f"❌ 此工具仅在完整版可用，当前为简化版")]
-                try:
-                    from mobile_mcp.core.ai.test_generator_from_history import TestGeneratorFromHistory
-                    from mobile_mcp.core.utils.operation_history_manager import OperationHistoryManager
-                    
-                    history_manager = OperationHistoryManager()
-                    operation_history = history_manager.get_all()
-                    
-                    if not operation_history:
-                        return [TextContent(type="text", text="❌ 没有操作历史，请先执行一些操作")]
-                    
-                    generator = TestGeneratorFromHistory()
-                    script = generator.generate_from_history(
-                        test_name=arguments["test_name"],
-                        package_name=arguments["package_name"],
-                        operation_history=operation_history
-                    )
-                    
-                    output_dir = arguments.get("output_dir", "tests")
-                    filename = arguments["filename"]
-                    if not filename.endswith('.py'):
-                        filename = f"{filename}.py"
-                    
-                    from pathlib import Path
-                    output_path = Path(output_dir) / filename
-                    output_path.parent.mkdir(parents=True, exist_ok=True)
-                    generator.save(str(output_path), script)
-                    
-                    return [TextContent(type="text", text=f"✅ 测试脚本已生成: {output_path}\n\n{script[:500]}...")]
-                except ImportError as e:
-                    return [TextContent(type="text", text=f"❌ 模块导入失败: {str(e)}")]
-                except Exception as e:
-                    return [TextContent(type="text", text=f"❌ 脚本生成失败: {str(e)}")]
-            
-            else:
-                return [TextContent(type="text", text=f"❌ 未知工具: {name}")]
-        
-        except Exception as e:
-            error_msg = str(e)
-            return [TextContent(type="text", text=f"❌ 执行失败: {error_msg}")]
-
-
-async def main():
-    """启动 MCP Server"""
-    server = MobileMCPServer()
-    mcp_server = Server("mobile-mcp")
-    
-    @mcp_server.list_tools()
-    async def list_tools():
-        return server.get_tools()
-    
-    @mcp_server.call_tool()
-    async def call_tool(name: str, arguments: dict):
-        return await server.handle_tool_call(name, arguments)
-    
-    mode_name = "完整版 (39工具)" if SERVER_MODE == "full" else "简化版 (32工具)"
-    print(f"🚀 Mobile MCP Server v2.2.6 启动中... [{mode_name}]", file=sys.stderr)
-    print(f"📋 运行模式: {SERVER_MODE.upper()}", file=sys.stderr)
-    if SERVER_MODE == "simple":
-        print("💡 提示: 使用完整版可获得更多功能（操作历史、动态配置等）", file=sys.stderr)
-    
-    async with stdio_server() as (read_stream, write_stream):
-        await mcp_server.run(read_stream, write_stream, mcp_server.create_initialization_options())
-
-
-if __name__ == "__main__":
-    asyncio.run(main())
-