#!/usr/bin/env python3
"""
腾讯混元3D API 拦截分析工具 (CloakBrowser)
利用持久化登录状态，自动捕获所有 API 请求和响应
"""

import json
import os
import sys
import time
from datetime import datetime
from cloakbrowser import launch_persistent_context

from ..config import get_profile_dir

PROFILE_DIR = str(get_profile_dir())
API_LOG_FILE = "./api_requests.log.json"


def main():
    logs = []

    def log_request(request):
        url = request.url
        # 只关注同域 API 和关键第三方接口
        if "/api/" in url or "hunyuan" in url:
            entry = {
                "time": datetime.now().isoformat(),
                "type": "request",
                "method": request.method,
                "url": url,
                "headers": dict(request.headers) if hasattr(request, "headers") else {},
            }
            # 尝试获取 POST body
            if request.method in ("POST", "PUT", "PATCH") and hasattr(request, "post_data"):
                try:
                    entry["body"] = request.post_data
                except Exception:
                    pass
            logs.append(entry)
            print(f"[REQ] {request.method} {url}")

    def log_response(response):
        url = response.url
        if "/api/" in url or "hunyuan" in url:
            entry = {
                "time": datetime.now().isoformat(),
                "type": "response",
                "status": response.status,
                "url": url,
            }
            # 尝试读取响应体
            try:
                # 只读取 JSON 响应
                content_type = response.headers.get("content-type", "")
                if "json" in content_type:
                    body = response.json()
                    entry["body"] = body
                else:
                    text = response.text()
                    # 限制文本长度，避免过大
                    entry["body_preview"] = text[:500]
            except Exception as e:
                entry["body_error"] = str(e)
            logs.append(entry)
            print(f"[RES] {response.status} {url}")

    if not os.path.exists(PROFILE_DIR):
        print(f"错误: 未找到持久化目录 {PROFILE_DIR}")
        print("请先运行 hunyuan3dweb-login 完成登录")
        sys.exit(1)

    context = launch_persistent_context(PROFILE_DIR, headless=False)
    page = context.new_page()

    page.on("request", log_request)
    page.on("response", log_response)

    print("正在打开腾讯混元3D并捕获 API...")
    page.goto("https://3d.hunyuan.tencent.com/")
    time.sleep(3)

    # 检查是否已登录
    login_btn = page.locator("button").filter(has_text="登录").first
    if login_btn.count() > 0:
        print("\n警告: 当前未检测到登录状态，API 可能返回未授权")
    else:
        print("\n已检测到登录状态，开始捕获 API...")

    print("\n你可以手动在浏览器中操作（切换页面、生成3D等）")
    print("所有 API 请求会实时打印在终端中")
    print("按 Enter 停止捕获并保存结果...\n")
    input()

    # 保存日志
    with open(API_LOG_FILE, "w", encoding="utf-8") as f:
        json.dump(logs, f, ensure_ascii=False, indent=2)

    print(f"\n共捕获 {len([l for l in logs if l['type'] == 'request'])} 个请求")
    print(f"结果已保存到: {os.path.abspath(API_LOG_FILE)}")

    context.close()


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\n已取消")
        sys.exit(0)