# 从根目录 cookies.txt 加载并注入 cookies（支持 Netscape 与 nodriver 原生格式）

from pathlib import Path

# 项目根目录（本文件所在目录的父目录即 audcf）
ROOT_DIR = Path(__file__).resolve().parent
COOKIES_FILE = ROOT_DIR / "cookies.txt"


def get_cookies_path() -> Path:
    return COOKIES_FILE


def cookies_file_exists() -> bool:
    return COOKIES_FILE.is_file()


def is_netscape_format(path: Path) -> bool:
    """根据首行判断是否为 Netscape cookies.txt 格式。"""
    if not path.is_file():
        return False
    try:
        with open(path, "r", encoding="utf-8", errors="ignore") as f:
            first = f.readline().strip()
        return first.startswith("# Netscape") or first.startswith("# HTTP Cookie File")
    except Exception:
        return False


def parse_netscape_cookies(path: Path) -> list[dict]:
    """
    解析 Netscape 格式 cookies.txt（含 Cookie-Editor 等导出的多行注释）。
    每行（非注释）：domain, subdomain, path, secure, expires, name, value（制表符分隔；value 可含制表符故用 maxsplit=6）
    返回字典列表供转为 CookieParam。
    """
    cookies = []
    with open(path, "r", encoding="utf-8", errors="ignore") as f:
        for line in f:
            line = line.strip()
            if not line or line.startswith("#"):
                continue
            parts = line.split("\t", 6)
            if len(parts) < 7:
                continue
            domain, _subdomain, path_val, secure_val, expires_str, name, value = parts
            secure = secure_val.upper() == "TRUE"
            try:
                expires = float(expires_str) if expires_str and expires_str != "0" else None
            except ValueError:
                expires = None
            cookies.append({
                "name": name,
                "value": value,
                "domain": domain,
                "path": path_val or "/",
                "expires": expires,
                "http_only": False,
                "secure": secure,
                "same_site": "Lax",
            })
    return cookies


async def load_cookies_nodriver_native(browser, path: Path) -> bool:
    """
    使用 nodriver 原生格式加载 cookies（browser.cookies.load）。
    成功返回 True，失败返回 False。
    """
    try:
        await browser.cookies.load(file=str(path))
        return True
    except Exception:
        return False


def _dict_to_cookie_param(d: dict):
    """将解析后的 cookie 字典转为 nodriver CDP 所需的 CookieParam。"""
    from nodriver.cdp import network
    expires = d.get("expires")
    if expires is not None:
        expires = network.TimeSinceEpoch(expires)
    same_site = d.get("same_site")
    if same_site:
        same_site = network.CookieSameSite(same_site)
    return network.CookieParam(
        name=d["name"],
        value=d["value"],
        domain=d.get("domain"),
        path=d.get("path") or "/",
        secure=d.get("secure"),
        http_only=d.get("http_only"),
        same_site=same_site,
        expires=expires,
    )


async def inject_netscape_cookies(tab, path: Path) -> bool:
    """
    解析 Netscape 格式的 cookies.txt 并通过 CDP 注入到当前 tab。
    应在已打开目标域页面后调用。
    成功返回 True，失败返回 False。
    """
    try:
        from nodriver import cdp
        cookies_list = parse_netscape_cookies(path)
        if not cookies_list:
            return True
        cookies_param = [_dict_to_cookie_param(d) for d in cookies_list]
        await tab.send(cdp.storage.set_cookies(cookies=cookies_param))
        return True
    except Exception:
        return False