""" 连接守护技能 - 确保设备与服务器始终保持连接 职责： 1. 自动处理系统弹窗（USB调试授权、权限请求、系统更新提示等） 2. 监控 ADB / WiFi / 网络连通性 3. 检测并恢复 uiautomator2 服务 4. 检测屏幕状态，必要时亮屏 5. 与 Hawk 网络层协同做断网恢复 这是一个「常驻」技能，Agent 启动后以后台循环方式持续运行。 """ import logging import time import re from typing import Dict, Any, List, Optional import sys import os _agent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) if _agent_dir not in sys.path: sys.path.insert(0, _agent_dir) try: from skills.base import BaseSkill except ImportError: from ..base import BaseSkill logger = logging.getLogger(__name__) # ═══════════════════════════════════════════════════════════════ # 弹窗规则库 — 覆盖主流安卓品牌：小米/华为/OPPO/vivo/三星/一加/荣耀/realme # # 每条规则: detect = 界面关键词（正则）, click = 要点击的按钮文案 # 规则按优先级排列，匹配到即执行，不会重复点击 # ═══════════════════════════════════════════════════════════════ _POPUP_DISMISS_RULES: List[Dict[str, Any]] = [ # ─── 1. USB 调试授权（所有品牌） ─── {"detect": [ "USB调试", "USB debugging", "允许USB调试", "Allow USB debugging", "USB 偵錯", # 繁体 "USB调试（安全设置）", # MIUI 特有 "RSA.*密钥指纹", "RSA.*key fingerprint", # 弹窗正文 "允许通过USB调试", # 华为 "이 컴퓨터에서 USB 디버깅", # 三星韩语 ], "click": [ "确定", "允许", "OK", "Allow", "始终允许", "Always allow", "확인", # 三星韩语 "允許", # 繁体 ]}, # ─── 1b. USB 连接模式（仅充电 → 文件传输，防 MIUI 间歇掉 ADB） ─── {"detect": [ "USB 用于", "USB用途", "USB 用途", "连接方式", "仅充电", "充电模式", "Charging only", "Charge only", "USB 偏好设置", "Use USB for", "传输文件", "文件传输", "Transfer files", ], "click": [ "传输文件", "文件传输", "Transfer files", "File transfer", "传输照片", "Photo transfer", "MTP", ]}, # ─── 2. MIUI 安全弹窗（小米/红米特有） ─── {"detect": [ "安全警告", "此操作具有风险", "模拟点击", "模拟输入", # MIUI「检测到模拟操作」 "检测到.*模拟", "监控.*辅助", "开发者选项.*安全", "安全设置已关闭", ], "click": ["确定", "继续", "我已知晓", "允许", "仍然允许"]}, # ─── 3. 华为/荣耀 特有弹窗 ─── {"detect": [ "允许HiSuite通过", "HiSuite", "仅充电", "传输文件", "传输照片", # USB 模式选择 "华为手机助手", "应用.*未通过安全检测", # 华为安全中心 "可能会损坏您的设备", ], "click": ["允许", "继续安装", "我已知晓", "确定", "仍然安装"]}, # ─── 4. OPPO/realme/一加 特有弹窗 ─── {"detect": [ "ColorOS", "USB 传输", "允许.*调试", "您要允许.*USB调试", "开发者模式.*已开启", "USB用途", "连接方式", "充电模式.*传输", # OPPO USB 模式选择 ], "click": ["确定", "允许", "OK", "传输文件"]}, # ─── 5. vivo/iQOO 特有弹窗 ─── {"detect": [ "Funtouch", "OriginOS", "是否允许USB调试", "开启USB调试.*安全风险", "vivo.*安全", ], "click": ["确定", "允许", "OK", "已了解风险"]}, # ─── 6. 三星 特有弹窗 ─── {"detect": [ "One UI", "Samsung", "您是否要允许USB调试", "Allow USB debugging", "Knox.*安全", "设备保护", ], "click": ["确定", "允许", "OK", "Allow"]}, # ─── 7. ADB 通用权限 ─── {"detect": ["允许通过", "ADB", "adb"], "click": ["确定", "允许", "OK"]}, # ─── 8. 安装应用（所有品牌） ─── {"detect": [ "要安装", "允许安装", "安装应用", "安装未知应用", "Install unknown apps", "来源不明", "Unknown sources", "是否安装该应用", "Install this app", "外部来源", "第三方来源", "继续安装.*可能有风险", # 各品牌安全提示 "仍要安装", ], "click": [ "继续安装", "允许", "安装", "Install", "仍然安装", "仍要安装", "我已了解", ]}, # ─── 9. 权限请求弹窗（通用） ─── {"detect": [ "允许.*访问", "允许.*使用", "Allow.*access", "请求.*权限", "needs.*permission", "允许.*读取", "允许.*写入", "允许.*拍照", "允许.*录音", "允许.*定位", "Allow.*location", "允许.*通知", "Allow.*notification", "允许.*悬浮窗", "允许.*显示在其他应用", "Display over other apps", "允许.*存储", "Allow.*storage", ], "click": [ "允许", "始终允许", "Allow", "While using the app", "仅在使用时允许", "仅使用期间", "始终", "Always", ]}, # ─── 10. 系统更新/升级 ─── {"detect": [ "系统更新", "System update", "升级", "发现新版本", "New version", "现在更新", "Update now", "MIUI.*更新", "EMUI.*更新", "ColorOS.*更新", "OTA.*更新", ], "click": ["稍后", "以后", "取消", "Later", "Cancel", "Not now", "稍后再说"]}, # ─── 11. 电池优化（各品牌措辞不同） ─── {"detect": [ "电池优化", "Battery optimization", "忽略电池优化", "省电策略", "后台耗电", "电量管理", "自启动管理", "自启动", "是否允许.*后台运行", "关闭.*省电", "Ignore battery optimizations", "不再受限", # 华为 ], "click": ["允许", "是", "Yes", "Allow", "不受限制", "确定"]}, # ─── 12. 后台运行保活 ─── {"detect": [ "后台运行", "后台活动", "Background", "锁定.*后台", "保持运行", "允许.*自启动", "允许.*后台", "关闭后台清理", "电池管理.*允许", ], "click": ["允许", "Allow", "确定", "开启"]}, # ─── 13. 通知权限 ─── {"detect": [ "通知权限", "发送通知", "Post notifications", "允许.*推送", "开启通知", "Notification access", ], "click": ["允许", "Allow", "确定"]}, # ─── 14. 悬浮窗/无障碍 ─── {"detect": [ "悬浮窗", "Display over", "draw over", "无障碍", "Accessibility", "辅助功能", ], "click": ["允许", "Allow", "确定", "开启"]}, # ─── 15. Termux / 终端相关 ─── {"detect": ["Termux", "终端", "com.termux"], "click": ["允许", "Allow", "确定"]}, # ─── 16. 应用崩溃/ANR ─── {"detect": [ "ANR", "没有响应", "not responding", "停止运行", "已停止", "has stopped", "keeps stopping", "无响应.*关闭", "close app", ], "click": ["等待", "Wait", "再次打开", "Reopen"]}, # ─── 17. 隐私合规弹窗（国内APP常见） ─── {"detect": [ "隐私政策", "用户协议", "Privacy policy", "个人信息保护", "我已阅读", "同意并继续", "Agree.*continue", ], "click": ["同意", "同意并继续", "Agree", "我已阅读并同意", "确定"]}, # ─── 18. 广告/推广弹窗 ─── {"detect": [ "领取.*红包", "签到.*奖励", "今日推荐", "猜你喜欢", "开通.*会员", "免费试用", "跳过", "Skip", ], "click": ["跳过", "关闭", "Skip", "Close", "×", "✕"]}, # ─── 19. SIM卡/运营商弹窗 ─── {"detect": [ "SIM.*未插入", "SIM.*not inserted", "无SIM卡", "No SIM", "紧急呼叫", "Emergency", ], "click": ["确定", "OK", "关闭"]}, ] class ConnectionGuardSkill(BaseSkill): """ 连接守护技能 — 常驻后台，确保设备与服务器始终连接 核心能力： 1. dismiss_popups() — 自动关闭系统弹窗（USB授权、权限请求等） 2. ensure_screen_on() — 确保屏幕亮起 3. check_u2_alive() — 检测 uiautomator2 是否存活 4. check_network() — 检测网络连通性（委托 Hawk） 5. full_guard_cycle() — 执行一轮完整守护（上述全部） """ PACKAGE = "" NAME = "连接守护" def _shell(self, cmd: str) -> str: """安全执行 shell 并返回输出""" try: r = self.d.shell(cmd) return (getattr(r, "output", None) or str(r) or "").strip() except Exception: return "" # ========== 1. 弹窗自动处理 ========== def dismiss_popups(self) -> Dict[str, Any]: """ 扫描当前界面，自动点击 USB 授权/权限请求/系统弹窗的确认按钮。 返回本轮处理了哪些弹窗。 """ dismissed = [] if not self.d: return {"dismissed": dismissed, "count": 0} try: xml = self.d.dump_hierarchy() except Exception as e: logger.debug(f"dump_hierarchy 失败: {e}") return {"dismissed": dismissed, "count": 0, "error": str(e)} for rule in _POPUP_DISMISS_RULES: detected = False for keyword in rule["detect"]: if re.search(keyword, xml, re.IGNORECASE): detected = True break if not detected: continue for btn_text in rule["click"]: try: if self.d(text=btn_text).exists(timeout=0.5): self.d(text=btn_text).click() dismissed.append({"popup": rule["detect"][0], "clicked": btn_text}) logger.info(f"🛡️ 自动处理弹窗: 检测到「{rule['detect'][0]}」→ 点击「{btn_text}」") time.sleep(0.5) break except Exception: continue # textContains 兜底 try: if self.d(textContains=btn_text).exists(timeout=0.3): self.d(textContains=btn_text).click() dismissed.append({"popup": rule["detect"][0], "clicked": btn_text}) logger.info(f"🛡️ 自动处理弹窗: 检测到「{rule['detect'][0]}」→ 点击「{btn_text}」(contains)") time.sleep(0.5) break except Exception: continue return {"dismissed": dismissed, "count": len(dismissed)} # ========== 2. 屏幕状态 ========== def ensure_screen_on(self) -> Dict[str, Any]: """确保屏幕亮起（某些 ADB 操作需要屏幕点亮）""" if not self.d: return {"screen_on": False, "action": "no_device"} try: info = self.d.info if info.get("screenOn"): return {"screen_on": True, "action": "already_on"} self.d.press("power") time.sleep(0.5) # 如果有锁屏，向上滑动解锁 self.d.swipe_ext("up", scale=0.8) time.sleep(0.3) return {"screen_on": True, "action": "woke_up"} except Exception as e: logger.debug(f"ensure_screen_on 异常: {e}") return {"screen_on": False, "action": "error", "error": str(e)} # ========== 3. uiautomator2 存活检测 ========== def check_u2_alive(self) -> Dict[str, Any]: """检测 uiautomator2 是否正常工作，异常时尝试重启""" if not self.d: return {"alive": False, "action": "no_device"} try: info = self.d.info return {"alive": True, "package": info.get("currentPackageName", ""), "action": "ok"} except Exception as e: logger.warning(f"u2 异常: {e}，尝试重启服务") try: self._shell("am instrument -w -r -e debug false " "-e class androidx.test.runner.AndroidJUnitRunner " "io.appium.uiautomator2.server.test/" "androidx.test.runner.AndroidJUnitRunner 2>/dev/null &") time.sleep(3) info = self.d.info return {"alive": True, "action": "restarted"} except Exception as e2: return {"alive": False, "action": "restart_failed", "error": str(e2)} # ========== 4. 网络检测（委托 Hawk） ========== def check_network(self) -> Dict[str, Any]: """检测网络连通性，断网时委托 Hawk 尝试恢复""" try: from hawk import is_network_available, try_reconnect_network if is_network_available(self.d): return {"network": True, "action": "ok"} logger.info("🌐 检测到断网，启动 Hawk 网络恢复...") result = try_reconnect_network(self.d) return { "network": result.get("success", False), "action": "hawk_reconnect", "detail": result, } except ImportError: # Hawk 不可用时仅做 ping 检测 out = self._shell("ping -c 1 -W 2 8.8.8.8 2>/dev/null && echo ok || echo fail") return {"network": "ok" in out, "action": "ping_only"} except Exception as e: return {"network": False, "action": "error", "error": str(e)} # ========== 5. ADB 连接检测 ========== def check_adb_connection(self) -> Dict[str, Any]: """检测 ADB 连接状态""" out = self._shell("getprop ro.serialno 2>/dev/null || echo unknown") if out and out != "unknown": return {"adb": True, "serial": out} return {"adb": False, "serial": ""} # ========== 6. 完整守护周期 ========== def full_guard_cycle(self) -> Dict[str, Any]: """ 执行一轮完整守护检查（Agent 心跳循环中调用）： 1. 自动处理弹窗（USB授权等） 2. 确保屏幕亮起 3. 检测 u2 存活 4. 检测网络 5. 检测 ADB """ results = {} results["popups"] = self.dismiss_popups() results["screen"] = self.ensure_screen_on() results["u2"] = self.check_u2_alive() results["network"] = self.check_network() results["adb"] = self.check_adb_connection() all_ok = ( results["u2"].get("alive", False) and results["network"].get("network", False) ) results["all_ok"] = all_ok results["timestamp"] = int(time.time()) if not all_ok: issues = [] if not results["u2"].get("alive"): issues.append("u2不可用") if not results["network"].get("network"): issues.append("网络断开") logger.warning(f"⚠️ 守护检查发现问题: {', '.join(issues)}") else: popup_count = results["popups"].get("count", 0) if popup_count > 0: logger.info(f"🛡️ 守护周期完成: 处理了 {popup_count} 个弹窗，连接正常") return results