mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
|
import asyncio
|
||
|
|
|
||
|
|
import nonebot
|
||
|
|
from fastapi import APIRouter, FastAPI
|
||
|
|
from nonebot.adapters.onebot.v11 import Bot, MessageEvent
|
||
|
|
from nonebot.log import default_filter, default_format
|
||
|
|
from nonebot.matcher import Matcher
|
||
|
|
from nonebot.message import run_preprocessor
|
||
|
|
from nonebot.typing import T_State
|
||
|
|
|
||
|
|
from zhenxun.configs.config import Config as gConfig
|
||
|
|
from zhenxun.services.log import logger, logger_
|
||
|
|
|
||
|
|
from .api.logs import router as ws_log_routes
|
||
|
|
from .api.logs.log_manager import LOG_STORAGE
|
||
|
|
from .api.tabs.database import router as database_router
|
||
|
|
from .api.tabs.main import router as main_router
|
||
|
|
from .api.tabs.main import ws_router as status_routes
|
||
|
|
from .api.tabs.manage import router as manage_router
|
||
|
|
from .api.tabs.manage import ws_router as chat_routes
|
||
|
|
from .api.tabs.plugin_manage import router as plugin_router
|
||
|
|
from .api.tabs.system import router as system_router
|
||
|
|
from .auth import router as auth_router
|
||
|
|
|
||
|
|
driver = nonebot.get_driver()
|
||
|
|
|
||
|
|
gConfig.add_plugin_config("web-ui", "username", "admin", help="前端管理用户名")
|
||
|
|
|
||
|
|
gConfig.add_plugin_config("web-ui", "password", None, help="前端管理密码")
|
||
|
|
|
||
|
|
gConfig.set_name("web-ui", "web-ui")
|
||
|
|
|
||
|
|
|
||
|
|
BaseApiRouter = APIRouter(prefix="/zhenxun/api")
|
||
|
|
|
||
|
|
|
||
|
|
BaseApiRouter.include_router(auth_router)
|
||
|
|
BaseApiRouter.include_router(main_router)
|
||
|
|
BaseApiRouter.include_router(manage_router)
|
||
|
|
BaseApiRouter.include_router(database_router)
|
||
|
|
BaseApiRouter.include_router(plugin_router)
|
||
|
|
BaseApiRouter.include_router(system_router)
|
||
|
|
|
||
|
|
|
||
|
|
WsApiRouter = APIRouter(prefix="/zhenxun/socket")
|
||
|
|
|
||
|
|
WsApiRouter.include_router(ws_log_routes)
|
||
|
|
WsApiRouter.include_router(status_routes)
|
||
|
|
WsApiRouter.include_router(chat_routes)
|
||
|
|
|
||
|
|
|
||
|
|
@driver.on_startup
|
||
|
|
def _():
|
||
|
|
try:
|
||
|
|
|
||
|
|
async def log_sink(message: str):
|
||
|
|
loop = None
|
||
|
|
if not loop:
|
||
|
|
try:
|
||
|
|
loop = asyncio.get_running_loop()
|
||
|
|
except Exception as e:
|
||
|
|
logger.warning("Web Ui log_sink", e=e)
|
||
|
|
if not loop:
|
||
|
|
loop = asyncio.new_event_loop()
|
||
|
|
loop.create_task(LOG_STORAGE.add(message.rstrip("\n")))
|
||
|
|
|
||
|
|
logger_.add(
|
||
|
|
log_sink, colorize=True, filter=default_filter, format=default_format
|
||
|
|
)
|
||
|
|
|
||
|
|
app: FastAPI = nonebot.get_app()
|
||
|
|
app.include_router(BaseApiRouter)
|
||
|
|
app.include_router(WsApiRouter)
|
||
|
|
logger.info("<g>API启动成功</g>", "Web UI")
|
||
|
|
except Exception as e:
|
||
|
|
logger.error("<g>API启动失败</g>", "Web UI", e=e)
|