mirror of https://github.com/Mai-with-u/MaiBot.git
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
"""WebUI 管理器 - 处理开发/生产环境的 WebUI 启动"""
|
||
import os
|
||
from pathlib import Path
|
||
from src.common.logger import get_logger
|
||
from .token_manager import get_token_manager
|
||
|
||
logger = get_logger("webui")
|
||
|
||
|
||
def setup_webui(mode: str = "production") -> bool:
|
||
"""
|
||
设置 WebUI
|
||
|
||
Args:
|
||
mode: 运行模式，"development" 或 "production"
|
||
|
||
Returns:
|
||
bool: 是否成功设置
|
||
"""
|
||
# 初始化 Token 管理器（确保 token 文件存在）
|
||
token_manager = get_token_manager()
|
||
current_token = token_manager.get_token()
|
||
logger.info(f"🔑 WebUI Access Token: {current_token}")
|
||
logger.info("💡 请使用此 Token 登录 WebUI")
|
||
|
||
if mode == "development":
|
||
return setup_dev_mode()
|
||
else:
|
||
return setup_production_mode()
|
||
|
||
|
||
def setup_dev_mode() -> bool:
|
||
"""设置开发模式 - 仅启用 CORS，前端自行启动"""
|
||
logger.info("📝 WebUI 开发模式已启用")
|
||
logger.info("🌐 请手动启动前端开发服务器: cd webui && npm run dev")
|
||
logger.info("💡 前端将运行在 http://localhost:7999")
|
||
return True
|
||
|
||
|
||
def setup_production_mode() -> bool:
|
||
"""设置生产模式 - 挂载静态文件"""
|
||
try:
|
||
from src.common.server import get_global_server
|
||
from fastapi.staticfiles import StaticFiles
|
||
from fastapi.responses import FileResponse
|
||
|
||
server = get_global_server()
|
||
base_dir = Path(__file__).parent.parent.parent
|
||
static_path = base_dir / "webui" / "dist"
|
||
|
||
if not static_path.exists():
|
||
logger.warning(f"❌ WebUI 静态文件目录不存在: {static_path}")
|
||
logger.warning("💡 请先构建前端: cd webui && npm run build")
|
||
return False
|
||
|
||
if not (static_path / "index.html").exists():
|
||
logger.warning(f"❌ 未找到 index.html: {static_path / 'index.html'}")
|
||
logger.warning("💡 请确认前端已正确构建")
|
||
return False
|
||
|
||
# 挂载静态资源
|
||
if (static_path / "assets").exists():
|
||
server.app.mount(
|
||
"/assets",
|
||
StaticFiles(directory=str(static_path / "assets")),
|
||
name="assets"
|
||
)
|
||
|
||
# 处理 SPA 路由
|
||
@server.app.get("/{full_path:path}")
|
||
async def serve_spa(full_path: str):
|
||
"""服务单页应用"""
|
||
# API 路由不处理
|
||
if full_path.startswith("api/"):
|
||
return None
|
||
|
||
# 检查文件是否存在
|
||
file_path = static_path / full_path
|
||
if file_path.is_file():
|
||
return FileResponse(file_path)
|
||
|
||
# 返回 index.html（SPA 路由）
|
||
return FileResponse(static_path / "index.html")
|
||
|
||
host = os.getenv("HOST", "127.0.0.1")
|
||
port = os.getenv("PORT", "8000")
|
||
logger.info("✅ WebUI 生产模式已挂载")
|
||
logger.info(f"🌐 访问 http://{host}:{port} 查看 WebUI")
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"挂载 WebUI 静态文件失败: {e}")
|
||
return False
|