From c795269be12327620393b3e563e9fa4f855e7232 Mon Sep 17 00:00:00 2001
From: ThrRip <coding@thrrip.space>
Date: Tue, 17 Dec 2024 10:22:50 +0800
Subject: [PATCH] home: Add global omits for reducing SSR time when needed

This also replaced the monitoring system data collector
check based on the user agent string.
---
 .idea/dictionaries/ThrRip.xml                 |  1 +
 app/app.config.ts                             |  4 +-
 packages/home/app/app.vue                     | 11 ++++--
 packages/home/app/middleware/omit.global.ts   | 39 +++++++++++++++++++
 .../app/middleware/trackPageview.global.ts    |  7 ++--
 packages/home/app/utils/types.ts              | 10 +++++
 6 files changed, 63 insertions(+), 9 deletions(-)
 create mode 100644 packages/home/app/middleware/omit.global.ts
 create mode 100644 packages/home/app/utils/types.ts

diff --git a/.idea/dictionaries/ThrRip.xml b/.idea/dictionaries/ThrRip.xml
index e0ba72d..e7be944 100644
--- a/.idea/dictionaries/ThrRip.xml
+++ b/.idea/dictionaries/ThrRip.xml
@@ -8,6 +8,7 @@
       <w>fortawesome</w>
       <w>getline</w>
       <w>mktemp</w>
+      <w>mzgfan</w>
       <w>nuxtjs</w>
       <w>pageview</w>
       <w>singleline</w>
diff --git a/app/app.config.ts b/app/app.config.ts
index cfb76c8..aa12629 100644
--- a/app/app.config.ts
+++ b/app/app.config.ts
@@ -19,7 +19,5 @@ export default defineAppConfig({
 
   statsBase: 'https://stats.mzg.fan',
 
-  biliApiRoomPlayInfoEndpoint: 'https://api.live.bilibili.com/xlive/web-room/v2/index/getRoomPlayInfo',
-
-  monitoringDataCollectorUserAgentMatch: /^Zabbix$/
+  biliApiRoomPlayInfoEndpoint: 'https://api.live.bilibili.com/xlive/web-room/v2/index/getRoomPlayInfo'
 })
diff --git a/packages/home/app/app.vue b/packages/home/app/app.vue
index e9153c5..93350a1 100644
--- a/packages/home/app/app.vue
+++ b/packages/home/app/app.vue
@@ -264,11 +264,16 @@
 </template>
 
 <script setup lang="ts">
+import type { Omits } from './utils/types'
+
+const omits = useState<Omits>('omits')
+
 // Live status
 const { data: biliApiLiveStatus } = await useFetch(
   useAppConfig().biliApiRoomPlayInfoEndpoint,
   {
     query: { room_id: useAppConfig().appHomeBiliRoomId },
+    immediate: !omits.value.liveStatus,
     transform: (res: {
       data: {
         live_status: 0 | 1
@@ -294,18 +299,18 @@ const { data: backendPlaylist } = await useAsyncData<Playlist>(
   // @ts-expect-error
   async () => {
     const appConfig = useAppConfig()
-    const requestUserAgent = useRequestHeader('user-agent')
     const { Client, Databases, Query } = await import('appwrite')
     const client = new Client()
     const databases = new Databases(client)
     client.setEndpoint(appConfig.backendBase).setProject(appConfig.backendProjectId)
     return databases.listDocuments('home', 'playlist', [
-      appConfig.monitoringDataCollectorUserAgentMatch.test(requestUserAgent ?? '') ?
+      omits.value.playlistMin ?
         Query.limit(1) :
         Query.limit(appConfig.backendQueryResultsLimit)
     ])
   },
   {
+    immediate: !omits.value.playlist,
     default: () => [],
     transform: (res: { total: number, documents: Playlist }): Playlist => res.documents
   }
@@ -355,7 +360,7 @@ function viewPlaylistToggleSorting (column: PlaylistColumn) {
   viewPlaylistDataUpdate(['sort'])
 }
 
-const viewPlaylistData = useState<Playlist>('viewPlaylistData')
+const viewPlaylistData = useState<Playlist>('viewPlaylistData', () => shallowRef([]))
 let viewPlaylistDataShuffled: Playlist = []
 function viewPlaylistDataUpdate (tasks: Array<'shuffle' | 'sort'>) {
   // @ts-expect-error
diff --git a/packages/home/app/middleware/omit.global.ts b/packages/home/app/middleware/omit.global.ts
new file mode 100644
index 0000000..082ee60
--- /dev/null
+++ b/packages/home/app/middleware/omit.global.ts
@@ -0,0 +1,39 @@
+/*
+ * Copyright (c) 2024 ThrRip, XMing (洺知-故犯)
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+import type { OmitDirective, Omits } from '../utils/types'
+
+export default defineNuxtRouteMiddleware(() => {
+  if (import.meta.client) return
+
+  const omits = useState<Omits>('omits', () => shallowRef({}))
+
+  const header = useRequestHeader('x-mzgfan-home-omit')
+  if (!header) return
+  const headerDirectives = header.replaceAll(' ', '').split(',')
+  if (headerDirectives.includes('all')) {
+    omits.value = {
+      liveStatus: true,
+      playlist: true,
+      stats: true
+    }
+    return
+  }
+  type OmitHeaderDirective = 'live-status' | 'playlist' | 'playlist-min' | 'stats'
+  const directiveTable: Record<OmitHeaderDirective, OmitDirective> = {
+    'live-status': 'liveStatus',
+    'playlist': 'playlist',
+    'playlist-min': 'playlistMin',
+    'stats': 'stats'
+  }
+  Object.keys(directiveTable).forEach(directive => {
+    if (headerDirectives.includes(directive)) {
+      omits.value[directiveTable[directive as OmitHeaderDirective]] = true
+    }
+  })
+})
diff --git a/packages/home/app/middleware/trackPageview.global.ts b/packages/home/app/middleware/trackPageview.global.ts
index 9802f78..7900b51 100644
--- a/packages/home/app/middleware/trackPageview.global.ts
+++ b/packages/home/app/middleware/trackPageview.global.ts
@@ -6,14 +6,15 @@
  * file, You can obtain one at https://mozilla.org/MPL/2.0/.
  */
 
+import type { Omits } from '../utils/types'
+
 export default defineNuxtRouteMiddleware(() => {
   if (
     !useAppConfig().statsBase ||
+    useState<Omits>('omits').value.stats ||
     import.meta.client ||
     import.meta.dev ||
-    !useRequestHeader('user-agent') ||
-    // @ts-expect-error
-    useAppConfig().monitoringDataCollectorUserAgentMatch.test(useRequestHeader('user-agent'))
+    !useRequestHeader('user-agent')
   ) return
 
   const fetchHeaders: Record<string, string> = {}
diff --git a/packages/home/app/utils/types.ts b/packages/home/app/utils/types.ts
new file mode 100644
index 0000000..9c25efc
--- /dev/null
+++ b/packages/home/app/utils/types.ts
@@ -0,0 +1,10 @@
+/*
+ * Copyright (c) 2024 ThrRip, XMing (洺知-故犯)
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+export type OmitDirective = 'liveStatus' | 'playlist' | 'playlistMin' | 'stats'
+export type Omits = Partial<Record<OmitDirective, boolean>>