"""登录态路由：查询与重新加载账号登录态

抓取接口全部匿名，只有加购与下单需要账号。登录本身不在这里做——两站登录都要过
reCAPTCHA 与设备验证，由 `scripts/login.py` 起有头浏览器人工完成一次，
本服务只读取落盘的 cookie。这里提供的是运维视角的两个动作：

- `/api/auth/status`：现在还登录着吗（默认真实打一次请求探测，不看缓存）
- `/api/auth/reload`：人工重新登录后，免重启服务重新读取登录态
"""
from fastapi import APIRouter, Depends

from app.shared.api import ApiResponse, get_container, require_bearer_token
from app.trading.container import TradingContainer
from app.trading.models import (
    AuthReloadData,
    AuthReloadRequest,
    AuthSiteStatus,
    AuthStatusData,
    AuthStatusRequest,
)
from app.trading.services.auth_session import AuthStatus

router = APIRouter(prefix="/api/auth", tags=["auth"])


def _to_model(status: AuthStatus) -> AuthSiteStatus:
    return AuthSiteStatus(**status.to_dict())


@router.post(
    "/status",
    response_model=ApiResponse[AuthStatusData],
    dependencies=[Depends(require_bearer_token)],
)
async def auth_status(
    payload: AuthStatusRequest,
    container: TradingContainer = Depends(get_container),
) -> ApiResponse[AuthStatusData]:
    """查询账号登录态

    `refresh=true`（默认）时会真实访问站点探测——登录态过期没有可靠的本地判据，
    cookie 上的 expires 与服务端会话不是一回事，只能问站点。
    不需要这次网络往返时传 `refresh=false`，此时返回上一次探测的缓存结果
    （`logged_in` 为 null 表示从未探测过）。

    `logged_in=false` 时，加购与下单接口会直接返回 5001，需要重新跑
    `scripts/login.py --site <site>` 后调用 `/api/auth/reload`。
    """
    sites = [payload.site.value] if payload.site else list(container.auth_session.sites)
    statuses = []
    for site in sites:
        status = (
            await container.auth_session.check(site)
            if payload.refresh
            else container.auth_session.status(site)
        )
        statuses.append(_to_model(status))

    return ApiResponse[AuthStatusData](
        success=True,
        msg="success",
        data=AuthStatusData(sites=statuses),
        code=0,
    )


@router.post(
    "/reload",
    response_model=ApiResponse[AuthReloadData],
    dependencies=[Depends(require_bearer_token)],
)
async def auth_reload(
    payload: AuthReloadRequest,
    container: TradingContainer = Depends(get_container),
) -> ApiResponse[AuthReloadData]:
    """重新从磁盘加载登录态并立即探测

    人工跑完 `scripts/login.py` 后调用，避免为了换一套 cookie 重启整个服务。
    """
    sites = [payload.site.value] if payload.site else list(container.auth_session.sites)
    reloaded = {site: container.auth_session.reload(site) for site in sites}
    statuses = [_to_model(await container.auth_session.check(site)) for site in sites]

    return ApiResponse[AuthReloadData](
        success=True,
        msg="success",
        data=AuthReloadData(reloaded=reloaded, sites=statuses),
        code=0,
    )