feat: support track the time of tab groups (#481)

Author: sheepzh

src/api/chrome/permission.ts

@@ -36,5 +36,5 @@ export async function requestPerm(perm: chrome.runtime.ManifestPermissions): Pro
 }
 
 export async function onPermRemoved(callback: ArgCallback<chrome.permissions.Permissions>) {
-    chrome.permissions.onRemoved.addListener(callback)
+    chrome.permissions?.onRemoved.addListener(callback)
 }

src/background/index.ts

@@ -7,6 +7,7 @@
 
 import { listTabs } from "@api/chrome/tab"
 import { isNoneWindowId, onNormalWindowFocusChanged } from "@api/chrome/window"
+import optionHolder from "@service/components/option-holder"
 import { isBrowserUrl } from "@util/pattern"
 import { openLog } from "../common/logger"
 import ActiveTabListener from "./active-tab-listener"
@@ -78,3 +79,6 @@ onNormalWindowFocusChanged(async windowId => {
         badgeTextManager.updateFocus({ url, tabId })
     })
 })
+
+// listen permission change event
+optionHolder.listenPermChange()

src/background/track-server.ts

@@ -112,6 +112,7 @@ const RUN_TIME_END_CACHE: { [host: string]: number } = {}
 async function handleTrackRunTimeEvent(event: timer.core.Event): Promise<void> {
     const { start, end, url, host } = event || {}
     if (!host || !start || !end) return
+    if (whitelistHolder.contains(host, url)) return
     const realStart = Math.max(RUN_TIME_END_CACHE[host] ?? 0, start)
     const byDate = splitRunTime(realStart, end)
     if (!Object.keys(byDate).length) return

src/service/components/option-holder.ts

@@ -18,6 +18,12 @@ class OptionHolder {
         })
     }
 
+    listenPermChange() {
+        onPermRemoved(perm => {
+            perm.permissions?.includes('tabGroups') && this.set({ countTabGroup: false })
+        })
+    }
+
     private async reset(): Promise<DefaultOption> {
         const exist: Partial<timer.option.AllOption> = await db.getOption()
         const result = defaultOption()

test-e2e/common/whitelist.ts

@@ -16,7 +16,5 @@ export async function createWhitelist(context: LaunchContext, white: string) {
     await whitePage.click('.el-button:nth-child(3)')
     const checkBtn = await whitePage.waitForSelector('.el-overlay.is-message-box .el-button.el-button--primary')
     await checkBtn?.click()
-    await sleep(.2)
-
-    await whitePage.close()
+    setTimeout(() => whitePage.close(), 200)
 }

test-e2e/tracker/run-time.test.ts

@@ -14,8 +14,7 @@ async function clickRunTimeChange(siteHost: string): Promise<void> {
         const runTimeSwitch = document.querySelector<HTMLDivElement>('#site-manage-table-wrapper table > tbody > tr > td.el-table_1_column_7 .el-switch')
         runTimeSwitch?.click()
     })
-    await sleep(.2)
-    await sitePage.close()
+    setTimeout(() => sitePage.close(), 200)
 }
 
 describe('Run time tracking', () => {