NeteaseUrl/web/deps.py

# -*- coding: utf-8 -*-
"""FastAPI 依赖：会话解析、当前用户、登录校验。

与 auth_system 的语义对齐（共用同一 Cookie、同一站点库）：
- 会话 Cookie 经 core.session 解码（Flask 兼容签名），密钥同源
- 过期：签发时间超过 core.session.SESSION_MAX_AGE 视为失效（服务端把关，
  不依赖浏览器是否还留着 Cookie）
- 封禁账号（status 0/2）：按未登录处理
- 顶号校验：会话内的 `site_uid_seq` 必须与库中 `login_seq` 一致，否则视为失效
- 剩余次数/账本预扣属付费流程，由 web.billing 承担（此处只做身份解析）
"""
from dataclasses import dataclass
from typing import Annotated, Any

from fastapi import Cookie, Depends

from core import repo
from core import session as sess
from web.errors import AppError

_SESSION_COOKIE = sess.COOKIE_NAME


async def secret_key() -> str:
    """会话签名密钥（密钥来源：data/secret.key）。"""
    import auth_system
    return await auth_system.get_secret_key()


@dataclass
class SiteUser:
    id: int
    username: str
    role: str
    credits: int

    @property
    def is_admin(self) -> bool:
        return self.role == 'admin'

    @property
    def unlimited(self) -> bool:
        return self.is_admin or self.credits == -1


async def _parse_cookie(session_cookie: str | None) -> dict[str, Any] | None:
    # 服务端叠加签发时间校验（SESSION_MAX_AGE）：Cookie 自身的 max_age 只约束
    # 浏览器，被存下来的旧 Cookie 仍能通过签名校验，故过期必须由服务端把关。
    return sess.loads(await secret_key(), session_cookie or '', max_age=sess.SESSION_MAX_AGE)


async def current_user(
    session_cookie: Annotated[str | None, Cookie(alias=_SESSION_COOKIE)] = None,
) -> SiteUser | None:
    """当前登录用户（未登录/失效返回 None）。"""
    data = await _parse_cookie(session_cookie)
    uid = sess.get_uid(data)
    if not uid:
        return None
    row = await repo.get_user_by_id(uid)
    if row is None:
        return None
    status = row.get('status', 1)
    if status in (0, 2):  # 禁用/封禁：按未登录处理
        return None
    # 顶号校验：库有 login_seq 时会话版本必须严格一致
    db_seq = int(row.get('login_seq') or 0)
    sess_seq = int((data or {}).get(sess.SESSION_KEY + '_seq') or 0)
    if db_seq and sess_seq != db_seq:
        return None
    return SiteUser(
        id=int(row['id']),
        username=row['username'],
        role=row['role'],
        credits=int(row.get('duration_seconds') or 0),
    )


async def require_login(
    user: Annotated[SiteUser | None, Depends(current_user)],
) -> SiteUser:
    """登录闸：未登录抛 401，已登录返回用户。"""
    if user is None:
        raise AppError('请先登录后再使用解析功能', 401, error_code='unauthorized')
    return user


async def require_admin(
    user: Annotated[SiteUser | None, Depends(current_user)],
) -> SiteUser:
    """管理员闸：非管理员抛 403。"""
    if user is None or not user.is_admin:
        raise AppError('需要管理员权限', 403, error_code='forbidden')
    return user