1 files changed, 37 insertions, 0 deletions
diff --git a/ATRI/plugins/console/drivers.py b/ATRI/plugins/console/drivers.py
new file mode 100644
index 0000000..e691a7c
--- /dev/null
+++ b/ATRI/plugins/console/drivers.py
@@ -0,0 +1,37 @@
+from pathlib import Path
+
+from fastapi.staticfiles import StaticFiles
+from fastapi.middleware.cors import CORSMiddleware
+
+import ATRI
+from .view import handle_is_connect, handle_dashboard_info, handle_status
+
+
+origins = [
+    "http://localhost.tiangolo.com",
+    "https://localhost.tiangolo.com",
+    "http://localhost",
+    "http://localhost:8080",
+    "http://localhost:20000"
+]
+
+
+def register_route():
+    app = ATRI.driver().server_app
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=origins,
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"]
+    )
+    
+    static_path = str((Path(".") / "ATRI" / "plugins" / "console" / "atri-manege" / "dist").absolute())
+    
+    app.get("/bot/is_connect")(handle_is_connect)
+    app.get("/bot/status")(handle_status)
+    app.get("/bot/dashboard_info")(handle_dashboard_info)
+    
+    app.mount("/",
+              StaticFiles(directory=static_path, html=True),
+              name="bot")