"""辅助工具函数"""

import json
from pathlib import Path
from typing import List
from loguru import logger

from douyu.config import Account


def load_accounts(config_path: str = "config.yaml") -> List[Account]:
    """加载账号列表"""
    from douyu.config import Config

    config = Config(config_path)
    accounts = config.get_accounts()

    logger.info(f"加载了 {len(accounts)} 个账号")
    return accounts


def save_cookie(username: str, cookie: str, cookie_dir: str = "data/cookies") -> str:
    """
    保存Cookie到文件

    Args:
        username: 用户名
        cookie: Cookie字符串
        cookie_dir: Cookie存储目录

    Returns:
        文件路径
    """
    import time

    cookie_path = Path(cookie_dir)
    cookie_path.mkdir(parents=True, exist_ok=True)

    filepath = cookie_path / f"{username}.json"

    data = {
        'username': username,
        'cookie': cookie,
        'timestamp': int(time.time()),
    }

    with open(filepath, 'w', encoding='utf-8') as f:
        json.dump(data, f, ensure_ascii=False, indent=2)

    logger.info(f"Cookie已保存: {filepath}")
    return str(filepath)


def load_cookie(username: str, cookie_dir: str = "data/cookies") -> str:
    """
    从文件加载Cookie

    Args:
        username: 用户名
        cookie_dir: Cookie存储目录

    Returns:
        Cookie字符串，不存在返回空字符串
    """
    filepath = Path(cookie_dir) / f"{username}.json"

    if not filepath.exists():
        return ""

    try:
        with open(filepath, 'r', encoding='utf-8') as f:
            data = json.load(f)
        return data.get('cookie', '')
    except Exception as e:
        logger.error(f"加载Cookie失败: {e}")
        return ""


def format_cookie_for_browser(cookie_str: str) -> dict:
    """
    将Cookie字符串转换为浏览器格式

    Args:
        cookie_str: Cookie字符串

    Returns:
        Cookie字典
    """
    cookies = {}
    for item in cookie_str.split(';'):
        item = item.strip()
        if '=' in item:
            key, value = item.split('=', 1)
            cookies[key.strip()] = value.strip()
    return cookies