Archived
snowy-yang
9df0028195
迁移端点：
- POST /api/v1/download（单曲：file 后台任务 / json 同步交付+结算）
- POST /api/v1/download/batch_zip（按去重歌曲数计费）
- POST /api/v1/podcast/download、/podcast/download/batch_zip
- 新增 web/billing.py（预扣+开账同事务 / 只退 open 行 / rename 承接）
与 web/tasks_registry.py（任务登记/容量闸/账本撞名重试，与 main 同口径）
- worker 仍复用 main 既有实现（无 flask 依赖），账本与交付语义不变
修复我引入的真实缺陷：
- download 参数校验分支（ID/音质/格式非法）漏退款 → 预扣泄漏。改为统一 _fail() 退款
- web/routers/{auth,register}.py 在导入期绑定 deps.secret_key，导致 patch web.deps.secret_key
无效、且测试顺序相关。改为调用期 deps.secret_key() 解析
测试：tests/test_download_billing.py 7 项沙箱用例如实断言账本守恒
（预扣必退、无 open 行残留、次数不足不建任务）；
全套 pytest 69/69（顺序无关）；ruff All checks passed；pyright 0 errors；奇偶 36/36。
227 lines
9.1 KiB
Python
227 lines
9.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""认证操作端点（FastAPI 栈）——与 Flask 侧同名端点响应契约逐字段对齐。
|
||
|
||
已迁移：
|
||
- POST /api/v1/auth/logout ↔ Flask /auth/logout（顶号：递增 login_seq + 删 Cookie）
|
||
- POST /api/v1/auth/heartbeat ↔ Flask /auth/heartbeat
|
||
- POST /api/v1/auth/ann_seen ↔ Flask /auth/ann_seen
|
||
- POST /api/v1/auth/login ↔ Flask /auth/login（密码校验 + 失败锁定 + 管理员 2FA 转票据）
|
||
|
||
设计：
|
||
- 登录失败锁定与管理员 2FA 票据表**复用 auth_routes 的模块级状态**，两栈共享同一份
|
||
计数/票据，避免「换个框架打就绕过锁定」。
|
||
- 密码校验（PBKDF2，10 万次迭代，CPU 密集）与库操作经 asyncio.to_thread 执行，
|
||
不阻塞事件循环。
|
||
- 会话 Cookie 由 core.session 签发（与 Flask 同格式同密钥），删 Cookie 亦同口径。
|
||
"""
|
||
import asyncio
|
||
import os
|
||
import time
|
||
from typing import Annotated, Any
|
||
|
||
from fastapi import APIRouter, Depends, Request
|
||
from fastapi.responses import JSONResponse
|
||
|
||
import auth_system as auth
|
||
import web.deps as deps
|
||
from core import db, repo
|
||
from core import session as cs
|
||
from web import ratelimit
|
||
from web.deps import SiteUser, current_user
|
||
from web.response import error, success
|
||
|
||
router = APIRouter()
|
||
|
||
SESSION_COOKIE = 'session'
|
||
_SESSION_MAX_AGE = 7 * 86400
|
||
|
||
|
||
def _format_credits(count: int) -> str:
|
||
if count < 0:
|
||
return '不限次数'
|
||
if count <= 0:
|
||
return '已用完'
|
||
return f'{count} 次'
|
||
|
||
|
||
def _user_public(row: dict) -> dict:
|
||
return {
|
||
'id': row['id'],
|
||
'username': row['username'],
|
||
'email': row.get('email'),
|
||
'role': row['role'],
|
||
'duration_seconds': row['duration_seconds'],
|
||
'created_at': row.get('created_at'),
|
||
'last_login': row.get('last_login'),
|
||
'ann_seen': row.get('ann_seen'),
|
||
}
|
||
|
||
|
||
async def _request_data(request: Request) -> dict[str, Any]:
|
||
data: dict[str, Any] = {}
|
||
if request.method == 'GET':
|
||
data.update(dict(request.query_params))
|
||
return data
|
||
try:
|
||
body = await request.json()
|
||
if isinstance(body, dict):
|
||
data.update(body)
|
||
except Exception:
|
||
pass
|
||
if not data:
|
||
try:
|
||
form = await request.form()
|
||
data.update(dict(form.items()))
|
||
except Exception:
|
||
pass
|
||
return data
|
||
|
||
|
||
def _cookie_kwargs() -> dict:
|
||
secure = os.environ.get('ENABLE_SECURE_COOKIE', '').strip().lower() in ('1', 'true', 'yes')
|
||
return {'max_age': _SESSION_MAX_AGE, 'httponly': True, 'samesite': 'lax',
|
||
'path': '/', 'secure': secure}
|
||
|
||
|
||
def _login_response(payload: dict, message: str, uid: int, seq: int) -> JSONResponse:
|
||
"""构造带会话 Cookie 的成功响应（Cookie 格式与 Flask 一致）。"""
|
||
resp = JSONResponse(content={'status': 200, 'success': True, 'message': message, 'data': payload})
|
||
cookie = cs.dumps(deps.secret_key(), {
|
||
'site_uid': uid, 'site_uid_seq': seq, 'site_uid_ts': int(time.time()),
|
||
})
|
||
resp.set_cookie(SESSION_COOKIE, cookie, **_cookie_kwargs())
|
||
return resp
|
||
|
||
|
||
@router.post('/auth/logout')
|
||
async def logout(user: Annotated[SiteUser | None, Depends(current_user)]):
|
||
"""退出登录：递增 login_seq 使本账号全部已签发会话失效，并删除本机 Cookie。"""
|
||
if user is not None:
|
||
try:
|
||
await db.execute(
|
||
'UPDATE users SET login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?',
|
||
(user.id,))
|
||
except Exception:
|
||
pass
|
||
resp = JSONResponse(content={'status': 200, 'success': True, 'message': '已退出登录'})
|
||
resp.delete_cookie(SESSION_COOKIE)
|
||
return resp
|
||
|
||
|
||
@router.post('/auth/heartbeat')
|
||
async def heartbeat(user: Annotated[SiteUser | None, Depends(current_user)]):
|
||
"""会话保活：返回最新次数；超限/失效给 401（expired 语义与 Flask 同）。"""
|
||
if user is None:
|
||
return error('登录已失效', 401, error_code='unauthorized')
|
||
row = await repo.get_user_by_id(user.id)
|
||
if row is None:
|
||
return error('登录已失效', 401, error_code='unauthorized')
|
||
credits = int(row.get('duration_seconds') or 0)
|
||
if row.get('role') != 'admin' and credits != -1 and credits <= 0:
|
||
return error('剩余次数已用完', 401, error_code='expired')
|
||
return success({'duration_seconds': row.get('duration_seconds'),
|
||
'duration_text': _format_credits(credits)})
|
||
|
||
|
||
@router.post('/auth/ann_seen')
|
||
async def ann_seen(user: Annotated[SiteUser | None, Depends(current_user)]):
|
||
"""标记公告已读（记录当前公告版本）。"""
|
||
if user is not None:
|
||
ann = await repo.get_announcement()
|
||
await db.execute('UPDATE users SET ann_seen = ? WHERE id = ?',
|
||
(ann.get('updated_at', ''), user.id))
|
||
return success()
|
||
|
||
|
||
@router.post('/auth/login')
|
||
async def login(request: Request):
|
||
"""登录：密码校验（含邮箱登录）、失败锁定、管理员两步验证转票据。"""
|
||
from auth_routes import (
|
||
_admin2fa_issue,
|
||
_admin2fa_mask,
|
||
_login_fail_check,
|
||
_login_fail_clear,
|
||
_login_fail_locked_log_once,
|
||
_login_fail_record,
|
||
)
|
||
|
||
data = await _request_data(request)
|
||
username = str(data.get('username') or '').strip()
|
||
password = str(data.get('password') or '')
|
||
ip = ratelimit.client_ip(request)
|
||
|
||
if len(password) > 128:
|
||
# 超长密码不做 PBKDF2（CPU 放大器），直接按密码错误拒绝
|
||
return error('用户名或密码错误')
|
||
|
||
try:
|
||
# 邮箱登录：含 @ 按归一化邮箱解析真实账号；锁定 key 用真实用户名
|
||
if '@' in username:
|
||
iuser = await asyncio.to_thread(auth.get_user_by_email, username)
|
||
lock_name = iuser['username'] if iuser else (auth._normalize_email(username) or username)
|
||
else:
|
||
iuser = None
|
||
lock_name = username
|
||
|
||
if not _login_fail_check(lock_name):
|
||
if _login_fail_locked_log_once(lock_name):
|
||
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
||
'登录', '失败次数过多，临时锁定')
|
||
return error('尝试过于频繁，该账号已临时锁定，请 10 分钟后再试')
|
||
|
||
verify_name = iuser['username'] if iuser is not None else username
|
||
user = await asyncio.to_thread(auth.verify_login, verify_name, password)
|
||
if not user:
|
||
_login_fail_record(lock_name)
|
||
return error('用户名或密码错误')
|
||
|
||
keys = user.keys()
|
||
status = user['status'] if 'status' in keys else 1
|
||
if status == 2:
|
||
reason = user['ban_reason'] if 'ban_reason' in keys else ''
|
||
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
||
'登录', f'封禁账号尝试登录: {reason}')
|
||
return error('账号已被封禁' + (f'：{reason}' if reason else '') + '，请联系管理员')
|
||
if status == 0:
|
||
return error('账号已被禁用，请联系管理员')
|
||
|
||
_login_fail_clear(lock_name)
|
||
|
||
if user['role'] == 'admin':
|
||
# 管理员一律转两步验证通道（密码正确才签发票据；未绑定邮箱 fail-closed）
|
||
admin_email = user['email'] if 'email' in keys else ''
|
||
if not admin_email:
|
||
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
||
'管理员登录', '未绑定邮箱，两步验证不可用')
|
||
return error('管理员账号未绑定邮箱，请联系管理员先完成绑定')
|
||
ticket = _admin2fa_issue(user)
|
||
await asyncio.to_thread(auth.add_log, 'success', lock_name, ip,
|
||
'管理员登录', '密码验证通过，待邮箱验证')
|
||
return success({'ticket': ticket, 'email_masked': _admin2fa_mask(admin_email),
|
||
'username': user['username'], 'admin_2fa': True},
|
||
'密码验证通过，请完成邮箱验证')
|
||
|
||
# 普通用户：建立会话（原子递增 login_seq 顶号）
|
||
uid = int(user['id'])
|
||
now = int(time.time())
|
||
await db.execute(
|
||
'UPDATE users SET last_active = ?, last_login = ?, '
|
||
'login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?', (now, now, uid))
|
||
seq_row = await repo.get_user_by_id(uid)
|
||
seq = int((seq_row or {}).get('login_seq') or 1)
|
||
|
||
banned, _info = await asyncio.to_thread(auth.record_login_location, user, ip)
|
||
if banned:
|
||
await db.execute(
|
||
'UPDATE users SET login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?', (uid,))
|
||
return error('检测到账号在多个省份异常登录，已自动封禁，请联系管理员')
|
||
|
||
fresh = await repo.get_user_by_id(uid) or {}
|
||
return _login_response({
|
||
'user': _user_public(fresh),
|
||
'duration_text': _format_credits(int(fresh.get('duration_seconds') or 0)),
|
||
'redirect': 'admin' if fresh.get('role') == 'admin' else 'user',
|
||
}, '登录成功', uid, seq)
|
||
except Exception:
|
||
return error('登录失败，请稍后重试', 500)
|