From e90234845b67d0778126e513046c4e29b331bcae Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Sun, 18 Sep 2022 15:29:06 +0800
Subject: =?UTF-8?q?=F0=9F=9A=A7=20=E4=BC=98=E5=8C=96=E6=8E=A7=E5=88=B6?=
 =?UTF-8?q?=E5=8F=B0=E5=8A=A0=E8=BD=BD=E6=96=B9=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/console/driver/__init__.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'ATRI/plugins/console/driver')

diff --git a/ATRI/plugins/console/driver/__init__.py b/ATRI/plugins/console/driver/__init__.py
index 22afd4a..04c714d 100644
--- a/ATRI/plugins/console/driver/__init__.py
+++ b/ATRI/plugins/console/driver/__init__.py
@@ -1,7 +1,9 @@
 from nonebot.drivers.fastapi import Driver
 
+from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
 
+from ATRI.plugins.console.data_source import FRONTEND_DIR
 from .view import (
     handle_auther,
     handle_base_uri,
@@ -49,8 +51,15 @@ def register_routes(driver: Driver):
     app.get(CONSOLE_API_BLOCK_LIST_URI)(handle_get_block_list)
     app.get(CONSOLE_API_BLOCK_EDIT_URI)(handle_edit_block)
 
+    static_path = str(FRONTEND_DIR)
+    app.mount(
+        "/",
+        StaticFiles(directory=static_path, html=True),
+        name="atri-console",
+    )
+
 
-def init():
+def init_driver():
     from ATRI import driver
 
     register_routes(driver())  # type: ignore
-- 
cgit v1.2.3