From 975663f0ffd3e1000030f2a44a89933ae4013685 Mon Sep 17 00:00:00 2001
From: ssongliu <73214554+ssongliu@users.noreply.github.com>
Date: Fri, 14 Apr 2023 15:06:55 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E8=87=AA=E5=8A=A8?=
 =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=A3=80=E6=B5=8B=20(#621)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 frontend/src/components/system-upgrade/index.vue | 15 +++++++++++++--
 frontend/src/store/index.ts                      |  1 +
 frontend/src/store/interface/index.ts            |  1 +
 frontend/src/views/home/index.vue                | 13 +++++++++++++
 4 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/frontend/src/components/system-upgrade/index.vue b/frontend/src/components/system-upgrade/index.vue
index a0ca16e7b..288c2a37c 100644
--- a/frontend/src/components/system-upgrade/index.vue
+++ b/frontend/src/components/system-upgrade/index.vue
@@ -1,10 +1,21 @@
 <template>
     <div>
         <span class="version">{{ version }}</span>
-        <el-button v-if="version !== 'Waiting'" style="margin-top: -2px" type="primary" link @click="onLoadUpgradeInfo">
+        <el-badge is-dot class="item" v-if="version !== 'Waiting' && globalStore.hasNewVersion">
+            <el-button style="margin-top: -8px" type="primary" link @click="onLoadUpgradeInfo">
+                {{ $t('setting.upgradeCheck') }}
+            </el-button>
+        </el-badge>
+        <el-button
+            v-if="version !== 'Waiting' && !globalStore.hasNewVersion"
+            style="margin-top: -2px"
+            type="primary"
+            link
+            @click="onLoadUpgradeInfo"
+        >
             {{ $t('setting.upgradeCheck') }}
         </el-button>
-        <el-tag v-else round style="margin-left: 10px">{{ $t('setting.upgrading') }}</el-tag>
+        <el-tag v-if="version === 'Waiting'" round style="margin-left: 10px">{{ $t('setting.upgrading') }}</el-tag>
     </div>
 
     <el-drawer :close-on-click-modal="false" :key="refresh" v-model="drawerVisiable" size="50%" append-to-body>
diff --git a/frontend/src/store/index.ts b/frontend/src/store/index.ts
index d1a47e83e..04c54d6a5 100644
--- a/frontend/src/store/index.ts
+++ b/frontend/src/store/index.ts
@@ -21,6 +21,7 @@ export const GlobalStore = defineStore({
         },
         isFullScreen: false,
         agreeLicense: false,
+        hasNewVersion: false,
     }),
     getters: {},
     actions: {
diff --git a/frontend/src/store/interface/index.ts b/frontend/src/store/interface/index.ts
index 55cb4b0e2..44bf0bba6 100644
--- a/frontend/src/store/interface/index.ts
+++ b/frontend/src/store/interface/index.ts
@@ -17,6 +17,7 @@ export interface GlobalState {
     themeConfig: ThemeConfigProp;
     isFullScreen: boolean;
     agreeLicense: boolean;
+    hasNewVersion: boolean;
 }
 
 export interface MenuState {
diff --git a/frontend/src/views/home/index.vue b/frontend/src/views/home/index.vue
index fea70a529..b3a595800 100644
--- a/frontend/src/views/home/index.vue
+++ b/frontend/src/views/home/index.vue
@@ -210,7 +210,10 @@ import { useRouter } from 'vue-router';
 import RouterButton from '@/components/router-button/index.vue';
 import { loadBaseInfo, loadCurrentInfo } from '@/api/modules/dashboard';
 import { getIOOptions, getNetworkOptions } from '@/api/modules/monitor';
+import { loadUpgradeInfo } from '@/api/modules/setting';
+import { GlobalStore } from '@/store';
 const router = useRouter();
+const globalStore = GlobalStore();
 
 const statuRef = ref();
 const appRef = ref();
@@ -470,7 +473,17 @@ const loadData = async () => {
     }
 };
 
+const loadUpgradeStatus = async () => {
+    const res = await loadUpgradeInfo();
+    if (res.data) {
+        globalStore.hasNewVersion = true;
+    } else {
+        globalStore.hasNewVersion = false;
+    }
+};
+
 onMounted(() => {
+    loadUpgradeStatus();
     onLoadNetworkOptions();
     onLoadIOOptions();
     onLoadBaseInfo(true, 'all');