summaryrefslogtreecommitdiff
path: root/ATRI/plugins/console/driver
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/console/driver')
7 files changed, 346 insertions, 322 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")
diff --git a/ATRI/plugins/console/driver/api.py b/ATRI/plugins/console/driver/api.py
index 4750148..4b78906 100644
--- a/ATRI/plugins/console/driver/api.py
+++ b/ATRI/plugins/console/driver/api.py
@@ -1,202 +1,86 @@
-import os
-import json
-import time
-import psutil
-from pathlib import Path
-from datetime import datetime
+import asyncio
-from ATRI.service import ServiceTools, SERVICES_DIR
-from ATRI.exceptions import GetStatusError
-from ..models import PlatformRuntimeInfo, BotRuntimeInfo, ServiceInfo
+from fastapi import Depends, status
+from starlette.websockets import WebSocket, WebSocketState
+from websockets.exceptions import ConnectionClosedOK
-
-def get_processing_data() -> tuple:
- try:
- p_cpu = psutil.cpu_percent(interval=1)
- p_mem = psutil.virtual_memory().percent
- disk = psutil.disk_usage("/").percent
- inte_send = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore
- inte_recv = psutil.net_io_counters().bytes_recv / 1000000 # type: ignore
-
- process = psutil.Process(os.getpid())
- b_cpu = process.cpu_percent(interval=1)
- b_mem = process.memory_percent(memtype="rss")
-
- now = time.time()
- boot = psutil.boot_time()
- b = process.create_time()
- boot_time = str(
- datetime.utcfromtimestamp(now).replace(microsecond=0)
- - datetime.utcfromtimestamp(boot).replace(microsecond=0)
- )
- bot_time = str(
- datetime.utcfromtimestamp(now).replace(microsecond=0)
- - datetime.utcfromtimestamp(b).replace(microsecond=0)
- )
- except Exception:
- raise GetStatusError("Getting runtime failed.")
-
- if p_cpu > 90: # type: ignore
- msg = "咱感觉有些头晕..."
- if p_mem > 90:
- msg = "咱感觉有点头晕并且有点累..."
- elif p_mem > 90:
- msg = "咱感觉有点累..."
- elif disk > 90:
- msg = "咱感觉身体要被塞满了..."
- else:
- msg = "アトリは、高性能ですから!"
-
- return (
- PlatformRuntimeInfo(
- stat_msg=msg,
- cpu_percent=str(p_cpu),
- mem_percent=str(p_mem),
- disk_percent=str(disk),
- inte_send=str(inte_send),
- inte_recv=str(inte_recv),
- boot_time=boot_time,
- ).dict(),
- BotRuntimeInfo(
- cpu_percent=str(b_cpu), mem_percent=str(b_mem), bot_run_time=bot_time
- ).dict(),
- )
-
-
-def get_service_list() -> dict:
- result = dict()
-
- files = os.listdir(SERVICES_DIR)
- for f in files:
- # Thank you, MacOS
- if f == ".DS_Store":
- continue
-
- serv_path = SERVICES_DIR / f
- data = json.loads(serv_path.read_bytes())
-
- serv_name = data["service"]
- serv_docs = data["docs"]
- serv_is_enabled = data["enabled"]
- serv_disable_user = data["disable_user"]
- serv_disable_group = data["disable_group"]
-
- result[serv_name] = ServiceInfo(
- service_name=serv_name,
- service_docs=serv_docs,
- is_enabled=serv_is_enabled,
- disable_user=serv_disable_user,
- disable_group=serv_disable_group,
- ).dict()
-
- return result
-
-
-def control_service(
- serv_name: str, is_enab: int, enab_u: str, enab_g: str, disab_u: str, disab_g: str
-) -> tuple:
- try:
- serv_data = ServiceTools().load_service(serv_name)
- except Exception:
- return False, dict()
-
- if is_enab != 1:
- if is_enab == 0:
- serv_data["enabled"] = False
- else:
- serv_data["enabled"] = True
-
- if enab_u:
- if enab_u not in serv_data["disable_user"]:
- return False, {"msg": "Target not in list"}
- serv_data["disable_user"].remove(enab_u)
-
- if enab_g:
- if enab_g not in serv_data["disable_user"]:
- return False, {"msg": "Target not in list"}
- serv_data["disable_group"].remove(enab_g)
-
- if disab_u:
- if disab_u in serv_data["disable_user"]:
- return False, {"msg": "Target already exists in list"}
- serv_data["disable_user"].append(disab_u)
-
- if disab_g:
- if disab_g in serv_data["disable_user"]:
- return False, {"msg": "Target already exists in list"}
- serv_data["disable_group"].append(disab_g)
-
- try:
- ServiceTools().save_service(serv_data, serv_name)
- except Exception:
- return False, dict()
-
- return True, serv_data
-
-
-MANEGE_DIR = Path(".") / "data" / "plugins" / "manege"
+from .depends import *
+from .data_source import (
+ get_process_info,
+ get_service_list,
+ edit_service as _edit_service,
+ get_block_list,
+ edit_block_list as _edit_block_list,
+)
+from ..listener import get_message_info
-def get_block_list() -> dict:
- u_data = dict()
- g_data = dict()
+def base_url(_=Depends(http_author)):
+ return {"status": status.HTTP_204_NO_CONTENT, "msg": "该路径仅供控制台加载"}
- u_f = "block_user.json"
- path = MANEGE_DIR / u_f
- if path.is_file():
- u_data = json.loads(path.read_bytes())
- g_f = "block_group.json"
- path = MANEGE_DIR / g_f
- if path.is_file():
- g_data = json.loads(path.read_bytes())
+def auth_info(_=Depends(http_author)):
+ return {"status": status.HTTP_200_OK, "detail": "OK"}
- return {"user": u_data, "group": g_data}
-
-def edit_block_list(is_enab: bool, user_id: str, group_id: str) -> tuple:
- d = get_block_list()
- u_d = d["user"]
- g_d = d["group"]
-
- now_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-
- if is_enab:
- if user_id:
- if user_id in u_d:
- return False, {"msg": "Target already exists in list"}
- u_d[user_id] = now_time
-
- if group_id:
- if group_id in g_d:
- return False, {"msg": "Target already exists in list"}
- g_d[group_id] = now_time
- else:
- if user_id:
- if user_id not in u_d:
- return False, {"msg": "Target not in list"}
- del u_d[user_id]
-
- if group_id:
- if group_id not in g_d:
- return False, {"msg": "Target not in list"}
- del g_d[group_id]
+async def runtime_info(websocket: WebSocket, _pass=Depends(websocket_author)):
+ if not _pass:
+ return
+ await websocket.accept()
try:
- u_f = "block_user.json"
- path = MANEGE_DIR / u_f
- with open(path, "w", encoding="utf-8") as w:
- w.write(json.dumps(u_d))
-
- g_f = "block_group.json"
- path = MANEGE_DIR / g_f
- with open(path, "w", encoding="utf-8") as w:
- w.write(json.dumps(g_d))
- except Exception:
- return False, dict()
+ while websocket.client_state == WebSocketState.CONNECTED:
+ await websocket.send_json(get_process_info())
+ await asyncio.sleep(1)
+ except ConnectionClosedOK:
+ pass
+ finally:
+ await websocket.close()
+ return
- return True, {"user": u_d, "group": g_d}
+async def message_info(websocket: WebSocket, _pass=Depends(websocket_author)):
+ if not _pass:
+ return
+ await websocket.accept()
-def get_group_list():
- ...
+ try:
+ while websocket.client_state == WebSocketState.CONNECTED:
+ await websocket.send_json(get_message_info())
+ await asyncio.sleep(1)
+ except ConnectionClosedOK:
+ pass
+ finally:
+ await websocket.close()
+ return
+
+
+def service_list(_=Depends(http_author)):
+ return {"status": status.HTTP_200_OK, "data": get_service_list()}
+
+
+def edit_service(
+ service: str,
+ global_enabled: str = "2",
+ enabled: str = "2",
+ user: str = str(),
+ group: str = str(),
+ _=Depends(http_author),
+):
+ return {
+ "status": status.HTTP_200_OK,
+ "data": _edit_service(
+ service, int(global_enabled), bool(int(enabled)), user, group
+ ),
+ }
+
+
+def block_list_info(_=Depends(http_author)):
+ return {"status": status.HTTP_200_OK, "data": get_block_list()}
+
+
+def edit_block_list(enabled: str, user_id: str = str(), group_id: str = str(), _=Depends(http_author)):
+ return {
+ "status": status.HTTP_200_OK,
+ "data": _edit_block_list(bool(int(enabled)), user_id, group_id),
+ }
diff --git a/ATRI/plugins/console/driver/data_source.py b/ATRI/plugins/console/driver/data_source.py
new file mode 100644
index 0000000..12fb938
--- /dev/null
+++ b/ATRI/plugins/console/driver/data_source.py
@@ -0,0 +1,167 @@
+import os
+import json
+import psutil
+from pathlib import Path
+from datetime import datetime
+
+from ATRI.exceptions import GetStatusError
+
+from ATRI.utils import FileDealer
+from ATRI.service import ServiceTools, SERVICES_DIR
+
+from . import models
+
+
+def get_process_info() -> dict:
+ try:
+ platform_cpu = psutil.cpu_percent(interval=1)
+ platform_mem = psutil.virtual_memory().percent
+ platform_disk = psutil.disk_usage("/").percent
+ platform_net_sent = str(psutil.net_io_counters().bytes_sent / 1000000)
+ platform_net_recv = str(psutil.net_io_counters().bytes_recv / 1000000)
+
+ process = psutil.Process(os.getpid())
+ bot_cpu = str(process.cpu_percent(interval=1))
+ bot_mem = str(process.memory_percent(memtype="rss"))
+
+ now_time = datetime.now().timestamp()
+ _boot_time = psutil.boot_time()
+ _bot_run_time = process.create_time()
+ boot_time = str(
+ datetime.utcfromtimestamp(now_time).replace(microsecond=0)
+ - datetime.utcfromtimestamp(_boot_time).replace(microsecond=0)
+ )
+ bot_run_time = str(
+ datetime.utcfromtimestamp(now_time).replace(microsecond=0)
+ - datetime.utcfromtimestamp(_bot_run_time).replace(microsecond=0)
+ )
+ except Exception:
+ raise GetStatusError("获取实例运行信息失败")
+
+ stat_msg = "アトリは、高性能ですから!"
+ if platform_cpu > 90:
+ stat_msg = "咱感觉有些头晕..."
+ if platform_mem > 90:
+ stat_msg = "咱感觉有点头晕并且有点累..."
+ elif platform_mem > 90:
+ stat_msg = "咱感觉有点累..."
+ elif platform_disk > 90:
+ stat_msg = "咱感觉身体要被塞满了..."
+
+ platform_cpu = str(platform_cpu)
+ platform_mem = str(platform_mem)
+ platform_disk = str(platform_disk)
+
+ return models.RuntimeInfo(
+ status_message=stat_msg,
+ platform_info=models.PlatformRuntimeInfo(
+ cpu_percent=platform_cpu,
+ mem_percent=platform_mem,
+ disk_percent=platform_disk,
+ net_sent=platform_net_sent,
+ net_recv=platform_net_recv,
+ boot_time=boot_time,
+ ),
+ bot_info=models.BotRuntimeInfo(
+ cpu_percent=bot_cpu, mem_percent=bot_mem, run_time=bot_run_time
+ ),
+ ).dict()
+
+
+def get_service_list() -> dict:
+ result = dict()
+
+ files = os.listdir(SERVICES_DIR)
+ for file in files:
+ # Thank you, MacOS
+ if file == ".DS_Store":
+ continue
+
+ service_path = SERVICES_DIR / file
+ data = models.ServiceInfo.parse_file(service_path)
+ result[data.service] = data.dict()
+
+ return result
+
+
+def edit_service(
+ service: str, global_enabled: int, enabled: bool, user_id: str, group_id: str
+):
+ data = ServiceTools.load_service(service)
+
+ if global_enabled != 2 and global_enabled:
+ data.enabled = bool(global_enabled)
+ else:
+ data.enabled = False
+ if user_id or group_id:
+ if enabled:
+ if user_id not in data.disable_user:
+ return {"detail": "用户不存在于禁用名单"}
+ else:
+ data.disable_user.remove(user_id)
+
+ if group_id not in data.disable_group:
+ return {"detail": "群不存在于禁用名单"}
+ else:
+ data.disable_group.remove(group_id)
+ else:
+ if user_id not in data.disable_user:
+ data.disable_user.append(user_id)
+ else:
+ return {"detail": "用户已存在于禁用名单"}
+
+ if group_id not in data.disable_group:
+ data.disable_group.append(group_id)
+ else:
+ return {"detail": "群已存在于禁用名单"}
+
+ ServiceTools.save_service(data.dict(), service)
+
+ return {"detail": "操作完成~"}
+
+
+def get_block_list() -> models.BlockInfo:
+ file_dir = Path(".") / "data" / "plugins" / "manege"
+ path = file_dir / "block_user.json"
+ user_data = json.loads(path.read_bytes())
+
+ path = file_dir / "block_group.json"
+ group_data = json.loads(path.read_bytes())
+
+ return models.BlockInfo(user=user_data, group=group_data)
+
+
+async def edit_block_list(enabled: bool, user_id: str, group_id: str):
+ data = get_block_list()
+ now_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+ if enabled:
+ if user_id:
+ if user_id in data.user:
+ return {"detail": "用户已存在于黑名单"}
+ else:
+ data.user[user_id] = now_time
+ if group_id:
+ if group_id in data.group:
+ return {"detail": "群已存在于黑名单"}
+ else:
+ data.group[group_id] = now_time
+ else:
+ if user_id:
+ if user_id in data.user:
+ del data.user[user_id]
+ else:
+ return {"detail": "用户不存在于黑名单"}
+ if group_id:
+ if group_id in data.group:
+ del data.group[group_id]
+ else:
+ return {"detail": "群不存在于黑名单"}
+
+ file_dir = Path(".") / "data" / "plugins" / "manege"
+ path = file_dir / "block_user.json"
+ await FileDealer(path).write(json.dumps(data.user))
+
+ path = file_dir / "block_group.json"
+ await FileDealer(path).write(json.dumps(data.group))
+
+ return {"detail": "操作完成~"}
diff --git a/ATRI/plugins/console/driver/depends.py b/ATRI/plugins/console/driver/depends.py
new file mode 100644
index 0000000..8205f44
--- /dev/null
+++ b/ATRI/plugins/console/driver/depends.py
@@ -0,0 +1,38 @@
+from typing import Union
+from datetime import datetime
+
+from fastapi import Query, HTTPException, status
+from starlette.websockets import WebSocket
+
+from ..data_source import AuthDealer
+
+
+def http_author(token: Union[str, None] = Query(default=None)):
+ data = AuthDealer.get()
+ if data is None:
+ raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="验证信息不存在")
+
+ now_time = datetime.now().timestamp()
+ if token != data.token:
+ raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="密钥不匹配, 请检查")
+ elif now_time > data.dead_time:
+ raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="密钥已过期")
+ else:
+ return token
+
+
+async def websocket_author(
+ websocket: WebSocket, token: Union[str, None] = Query(default=None)
+):
+ data = AuthDealer.get()
+ if not data:
+ await websocket.close(code=status.WS_1008_POLICY_VIOLATION)
+ return
+
+ now_time = datetime.now().timestamp()
+ if token != data.token:
+ await websocket.close(code=status.WS_1008_POLICY_VIOLATION)
+ elif now_time > data.dead_time:
+ await websocket.close(code=status.WS_1008_POLICY_VIOLATION)
+ else:
+ return token
diff --git a/ATRI/plugins/console/driver/models.py b/ATRI/plugins/console/driver/models.py
new file mode 100644
index 0000000..cbc110d
--- /dev/null
+++ b/ATRI/plugins/console/driver/models.py
@@ -0,0 +1,38 @@
+from pydantic import BaseModel
+
+
+class PlatformRuntimeInfo(BaseModel):
+ cpu_percent: str
+ mem_percent: str
+ disk_percent: str
+ net_sent: str
+ net_recv: str
+ boot_time: str
+
+
+class BotRuntimeInfo(BaseModel):
+ cpu_percent: str
+ mem_percent: str
+ run_time: str
+
+
+class RuntimeInfo(BaseModel):
+ status_message: str
+ platform_info: PlatformRuntimeInfo
+ bot_info: BotRuntimeInfo
+
+
+class ServiceInfo(BaseModel):
+ service: str
+ docs: str
+ permission: list
+ cmd_list: dict
+ enabled: bool
+ only_admin: bool
+ disable_user: list
+ disable_group: list
+
+
+class BlockInfo(BaseModel):
+ user: dict
+ group: dict
diff --git a/ATRI/plugins/console/driver/path.py b/ATRI/plugins/console/driver/path.py
new file mode 100644
index 0000000..7009eb7
--- /dev/null
+++ b/ATRI/plugins/console/driver/path.py
@@ -0,0 +1,14 @@
+CONSOLE_BASE_URL = "/capi"
+
+CONSOLE_AUTH_URL = CONSOLE_BASE_URL + "/auth"
+
+CONSOLE_RUNTIME_INFO_URL = CONSOLE_BASE_URL + "/runtime"
+CONSOLE_MESSAGE_INFO_URL = CONSOLE_BASE_URL + "/message"
+
+CONSOLE_SERVICE_BASE_URL = CONSOLE_BASE_URL + "/service"
+CONSOLE_SERVICE_LIST_URL = CONSOLE_SERVICE_BASE_URL + "/list"
+CONSOLE_SERVICE_EDIT_URL = CONSOLE_SERVICE_BASE_URL + "/edit"
+
+CONSOLE_BLOCK_BASE_URL = CONSOLE_BASE_URL + "/block"
+CONSOLE_BLOCK_LIST_URL = CONSOLE_BLOCK_BASE_URL + "/list"
+CONSOLE_BLOCK_EDIT_URL = CONSOLE_BLOCK_BASE_URL + "/edit"
diff --git a/ATRI/plugins/console/driver/view.py b/ATRI/plugins/console/driver/view.py
deleted file mode 100644
index 624e41f..0000000
--- a/ATRI/plugins/console/driver/view.py
+++ /dev/null
@@ -1,101 +0,0 @@
-from ..data_source import Console
-from ..listener import get_message_deal_info
-from .api import (
- control_service,
- edit_block_list,
- get_block_list,
- get_processing_data,
- get_service_list,
-)
-
-
-def auth_token(token: str) -> tuple:
- auth_data: dict = Console().get_auth_info()
- if not auth_data.get("token", None):
- return False, {"status": 500, "msg": "This bot is not create auth data yet."}
- _token = auth_data["token"]
- if token != _token:
- return False, {"status": 403, "msg": "Token error, please check again."}
- else:
- return True, {"status": 200, "msg": "OK"}
-
-
-def handle_base_uri():
- return {"status": 204, "msg": "This path just for console load."}
-
-
-def handle_auther(token: str):
- auth, data = auth_token(token)
- return data if auth else data
-
-
-def handle_runtime_info(token: str):
- auth, data = auth_token(token)
- if not auth:
- return data
-
- plat, bot = get_processing_data()
- return {"status": 200, "data": {"platform": plat, "bot": bot}}
-
-
-def handle_message_deal_info(token: str):
- auth, data = auth_token(token)
- if not auth:
- return data
-
- return {"status": 200, "data": get_message_deal_info()}
-
-
-def handle_get_service_list(token: str):
- auth, data = auth_token(token)
- if not auth:
- return data
-
- return {"status": 200, "data": get_service_list()}
-
-
-def handle_control_service(
- token: str,
- service: str,
- is_enabled: int = 1,
- enabled_user: str = str(),
- enabled_group: str = str(),
- disable_user: str = str(),
- disable_group: str = str(),
-):
- auth, data = auth_token(token)
- if not auth:
- return data
-
- is_ok, data = control_service(
- service, is_enabled, enabled_user, enabled_group, disable_user, disable_group
- )
- if not is_ok:
- return {"status": 422, "msg": "Dealing service data failed"}
-
- return {"status": 200, "data": data}
-
-
-def handle_get_block_list(token: str):
- auth, data = auth_token(token)
- if not auth:
- return data
-
- return {"status": 200, "data": get_block_list()}
-
-
-def handle_edit_block(
- token: str,
- is_enabled: bool,
- user_id: str = str(),
- group_id: str = str(),
-):
- auth, data = auth_token(token)
- if not auth:
- return data
-
- is_ok, data = edit_block_list(is_enabled, user_id, group_id)
- if not is_ok:
- return {"status": 422, "msg": "Dealing block data failed"}
-
- return {"status": 200, "data": data}