1 files changed, 56 insertions, 0 deletions
diff --git a/ATRI/plugins/console/driver/__init__.py b/ATRI/plugins/console/driver/__init__.py
new file mode 100644
index 0000000..fe4398a
--- /dev/null
+++ b/ATRI/plugins/console/driver/__init__.py
@@ -0,0 +1,56 @@
+from nonebot import get_driver
+from nonebot.drivers.fastapi import Driver
+
+from fastapi.middleware.cors import CORSMiddleware
+
+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,
+)
+
+
+CONSOLE_API_URI = "/capi"  # base point
+CONSOLE_API_AUTH_URI = "/capi/auth"  # 验证后台许可
+CONSOLE_API_RUNTIME_URI = "/capi/runtime"  # 获取运行占用信息
+CONSOLE_API_MESSAGE_URI = "/capi/message"  # 获取信息处理信息
+
+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):
+    app = driver.server_app
+
+    origins = ["*"]
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=origins,
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+    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_API_SERVICE_LIST_URI)(handle_get_service_list)
+    app.get(CONSOLE_API_SERVICE_CONTROL_URI)(handle_control_service)
+
+    app.get(CONSOLE_API_BLOCK_LIST_URI)(handle_get_block_list)
+    app.get(CONSOLE_API_BLOCK_EDIT_URI)(handle_edit_block)
+
+
+def init():
+    driver = get_driver()
+    register_routes(driver)  # type: ignore