Add counter of visit

Author: hai-altx-soft

scripts/activity.js

@@ -6,6 +6,10 @@ class Activity {
             if (tab.id && (tab.id != 0)) {
                 tabs = tabs || [];
                 var domain = this.extractHostname(tab.url);
+                var isDifferentUrl = false;
+                if (currentTab !== tab.url) {
+                    isDifferentUrl = true;
+                }
                 this.setCurrentActiveTab(domain);
                 if (this.isNewUrl(domain) && !this.isInBlackList(domain)) {
                     var favicon = tab.favIconUrl;
@@ -15,6 +19,12 @@ class Activity {
                     var newTab = new Tab(domain, favicon);
                     tabs.push(newTab);
                 }
+
+                if (isDifferentUrl) {
+                    var tabUrl = this.getTab(domain);
+                    if (tabUrl !== undefined)
+                        tabUrl.incCounter();
+                }
             }
         }
         else this.clearCurrentActiveTab();

scripts/background.js

@@ -60,8 +60,9 @@ function mainTRacker(activeUrl, tab, activeTab) {
     if (activity.isLimitExceeded(activeUrl, tab)) {
         setBlockPageToCurrent(activeUrl);
     }
-    if (!activity.isInBlackList(activeUrl))
+    if (!activity.isInBlackList(activeUrl)) {
         tab.incSummaryTime();
+    }
     if (setting_view_in_badge === true) {
         if (activity.isInBlackList(activeUrl)) {
             chrome.browserAction.setBadgeBackgroundColor({ color: '#FF0000' })
@@ -108,18 +109,18 @@ function checkDOM(state, activeUrl, tab, activeTab) {
     }
 }
 
-function checkPermissions(callback, activeUrl, tab, activeTab){
+function checkPermissions(callback, activeUrl, tab, activeTab) {
     chrome.permissions.contains({
         permissions: ['tabs'],
         origins: ["http://*/*", "https://*/*"]
-      }, function(result) {
+    }, function (result) {
         if (result) {
             chrome.tabs.executeScript({ code: "var videoElement = document.getElementsByTagName('video')[0]; (videoElement !== undefined && videoElement.currentTime > 0 && !videoElement.paused && !videoElement.ended && videoElement.readyState > 2);" }, (results) => {
                 if (results !== undefined && results[0] !== undefined && results[0] === true)
-                callback(activeUrl, tab, activeTab);
+                    callback(activeUrl, tab, activeTab);
             });
         }
-      });
+    });
 }
 
 function backgroundUpdateStorage() {
@@ -168,7 +169,7 @@ function loadTabs() {
     storage.loadTabs(STORAGE_TABS, function (items) {
         tabs = [];
         for (var i = 0; i < items.length; i++) {
-            tabs.push(new Tab(items[i].url, items[i].favicon, items[i].days, items[i].summaryTime));
+            tabs.push(new Tab(items[i].url, items[i].favicon, items[i].days, items[i].summaryTime, items[i].counter));
         }
     });
 }

scripts/tab.js

@@ -1,13 +1,17 @@
 'use strict';
 
 class Tab {
-    constructor(url, favicon, days, summary) {
+    constructor(url, favicon, days, summary, counter) {
         this.url = url;
         this.favicon = favicon;
         if (summary !== undefined)
             this.summaryTime = summary;
         else
             this.summaryTime = 0;
+        if (counter !== undefined)
+            this.counter = counter;
+        else
+            this.counter = 0;
         if (days !== undefined)
             this.days = days;
         else
@@ -27,11 +31,25 @@ class Tab {
         }
     }
 
+    incCounter(){
+        this.counter +=1;
+
+        var today = new Date().toLocaleDateString();
+        var day = this.days.find(x => x.date == today);
+        if (day === undefined) {
+            this.addNewDay(today);
+        }
+        else {
+            day['counter'] += 1;
+        }
+    }
+
     addNewDay(today) {
         this.days.push(
             {
                 'date': today,
-                'summary': 1
+                'summary': 1,
+                'counter': 1
             }
         );
     }