update：优化http_server代码路径

2026-07-23 23:53:55 +08:00 · 2025-02-15 20:31:55 +08:00
parent 2baba43946
commit 2fe4f44529
5 changed files with 2 additions and 67 deletions
@@ -0,0 +1,92 @@
+import logging
+import os
+from aiohttp import web
+from core.utils.util import get_local_ip, get_project_dir, check_password
+from manager.api.prompt import PromptApi
+from manager.api.auth import AuthApi
+from aiohttp.web_middlewares import middleware
+
+logger = logging.getLogger(__name__)
+
+
+@middleware
+async def cors_middleware(request, handler):
+    # 预检请求处理
+    if request.method == 'OPTIONS':
+        return web.Response(
+            status=204,
+            headers={
+                'Access-Control-Allow-Origin': '*',
+                'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, OPTIONS',
+                'Access-Control-Allow-Headers': 'Content-Type, Authorization, Token',
+                'Access-Control-Max-Age': '86400',
+            }
+        )
+
+    try:
+        response = await handler(request)
+    except web.HTTPException as ex:
+        response = ex
+
+    # 添加CORS头到所有响应
+    cors_headers = {
+        'Access-Control-Allow-Origin': '*',
+        'Access-Control-Expose-Headers': 'Content-Type, Authorization, Token',
+        'Vary': 'Origin'  # 避免缓存问题
+    }
+    response.headers.update(cors_headers)
+
+    return response
+
+
+class ConfigServer:
+    def __init__(self, config: dict):
+        self.config = config
+        self.app = web.Application(middlewares=[cors_middleware])  # 注册中间件
+
+        # 初始化接口处理器
+        self.prompt_handler = PromptApi(config)
+        self.auth_handler = AuthApi(config)
+        self.setup_routes()
+
+    def setup_routes(self):
+        # 注册prompt接口
+        self.app.router.add_get('/api/prompt', self.prompt_handler.get_prompt)
+        self.app.router.add_post('/api/prompt', self.prompt_handler.update_prompt)
+
+        # 注册auth接口
+        self.app.router.add_post('/api/login', self.auth_handler.login)
+
+        # 注册静态文件路由
+        static_dir = os.path.join(get_project_dir(), 'manager/static')  # 获取static目录绝对路径
+        self.app.router.add_static(
+            prefix='/manager/',  # 匹配前缀
+            path=static_dir,  # 静态文件目录
+            name='static'
+        )
+        self.app.router.add_get('/manager', self.redirect_to_index)
+
+    async def redirect_to_index(self, _):
+        raise web.HTTPFound('/manager/')
+
+    async def start(self):
+        try:
+            http_config = self.config['manager']
+            if not http_config.get('enabled', False):
+                logger.info("HTTP server is disabled")
+                return
+            token = self.config['manager']['token']
+            if not check_password(token):
+                logger.info("您设置的后台密码太弱了，启动后台管理失败！")
+                return
+
+            runner = web.AppRunner(self.app)
+            await runner.setup()
+            site = web.TCPSite(runner, http_config['ip'], http_config['port'])
+            await site.start()
+            logger.info(
+                f"Config HTTP server is running at http://{get_local_ip()}:{http_config['port']}/manager/login.html")
+            return runner  # 返回runner以便后续清理
+        except Exception as e:
+            logger.error(f"Failed to start HTTP server: {e}")
+            raise