Main tracker.ts

Author: Stigmatoz

src/tracker.ts

@@ -1,34 +1,72 @@
-import Browser from 'webextension-polyfill';
-import { isValidPage } from './compositions/valid-page';
-import { extractHostname } from './compositions/extract-hostname';
-import { injectTabsRepository } from './repository/inject-tabs-repository';
-import { isInBlackList } from './compositions/black-list';
-import { useBadge } from './compositions/set-badge';
-import { BadgeColor } from './compositions/types';
+import Browser from "webextension-polyfill";
+import { isValidPage } from "./compositions/valid-page";
+import { extractHostname } from "./compositions/extract-hostname";
+import { injectTabsRepository } from "./repository/inject-tabs-repository";
+import { isInBlackList } from "./compositions/black-list";
+import { useBadge } from "./compositions/set-badge";
+import { BadgeColor } from "./compositions/types";
+import { INTERVAL_INACTIVITY_DEFAULT, StorageParams,  } from "./storage/storage-params";
+import { injecStorage } from "./storage/inject-storage";
+import { closeInterval } from "./compositions/daily-intervals";
+import { ActiveTab } from "./compositions/activeTab";
+import { isLimitExceeded } from "./compositions/limit-list";
+import { Tab } from "./entity/tab";
+import { useBlockPage } from "./compositions/block-page";
 
-export async function initTracker(){
+async function trackTime(){
+    const activeTabInstance = ActiveTab.getInstance();
     const window = await Browser.windows.getLastFocused({ populate: true });
-    if (window.focused){
-        const activeTab = window.tabs?.find(t => t.active === true);
+    if (window.focused) {
+        const activeTab = window.tabs?.find((t) => t.active === true);
         if (isValidPage(activeTab)) {
             const activeUrl = extractHostname(activeTab!.url);
+            activeTabInstance.setActiveTab(activeUrl);
 
             if (await isInBlackList(activeUrl)) {
-                useBadge({ tabId: activeTab!.id!, text: 'n/a', color: BadgeColor.green });
-            } else{
+                useBadge({
+                    tabId: activeTab!.id!,
+                    text: "n/a",
+                    color: BadgeColor.red,
+                });
+            } else {
                 const repo = await injectTabsRepository();
-                let tab = repo.getTab(activeUrl);
-                if (tab === undefined)
-                    tab = await repo.addTab(activeTab!);
-
-                    if (currentTab !== tab.url) {
-                        activity.setCurrentActiveTab(tab.url);
-                    }
-                    const state = await Browser.idle.queryState(parseInt(setting_interval_inactivity));
-                        if (state === 'active') {
-                            mainTRacker(activeUrl, tab, activeTab);
-                        } else checkDOM(state, activeUrl, tab, activeTab);
+                const tab = repo.getTab(activeUrl);
+                if (tab == undefined) {
+                    await repo.addTab(activeTab!);
+                } else {
+                    const storage = injecStorage();
+                    const inactivityInterval = (await storage.getValue(
+                        StorageParams.INTERVAL_INACTIVITY,
+                        INTERVAL_INACTIVITY_DEFAULT
+                    )) as number;
+                    const state = await Browser.idle.queryState(inactivityInterval);
+                    mainTracker(state, activeTab!, activeUrl, tab);
+                }
             }
         }
-    } else activity.closeIntervalForCurrentTab(); 
-}
+    } else {
+        await closeInterval(activeTabInstance.getActiveTab());
+        activeTabInstance.setActiveTab(null);
+    }
+}
+
+async function mainTracker(state: Browser.Idle.IdleState, activeTab: Browser.Tabs.Tab, activeUrl: string, tab: Tab){
+    function isAudible(){
+        return state === 'idle' && activeTab.audible;
+    }
+
+    if (state === 'active' || isAudible()){
+        if (await isLimitExceeded(activeUrl, tab)){
+            const summaryTime = tab.days.at(-1)!.summary;
+            const summaryCounter = tab.days.at(-1)!.counter;
+            await useBlockPage(activeUrl, summaryTime, summaryCounter);
+            return;
+        }
+
+        tab.incSummaryTime();
+    }
+}
+
+export async function initTracker() {
+    setInterval(trackTime, 1000);
+}