Add Group Information to Tracking (#1266)

Co-authored-by: Isaac <NoSpam>

src/background/track.js

@@ -1,10 +1,13 @@
 import browser from "webextension-polyfill";
+import browserInfo from "browser-info";
 import log from "loglevel";
 import { getSettings } from "src/settings/settings";
+import { queryTabGroups } from "../common/tabGroups";
 import { updateSession } from "./save";
 import getSessions from "./getSessions";
 
 const logDir = "background/track";
+const isEnabledTabGroups = browserInfo().name == "Chrome" && browserInfo().version >= 89;
 
 const setTrackingInfo = async (trackingWindows, isTracking) => {
   await browser.storage.session.set({ trackingInfo: { trackingWindows, isTracking } });
@@ -47,6 +50,17 @@ export const updateTrackingSession = async (tempSession) => {
       trackedSession.tabsNumber += Object.keys(win).length;
     }
 
+    //Update Group Information
+    if (isEnabledTabGroups && getSettings("saveTabGroups")) {
+      const filteredWindows = Object.values(trackedSession.windowsInfo).filter(window => window.type === "normal");
+      const tabGroups = await Promise.all(filteredWindows.map(window => queryTabGroups({
+        windowId: window.id,
+      })));
+      const filteredTabGroups = tabGroups.flat().filter(tabGroup =>
+        Object.keys(trackedSession.windows).includes(String(tabGroup.windowId)));
+      if (filteredTabGroups.length > 0) trackedSession.tabGroups = filteredTabGroups;
+    }
+
     await updateSession(trackedSession);
   }
 };