#!/usr/bin/env python3
"""
测试登录功能
"""

import sys
from pathlib import Path

# 添加项目根目录到path
sys.path.insert(0, str(Path(__file__).parent))

from douyu.config import Config, Account
from douyu.crypto import encrypt_password, encrypt_nickname_or_phone
from douyu.email_verifier import EmailVerifier
from utils.logger import setup_logger


def test_crypto():
    """测试加密功能"""
    print("=== 测试加密功能 ===")

    # 测试密码加密
    password = "test_password"
    encrypted = encrypt_password(password)
    print(f"密码: {password}")
    print(f"MD5: {encrypted}")

    # 测试用户名加密
    username = "test_user"
    encrypted = encrypt_nickname_or_phone(username)
    print(f"用户名: {username}")
    print(f"加密后: {encrypted}")

    print()


def test_config():
    """测试配置加载"""
    print("=== 测试配置加载 ===")

    try:
        config = Config("config.yaml")
        accounts = config.get_accounts()

        print(f"加载了 {len(accounts)} 个账号")
        for i, acc in enumerate(accounts):
            print(f"  [{i}] {acc.username} - {acc.email}")

        proxy = config.get_proxy()
        print(f"代理: {'启用' if proxy.enabled else '禁用'}")

    except FileNotFoundError as e:
        print(f"配置文件不存在: {e}")
    except Exception as e:
        print(f"加载配置失败: {e}")

    print()


def test_email_verifier():
    """测试邮箱验证器（不实际连接）"""
    print("=== 测试邮箱验证器 ===")

    # 测试验证码提取
    verifier = EmailVerifier("", "", "", "")

    # 模拟邮件内容
    test_cases = [
        "您的验证码是：123456，请在5分钟内完成验证。",
        "验证码:654321",
        "Your verification code is 789012",
        "【斗鱼】安全验证码：345678，切勿泄露给他人！",
    ]

    for text in test_cases:
        code = verifier._extract_verification_code(text)
        print(f"文本: {text[:30]}...")
        print(f"验证码: {code}")
        print()

    print()


def main():
    """运行所有测试"""
    setup_logger(level="INFO")

    print("斗鱼自动登录工具 - 测试\n")

    test_crypto()
    test_config()
    test_email_verifier()

    print("测试完成!")


if __name__ == '__main__':
    main()