Add content script for tracking

Author: Stigmatoz

src/background.ts

@@ -1,6 +1,6 @@
 import browser from "webextension-polyfill";
 
-console.log("Hello from the background!");
+console.log("Start background script");
 
 browser.runtime.onInstalled.addListener((details) => {
   console.log("Extension installed:", details);

src/manifest.json

@@ -3,8 +3,8 @@
     "{{firefox}}.manifest_version": 2,
     "name": "Web Activity Time Tracker",
     "short_name": "Web Tracker",
-    "version": "1.7.2",
-    "description": "Track and limit time your activity in the browser every day.",
+    "version": "2.0.0",
+    "description": "Track and limit time your activity in the browser every day. Control how much time you spend on websites.",
     "options_page": "src/options.html",
     "icons": {
         "16": "src/icons/16x16.png",
@@ -27,6 +27,14 @@
         "https://www.netflix.com/*"
     ],
     "offline_enabled": true,
+    "content_scripts": [
+        {
+          "matches": ["*://*/*"],
+          "js": ["src/tracker.ts"],
+          "all_frames": true,
+          "run_at": "document_start"
+        }
+    ],
     "background": {
         "{{chrome}}.service_worker": "src/background.ts",
         "{{firefox}}.scripts": ["src/background.ts"]

src/tracker.ts

@@ -0,0 +1,36 @@
+import Browser from 'webextension-polyfill';
+import { isValidPage } from './compositions/valid-page';
+import { extractHostname } from './compositions/extract-hostname';
+
+export async function initTracker(){
+    const window = await Browser.windows.getLastFocused({ populate: true });
+    if (window.focused){
+        const activeTab = window.tabs?.find(t => t.active === true);
+        if (isValidPage(activeTab)) {
+            var activeUrl = extractHostname(activeTab?.url);
+            var tab = activity.getTab(activeUrl);
+            if (tab === undefined) {
+                activity.addTab(activeTab);
+            }
+
+            if (activity.isInBlackList(activeUrl)) {
+                chrome.action.setBadgeBackgroundColor({ color: '#fdb8b8' })
+                chrome.action.setBadgeText({
+                    tabId: activeTab.id,
+                    text: 'n/a'
+                });
+            } else {
+                if (tab !== undefined) {
+                    if (currentTab !== tab.url) {
+                        activity.setCurrentActiveTab(tab.url);
+                    }
+                    chrome.idle.queryState(parseInt(setting_interval_inactivity), function(state) {
+                        if (state === 'active') {
+                            mainTRacker(activeUrl, tab, activeTab);
+                        } else checkDOM(state, activeUrl, tab, activeTab);
+                    });
+                }
+            }
+        } 
+    } else activity.closeIntervalForCurrentTab(); 
+}