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/ 指纹未变；忽略注释的逐字等价性校验通过。
79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""音频流代理。
|
||
|
||
要点：
|
||
- 签名校验：token 由 core.security 校验（密钥同源）；失效 403。
|
||
- HEAD 无正文：直接 200 + Accept-Ranges，不请求上游。
|
||
- Range 透传：客户端 Range 原样转发上游，206/Content-Range 原样回传（拖动进度可用）。
|
||
- 编码透明：上游若带 Content-Encoding（aiohttp 会解压），不得沿用压缩前的
|
||
Content-Length/Content-Range（否则头体不一致）。
|
||
- 流式转发：16KB 分块，不整段入内存；下游中断时 finally 关闭上游响应。
|
||
- Cache-Control: no-store。
|
||
"""
|
||
from fastapi import APIRouter, Request
|
||
from fastapi.responses import Response, StreamingResponse
|
||
|
||
from core import security
|
||
from core.http_client import client as http_client
|
||
from web.response import error
|
||
|
||
router = APIRouter()
|
||
|
||
_UPSTREAM_HEADERS = {
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
|
||
'Referer': 'https://music.163.com/',
|
||
}
|
||
|
||
_CHUNK = 16 * 1024
|
||
|
||
|
||
@router.api_route('/audio_stream', methods=['GET', 'HEAD'])
|
||
async def audio_stream(request: Request):
|
||
"""签名音频代理流（支持 Range 拖动进度）。"""
|
||
if request.method == 'HEAD':
|
||
return Response(status_code=200, headers={'Accept-Ranges': 'bytes'})
|
||
|
||
token = request.query_params.get('token', '')
|
||
url = security.verify_audio_token(token)
|
||
if not url:
|
||
return error('音频链接已过期或无效，请重新解析', 403)
|
||
|
||
upstream_headers = dict(_UPSTREAM_HEADERS)
|
||
range_header = request.headers.get('range')
|
||
if range_header:
|
||
upstream_headers['Range'] = range_header
|
||
|
||
try:
|
||
upstream = await http_client.open('GET', url, headers=upstream_headers, timeout=20)
|
||
except Exception:
|
||
return error('音频获取失败', 502)
|
||
|
||
if upstream.status >= 400:
|
||
upstream.close()
|
||
return error('音频获取失败', 502)
|
||
|
||
upstream_encoded = bool(upstream.headers.get('Content-Encoding'))
|
||
pass_headers: dict[str, str] = {}
|
||
if not upstream_encoded:
|
||
for h in ('Content-Length', 'Content-Range', 'Accept-Ranges'):
|
||
if h in upstream.headers:
|
||
pass_headers[h] = upstream.headers[h]
|
||
elif 'Accept-Ranges' in upstream.headers:
|
||
pass_headers['Accept-Ranges'] = upstream.headers['Accept-Ranges']
|
||
pass_headers['Cache-Control'] = 'no-store'
|
||
|
||
async def _body():
|
||
try:
|
||
async for chunk in upstream.content.iter_chunked(_CHUNK):
|
||
yield chunk
|
||
finally:
|
||
# 下游断开/提前关闭时也释放上游连接
|
||
upstream.close()
|
||
|
||
return StreamingResponse(
|
||
_body(),
|
||
status_code=upstream.status,
|
||
media_type=upstream.headers.get('Content-Type', 'audio/mpeg'),
|
||
headers=pass_headers,
|
||
)
|