mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import os
|
|
|
|
import quart
|
|
import quart_cors
|
|
|
|
from ....core import app, entities as core_entities
|
|
from .groups import logs, system, settings, plugins, stats, user
|
|
from . import group
|
|
|
|
|
|
class HTTPController:
|
|
|
|
ap: app.Application
|
|
|
|
quart_app: quart.Quart
|
|
|
|
def __init__(self, ap: app.Application) -> None:
|
|
self.ap = ap
|
|
self.quart_app = quart.Quart(__name__)
|
|
quart_cors.cors(self.quart_app, allow_origin="*")
|
|
|
|
async def initialize(self) -> None:
|
|
await self.register_routes()
|
|
|
|
async def run(self) -> None:
|
|
if self.ap.system_cfg.data["http-api"]["enable"]:
|
|
|
|
async def shutdown_trigger_placeholder():
|
|
while True:
|
|
await asyncio.sleep(1)
|
|
|
|
async def exception_handler(*args, **kwargs):
|
|
try:
|
|
await self.quart_app.run_task(
|
|
*args, **kwargs
|
|
)
|
|
except Exception as e:
|
|
self.ap.logger.error(f"启动 HTTP 服务失败： {e}")
|
|
|
|
self.ap.task_mgr.create_task(
|
|
exception_handler(
|
|
host=self.ap.system_cfg.data["http-api"]["host"],
|
|
port=self.ap.system_cfg.data["http-api"]["port"],
|
|
shutdown_trigger=shutdown_trigger_placeholder,
|
|
),
|
|
name="http-api-quart",
|
|
scopes=[core_entities.LifecycleControlScope.APPLICATION],
|
|
)
|
|
|
|
# await asyncio.sleep(5)
|
|
|
|
async def register_routes(self) -> None:
|
|
|
|
@self.quart_app.route("/healthz")
|
|
async def healthz():
|
|
return {"code": 0, "msg": "ok"}
|
|
|
|
for g in group.preregistered_groups:
|
|
ginst = g(self.ap, self.quart_app)
|
|
await ginst.initialize()
|
|
|
|
frontend_path = "web/dist"
|
|
|
|
@self.quart_app.route("/")
|
|
async def index():
|
|
return await quart.send_from_directory(
|
|
frontend_path,
|
|
"index.html",
|
|
mimetype="text/html"
|
|
)
|
|
|
|
@self.quart_app.route("/<path:path>")
|
|
async def static_file(path: str):
|
|
|
|
mimetype = None
|
|
|
|
if path.endswith(".html"):
|
|
mimetype = "text/html"
|
|
elif path.endswith(".js"):
|
|
mimetype = "application/javascript"
|
|
elif path.endswith(".css"):
|
|
mimetype = "text/css"
|
|
elif path.endswith(".png"):
|
|
mimetype = "image/png"
|
|
elif path.endswith(".jpg"):
|
|
mimetype = "image/jpeg"
|
|
elif path.endswith(".jpeg"):
|
|
mimetype = "image/jpeg"
|
|
elif path.endswith(".gif"):
|
|
mimetype = "image/gif"
|
|
elif path.endswith(".svg"):
|
|
mimetype = "image/svg+xml"
|
|
elif path.endswith(".ico"):
|
|
mimetype = "image/x-icon"
|
|
elif path.endswith(".json"):
|
|
mimetype = "application/json"
|
|
elif path.endswith(".txt"):
|
|
mimetype = "text/plain"
|
|
|
|
return await quart.send_from_directory(
|
|
frontend_path,
|
|
path,
|
|
mimetype=mimetype
|
|
) |