From 1fbc92bc6d159bf593dfba42891b40081138e96a Mon Sep 17 00:00:00 2001
From: Junyan Qin <1010553892@qq.com>
Date: Mon, 14 Oct 2024 21:18:36 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E9=A6=96=E9=A1=B5=E5=B1=95=E7=A4=BA?=
 =?UTF-8?q?=E7=89=88=E6=9C=AC=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/api/http/controller/group.py         |  2 +-
 pkg/api/http/controller/groups/system.py | 19 ++++++++++
 pkg/api/http/controller/main.py          |  2 +-
 pkg/core/boot.py                         |  6 +++
 pkg/core/bootutils/log.py                |  4 +-
 pkg/utils/constants.py                   |  2 +
 web/src/App.vue                          | 47 ++++++++++++++++++++++--
 web/src/pages/Logs.vue                   |  4 +-
 web/src/plugins/index.js                 |  1 +
 web/src/store/index.js                   | 14 ++++++-
 10 files changed, 90 insertions(+), 11 deletions(-)
 create mode 100644 pkg/api/http/controller/groups/system.py

diff --git a/pkg/api/http/controller/group.py b/pkg/api/http/controller/group.py
index b9533567..4f253e05 100644
--- a/pkg/api/http/controller/group.py
+++ b/pkg/api/http/controller/group.py
@@ -54,7 +54,7 @@ class RouterGroup(abc.ABC):
                     return self.http_status(500, -2, str(e))
                 
             new_f = handler_error
-            new_f.__name__ = f.__name__
+            new_f.__name__ = (self.name + rule).replace('/', '__')
             new_f.__doc__ = f.__doc__
 
             self.quart_app.route(rule, **options)(new_f)
diff --git a/pkg/api/http/controller/groups/system.py b/pkg/api/http/controller/groups/system.py
new file mode 100644
index 00000000..43b06ddd
--- /dev/null
+++ b/pkg/api/http/controller/groups/system.py
@@ -0,0 +1,19 @@
+import quart
+
+from .....core import app
+from .. import group
+from .....utils import constants
+
+
+@group.group_class('system', '/api/v1/system')
+class SystemRouterGroup(group.RouterGroup):
+    
+    async def initialize(self) -> None:
+        @self.route('/info', methods=['GET'])
+        async def _() -> str:
+            return self.success(
+                data={
+                    "version": constants.semantic_version,
+                    "debug": constants.debug_mode
+                }
+            )
\ No newline at end of file
diff --git a/pkg/api/http/controller/main.py b/pkg/api/http/controller/main.py
index 6c7c72f2..76545832 100644
--- a/pkg/api/http/controller/main.py
+++ b/pkg/api/http/controller/main.py
@@ -5,7 +5,7 @@ import asyncio
 import quart
 
 from ....core import app
-from .groups import logs
+from .groups import logs, system
 from . import group
 
 
diff --git a/pkg/core/boot.py b/pkg/core/boot.py
index 1e5f1052..0f0ab3ae 100644
--- a/pkg/core/boot.py
+++ b/pkg/core/boot.py
@@ -2,10 +2,12 @@ from __future__ import print_function
 
 import traceback
 import asyncio
+import os
 
 from . import app
 from ..audit import identifier
 from . import stage
+from ..utils import constants
 
 # 引入启动阶段实现以便注册
 from .stages import load_config, setup_logger, build_app, migrate, show_notes
@@ -25,6 +27,10 @@ async def make_app(loop: asyncio.AbstractEventLoop) -> app.Application:
     # 生成标识符
     identifier.init()
 
+    # 确定是否为调试模式
+    if "DEBUG" in os.environ and os.environ["DEBUG"] in ["true", "1"]:
+        constants.debug_mode = True
+
     ap = app.Application()
 
     ap.event_loop = loop
diff --git a/pkg/core/bootutils/log.py b/pkg/core/bootutils/log.py
index ad8252df..36bc51a0 100644
--- a/pkg/core/bootutils/log.py
+++ b/pkg/core/bootutils/log.py
@@ -5,6 +5,8 @@ import time
 
 import colorlog
 
+from ...utils import constants
+
 
 log_colors_config = {
     "DEBUG": "green",  # cyan white
@@ -22,7 +24,7 @@ async def init_logging(extra_handlers: list[logging.Handler] = None) -> logging.
 
     level = logging.INFO
 
-    if "DEBUG" in os.environ and os.environ["DEBUG"] in ["true", "1"]:
+    if constants.debug_mode:
         level = logging.DEBUG
 
     log_file_name = "data/logs/qcg-%s.log" % time.strftime(
diff --git a/pkg/utils/constants.py b/pkg/utils/constants.py
index 74437702..79b470e6 100644
--- a/pkg/utils/constants.py
+++ b/pkg/utils/constants.py
@@ -1 +1,3 @@
 semantic_version = "v3.3.1.1"
+
+debug_mode = False
\ No newline at end of file
diff --git a/web/src/App.vue b/web/src/App.vue
index 7ba342ba..59f01d97 100644
--- a/web/src/App.vue
+++ b/web/src/App.vue
@@ -5,7 +5,13 @@
       <v-navigation-drawer id="navigation-drawer" :width="160" app permanent rail>
         <v-list-item id="logo-list-item">
           <template v-slot:prepend>
-            <v-img src="@/assets/langbot-logo-block.png" height="32" width="32"></v-img>
+            <div id="logo-container">
+              <v-img id="logo-img" src="@/assets/langbot-logo-block.png" height="32" width="32"></v-img>
+
+              <div id="version-chip" v-tooltip="proxy.$store.state.version + (proxy.$store.state.debug ? ' (调试模式已启用)' : '')" :style="{ 'background-color': proxy.$store.state.debug ? '#c79a47' : '#1e9ae2' }">
+                {{ proxy.$store.state.version }}
+              </div>
+            </div>
           </template>
         </v-list-item>
         <v-divider></v-divider>
@@ -23,7 +29,7 @@
         <template v-slot:append>
           <div>
             <v-list density="compact" nav>
-              <v-list-item id="about-list-item" title="关于" prepend-icon="mdi-cog-outline" v-tooltip="关于">
+              <v-list-item id="about-list-item" title="系统信息" prepend-icon="mdi-cog-outline" v-tooltip="系统信息">
               </v-list-item>
             </v-list>
           </div>
@@ -38,6 +44,9 @@
 </template>
 
 <script setup>
+import { getCurrentInstance } from 'vue'
+
+const { proxy } = getCurrentInstance()
 </script>
 
 <style scoped>
@@ -46,9 +55,39 @@
   flex-direction: column;
 }
 
-#logo-list-item {
+#logo-container {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  justify-content: center;
   margin-left: -0.2rem;
-  margin-block: 0.5rem;
+}
+
+#logo-img {
+  /* margin-left: -0.2rem; */
+}
+
+#logo-list-item {
+  margin-top: 0.5rem;
+  margin-bottom: 1.5rem;
+}
+
+#version-chip {
+  position: absolute;
+  top: 0;
+  right: 0;
+  margin-top: 2.4rem;
+  margin-right: 0.3rem;
+  font-size: 0.55rem;
+  font-weight: 500;
+  padding-inline: 0.4rem;
+  text-align: center;
+  background-color: #c79a47;
+  color: white;
+  border-radius: 0.5rem;
+  border: 1px solid #fff;
+  box-shadow: 0 0 0.1rem 0 rgba(0, 0, 0, 0.5);
+  user-select: none;
 }
 
 #about-list-item {
diff --git a/web/src/pages/Logs.vue b/web/src/pages/Logs.vue
index a4edd464..aeaff77d 100644
--- a/web/src/pages/Logs.vue
+++ b/web/src/pages/Logs.vue
@@ -32,7 +32,7 @@ let logPointer = {
 }
 
 const refreshLog = () => {
-    proxy.$axios.get(`${proxy.$store.state.apiBaseUrl}/logs`, {
+    proxy.$axios.get(`/logs`, {
         params: {
             start_page_number: logPointer.start_page_number,
             start_offset: logPointer.start_offset
@@ -79,7 +79,7 @@ onUnmounted(() => {
 #log-card {
     margin: 1rem;
     margin-top: 1rem;
-    height: calc(100vh - 10rem);
+    height: calc(100vh - 9.5rem);
     border-radius: 1rem;
 }
 
diff --git a/web/src/plugins/index.js b/web/src/plugins/index.js
index 9d8b9d4e..b359e1f8 100644
--- a/web/src/plugins/index.js
+++ b/web/src/plugins/index.js
@@ -17,4 +17,5 @@ export function registerPlugins (app) {
     .use(store)
 
   app.config.globalProperties.$axios = axios
+  store.commit('initializeFetch')
 }
diff --git a/web/src/store/index.js b/web/src/store/index.js
index b18cfcbc..b7fed0d6 100644
--- a/web/src/store/index.js
+++ b/web/src/store/index.js
@@ -6,8 +6,18 @@ export default createStore({
   state: {
     apiBaseUrl: 'http://localhost:5300/api/v1',
     autoRefreshLog: false,
-    version: '0.0.1'
+    version: 'v0.0.0',
+    debug: false
+  },
+  mutations: {
+    initializeFetch() {
+      axios.defaults.baseURL = this.state.apiBaseUrl
+
+      axios.get('/system/info').then(response => {
+        this.state.version = response.data.data.version
+        this.state.debug = response.data.data.debug
+      })
+    }
   },
-  mutations: {},
   actions: {},
 })