Archived
snowy-yang
6288a98ecc
为什么：单文件 1100~1900 行的巨石模块让「改一处要通读全文」；同时注释里残留 大量已删台账的编号，以及「Flask/FastAPI 双栈迁移期」这类已失实的架构描述， 会误导后来者按不复存在的架构去改代码。 拆分（对外导入路径不变，行为零变更）： - music_downloader.py -> music_downloader/（models/downloader/cover/_http/_log/_concurrency） - auth_system.py -> auth_system/（db/security/users/credits/ledger/mail/cards/backup...） - music_api.py -> music_api/（api/client/crypto/qr/cookies/compat/constants/errors） - services/runtime.py -> services/（api_service/_shared/_taskcore/downloads/batch/podcast/ streaming/tasks）；services/runtime.py 保留为兼容门面 兼容性做法：各包用 __init__.py 以 __all__ 原样再导出全部符号，既有调用点 （auth_system.xxx / services.xxx / from music_api import Y）零改动；路径常量改为经包属性 动态读取（auth_system.DB_PATH），否则测试的 monkeypatch 会静默失效并写到生产库。 修复 9 处代码损伤（上一轮注释清理把 `()` 当「空括号残留」删掉所致；用「与拆分前逐字 等价」校验逐一定位，静态检查只会报出其中 4 处）： - core/config.py ).strip() or '*' - core/state.py threading.Lock() x2 - qr_login.py datetime.now() / .strip() - music_downloader/downloader.py .lower() - services/podcast.py set() / time.time() x2 清理与删除： - 删除 部署文档.md / 功能介绍.md / README_私有包警告.txt - 清理指向已删台账与文档的过时编号注释；把「迁移期 / 两栈 / 双框架 / 已迁移」等 失实描述改写为现状描述（保留 `<-> Flask /path` 映射表等仍有价值的契约来源说明） - _mkpkg.py：公开包剔除清单去掉已不存在的文档名；私有包警告改为自包含 - pyproject.toml：补 pythonpath = ["."]，使裸跑 pytest 不再 import core 失败 验证：ruff 全通过；pyright 0 errors / 0 warnings（102 文件）；pytest 144 passed； 真实 uvicorn 冒烟 6 条路由符合预期且生产 data/ 指纹未变；忽略注释的逐字等价性校验通过。
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""音频直链多重防护（HMAC 签名 + 时效 + 代理流）——框架无关。
|
||
|
||
密钥来源为**启动时注入**（`init_secret_key`），不依赖任何 Web 框架；
|
||
签名与校验算法逐字保持与历史实现一致，
|
||
故存量客户端已持有的直链 token 仍然有效。
|
||
"""
|
||
|
||
import base64
|
||
import hashlib
|
||
import hmac
|
||
import secrets
|
||
import time
|
||
from typing import Any
|
||
|
||
from core.config import AUDIO_HOST_MARKS, AUDIO_TOKEN_TTL, AUDIO_URL_KEYS
|
||
|
||
# 由应用启动时注入（来源：auth_system.get_secret_key() → app.secret_key）
|
||
_SECRET_KEY: bytes = b''
|
||
|
||
|
||
def init_secret_key(key) -> None:
|
||
"""注入签名密钥（str 或 bytes）。须在生成/校验音频 token 前调用一次。"""
|
||
global _SECRET_KEY
|
||
_SECRET_KEY = key.encode('utf-8') if isinstance(key, str) else key
|
||
|
||
|
||
def is_audio_url(v) -> bool:
|
||
return isinstance(v, str) and v.startswith('http') and any(m in v for m in AUDIO_HOST_MARKS)
|
||
|
||
|
||
def sign_payload(payload: str) -> str:
|
||
return hmac.new(_SECRET_KEY, payload.encode('utf-8'), hashlib.sha256).hexdigest()[:32]
|
||
|
||
|
||
def make_audio_token(url: str) -> str:
|
||
exp = int(time.time()) + AUDIO_TOKEN_TTL
|
||
nonce = secrets.token_hex(4)
|
||
payload = base64.urlsafe_b64encode(f'{exp}|{nonce}|{url}'.encode()).decode().rstrip('=')
|
||
return payload + '.' + sign_payload(payload)
|
||
|
||
|
||
def verify_audio_token(token: str) -> str | None:
|
||
try:
|
||
payload, sig = token.rsplit('.', 1)
|
||
if not hmac.compare_digest(sig, sign_payload(payload)):
|
||
return None
|
||
pad = '=' * (-len(payload) % 4)
|
||
exp, _nonce, url = base64.urlsafe_b64decode(payload + pad).decode('utf-8').split('|', 2)
|
||
if int(exp) < time.time():
|
||
return None
|
||
return url
|
||
except Exception:
|
||
return None
|
||
|
||
|
||
def protect_audio_urls(obj: Any) -> bool:
|
||
"""递归就地替换 dict/list 中的音频直链为签名代理地址，返回是否发生替换"""
|
||
hit = {'v': False}
|
||
|
||
def walk(o):
|
||
if isinstance(o, dict):
|
||
for k, v in list(o.items()):
|
||
if k in AUDIO_URL_KEYS and is_audio_url(v):
|
||
o[k] = '/audio_stream?token=' + make_audio_token(v)
|
||
hit['v'] = True
|
||
else:
|
||
walk(v)
|
||
elif isinstance(o, list):
|
||
for it in o:
|
||
walk(it)
|
||
|
||
walk(obj)
|
||
return hit['v']
|