1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
from nonebot.drivers.fastapi import Driver
from fastapi.middleware.cors import CORSMiddleware
from ATRI import driver
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():
register_routes(driver) # type: ignore
driver().on_startup(init)
|
