From 7042d94213532191a0c72d34d7c85193184c079f Mon Sep 17 00:00:00 2001
From: Kyomotoi <kyomotoiowo@gmail.com>
Date: Wed, 4 May 2022 16:07:18 +0800
Subject: =?UTF-8?q?=E2=9C=A8=20=E4=B8=BA=E5=89=8D=E7=AB=AF=E9=A1=B5?=
 =?UTF-8?q?=E9=9D=A2=E6=89=93=E5=9F=BA=E7=A1=80,=20=E6=9B=B4=E6=96=B0?=
 =?UTF-8?q?=E5=91=BD=E4=BB=A4=E5=8F=8AAPI?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/status/driver/view.py | 39 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 ATRI/plugins/status/driver/view.py

(limited to 'ATRI/plugins/status/driver/view.py')

diff --git a/ATRI/plugins/status/driver/view.py b/ATRI/plugins/status/driver/view.py
new file mode 100644
index 0000000..5d8c56f
--- /dev/null
+++ b/ATRI/plugins/status/driver/view.py
@@ -0,0 +1,39 @@
+from ..data_source import Status
+from ..listener import get_message_deal_info
+
+
+def handle_base_uri():
+    return {"status": 204, "msg": "This path just for console load."}
+
+
+def handle_runtime_info(token: str):
+    auth, data = auth_token(token)
+    plat, bot = Status().get_status(True)
+    if auth:
+        return {"status": 200, "data": {"platform": plat, "bot": bot}}
+    else:
+        return data
+
+
+def handle_message_deal_info(token: str):
+    auth, data = auth_token(token)
+    if auth:
+        return {"status": 200, "data": get_message_deal_info()}
+    else:
+        return data
+
+
+def handle_auther(token: str):
+    auth, data = auth_token(token)
+    return data if auth else data
+
+
+def auth_token(token: str) -> tuple:
+    auth_data: dict = Status().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"}
-- 
cgit v1.2.3