"""
一键测试脚本：验证 .env 配置和 LLM API 连通性。
用法: python test_api.py
"""

import sys
from pathlib import Path

# 加载 .env
from dotenv import load_dotenv
load_dotenv(Path(__file__).parent / ".env")

from get_notes.config import AppConfig

config = AppConfig()

print("=" * 55)
print("  Get笔记 — API 连通性测试")
print("=" * 55)
print()

# 检查配置
print(f"  LLM_BASE_URL: {config.llm.base_url}")
print(f"  LLM_MODEL:    {config.llm.model}")
key_display = config.llm.api_key[:8] + "..." if len(config.llm.api_key) > 8 else "(未设置)"
print(f"  LLM_API_KEY:  {key_display}")
print()

if not config.llm.api_key or "填入" in config.llm.api_key:
    print("  [!] API Key 未设置")
    print("  请编辑 .env 文件，将 LLM_API_KEY 替换为你的真实 Key")
    print()
    print("  vim .env")
    print("  # 或")
    print("  open .env")
    sys.exit(1)

# 测试 API 调用
print("  发送测试请求...")
import requests

try:
    resp = requests.post(
        f"{config.llm.base_url}/chat/completions",
        headers={
            "Authorization": f"Bearer {config.llm.api_key}",
            "Content-Type": "application/json",
        },
        json={
            "model": config.llm.model,
            "temperature": 0.3,
            "max_tokens": 100,
            "messages": [
                {"role": "user", "content": "用一句话介绍手冲咖啡"},
            ],
        },
        timeout=30,
    )
    resp.raise_for_status()
    data = resp.json()
    content = data["choices"][0]["message"]["content"]
    model_used = data.get("model", "unknown")
    usage = data.get("usage", {})

    print()
    print(f"  [OK] API 调用成功!")
    print(f"  模型: {model_used}")
    print(f"  Token: {usage.get('prompt_tokens', '?')} + {usage.get('completion_tokens', '?')} = {usage.get('total_tokens', '?')}")
    print(f"  回复: {content}")
    print()
    print("=" * 55)
    print("  一切就绪! 你可以运行:")
    print()
    print('  python -m get_notes "你的链接" -v')
    print()
    print("  或交互模式:")
    print()
    print("  python -m get_notes --interactive")
    print("=" * 55)

except requests.exceptions.HTTPError as e:
    print()
    print(f"  [FAIL] API 返回错误: {e}")
    try:
        error_detail = resp.json()
        print(f"  详情: {error_detail}")
    except Exception:
        print(f"  响应: {resp.text[:300]}")
    sys.exit(1)

except requests.exceptions.ConnectionError:
    print()
    print("  [FAIL] 无法连接到 API 服务器")
    print(f"  请检查 BASE_URL: {config.llm.base_url}")
    sys.exit(1)

except Exception as e:
    print()
    print(f"  [FAIL] 发生错误: {e}")
    sys.exit(1)