#!/usr/bin/env python3
"""
工作手机SDK v3.0 - 完整系统测试
测试所有接口和设备控制功能

运行: python3 tests/test_full_system.py
（勿用 pytest 收集本文件：需 SDK 与 ADB 在线，由 main() 显式执行）
"""

import os
import sys
import time

import requests

BASE_URL = os.environ.get("SDK_BASE_URL", "http://localhost:8899")
API_KEY = os.environ.get("SDK_API_KEY", "workphone-secret-key")

HEADERS = {
    "Content-Type": "application/json",
    "X-API-Key": API_KEY,
}

passed = 0
failed = 0
total = 0


def run_check(name: str, func):
    """运行单条检查（原 test 名与 pytest 冲突，已改名）"""
    global passed, failed, total
    total += 1
    try:
        result = func()
        if result:
            passed += 1
            print(f"  ✅ {name}")
        else:
            failed += 1
            print(f"  ❌ {name}")
    except Exception as e:
        failed += 1
        print(f"  ❌ {name} - 异常: {e}")


def main() -> int:
    global passed, failed, total
    passed = failed = total = 0

    print("\n🔧 === 基础接口测试 ===\n")

    def check_health():
        r = requests.get(f"{BASE_URL}/health", timeout=10)
        d = r.json()
        return d.get("status") == "healthy" and d.get("adb_devices", 0) >= 1

    def check_root():
        r = requests.get(f"{BASE_URL}/", timeout=10)
        return r.status_code == 200

    def check_docs():
        r = requests.get(f"{BASE_URL}/docs", timeout=10)
        return r.status_code == 200

    run_check("健康检查 (/health)", check_health)
    run_check("根路由 (/)", check_root)
    run_check("API文档 (/docs)", check_docs)

    print("\n📱 === ADB设备控制测试 ===\n")

    device_serial = None

    def check_adb_list():
        nonlocal device_serial
        r = requests.get(f"{BASE_URL}/api/v3/adb/devices", timeout=15)
        d = r.json()
        devices = d.get("data", [])
        if devices:
            device_serial = devices[0]["serial"]
            return devices[0]["status"] == "online"
        return False

    def check_adb_info():
        r = requests.get(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}", timeout=15
        )
        d = r.json()
        return d.get("code") == 200 and bool(
            d.get("data", {}).get("android_version")
        )

    def check_adb_screenshot():
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/screenshot", timeout=30
        )
        d = r.json()
        return d.get("code") == 200 and d.get("data", {}).get("size", 0) > 10000

    def check_adb_uitree():
        r = requests.get(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/ui-tree", timeout=30
        )
        d = r.json()
        return d.get("code") == 200 and d.get("data", {}).get("length", 0) > 100

    def check_adb_click():
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/click",
            json={"x": 540, "y": 1200},
            timeout=15,
        )
        return r.json().get("code") == 200

    def check_adb_swipe():
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/swipe",
            json={"direction": "up"},
            timeout=15,
        )
        return r.json().get("code") == 200

    def check_adb_key_home():
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/key",
            json={"key": "home"},
            timeout=15,
        )
        return r.json().get("code") == 200

    def check_adb_key_back():
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/key",
            json={"key": "back"},
            timeout=15,
        )
        return r.json().get("code") == 200

    def check_adb_current_app():
        r = requests.get(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/app/current",
            timeout=15,
        )
        d = r.json()
        return d.get("code") == 200 and d.get("data", {}).get("package")

    def check_adb_app_list():
        r = requests.get(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/app/list", timeout=30
        )
        return r.json().get("code") == 200

    def check_adb_input():
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/input",
            json={"text": "test", "clear": True},
            timeout=30,
        )
        return r.json().get("code") == 200

    def check_adb_click_text():
        try:
            requests.post(
                f"{BASE_URL}/api/v3/adb/devices/{device_serial}/key",
                json={"key": "home"},
                timeout=30,
            )
        except Exception:
            pass
        time.sleep(1)
        r = requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/click-text",
            json={"text": "Chrome"},
            timeout=30,
        )
        d = r.json()
        return d.get("code") in [200, 404]

    run_check("ADB设备列表", check_adb_list)
    run_check("ADB设备信息", check_adb_info)
    run_check("ADB截图", check_adb_screenshot)
    run_check("ADB获取UI树", check_adb_uitree)
    run_check("ADB点击坐标", check_adb_click)
    run_check("ADB滑动", check_adb_swipe)
    run_check("ADB按Home键", check_adb_key_home)
    run_check("ADB按返回键", check_adb_key_back)
    run_check("ADB获取当前APP", check_adb_current_app)
    run_check("ADB获取APP列表", check_adb_app_list)
    run_check("ADB输入文字", check_adb_input)
    run_check("ADB点击文字", check_adb_click_text)

    if device_serial:
        try:
            requests.post(
                f"{BASE_URL}/api/v3/adb/devices/{device_serial}/key",
                json={"key": "home"},
                timeout=30,
            )
        except Exception:
            pass
        time.sleep(0.5)

    print("\n🌐 === 统一API测试 ===\n")

    def check_unified_send():
        if not device_serial:
            return False
        r = requests.post(
            f"{BASE_URL}/api/v3/message/send",
            json={
                "device_id": device_serial,
                "platform": "wechat",
                "to_id": "测试",
                "content": "测试消息",
                "msg_type": "text",
            },
            timeout=90,
        )
        d = r.json()
        return d.get("code") == 200

    run_check("统一发送消息", check_unified_send)

    if device_serial:
        requests.post(
            f"{BASE_URL}/api/v3/adb/devices/{device_serial}/key",
            json={"key": "home"},
            timeout=10,
        )

    print(f"\n{'='*50}")
    print(f"📊 测试结果: {passed}/{total} 通过, {failed} 失败")
    print(f"{'='*50}")

    if failed == 0:
        print("🎉 所有测试通过！工作手机SDK完全可用！")
    else:
        print(f"⚠️ 有 {failed} 个测试失败，需要排查")

    return 0 if failed == 0 else 1


if __name__ == "__main__":
    sys.exit(main())