"""多账号管理，对应独立的账号配置管理。"""

from __future__ import annotations

import json
import logging
import os
from pathlib import Path

logger = logging.getLogger(__name__)

# 账号配置文件路径
_CONFIG_DIR = Path.home() / ".xhs"
_ACCOUNTS_FILE = _CONFIG_DIR / "accounts.json"


def _load_config() -> dict:
    """加载账号配置。"""
    if not _ACCOUNTS_FILE.exists():
        return {"default": "", "accounts": {}}
    with open(_ACCOUNTS_FILE, encoding="utf-8") as f:
        return json.load(f)


def _save_config(config: dict) -> None:
    """保存账号配置。"""
    _CONFIG_DIR.mkdir(parents=True, exist_ok=True)
    with open(_ACCOUNTS_FILE, "w", encoding="utf-8") as f:
        json.dump(config, f, ensure_ascii=False, indent=2)


def list_accounts() -> list[dict]:
    """列出所有账号。"""
    config = _load_config()
    default = config.get("default", "")
    accounts = config.get("accounts", {})
    result = []
    for name, info in accounts.items():
        result.append(
            {
                "name": name,
                "description": info.get("description", ""),
                "is_default": name == default,
                "profile_dir": _get_profile_dir(name),
            }
        )
    return result


def add_account(name: str, description: str = "") -> None:
    """添加账号。"""
    config = _load_config()
    accounts = config.setdefault("accounts", {})
    if name in accounts:
        raise ValueError(f"账号 '{name}' 已存在")

    accounts[name] = {"description": description}

    # 如果是第一个账号，设为默认
    if not config.get("default"):
        config["default"] = name

    _save_config(config)

    # 创建 Profile 目录
    profile_dir = _get_profile_dir(name)
    os.makedirs(profile_dir, exist_ok=True)

    logger.info("添加账号: %s", name)


def remove_account(name: str) -> None:
    """删除账号。"""
    config = _load_config()
    accounts = config.get("accounts", {})
    if name not in accounts:
        raise ValueError(f"账号 '{name}' 不存在")

    del accounts[name]

    # 如果删除的是默认账号，清除默认
    if config.get("default") == name:
        config["default"] = next(iter(accounts), "")

    _save_config(config)
    logger.info("删除账号: %s", name)


def set_default_account(name: str) -> None:
    """设置默认账号。"""
    config = _load_config()
    accounts = config.get("accounts", {})
    if name not in accounts:
        raise ValueError(f"账号 '{name}' 不存在")

    config["default"] = name
    _save_config(config)
    logger.info("默认账号设置为: %s", name)


def get_default_account() -> str:
    """获取默认账号名称。"""
    config = _load_config()
    return config.get("default", "")


def _get_profile_dir(account: str) -> str:
    """获取账号的 Chrome Profile 目录。"""
    return str(_CONFIG_DIR / "accounts" / account / "chrome-profile")