From b211cd8fff64f4abbc9c9832106411fa8c63fa8d Mon Sep 17 00:00:00 2001
From: Shigma <1700011071@pku.edu.cn>
Date: Fri, 23 Apr 2021 17:58:56 +0800
Subject: [PATCH] feat(webui): support webui without database, fix #234

---
 packages/koishi-core/src/context.ts           | 17 ++++++++-----
 packages/plugin-teach/client/teach.vue        |  2 +-
 .../client/components/numeric.vue             |  9 +++----
 packages/plugin-webui/client/main.ts          |  6 +++--
 packages/plugin-webui/client/views/bots.vue   |  4 ++--
 .../plugin-webui/client/views/home/home.vue   | 24 +++++++++++--------
 packages/plugin-webui/src/data.ts             |  2 +-
 packages/plugin-webui/src/server.ts           | 10 ++++++--
 packages/plugin-webui/src/stats.ts            | 11 ++++++---
 9 files changed, 54 insertions(+), 31 deletions(-)

diff --git a/packages/koishi-core/src/context.ts b/packages/koishi-core/src/context.ts
index 9d94390fef..3fea0e003b 100644
--- a/packages/koishi-core/src/context.ts
+++ b/packages/koishi-core/src/context.ts
@@ -317,19 +317,21 @@ export class Context {
     }
   }
 
-  on<K extends EventName>(name: K, listener: EventMap[K], prepend = false) {
+  on<K extends EventName>(name: K, listener: EventMap[K], prepend?: boolean): () => boolean
+  on(name: string & EventName, listener: Disposable, prepend = false) {
     const method = prepend ? 'unshift' : 'push'
 
-    // handle plugin-related events
-    const _listener = listener as Disposable
+    // handle special events
     if (name === 'connect' && this.app.status === App.Status.open) {
-      return _listener(), () => false
+      return listener(), () => false
     } else if (name === 'before-disconnect') {
-      this.state.disposables[method](_listener)
-      return () => remove(this.state.disposables, _listener)
+      this.state.disposables[method](listener)
+      return () => remove(this.state.disposables, listener)
     } else if (name === 'before-connect') {
       // before-connect is side effect
       this.addSideEffect()
+    } else if (typeof name === 'string' && name.startsWith('delegate/')) {
+      if (this[name.slice(9)]) return listener(), () => false
     }
 
     const hooks = this.app._hooks[name] ||= []
@@ -513,6 +515,7 @@ export class Context {
         return value
       },
       set(value) {
+        if (!this.app[privateKey]) this.emit('delegate/' + key)
         defineProperty(this.app, privateKey, value)
       },
     })
@@ -544,6 +547,8 @@ export interface EventMap extends SessionEventMap {
   [Context.middleware]: Middleware
 
   // Koishi events
+  'delegate/assets'(): void
+  'delegate/database'(): void
   'appellation'(name: string, session: Session): string
   'before-parse'(content: string, session: Session): Argv
   'parse'(argv: Argv, session: Session): string
diff --git a/packages/plugin-teach/client/teach.vue b/packages/plugin-teach/client/teach.vue
index 8825b33abe..b5f5f7cb1c 100644
--- a/packages/plugin-teach/client/teach.vue
+++ b/packages/plugin-teach/client/teach.vue
@@ -2,7 +2,7 @@
   <div class="card-grid profile-grid">
     <k-numeric title="总问题数量" icon="quote-left">{{ meta.questions }}</k-numeric>
     <k-numeric title="总问答数量" icon="quote-right">{{ meta.dialogues }}</k-numeric>
-    <k-numeric title="图片服务器" icon="hdd" type="size" :value="meta.assetSize"/>
+    <k-numeric title="图片服务器" icon="hdd" type="size" :value="meta.assetSize" fallback="暂无数据"/>
   </div>
   <div class="card-grid chart-grid">
     <word-cloud/>
diff --git a/packages/plugin-webui/client/components/numeric.vue b/packages/plugin-webui/client/components/numeric.vue
index 389edc9f6e..f5a2fa24a8 100644
--- a/packages/plugin-webui/client/components/numeric.vue
+++ b/packages/plugin-webui/client/components/numeric.vue
@@ -17,17 +17,18 @@ const props = defineProps<{
   icon: string
   type?: 'size'
   value?: number
+  fallback?: string
 }>()
 
 const text = computed(() => {
-  if (!props.value) return '暂无数据'
+  if (!props.value) return props.fallback
   if (props.type === 'size') {
     if (props.value >= (1 << 20) * 1000) {
-      return +(props.value / (1 << 30)).toFixed(1) + ' GB'
+      return (props.value / (1 << 30)).toFixed(1) + ' GB'
     } else if (props.value >= (1 << 10) * 1000) {
-      return +(props.value / (1 << 20)).toFixed(1) + ' MB'
+      return (props.value / (1 << 20)).toFixed(1) + ' MB'
     } else {
-      return +(props.value / (1 << 10)).toFixed(1) + ' KB'
+      return (props.value / (1 << 10)).toFixed(1) + ' KB'
     }
   }
 })
diff --git a/packages/plugin-webui/client/main.ts b/packages/plugin-webui/client/main.ts
index 9f4da23515..2d4a8b0940 100644
--- a/packages/plugin-webui/client/main.ts
+++ b/packages/plugin-webui/client/main.ts
@@ -26,17 +26,19 @@ self['KoishiClient'] = client
 
 const app = Vue.createApp(App)
 
+const statsPrerequisite: 'stats'[] = KOISHI_CONFIG.database ? ['stats'] : []
+
 router.addRoute({
   path: '/',
   name: '仪表盘',
-  meta: { icon: 'tachometer-alt', require: ['stats', 'meta', 'profile', 'registry'] },
+  meta: { icon: 'tachometer-alt', require: [...statsPrerequisite, 'meta', 'profile', 'registry'] },
   component: () => import('./views/home/home.vue'),
 })
 
 router.addRoute({
   path: '/bots',
   name: '机器人',
-  meta: { icon: 'robot', require: ['stats', 'profile'] },
+  meta: { icon: 'robot', require: [...statsPrerequisite, 'profile'] },
   component: () => import('./views/bots.vue'),
 })
 
diff --git a/packages/plugin-webui/client/views/bots.vue b/packages/plugin-webui/client/views/bots.vue
index fc9b420330..0cded4ce2f 100644
--- a/packages/plugin-webui/client/views/bots.vue
+++ b/packages/plugin-webui/client/views/bots.vue
@@ -6,14 +6,14 @@
         <th>用户名</th>
         <th>运行状态</th>
         <th>当前消息频率</th>
-        <th>近期消息频率</th>
+        <th v-if="stats">近期消息频率</th>
       </tr>
       <tr v-for="{ platform, username, selfId, code, currentRate } in profile.bots">
         <td>{{ platform }}</td>
         <td>{{ username }}</td>
         <td>{{ codes[code] }}</td>
         <td>发送 {{ currentRate[0] }}/min，接收 {{ currentRate[1] }}/min</td>
-        <td>发送 {{ stats.botSend[`${platform}:${selfId}`] || 0 }}/d，接收 {{ stats.botReceive[`${platform}:${selfId}`] || 0 }}/d</td>
+        <td v-if="stats">发送 {{ stats.botSend[`${platform}:${selfId}`] || 0 }}/d，接收 {{ stats.botReceive[`${platform}:${selfId}`] || 0 }}/d</td>
       </tr>
     </table>
     <p v-else>暂无数据。</p>
diff --git a/packages/plugin-webui/client/views/home/home.vue b/packages/plugin-webui/client/views/home/home.vue
index fda3059560..e559edb5e2 100644
--- a/packages/plugin-webui/client/views/home/home.vue
+++ b/packages/plugin-webui/client/views/home/home.vue
@@ -1,18 +1,20 @@
 <template>
   <div class="card-grid profile-grid">
     <k-numeric title="当前消息频率" icon="paper-plane">{{ currentRate }} / min</k-numeric>
-    <k-numeric title="近期消息频率" icon="history">{{ recentRate }} / d</k-numeric>
+    <k-numeric title="近期消息频率" icon="history" v-if="config.database">{{ recentRate }} / d</k-numeric>
     <k-numeric title="命名插件数量" icon="plug">{{ registry.pluginCount }}</k-numeric>
-    <k-numeric title="数据库体积" icon="database" type="size" :value="meta.storageSize"/>
-    <k-numeric title="活跃用户数量" icon="heart">{{ meta.activeUsers }}</k-numeric>
-    <k-numeric title="活跃群数量" icon="users">{{ meta.activeGroups }}</k-numeric>
-  </div>
-  <load-chart/>
-  <div class="card-grid chart-grid">
-    <history-chart/>
-    <hour-chart/>
-    <group-chart/>
+    <k-numeric title="数据库体积" icon="database" type="size" :value="meta.storageSize" v-if="meta" fallback="未安装"/>
+    <k-numeric title="活跃用户数量" icon="heart" v-if="config.database">{{ meta.activeUsers }}</k-numeric>
+    <k-numeric title="活跃群数量" icon="users" v-if="config.database">{{ meta.activeGroups }}</k-numeric>
   </div>
+  <template v-if="config.database">
+    <load-chart/>
+    <div class="card-grid chart-grid">
+      <history-chart/>
+      <hour-chart/>
+      <group-chart/>
+    </div>
+  </template>
 </template>
 
 <script setup lang="ts">
@@ -24,6 +26,8 @@ import HistoryChart from './history-chart.vue'
 import HourChart from './hour-chart.vue'
 import LoadChart from './load-chart.vue'
 
+const config = KOISHI_CONFIG
+
 const currentRate = computed(() => {
   return profile.value.bots.reduce((sum, bot) => sum + bot.currentRate[0], 0)
 })
diff --git a/packages/plugin-webui/src/data.ts b/packages/plugin-webui/src/data.ts
index 48044c3413..6759e3acec 100644
--- a/packages/plugin-webui/src/data.ts
+++ b/packages/plugin-webui/src/data.ts
@@ -143,7 +143,7 @@ export class Meta implements DataSource<Meta.Payload> {
 
   constructor(private ctx: Context, public config: Meta.Config) {
     this.extend(async () => ctx.assets?.stats())
-    this.extend(async () => ctx.database.getStats())
+    this.extend(async () => ctx.database?.getStats())
 
     ctx.all().on('command', ({ session }: Argv<'lastCall'>) => {
       session.user.lastCall = new Date()
diff --git a/packages/plugin-webui/src/server.ts b/packages/plugin-webui/src/server.ts
index 74419cf307..e045536c0f 100644
--- a/packages/plugin-webui/src/server.ts
+++ b/packages/plugin-webui/src/server.ts
@@ -24,6 +24,7 @@ export interface Config extends BaseConfig, Profile.Config, Meta.Config, Registr
 }
 
 export interface ClientConfig extends Required<BaseConfig> {
+  database: boolean
   endpoint: string
   extensions: string[]
 }
@@ -71,7 +72,7 @@ export class WebServer extends Adapter {
 
     const { apiPath, uiPath, devMode, selfUrl, title } = config
     const endpoint = selfUrl + apiPath
-    this.global = { title, uiPath, endpoint, devMode, extensions: [] }
+    this.global = { title, uiPath, endpoint, devMode, extensions: [], database: false }
     this.root = resolve(__dirname, '..', devMode ? 'client' : 'dist')
 
     this.server = new WebSocket.Server({
@@ -100,6 +101,10 @@ export class WebServer extends Adapter {
 
     ctx.on('connect', () => this.start())
     ctx.before('disconnect', () => this.stop())
+
+    ctx.on('delegate/database', () => {
+      this.global.database = !!ctx.database
+    })
   }
 
   broadcast(type: string, body: any) {
@@ -165,6 +170,7 @@ export class WebServer extends Adapter {
     })
 
     socket.on('message', async (data) => {
+      if (!this.ctx.database) return
       const { type, body } = JSON.parse(data.toString())
       const method = WebServer.listeners[type]
       if (method) {
@@ -240,7 +246,7 @@ export class WebServer extends Adapter {
 export namespace WebServer {
   export interface Sources extends Record<string, DataSource> {
     meta: Meta
-    stats: Statistics
+    stats?: Statistics
     profile: Profile
     registry: Registry
   }
diff --git a/packages/plugin-webui/src/stats.ts b/packages/plugin-webui/src/stats.ts
index cbe58c45fe..84730e1b08 100644
--- a/packages/plugin-webui/src/stats.ts
+++ b/packages/plugin-webui/src/stats.ts
@@ -84,8 +84,6 @@ export class Statistics implements DataSource<Statistics.Payload> {
   average = average
 
   constructor(private ctx: Context, public config: Statistics.Config = {}) {
-    this.sync = ctx.database.createSynchronizer()
-
     if (config.handleSignals !== false) {
       const handleSignal = (signal: NodeJS.Signals) => {
         new Logger('app').info(`terminated by ${signal}`)
@@ -103,6 +101,10 @@ export class Statistics implements DataSource<Statistics.Payload> {
       })
     }
 
+    ctx.on('delegate/database', () => {
+      this.sync = ctx.database.createSynchronizer()
+    })
+
     ctx.before('disconnect', async () => {
       // rollback to default implementation to prevent infinite call stack
       if (Session.prototype.send[customTag]) {
@@ -111,6 +113,8 @@ export class Statistics implements DataSource<Statistics.Payload> {
       await this.upload(true)
     })
 
+    ctx = ctx.select('database')
+
     ctx.before('command', ({ command, session }) => {
       if (command.parent?.name !== 'test') {
         const [name] = command.name.split('.', 1)
@@ -150,7 +154,7 @@ export class Statistics implements DataSource<Statistics.Payload> {
     if (forced || +date - +this.lastUpdate > REFRESH_INTERVAL || dateHour !== this.updateHour) {
       this.lastUpdate = date
       this.updateHour = dateHour
-      await this.sync.upload(date)
+      await this.sync?.upload(date)
     }
   }
 
@@ -229,6 +233,7 @@ export class Statistics implements DataSource<Statistics.Payload> {
   }
 
   async get() {
+    if (!this.sync) return
     const date = new Date()
     const dateNumber = Time.getDateNumber(date, date.getTimezoneOffset())
     if (dateNumber !== this.cachedDate) {