Badge support (#28)

* add badge display for selected user

* feat: while the selected role is disabled, the first role will be used by default

* feat: UI-only refresh support

Co-authored-by: daidr <daidr@daidr.me>

Author: lingyired

src/background/index.ts

@@ -1,5 +1,5 @@
 import { onMessage } from 'webext-bridge'
-import { alarms, cookies, i18n, notifications, runtime, storage } from 'webextension-polyfill'
+import { action, alarms, cookies, i18n, notifications, runtime, storage } from 'webextension-polyfill'
 import type { Cookies, Notifications } from 'webextension-polyfill'
 import type { IAlertSetting, IAlertStatus, IRoleDataItem, IUserData, IUserDataItem } from '~/types'
 import { getRoleDataByCookie, getRoleInfoByCookie } from '~/utils'
@@ -163,6 +163,9 @@ const writeDataToStorage = async function <T>(key: string, data: T) {
   await storage.local.set({ [key]: data })
 }
 
+// selected uid
+let selectedUid = ''
+
 const targetPages = [
   'https://api-os-takumi.mihoyo.com/binding/api/getUserGameRolesByCookie?game_biz=hk4e_global&asource=paimon',
   'https://api-takumi.mihoyo.com/binding/api/getUserGameRolesByCookie?game_biz=hk4e_cn&asource=paimon',
@@ -230,6 +233,10 @@ const readDataFromStorage = async function <T>(key: string, defaultVal: T): Prom
     return defaultVal
 }
 
+const getSelectedUid = async () => {
+  return await readDataFromStorage<string>('selectedRole', '')
+}
+
 // 获取国服cookie
 const getMiHoYoCookie = async function () {
   let cookieString = ''
@@ -372,36 +379,61 @@ const doAlertCheck = async function (roleInfo: IUserDataItem) {
   }
 }
 
-const refreshData = async function () {
+const refreshData = async function (uiOnly = false) {
   // 取出原始 roleList
   const originRoleList = await readDataFromStorage<IUserDataItem[]>('roleList', [])
   // 取出启用的 role
   const enabledRoleList = originRoleList.filter((item) => {
     return item.isEnabled
   })
 
+  // 如果当前还没有 selectedUid 则获取一个
+  let _selectedUid = selectedUid || await getSelectedUid()
+
+  if (enabledRoleList.length > 0) {
+    // 查看选中uid是否已经启用
+    if (!enabledRoleList.find(item => item.uid === _selectedUid)) {
+      // 未启用则选中第一个
+      _selectedUid = enabledRoleList[0]?.uid
+    }
+  }
+
   const setCookie = async (cookie: string) => {
     currentCookie = cookie
     await updateRules()
   }
 
+  let hasUpdatedBadge = false
   // 遍历启用的 role
   for (const role of enabledRoleList) {
-    const data = await getRoleDataByCookie(role.serverType === 'os', role.cookie, role.uid, role.serverRegion, setCookie)
+    const data = uiOnly ? role.data : await getRoleDataByCookie(role.serverType === 'os', role.cookie, role.uid, role.serverRegion, setCookie)
+
     if (data) {
       // 更新 roleList
-      const index = originRoleList.findIndex((item) => {
+      const roleIndex = originRoleList.findIndex((item) => {
         return item.uid === role.uid
       })
 
-      originRoleList.splice(index, 1, {
+      originRoleList.splice(roleIndex, 1, {
         ...role,
         data,
         isError: false,
         errorMessage: '',
         updateTimestamp: Date.now(),
       })
       doAlertCheck(role)
+
+      if (!hasUpdatedBadge) {
+        let shouldUpdateBadge = false
+        if (_selectedUid && _selectedUid === role.uid)
+          shouldUpdateBadge = true // 更新 _selectedUid 当前的 resin 数据到 badge
+
+        if (shouldUpdateBadge && role?.data?.current_resin) {
+          action.setBadgeText({ text: `${role.data.current_resin}` })
+          action.setBadgeBackgroundColor({ color: '#6F9FDF' })
+          hasUpdatedBadge = true
+        }
+      }
     }
     else {
       // 获取失败，写入错误信息
@@ -457,7 +489,7 @@ onMessage('get_role_list', async () => {
 })
 
 onMessage('get_selected_role', async () => {
-  return await readDataFromStorage<string>('selectedRole', '')
+  return await getSelectedUid()
 })
 
 onMessage('refresh_request', async () => {
@@ -466,7 +498,9 @@ onMessage('refresh_request', async () => {
 })
 
 onMessage<{ uid: string }, 'set_selected_role'>('set_selected_role', async ({ data: { uid } }) => {
+  selectedUid = uid // update cache
   await writeDataToStorage('selectedRole', uid)
+  refreshData()
 })
 
 onMessage<{ uid: string; status: boolean }, 'set_role_status'>('set_role_status', async ({ data: { uid, status } }) => {
@@ -479,6 +513,8 @@ onMessage<{ uid: string; status: boolean }, 'set_role_status'>('set_role_status'
   clearNotifications(originRoleList[index].alertStatus)
   originRoleList[index].alertStatus = defaultAlertStatus
   await writeDataToStorage('roleList', originRoleList)
+  // 刷新一次数据（仅刷新ui，例如badge/notification）
+  refreshData(true)
 })
 
 onMessage<{ uid: string; status: boolean }, 'set_role_alert_status'>('set_role_alert_status', async ({ data: { uid, status } }) => {