151 lines
4.6 KiB
Python
151 lines
4.6 KiB
Python
from fastapi import FastAPI, Request
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from fastapi.staticfiles import StaticFiles
|
||
import os
|
||
import logging
|
||
from app.core.config import settings
|
||
from app.api.v1 import (
|
||
auth,
|
||
client,
|
||
admin,
|
||
analytics,
|
||
admin_config,
|
||
feature,
|
||
checkin,
|
||
user_profile,
|
||
stats,
|
||
algorithm,
|
||
logs,
|
||
ai_chat,
|
||
ai_pattern,
|
||
ai_identify,
|
||
)
|
||
from app.db import init_db
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
app = FastAPI(title=settings.PROJECT_NAME)
|
||
log = logging.getLogger("designercep.startup")
|
||
|
||
SERVER_DIR = Path(__file__).resolve().parents[2]
|
||
ARCHIVES_DIR = SERVER_DIR / "archives"
|
||
|
||
# Ensure archives directory exists
|
||
os.makedirs(ARCHIVES_DIR, exist_ok=True)
|
||
|
||
# ========== CORS 配置 ==========
|
||
IS_DEV = settings.ENV == "development"
|
||
|
||
if IS_DEV:
|
||
# 开发环境：保持宽松
|
||
print("⚠️ Running in Development Mode: CORS allows *")
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
else:
|
||
# 生产环境：严格配置
|
||
allowed_origins = settings.ALLOWED_ORIGINS.split(",")
|
||
print(f"🔒 Running in Production Mode: CORS allowed origins: {allowed_origins}")
|
||
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=allowed_origins,
|
||
allow_credentials=True,
|
||
allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
# ✅ CEP 环境特殊处理 (Origin: null or cep://)
|
||
@app.middleware("http")
|
||
async def cep_cors_middleware(request: Request, call_next):
|
||
origin = request.headers.get("origin")
|
||
|
||
# CEP 的 Origin 是 null 或 cep://
|
||
if origin in ["null", None] or (origin and origin.startswith("cep://")):
|
||
response = await call_next(request)
|
||
response.headers["Access-Control-Allow-Origin"] = "*"
|
||
response.headers["Access-Control-Allow-Credentials"] = "true"
|
||
response.headers["Access-Control-Allow-Methods"] = (
|
||
"GET, POST, PUT, DELETE, OPTIONS"
|
||
)
|
||
response.headers["Access-Control-Allow-Headers"] = "*"
|
||
return response
|
||
|
||
return await call_next(request)
|
||
|
||
|
||
app.include_router(
|
||
auth.router, prefix=f"{settings.API_V1_STR}/auth", tags=["authentication"]
|
||
)
|
||
app.include_router(
|
||
client.router, prefix=f"{settings.API_V1_STR}/client", tags=["client"]
|
||
)
|
||
app.include_router(admin.router, prefix=f"{settings.API_V1_STR}/admin", tags=["admin"])
|
||
app.include_router(
|
||
analytics.router, prefix=f"{settings.API_V1_STR}/analytics", tags=["analytics"]
|
||
)
|
||
|
||
# 新增路由
|
||
app.include_router(
|
||
admin_config.router, prefix=settings.API_V1_STR, tags=["admin-config"]
|
||
)
|
||
app.include_router(feature.router, prefix=settings.API_V1_STR, tags=["feature"])
|
||
app.include_router(checkin.router, prefix=settings.API_V1_STR, tags=["checkin"])
|
||
app.include_router(
|
||
user_profile.router, prefix=settings.API_V1_STR, tags=["user-profile"]
|
||
)
|
||
app.include_router(stats.router, prefix=settings.API_V1_STR, tags=["stats"])
|
||
app.include_router(algorithm.router, prefix=settings.API_V1_STR, tags=["algorithm"])
|
||
app.include_router(logs.router, prefix=settings.API_V1_STR, tags=["logs"])
|
||
app.include_router(ai_chat.router, prefix=settings.API_V1_STR, tags=["ai-chat"])
|
||
app.include_router(ai_pattern.router, prefix=settings.API_V1_STR, tags=["ai-pattern"])
|
||
app.include_router(ai_identify.router, prefix=settings.API_V1_STR, tags=["ai-identify"])
|
||
|
||
|
||
# Health Check
|
||
@app.get("/health")
|
||
def health_check():
|
||
return {
|
||
"status": "healthy",
|
||
"timestamp": datetime.now().isoformat(),
|
||
"env": "development" if IS_DEV else "production",
|
||
}
|
||
|
||
|
||
# ========== Static Files (Development Only) ==========
|
||
if IS_DEV:
|
||
# Mount archives directory for download
|
||
app.mount("/download", StaticFiles(directory=ARCHIVES_DIR), name="download")
|
||
else:
|
||
print(
|
||
"ℹ️ Production Mode: Static files are NOT mounted by FastAPI (handled by Caddy/Nginx)."
|
||
)
|
||
|
||
|
||
@app.get("/")
|
||
def read_root():
|
||
from fastapi.responses import RedirectResponse
|
||
|
||
if IS_DEV:
|
||
# 重定向到 Shell 登录页
|
||
return RedirectResponse(url="/shell/index.html")
|
||
return {"message": "DesignerCEP API is running"}
|
||
|
||
|
||
@app.on_event("startup")
|
||
def on_startup():
|
||
# 应用启动时初始化数据库（创建表）
|
||
init_db()
|
||
for warning in settings.runtime_warnings():
|
||
log.warning("[Config] %s", warning)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
|
||
uvicorn.run("app.main:app", host="0.0.0.0", port=8000, reload=True)
|