From 7e6afd35279a530a153a9dda83fc098c2da0a1f9 Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 28 Apr 2023 09:33:12 -0500
Subject: [PATCH 1/7] Assert accessing id on an actual tab

---
 src/background.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/background.ts b/src/background.ts
index 7a2d2fb..b778ca4 100644
--- a/src/background.ts
+++ b/src/background.ts
@@ -43,9 +43,12 @@ browser.windows.onFocusChanged.addListener(async (windowId) => {
 browser.tabs.onUpdated.addListener(async (tabId, changeInfo) => {
   if (changeInfo.status === 'complete') {
     // Get current tab URL.
-    const [tab] = await browser.tabs.query({ active: true, currentWindow: true });
+    const tabs: browser.Tabs.Tab[] = await browser.tabs.query({
+      active: true,
+      currentWindow: true,
+    });
     // If tab updated is the same as active tab
-    if (tabId == tab.id) {
+    if (tabId == tabs[0]?.id) {
       await WakaTimeCore.recordHeartbeat();
     }
   }

From 6b9dd5c26e8b5d93ed68610cda1fec33557b8b00 Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 28 Apr 2023 09:49:51 -0500
Subject: [PATCH 2/7] Display track social media input check correctly

---
 src/components/Options.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/components/Options.tsx b/src/components/Options.tsx
index c785c5c..d6aeb68 100644
--- a/src/components/Options.tsx
+++ b/src/components/Options.tsx
@@ -302,7 +302,7 @@ export default function Options(): JSX.Element {
                 <div
                   onClick={() => setState({ ...state, trackSocialMedia: !state.trackSocialMedia })}
                 >
-                  <input type="checkbox" defaultChecked={state.trackSocialMedia} />
+                  <input type="checkbox" checked={state.trackSocialMedia} />
                   <span>Track social media sites</span>
                 </div>
                 <button

From 9898339a33a5d48fe9f80a4ca606cc03028e306a Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 28 Apr 2023 13:18:06 -0500
Subject: [PATCH 3/7] chore: Change dev and social sites config and option from
 string to list of strings

---
 src/components/Options.tsx    | 14 +++++------
 src/config/config.test.ts     | 44 +++++++++++++++++++----------------
 src/config/config.ts          | 31 ++++++++++++++++++++----
 src/core/WakaTimeCore.ts      |  5 ++--
 src/utils/getDomainFromUrl.ts |  9 +++++++
 5 files changed, 69 insertions(+), 34 deletions(-)

diff --git a/src/components/Options.tsx b/src/components/Options.tsx
index d6aeb68..a0f8107 100644
--- a/src/components/Options.tsx
+++ b/src/components/Options.tsx
@@ -16,7 +16,7 @@ interface State {
   loading: boolean;
   loggingStyle: string;
   loggingType: string;
-  socialMediaSites: string;
+  socialMediaSites: string[];
   theme: string;
   trackSocialMedia: boolean;
   whitelist: string;
@@ -62,7 +62,7 @@ export default function Options(): JSX.Element {
       hostname: items.hostname as string,
       loggingStyle: items.loggingStyle as string,
       loggingType: items.loggingType as string,
-      socialMediaSites: items.socialMediaSites as string,
+      socialMediaSites: items.socialMediaSites as string[],
       theme: items.theme as string,
       trackSocialMedia: items.trackSocialMedia as boolean,
       whitelist: items.whitelist as string,
@@ -336,14 +336,14 @@ export default function Options(): JSX.Element {
                       </div>
                       <div className="modal-body">
                         <SitesList
-                          handleChange={(sites: string) =>
+                          handleChange={(sites: string) => {
                             setState({
                               ...state,
-                              socialMediaSites: sites,
-                            })
-                          }
+                              socialMediaSites: sites.split('\n'),
+                            });
+                          }}
                           label="Social"
-                          sites={state.socialMediaSites}
+                          sites={state.socialMediaSites.join('\n')}
                           helpText="Sites that you don't want to show in your reports."
                           rows={5}
                         />
diff --git a/src/config/config.test.ts b/src/config/config.test.ts
index 623fac0..98d2c35 100644
--- a/src/config/config.test.ts
+++ b/src/config/config.test.ts
@@ -34,16 +34,18 @@ describe('wakatime config', () => {
         },
         "currentUserApiEndPoint": "/users/current",
         "detectionIntervalInSeconds": 60,
-        "devSites": "https://codepen.io/
-      https://www.codewars.com/
-      https://dev.to/
-      https://github.com/
-      https://www.hackerrank.com/
-      https://leetcode.com/
-      https://developer.mozilla.org/en-US/
-      https://stackoverflow.com/
-      https://www.udemy.com/
-      https://www.w3schools.com/",
+        "devSites": [
+          "codepen.io",
+          "codewars.com",
+          "dev.to",
+          "github.com",
+          "hackerrank.com",
+          "leetcode.com",
+          "developer.mozilla.org",
+          "stackoverflow.com",
+          "udemy.com",
+          "w3schools.com",
+        ],
         "heartbeatApiEndPoint": "/users/current/heartbeats",
         "hostname": "",
         "loggingEnabled": true,
@@ -51,16 +53,18 @@ describe('wakatime config', () => {
         "loggingType": "domain",
         "logoutUserUrl": "https://wakatime.com/logout",
         "name": "WakaTime",
-        "socialMediaSites": "https://www.facebook.com/
-      https://www.instagram.com/
-      https://www.linkedin.com/
-      https://www.pinterest.com/
-      https://www.reddit.com/
-      https://www.snapchat.com/
-      https://www.tiktok.com/
-      https://twitter.com/
-      https://www.whatsapp.com/
-      https://www.youtube.com/",
+        "socialMediaSites": [
+          "facebook.com",
+          "instagram.com",
+          "linkedin.com",
+          "pinterest.com",
+          "reddit.com",
+          "snapchat.com",
+          "tiktok.com",
+          "twitter.com",
+          "whatsapp.com",
+          "youtube.com",
+        ],
         "states": [
           "allGood",
           "notLogging",
diff --git a/src/config/config.ts b/src/config/config.ts
index 08406ed..32bc0bf 100644
--- a/src/config/config.ts
+++ b/src/config/config.ts
@@ -66,7 +66,7 @@ export interface Config {
    */
   detectionIntervalInSeconds: number;
 
-  devSites: string;
+  devSites: string[];
   /**
    * Url to which to send the heartbeat
    */
@@ -87,7 +87,7 @@ export interface Config {
    * Extension name
    */
   name: string;
-  socialMediaSites: string;
+  socialMediaSites: string[];
   states: ApiStates[];
   /**
    * Get stats from the wakatime api
@@ -132,8 +132,18 @@ const config: Config = {
 
   detectionIntervalInSeconds: 60,
 
-  devSites:
-    'https://codepen.io/\nhttps://www.codewars.com/\nhttps://dev.to/\nhttps://github.com/\nhttps://www.hackerrank.com/\nhttps://leetcode.com/\nhttps://developer.mozilla.org/en-US/\nhttps://stackoverflow.com/\nhttps://www.udemy.com/\nhttps://www.w3schools.com/',
+  devSites: [
+    'codepen.io',
+    'codewars.com',
+    'dev.to',
+    'github.com',
+    'hackerrank.com',
+    'leetcode.com',
+    'developer.mozilla.org',
+    'stackoverflow.com',
+    'udemy.com',
+    'w3schools.com',
+  ],
 
   heartbeatApiEndPoint: process.env.HEARTBEAT_API_URL ?? '/users/current/heartbeats',
 
@@ -149,7 +159,18 @@ const config: Config = {
 
   name: 'WakaTime',
 
-  socialMediaSites: `https://www.facebook.com/\nhttps://www.instagram.com/\nhttps://www.linkedin.com/\nhttps://www.pinterest.com/\nhttps://www.reddit.com/\nhttps://www.snapchat.com/\nhttps://www.tiktok.com/\nhttps://twitter.com/\nhttps://www.whatsapp.com/\nhttps://www.youtube.com/`,
+  socialMediaSites: [
+    'facebook.com',
+    'instagram.com',
+    'linkedin.com',
+    'pinterest.com',
+    'reddit.com',
+    'snapchat.com',
+    'tiktok.com',
+    'twitter.com',
+    'whatsapp.com',
+    'youtube.com',
+  ],
 
   states: ['allGood', 'notLogging', 'notSignedIn', 'blacklisted', 'whitelisted'],
 
diff --git a/src/core/WakaTimeCore.ts b/src/core/WakaTimeCore.ts
index 793191c..68a837a 100644
--- a/src/core/WakaTimeCore.ts
+++ b/src/core/WakaTimeCore.ts
@@ -12,7 +12,7 @@ import { IS_FIREFOX, generateProjectFromDevSites } from '../utils';
 import { getApiKey } from '../utils/apiKey';
 import changeExtensionState from '../utils/changeExtensionState';
 import contains from '../utils/contains';
-import getDomainFromUrl from '../utils/getDomainFromUrl';
+import getDomainFromUrl, { getDomainFromUrlWithoutProtocol } from '../utils/getDomainFromUrl';
 
 class WakaTimeCore {
   tabsWithDevtoolsOpen: Tabs.Tab[];
@@ -149,8 +149,9 @@ class WakaTimeCore {
         url = document.URL;
       }
 
+      const hostname = getDomainFromUrlWithoutProtocol(url).replace('www.', '');
       if (!items.trackSocialMedia) {
-        if (contains(url, items.socialMediaSites as string)) {
+        if ((items.socialMediaSites as string[]).includes(hostname)) {
           return changeExtensionState('blacklisted');
         }
       }
diff --git a/src/utils/getDomainFromUrl.ts b/src/utils/getDomainFromUrl.ts
index 4c45620..8f19f4f 100644
--- a/src/utils/getDomainFromUrl.ts
+++ b/src/utils/getDomainFromUrl.ts
@@ -6,3 +6,12 @@ export default function getDomainFromUrl(url: string): string {
 
   return parts[0] + '//' + parts[2];
 }
+
+/**
+ * Returns domain from given URL.
+ */
+export function getDomainFromUrlWithoutProtocol(url: string): string {
+  const parts = url.split('/');
+
+  return parts[2];
+}

From b2d1fdc9c65b83ede6ec7e08094035f56bed93d2 Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 28 Apr 2023 13:28:58 -0500
Subject: [PATCH 4/7] Toggle Social Media

---
 src/components/Options.tsx | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/components/Options.tsx b/src/components/Options.tsx
index a0f8107..bce9fce 100644
--- a/src/components/Options.tsx
+++ b/src/components/Options.tsx
@@ -143,6 +143,10 @@ export default function Options(): JSX.Element {
     });
   };
 
+  const toggleSocialMedia = () => {
+    setState({ ...state, trackSocialMedia: !state.trackSocialMedia });
+  };
+
   const loggingStyle = function () {
     if (state.loggingStyle == 'blacklist') {
       return (
@@ -299,11 +303,13 @@ export default function Options(): JSX.Element {
 
             <div className="form-group row">
               <div className="col-lg-10 col-lg-offset-2 space-between align-items-center">
-                <div
-                  onClick={() => setState({ ...state, trackSocialMedia: !state.trackSocialMedia })}
-                >
-                  <input type="checkbox" checked={state.trackSocialMedia} />
-                  <span>Track social media sites</span>
+                <div>
+                  <input
+                    type="checkbox"
+                    checked={state.trackSocialMedia}
+                    onChange={toggleSocialMedia}
+                  />
+                  <span onClick={toggleSocialMedia}>Track social media sites</span>
                 </div>
                 <button
                   type="button"

From 673fd0d63f542a0914bd06d035a2426664a73043 Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 28 Apr 2023 13:37:38 -0500
Subject: [PATCH 5/7] Don't send a heartbeat on sites like 'chrome://newtab/'
 or 'about:newtab'

---
 src/config/config.test.ts | 4 ++++
 src/config/config.ts      | 3 +++
 src/core/WakaTimeCore.ts  | 7 +++++++
 3 files changed, 14 insertions(+)

diff --git a/src/config/config.test.ts b/src/config/config.test.ts
index 98d2c35..7987e4b 100644
--- a/src/config/config.test.ts
+++ b/src/config/config.test.ts
@@ -53,6 +53,10 @@ describe('wakatime config', () => {
         "loggingType": "domain",
         "logoutUserUrl": "https://wakatime.com/logout",
         "name": "WakaTime",
+        "nonTrackableSites": [
+          "chrome://",
+          "about:",
+        ],
         "socialMediaSites": [
           "facebook.com",
           "instagram.com",
diff --git a/src/config/config.ts b/src/config/config.ts
index 32bc0bf..1f24c91 100644
--- a/src/config/config.ts
+++ b/src/config/config.ts
@@ -87,6 +87,7 @@ export interface Config {
    * Extension name
    */
   name: string;
+  nonTrackableSites: string[];
   socialMediaSites: string[];
   states: ApiStates[];
   /**
@@ -159,6 +160,8 @@ const config: Config = {
 
   name: 'WakaTime',
 
+  nonTrackableSites: ['chrome://', 'about:'],
+
   socialMediaSites: [
     'facebook.com',
     'instagram.com',
diff --git a/src/core/WakaTimeCore.ts b/src/core/WakaTimeCore.ts
index 68a837a..b12a2dd 100644
--- a/src/core/WakaTimeCore.ts
+++ b/src/core/WakaTimeCore.ts
@@ -149,6 +149,13 @@ class WakaTimeCore {
         url = document.URL;
       }
 
+      for (const site of config.nonTrackableSites) {
+        if (url.startsWith(site)) {
+          // Don't send a heartbeat on sites like 'chrome://newtab/' or 'about:newtab'
+          return;
+        }
+      }
+
       const hostname = getDomainFromUrlWithoutProtocol(url).replace('www.', '');
       if (!items.trackSocialMedia) {
         if ((items.socialMediaSites as string[]).includes(hostname)) {

From a368c6651463d93fe01fac3579c233abfe5e075f Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 28 Apr 2023 13:52:47 -0500
Subject: [PATCH 6/7] bump v3.0.13

---
 src/manifests/chrome.json  | 2 +-
 src/manifests/firefox.json | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/manifests/chrome.json b/src/manifests/chrome.json
index 4b0583c..7b7a6ed 100644
--- a/src/manifests/chrome.json
+++ b/src/manifests/chrome.json
@@ -33,5 +33,5 @@
     "page": "options.html"
   },
   "permissions": ["alarms", "tabs", "storage", "idle"],
-  "version": "3.0.12"
+  "version": "3.0.13"
 }
diff --git a/src/manifests/firefox.json b/src/manifests/firefox.json
index 7c7f5df..7511ed7 100644
--- a/src/manifests/firefox.json
+++ b/src/manifests/firefox.json
@@ -46,5 +46,5 @@
     "storage",
     "idle"
   ],
-  "version": "3.0.12"
+  "version": "3.0.13"
 }

From 9fc39c77f66bd941a8e1e977966d84e22cee10ed Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Mon, 8 May 2023 16:37:19 -0500
Subject: [PATCH 7/7] check the domains by removing the subdomain

---
 src/core/WakaTimeCore.ts      |  4 ++--
 src/utils/getDomainFromUrl.ts | 10 +++++++---
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/core/WakaTimeCore.ts b/src/core/WakaTimeCore.ts
index b12a2dd..f5f256b 100644
--- a/src/core/WakaTimeCore.ts
+++ b/src/core/WakaTimeCore.ts
@@ -12,7 +12,7 @@ import { IS_FIREFOX, generateProjectFromDevSites } from '../utils';
 import { getApiKey } from '../utils/apiKey';
 import changeExtensionState from '../utils/changeExtensionState';
 import contains from '../utils/contains';
-import getDomainFromUrl, { getDomainFromUrlWithoutProtocol } from '../utils/getDomainFromUrl';
+import getDomainFromUrl, { getDomain } from '../utils/getDomainFromUrl';
 
 class WakaTimeCore {
   tabsWithDevtoolsOpen: Tabs.Tab[];
@@ -156,7 +156,7 @@ class WakaTimeCore {
         }
       }
 
-      const hostname = getDomainFromUrlWithoutProtocol(url).replace('www.', '');
+      const hostname = getDomain(url);
       if (!items.trackSocialMedia) {
         if ((items.socialMediaSites as string[]).includes(hostname)) {
           return changeExtensionState('blacklisted');
diff --git a/src/utils/getDomainFromUrl.ts b/src/utils/getDomainFromUrl.ts
index 8f19f4f..9c1f35a 100644
--- a/src/utils/getDomainFromUrl.ts
+++ b/src/utils/getDomainFromUrl.ts
@@ -10,8 +10,12 @@ export default function getDomainFromUrl(url: string): string {
 /**
  * Returns domain from given URL.
  */
-export function getDomainFromUrlWithoutProtocol(url: string): string {
-  const parts = url.split('/');
+export function getDomain(url: string): string {
+  const splittedUrl = url.replace(/(https?:\/\/)?(www.)?/i, '').split('.');
+  const domain = splittedUrl.slice(splittedUrl.length - 2).join('.');
+  if (domain.includes('/')) {
+    return domain.split('/')[0];
+  }
 
-  return parts[2];
+  return domain;
 }