# -*- coding: utf-8 -*-
# /// script
# requires-python = ">=3.10"
# dependencies = [
#     "playwright>=1.40",
# ]
# ///
r"""
调试用户目录 .profile 生成 + YouTube 账号登录脚本（yt-studio-auth-prep 技能）。

职责单一：在指定目录创建（或复用）一个独立的浏览器用户数据目录，打开可见浏览器
窗口交给用户手动完成 Google/YouTube 登录，轮询确认 studio.youtube.com 不再跳转
登录页后关闭窗口退出。登录态持久化在该目录，供下游脚本复用：

  - youtube-studio-csv-download : --user-data-dir "<place>/.profile" --channel <brand>
  - yt-studio-groupid-lookup    : 同上（在线捕获套件时）

用法（cwd 任意；目录不存在会自动创建）：
  python login_profile.py --channel chrome --user-data-dir "<place>/.profile"
  python login_profile.py --channel msedge --user-data-dir "<place>/.profile" --timeout-seconds 600

注意：
  - channel 一经选定，整个工作区所有下游命令必须用同一品牌（chrome/msedge）；
    换品牌 = 删除 .profile 后重跑本脚本，并重新捕获 bundles.json。
  - 登录态失效（下游报跳 Google 登录页）时直接重跑本脚本补登录，.profile 保留原数据。
"""

from __future__ import annotations

import argparse
import os
import time

STUDIO_URL = "https://studio.youtube.com/"
DEFAULT_TIMEOUT_SECONDS = 600


def classify_url(url: str) -> str:
    """当前页面 URL 归类：google=登录页，studio=已进 Studio，other=其他。"""
    u = (url or "").lower()
    if "accounts.google" in u:
        return "google"
    if "studio.youtube.com" in u:
        return "studio"
    return "other"


def selftest():
    assert classify_url("https://accounts.google.com/v3/signin/identifier?x=1") == "google"
    assert classify_url("https://studio.youtube.com/channel/UCxx/analytics") == "studio"
    assert classify_url("https://www.youtube.com/") == "other"
    assert classify_url("") == "other"
    print("selftest OK：URL 分类逻辑通过")


def wait_login(page, start_url: str, timeout_seconds: int) -> bool:
    """轮询等待用户完成登录；首次到达 studio 后回访一次，防会话弹回登录页。"""
    page.goto(start_url, wait_until="domcontentloaded")
    deadline = time.time() + timeout_seconds
    hinted = False
    last_report = time.time()
    while time.time() < deadline:
        kind = classify_url(page.url)
        if kind == "studio":
            page.wait_for_timeout(3000)
            page.goto(start_url, wait_until="domcontentloaded")
            if classify_url(page.url) == "studio":
                return True
        elif kind == "google":
            if not hinted:
                print("[登录] 请在打开的浏览器窗口中完成 Google/YouTube 登录"
                      "（支持两步验证），完成后本脚本自动检测……")
                hinted = True
            page.wait_for_timeout(2000)
        else:
            page.wait_for_timeout(3000)
            page.goto(start_url, wait_until="domcontentloaded")
        if time.time() - last_report >= 30:
            print(f"[登录] 仍在等待登录……剩余 {max(0, int(deadline - time.time()))} 秒")
            last_report = time.time()
    return False


def main(argv=None):
    ap = argparse.ArgumentParser(
        description="创建调试用户目录 .profile 并完成 YouTube 账号登录")
    ap.add_argument("--selftest", action="store_true",
                    help="仅跑 URL 分类自测（无需浏览器）")
    ap.add_argument("--user-data-dir", help="调试用户目录路径，如 <place>/.profile")
    ap.add_argument("--channel", choices=["chrome", "msedge"], default="chrome",
                    help="浏览器品牌，默认 %(default)s；选定后全工作区统一")
    ap.add_argument("--start-url", default=STUDIO_URL,
                    help="登录检测起始页，默认 %(default)s")
    ap.add_argument("--timeout-seconds", type=int, default=DEFAULT_TIMEOUT_SECONDS,
                    help="等待登录的最长秒数，默认 %(default)s")
    args = ap.parse_args(argv)

    if args.selftest:
        selftest()
        return
    if not args.user_data_dir:
        ap.error("--user-data-dir 必填，如 <place>/.profile")

    os.makedirs(args.user_data_dir, exist_ok=True)
    print(f"[登录] 打开 {args.channel}（profile={args.user_data_dir}），"
          f"最长等待 {args.timeout_seconds} 秒……")
    try:
        from playwright.sync_api import sync_playwright
    except ImportError:
        raise SystemExit("[!] 缺少 playwright：在工作区（yt-download-place）内用"
                         " uv run python 重试，或先 uv sync。")

    with sync_playwright() as p:
        try:
            context = p.chromium.launch_persistent_context(
                user_data_dir=args.user_data_dir,
                channel=args.channel,
                headless=False,
                args=["--disable-blink-features=AutomationControlled"],
            )
        except Exception as e:  # noqa: BLE001
            raise SystemExit(
                f"[!] 启动浏览器失败: {e}\n"
                f"    多为 {args.channel} 正被占用（上次窗口未关/手动开过 .profile）。"
                "任务管理器彻底退出该品牌浏览器进程后重跑。")
        try:
            page = context.new_page()
            ok = wait_login(page, args.start_url, args.timeout_seconds)
        finally:
            context.close()

    if not ok:
        raise SystemExit(
            f"[!] {args.timeout_seconds} 秒内未完成登录。.profile 已保留进度，"
            "直接重跑同一命令继续；等待不够就加大 --timeout-seconds。")
    print(f"[登录] 登录成功：{args.user_data_dir}（channel={args.channel}）")
    print(f"[i] 该工作区所有下游命令统一使用 --channel {args.channel} "
          f"--user-data-dir {args.user_data_dir}")


if __name__ == "__main__":
    main()