Initial commit

plugin/spa/spa_proxy.py

@@ -0,0 +1,50 @@
+import os
+
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
+
+from plugin.base import Plugin
+from uvicorn.server import logger
+
+
+class SpaProxy(Plugin):
+    def __init__(self, app: FastAPI, dist: str = "dist"):
+        self.app = app
+        self.dist = dist
+        self.name = "SPA Proxy"
+        self.version = "1.0.0"
+
+    def install(self):
+        # --- [新增] 挂载前端静态文件 --
+        frontend_dir = os.path.join(os.getcwd(), self.dist)
+        if not os.path.isdir(frontend_dir):
+            return
+
+        # 1. 挂载静态资源（如 /assets/...）
+        assets_dir = os.path.join(frontend_dir, "assets")
+        if os.path.isdir(assets_dir):
+            self.app.mount("/assets", StaticFiles(directory=assets_dir), name="assets")
+
+        # 2. 【关键】兜底路由：处理所有未被 API 匹配的路径（支持前端路由如 /analysis）
+        @self.app.get("/{full_path:path}")
+        async def serve_frontend(full_path: str):
+            # 如果请求的是 API 路径，不应走到这里（因为 API 路由已先注册）
+            index_path = os.path.join(frontend_dir, "index.html")
+            if os.path.isfile(index_path):
+                return FileResponse(index_path)
+            raise HTTPException(status_code=404, detail="Frontend not found")
+
+        # 3. 显式处理根路径（可选，但更清晰）
+        @self.app.get("/")
+        async def root():
+            return await serve_frontend("")
+
+        # self.app.add_api_route(
+        #     "/{full_path:path}",
+        #     serve_frontend,
+        #     methods=["GET"],
+        #     response_class=FileResponse,
+        # )
+
+        logger.info(f"前端静态文件挂载: {frontend_dir} ✅")