From 750b91db6649a6aaf56fffa03d3012ebced34403 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Thu, 12 Jun 2025 22:58:16 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=9C=AA=E8=AF=BB=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=E8=AE=A1=E6=95=B0=E5=92=8C=E6=A1=8C=E9=9D=A2=E5=9B=BE?=
 =?UTF-8?q?=E6=A0=87=E5=BE=BD=E7=AB=A0=E6=9B=B4=E6=96=B0=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/types.ts                            |  2 +
 src/layouts/components/UserNotification.vue | 41 ++++++++++---
 src/service-worker.ts                       | 60 +++++++++++++++++-
 src/types/global.d.ts                       | 17 ++++++
 src/utils/badge.ts                          | 68 +++++++++++++++++++++
 5 files changed, 178 insertions(+), 10 deletions(-)
 create mode 100644 src/types/global.d.ts
 create mode 100644 src/utils/badge.ts

diff --git a/src/api/types.ts b/src/api/types.ts
index c25359f8..1eaa6d4f 100644
--- a/src/api/types.ts
+++ b/src/api/types.ts
@@ -1008,6 +1008,8 @@ export interface SystemNotification {
   text: string
   // 通知时间
   date: string
+  // 是否已读
+  read?: boolean
 }
 
 // 下载器配置
diff --git a/src/layouts/components/UserNotification.vue b/src/layouts/components/UserNotification.vue
index 78a040ab..02894a3f 100644
--- a/src/layouts/components/UserNotification.vue
+++ b/src/layouts/components/UserNotification.vue
@@ -2,6 +2,7 @@
 import { formatDateDifference } from '@core/utils/formatters'
 import { SystemNotification } from '@/api/types'
 import { useI18n } from 'vue-i18n'
+import { updateAppBadge, clearAppBadge } from '@/utils/badge'
 
 const { t } = useI18n()
 
@@ -17,6 +18,34 @@ let eventSource: EventSource | null = null
 // 弹窗
 const appsMenu = ref(false)
 
+// 未读消息数量
+const unreadCount = computed(() => {
+  return notificationList.value.filter(item => !item.read).length
+})
+
+// 更新桌面图标徽章
+async function updateBadge() {
+  const count = unreadCount.value
+  await updateAppBadge(count)
+}
+
+// 清除桌面图标徽章
+async function clearBadge() {
+  await clearAppBadge()
+}
+
+// 标记所有消息为已读
+function markAllAsRead() {
+  hasNewMessage.value = false
+  // 标记所有消息为已读
+  notificationList.value.forEach(item => {
+    item.read = true
+  })
+  // 清除桌面图标徽章
+  clearBadge()
+  appsMenu.value = false
+}
+
 // SSE持续接收消息
 function startSSEMessager() {
   // 延迟 3 秒启动 SSE，避免相关认证信息尚未写入 Cookie 导致 403
@@ -27,6 +56,8 @@ function startSSEMessager() {
         const noti: SystemNotification = JSON.parse(event.data)
         notificationList.value.unshift(noti)
         hasNewMessage.value = true
+        // 更新桌面图标徽章
+        updateBadge()
       }
     })
   }, 3000)
@@ -70,15 +101,7 @@ onBeforeUnmount(() => {
         <template #append>
           <VTooltip :text="t('notification.markRead')">
             <template #activator="{ props }">
-              <IconBtn
-                v-bind="props"
-                @click="
-                  () => {
-                    hasNewMessage = false
-                    appsMenu = false
-                  }
-                "
-              >
+              <IconBtn v-bind="props" @click="markAllAsRead">
                 <VIcon icon="mdi-email-check-outline" size="20" />
               </IconBtn>
             </template>
diff --git a/src/service-worker.ts b/src/service-worker.ts
index 4ddfa5ac..73bc9ae2 100644
--- a/src/service-worker.ts
+++ b/src/service-worker.ts
@@ -11,6 +11,9 @@ precacheAndRoute(self.__WB_MANIFEST)
 // to allow work offline
 registerRoute(new NavigationRoute(createHandlerBoundToURL('index.html'), { denylist: [/^(\/[\w-]+)*\/api/] }))
 
+// 消息计数器，用于存储未读消息数量
+let unreadCount = 0
+
 // 通知选项
 const options = {
   icon: '/logo.png',
@@ -18,6 +21,33 @@ const options = {
   actions: [{ action: 'close', title: '关闭' }],
 }
 
+// 更新桌面图标徽章
+async function updateBadge(count: number) {
+  if ('setAppBadge' in navigator) {
+    try {
+      if (count > 0) {
+        await navigator.setAppBadge(count)
+      } else {
+        await navigator.clearAppBadge()
+      }
+    } catch (error) {
+      console.error('Failed to update app badge:', error)
+    }
+  }
+}
+
+// 清除桌面图标徽章
+async function clearBadge() {
+  if ('clearAppBadge' in navigator) {
+    try {
+      await navigator.clearAppBadge()
+      unreadCount = 0
+    } catch (error) {
+      console.error('Failed to clear app badge:', error)
+    }
+  }
+}
+
 // 监听 push 事件，显示通知
 self.addEventListener('push', function (event) {
   console.log('notification push')
@@ -43,7 +73,12 @@ self.addEventListener('push', function (event) {
       data: { url: payload.url },
       actions: options.actions,
     }
-    event.waitUntil(self.registration.showNotification(payload.title, content))
+
+    // 增加未读消息计数
+    unreadCount++
+
+    // 更新桌面图标徽章
+    event.waitUntil(Promise.all([self.registration.showNotification(payload.title, content), updateBadge(unreadCount)]))
   } catch (e) {
     console.error(e)
   }
@@ -71,3 +106,26 @@ self.addEventListener('notificationclick', function (event) {
     event.waitUntil(self.clients.openWindow(info.data?.url))
   }
 })
+
+// 监听来自主应用的消息，用于清除徽章或更新徽章数量
+self.addEventListener('message', function (event) {
+  console.log('service worker received message:', event.data)
+
+  if (event.data && event.data.type === 'CLEAR_BADGE') {
+    // 清除徽章
+    clearBadge()
+    event.ports[0]?.postMessage({ success: true })
+  } else if (event.data && event.data.type === 'UPDATE_BADGE') {
+    // 更新徽章数量
+    const count = event.data.count || 0
+    unreadCount = count
+    updateBadge(count)
+      .then(() => {
+        event.ports[0]?.postMessage({ success: true })
+      })
+      .catch(error => {
+        console.error('Failed to update badge:', error)
+        event.ports[0]?.postMessage({ success: false, error: error.message })
+      })
+  }
+})
diff --git a/src/types/global.d.ts b/src/types/global.d.ts
new file mode 100644
index 00000000..01ebeff1
--- /dev/null
+++ b/src/types/global.d.ts
@@ -0,0 +1,17 @@
+// PWA Badge API 类型定义
+declare global {
+  interface Navigator {
+    /**
+     * 设置应用徽章数量
+     * @param contents 要显示的数量，可选
+     */
+    setAppBadge(contents?: number): Promise<void>
+
+    /**
+     * 清除应用徽章
+     */
+    clearAppBadge(): Promise<void>
+  }
+}
+
+export {}
diff --git a/src/utils/badge.ts b/src/utils/badge.ts
new file mode 100644
index 00000000..ee3b7739
--- /dev/null
+++ b/src/utils/badge.ts
@@ -0,0 +1,68 @@
+/**
+ * PWA 徽章管理工具
+ */
+
+// 清除桌面图标徽章
+export async function clearAppBadge(): Promise<boolean> {
+  try {
+    // 如果浏览器支持原生Badge API，直接调用
+    if ('clearAppBadge' in navigator) {
+      await navigator.clearAppBadge()
+    }
+
+    // 向service worker发送清除徽章消息
+    if ('serviceWorker' in navigator && navigator.serviceWorker.controller) {
+      const messageChannel = new MessageChannel()
+
+      return new Promise(resolve => {
+        messageChannel.port1.onmessage = event => {
+          resolve(event.data.success)
+        }
+
+        navigator.serviceWorker.controller?.postMessage({ type: 'CLEAR_BADGE' }, [messageChannel.port2])
+      })
+    }
+
+    return true
+  } catch (error) {
+    console.error('Failed to clear app badge:', error)
+    return false
+  }
+}
+
+// 更新桌面图标徽章数量
+export async function updateAppBadge(count: number): Promise<boolean> {
+  try {
+    // 如果浏览器支持原生Badge API，直接调用
+    if ('setAppBadge' in navigator) {
+      if (count > 0) {
+        await navigator.setAppBadge(count)
+      } else {
+        await navigator.clearAppBadge()
+      }
+    }
+
+    // 向service worker发送更新徽章消息
+    if ('serviceWorker' in navigator && navigator.serviceWorker.controller) {
+      const messageChannel = new MessageChannel()
+
+      return new Promise(resolve => {
+        messageChannel.port1.onmessage = event => {
+          resolve(event.data.success)
+        }
+
+        navigator.serviceWorker.controller?.postMessage({ type: 'UPDATE_BADGE', count }, [messageChannel.port2])
+      })
+    }
+
+    return true
+  } catch (error) {
+    console.error('Failed to update app badge:', error)
+    return false
+  }
+}
+
+// 检查浏览器是否支持Badge API
+export function supportsBadgeAPI(): boolean {
+  return 'setAppBadge' in navigator && 'clearAppBadge' in navigator
+}