orion
5e4c3ed5e3
为 FastAPI 应用启用 Gzip，并兼容 Vite 在 /theme-assets/ 基址下生成的 assets 子路径。 - 压缩超过 1KB 的静态资源和 API 响应，保留不可变缓存策略 - 同时兼容 /theme-assets/<file> 与 /theme-assets/assets/<file> - 保留 /assets 旧入口，避免影响历史主题 - 已通过 Python 编译检查
244 lines
7.3 KiB
Python
244 lines
7.3 KiB
Python
# @Time : 2023/8/9 23:23
|
|
# @Author : Lan
|
|
# @File : main.py
|
|
# @Software: PyCharm
|
|
import asyncio
|
|
import re
|
|
import time
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import FileResponse, HTMLResponse, PlainTextResponse
|
|
from starlette.middleware.gzip import GZipMiddleware
|
|
from tortoise import Tortoise
|
|
from tortoise.contrib.fastapi import register_tortoise
|
|
|
|
from apps.admin.views import admin_api
|
|
from apps.base.models import KeyValue
|
|
from apps.base.utils import ip_limit
|
|
from apps.base.views import share_api, chunk_api, presign_api
|
|
from core.config import ensure_settings_row, refresh_settings
|
|
from core.database import db_startup_lock, get_db_config, init_db
|
|
from core.logger import logger
|
|
from core.response import APIResponse
|
|
from core.settings import settings, BASE_DIR, DEFAULT_CONFIG
|
|
from core.tasks import delete_expire_files, clean_incomplete_uploads
|
|
from core.utils import hash_password, is_password_hashed
|
|
from core.version import APP_VERSION
|
|
|
|
|
|
def build_public_config() -> dict:
|
|
return {
|
|
"name": settings.name,
|
|
"description": settings.description,
|
|
"explain": settings.page_explain,
|
|
"uploadSize": settings.uploadSize,
|
|
"expireStyle": settings.expireStyle,
|
|
"enableChunk": settings.enableChunk,
|
|
"openUpload": settings.openUpload,
|
|
"notify_title": settings.notify_title,
|
|
"notify_content": settings.notify_content,
|
|
"show_admin_address": settings.showAdminAddr,
|
|
"max_save_seconds": settings.max_save_seconds,
|
|
}
|
|
|
|
|
|
def build_public_meta() -> dict:
|
|
return {
|
|
"version": APP_VERSION,
|
|
"api": {
|
|
"legacyConfig": "/",
|
|
"publicConfig": "/api/v1/config",
|
|
"health": "/health",
|
|
},
|
|
"features": {
|
|
"chunkUpload": bool(settings.enableChunk),
|
|
"guestUpload": bool(settings.openUpload),
|
|
"adminAddressVisible": bool(settings.showAdminAddr),
|
|
"expirationModes": settings.expireStyle,
|
|
},
|
|
"limits": {
|
|
"uploadSize": settings.uploadSize,
|
|
"maxSaveSeconds": settings.max_save_seconds,
|
|
"uploadWindowMinutes": settings.uploadMinute,
|
|
"uploadWindowCount": settings.uploadCount,
|
|
},
|
|
}
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
logger.info("正在初始化应用...")
|
|
# 初始化数据库
|
|
await init_db()
|
|
|
|
# 加载配置（多进程下串行化启动写操作）
|
|
async with db_startup_lock():
|
|
await load_config()
|
|
# 启动后台任务
|
|
task = asyncio.create_task(delete_expire_files())
|
|
chunk_cleanup_task = asyncio.create_task(clean_incomplete_uploads())
|
|
logger.info("应用初始化完成")
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
# 清理操作
|
|
logger.info("正在关闭应用...")
|
|
task.cancel()
|
|
chunk_cleanup_task.cancel()
|
|
await asyncio.gather(task, chunk_cleanup_task, return_exceptions=True)
|
|
await Tortoise.close_connections()
|
|
logger.info("应用已关闭")
|
|
|
|
|
|
async def load_config():
|
|
await ensure_settings_row()
|
|
await KeyValue.update_or_create(
|
|
key="sys_start", defaults={"value": int(time.time() * 1000)}
|
|
)
|
|
await refresh_settings()
|
|
|
|
await migrate_password_to_hash()
|
|
|
|
ip_limit["error"].minutes = settings.errorMinute
|
|
ip_limit["error"].count = settings.errorCount
|
|
ip_limit["upload"].minutes = settings.uploadMinute
|
|
ip_limit["upload"].count = settings.uploadCount
|
|
|
|
|
|
async def migrate_password_to_hash():
|
|
if not is_password_hashed(settings.admin_token):
|
|
hashed = hash_password(settings.admin_token)
|
|
settings.admin_token = hashed
|
|
config_record = await KeyValue.filter(key="settings").first()
|
|
if config_record and config_record.value:
|
|
config_record.value["admin_token"] = hashed
|
|
await config_record.save()
|
|
logger.info("已将管理员密码迁移为哈希存储")
|
|
|
|
|
|
app = FastAPI(lifespan=lifespan)
|
|
|
|
app.add_middleware(GZipMiddleware, minimum_size=1024, compresslevel=6)
|
|
|
|
@app.middleware("http")
|
|
async def refresh_settings_middleware(request, call_next):
|
|
await refresh_settings()
|
|
return await call_next(request)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# 使用 register_tortoise 来添加异常处理器
|
|
register_tortoise(
|
|
app,
|
|
config=get_db_config(),
|
|
generate_schemas=False,
|
|
add_exception_handlers=True,
|
|
)
|
|
|
|
app.include_router(share_api)
|
|
app.include_router(chunk_api)
|
|
app.include_router(presign_api)
|
|
app.include_router(presign_api, prefix="/api")
|
|
app.include_router(admin_api)
|
|
|
|
|
|
@app.get("/theme-assets/{asset_path:path}")
|
|
@app.get("/assets/{asset_path:path}")
|
|
async def theme_asset(asset_path: str):
|
|
# Vite with base=/theme-assets/ emits /theme-assets/assets/<file> while
|
|
# legacy themes use /theme-assets/<file>. Keep both forms compatible.
|
|
normalized_asset_path = asset_path.removeprefix("assets/")
|
|
theme_assets = (BASE_DIR / settings.themesSelect / "assets").resolve()
|
|
target = (theme_assets / normalized_asset_path).resolve()
|
|
try:
|
|
target.relative_to(theme_assets)
|
|
except ValueError:
|
|
return PlainTextResponse("Not found", status_code=404)
|
|
if not target.is_file():
|
|
return PlainTextResponse("Not found", status_code=404)
|
|
return FileResponse(
|
|
target,
|
|
headers={"Cache-Control": "public, max-age=31536000, immutable"},
|
|
)
|
|
|
|
|
|
@app.exception_handler(404)
|
|
@app.get("/")
|
|
async def index(request=None, exc=None):
|
|
theme_dir = BASE_DIR / settings.themesSelect
|
|
asset_version = str(int((theme_dir / "index.html").stat().st_mtime))
|
|
content = (
|
|
open(theme_dir / "index.html", "r", encoding="utf-8")
|
|
.read()
|
|
.replace("{{title}}", str(settings.name))
|
|
.replace("{{description}}", str(settings.description))
|
|
.replace("{{keywords}}", str(settings.keywords))
|
|
.replace("{{opacity}}", str(settings.opacity))
|
|
.replace('"/assets/', '"/theme-assets/')
|
|
.replace('"assets/', '"/theme-assets/')
|
|
.replace("{{background}}", str(settings.background))
|
|
)
|
|
content = re.sub(
|
|
r'((?:theme-)?assets/[^"\']+\.(?:css|js|json|png))(["\'])',
|
|
rf"\1?v={asset_version}\2",
|
|
content,
|
|
)
|
|
return HTMLResponse(
|
|
content=content,
|
|
media_type="text/html",
|
|
headers={"Cache-Control": "no-cache"},
|
|
)
|
|
|
|
|
|
@app.get("/robots.txt")
|
|
async def robots():
|
|
return HTMLResponse(content=settings.robotsText, media_type="text/plain")
|
|
|
|
|
|
@app.post("/")
|
|
async def get_config():
|
|
return APIResponse(detail=build_public_config())
|
|
|
|
|
|
@app.get("/api/v1/config")
|
|
async def get_public_config():
|
|
return APIResponse(
|
|
detail={
|
|
"config": build_public_config(),
|
|
"meta": build_public_meta(),
|
|
}
|
|
)
|
|
|
|
|
|
@app.get("/health")
|
|
async def health_check():
|
|
return APIResponse(
|
|
detail={
|
|
"status": "ok",
|
|
"version": APP_VERSION,
|
|
"storage": settings.file_storage,
|
|
"theme": settings.themesSelect,
|
|
}
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run(
|
|
app="main:app",
|
|
host=settings.serverHost,
|
|
port=settings.serverPort,
|
|
reload=False,
|
|
workers=settings.serverWorkers,
|
|
)
|