From da9cd9dd8e9b16f995dd01e7ca4dc5f985673425 Mon Sep 17 00:00:00 2001
From: Shugen <shugen002@users.noreply.github.com>
Date: Sun, 31 Mar 2024 01:52:52 +0800
Subject: [PATCH] feat: add ignore this version

---
 src/components/UpdateNotifier.vue | 74 ++++++++++++++++++++++++++++---
 src/const.ts                      |  1 +
 2 files changed, 70 insertions(+), 5 deletions(-)

diff --git a/src/components/UpdateNotifier.vue b/src/components/UpdateNotifier.vue
index 54e3bc8..b36eeaf 100644
--- a/src/components/UpdateNotifier.vue
+++ b/src/components/UpdateNotifier.vue
@@ -1,7 +1,7 @@
 <script lang="ts">
 /* eslint-disable vue/valid-template-root */
 import { h, onMounted } from 'vue';
-import { VERSION } from '../const';
+import { STORAGE_UPDATE_IGNORE, VERSION } from '../const';
 import { useNotification, NButton } from 'naive-ui';
 import { getRecorderLatestVersion } from '../utils/version';
 import { compare as compareVersion } from 'semver';
@@ -16,11 +16,20 @@ onMounted(() => {
   getRecorderLatestVersion().then((version) => {
     try {
       if (compareVersion(version.webui.version, VERSION) > 0) {
+        const store = JSON.parse(localStorage.getItem(STORAGE_UPDATE_IGNORE) || '{}');
+        if (store.webui && store.webui === version.webui.version) {
+          return;
+        }
         const n = notification.warning({
           title: '更新提醒',
           content: `检测到新版本WebUI：${version.webui.version}，请及时更新！`,
           duration: 0,
-          action: () =>
+          action: () => h('div', {
+            style: {
+              display: 'flex',
+              gap: '1rem',
+            },
+          }, [
             h(NButton, {
               type: 'primary',
               text: true,
@@ -29,6 +38,28 @@ onMounted(() => {
                 window.open(version.webui.url);
               },
             }, () => '立即更新'),
+            h(NButton, {
+              type: 'default',
+              text: true,
+              onClick: () => {
+                n.destroy();
+              },
+            }, () => '忽略'),
+            h(NButton, {
+              type: 'default',
+              text: true,
+              onClick: () => {
+                n.destroy();
+                try {
+                  const store = JSON.parse(localStorage.getItem(STORAGE_UPDATE_IGNORE) || '{}');
+                  store.webui = version.webui.version;
+                  localStorage.setItem(STORAGE_UPDATE_IGNORE, JSON.stringify(store));
+                } catch (error) {
+                  // ignore
+                }
+              },
+            }, () => '忽略此版本'),
+          ]),
         });
       }
     } catch (e) {
@@ -41,15 +72,21 @@ const notified: { [key: string]: boolean } = {};
 
 function onRecorderChange() {
   if (recorderController.recorder !== null) {
-    if (notified[recorderController.recorder.meta.id]) {
+    const recorderMeta = recorderController.recorder.meta;
+
+    if (notified[recorderMeta.id]) {
       return;
     }
-    const recorderMeta = recorderController.recorder.meta;
+
     recorderController.recorder.getVersion().then((v) => {
       const serverVersion = v.fullSemVer;
       getRecorderLatestVersion().then((version) => {
         try {
           if (compareVersion(version.recorder.version, serverVersion) > 0) {
+            const store = JSON.parse(localStorage.getItem(STORAGE_UPDATE_IGNORE) || '{}');
+            if (store[recorderMeta.id] && store[recorderMeta.id] === version.webui.version) {
+              return;
+            }
             notified[recorderMeta.id] = true;
             const n = notification.warning({
               title: '更新提醒',
@@ -65,7 +102,12 @@ function onRecorderChange() {
                 ];
               },
               duration: 0,
-              action: () =>
+              action: () => h('div', {
+                style: {
+                  display: 'flex',
+                  gap: '1rem',
+                },
+              }, [
                 h(NButton, {
                   type: 'primary',
                   text: true,
@@ -74,6 +116,28 @@ function onRecorderChange() {
                     window.open(version.recorder.url);
                   },
                 }, () => '立即更新'),
+                h(NButton, {
+                  type: 'default',
+                  text: true,
+                  onClick: () => {
+                    n.destroy();
+                  },
+                }, () => '忽略'),
+                h(NButton, {
+                  type: 'default',
+                  text: true,
+                  onClick: () => {
+                    n.destroy();
+                    try {
+                      const store = JSON.parse(localStorage.getItem(STORAGE_UPDATE_IGNORE) || '{}');
+                      store[recorderMeta.id] = version.webui.version;
+                      localStorage.setItem(STORAGE_UPDATE_IGNORE, JSON.stringify(store));
+                    } catch (error) {
+                      // ignore
+                    }
+                  },
+                }, () => '忽略此版本'),
+              ]),
             });
           }
         } catch (e) {
diff --git a/src/const.ts b/src/const.ts
index 3b231be..86213e1 100644
--- a/src/const.ts
+++ b/src/const.ts
@@ -7,3 +7,4 @@ export const STORAGE_THEME = 'brec.theme';
 export const STORAGE_LATEST_VERSION = 'brec.latestVersion';
 export const STORAGE_FILES_SORT_OPTION = 'brec.filesSortOption';
 export const STORAGE_ROOM_ORDER_METHOD = 'brec.roomOrderMethod';
+export const STORAGE_UPDATE_IGNORE = 'brec.updateIgnore';