summaryrefslogtreecommitdiff
path: root/ATRI/plugins/console/driver/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/console/driver/__init__.py')
1 files changed, 16 insertions, 32 deletions
diff --git a/ATRI/plugins/console/driver/__init__.py b/ATRI/plugins/console/driver/__init__.py
index e5dfdc6..b6b9f89 100644
--- a/ATRI/plugins/console/driver/__init__.py
+++ b/ATRI/plugins/console/driver/__init__.py
@@ -5,32 +5,14 @@ from fastapi.middleware.cors import CORSMiddleware
from ATRI import conf
from ATRI.log import log
-from ATRI.plugins.console.data_source import FRONTEND_DIR
-from .view import (
- handle_auther,
- handle_base_uri,
- handle_control_service,
- handle_edit_block,
- handle_get_block_list,
- handle_get_service_list,
- handle_runtime_info,
- handle_message_deal_info,
-)
+from .path import *
+from .api import *
-CONSOLE_API_URI = "/capi" # base point
-CONSOLE_API_AUTH_URI = "/capi/auth" # 验证后台许可
-CONSOLE_API_RUNTIME_URI = "/capi/runtime" # 获取运行占用信息
-CONSOLE_API_MESSAGE_URI = "/capi/message" # 获取信息处理信息
+from ..data_source import FRONTEND_DIR
-CONSOLE_API_SERVICE_LIST_URI = "/capi/service/list" # 获取服务列表
-CONSOLE_API_SERVICE_CONTROL_URI = "/capi/service/control" # 对服务作出修改
-CONSOLE_API_BLOCK_LIST_URI = "/capi/block/list" # 获取封禁列表
-CONSOLE_API_BLOCK_EDIT_URI = "/capi/block/edit" # 编辑封禁列表
-
-
-def register_routes(driver: Driver):
+def register_driver(driver: Driver):
app = driver.server_app
origins = ["*"]
@@ -42,16 +24,18 @@ def register_routes(driver: Driver):
allow_headers=["Content-Type"],
)
- app.get(CONSOLE_API_URI)(handle_base_uri)
- app.get(CONSOLE_API_AUTH_URI)(handle_auther)
- app.get(CONSOLE_API_RUNTIME_URI)(handle_runtime_info)
- app.get(CONSOLE_API_MESSAGE_URI)(handle_message_deal_info)
+ app.get(CONSOLE_BASE_URL)(base_url)
+
+ app.get(CONSOLE_AUTH_URL)(auth_info)
+
+ app.websocket(CONSOLE_RUNTIME_INFO_URL)(runtime_info)
+ app.websocket(CONSOLE_MESSAGE_INFO_URL)(message_info)
- app.get(CONSOLE_API_SERVICE_LIST_URI)(handle_get_service_list)
- app.get(CONSOLE_API_SERVICE_CONTROL_URI)(handle_control_service)
+ app.get(CONSOLE_SERVICE_LIST_URL)(service_list)
+ app.get(CONSOLE_SERVICE_EDIT_URL)(edit_service)
- app.get(CONSOLE_API_BLOCK_LIST_URI)(handle_get_block_list)
- app.get(CONSOLE_API_BLOCK_EDIT_URI)(handle_edit_block)
+ app.get(CONSOLE_BLOCK_LIST_URL)(block_list_info)
+ app.get(CONSOLE_BLOCK_EDIT_URL)(edit_block_list)
static_path = str(FRONTEND_DIR)
app.mount(
@@ -64,6 +48,6 @@ def register_routes(driver: Driver):
def init_driver():
from ATRI import driver
+ register_driver(driver()) # type: ignore
c_url = f"{conf.BotConfig.host}:{conf.BotConfig.port}"
- log.info(f"控制台将运行于: http://{c_url} 对应API节点为: /capi")
- register_routes(driver()) # type: ignore
+ log.success(f"控制台将运行于: http://{c_url} 对应API节点为: /capi")