Version 1.9.4

core_js/popup.js

@@ -32,65 +32,6 @@ var loggingStatus;
 var statisticsStatus;
 var currentURL;
 
-async function getData()
-{
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["globalCounter"]
-    }).then((data) => {
-        globalCounter = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["globalurlcounter"]
-    }).then((data) => {
-        globalurlcounter = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["globalStatus"]
-    }).then((data) => {
-        globalStatus = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["badgedStatus"]
-    }).then((data) => {
-        badgedStatus = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["hashStatus"]
-    }).then((data) => {
-        hashStatus = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["loggingStatus"]
-    }).then((data) => {
-        loggingStatus = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getData",
-        params: ["statisticsStatus"]
-    }).then((data) => {
-        statisticsStatus = data.response;
-    });
-
-    await browser.runtime.sendMessage({
-        function: "getCurrentURL",
-        params: []
-    }).then((data) => {
-        currentURL = data.response;
-    });
-}
-
 /**
 * Initialize the UI.
 *
@@ -240,19 +181,26 @@ function resetGlobalCounter(){
 }
 
 $(document).ready(function(){
-    getData().then(() => {
-        init();
-        $('#reset_counter_btn').on("click", resetGlobalCounter);
-        changeSwitchButton("globalStatus", "globalStatus");
-        changeSwitchButton("tabcounter", "badgedStatus");
-        changeSwitchButton("logging", "loggingStatus");
-        changeSwitchButton("statistics", "statisticsStatus");
-        $('#loggingPage').attr('href', browser.extension.getURL('./html/log.html'));
-        $('#settings').attr('href', browser.extension.getURL('./html/settings.html'));
-        $('#cleaning_tools').attr('href', browser.extension.getURL('./html/cleaningTool.html'));
-        setText();
-    });
-
+    loadData("globalCounter")
+        .then(() => loadData("globalurlcounter"))
+        .then(() => loadData("globalStatus"))
+        .then(() => loadData("badgedStatus"))
+        .then(() => loadData("hashStatus"))
+        .then(() => loadData("loggingStatus"))
+        .then(() => loadData("statisticsStatus"))
+        .then(() => loadData("getCurrentURL", "currentURL"))
+        .then(() => {
+            init();
+            $('#reset_counter_btn').on("click", resetGlobalCounter);
+            changeSwitchButton("globalStatus", "globalStatus");
+            changeSwitchButton("tabcounter", "badgedStatus");
+            changeSwitchButton("logging", "loggingStatus");
+            changeSwitchButton("statistics", "statisticsStatus");
+            $('#loggingPage').attr('href', browser.extension.getURL('./html/log.html'));
+            $('#settings').attr('href', browser.extension.getURL('./html/settings.html'));
+            $('#cleaning_tools').attr('href', browser.extension.getURL('./html/cleaningTool.html'));
+            setText();
+        });
 });
 
 /**
@@ -299,6 +247,25 @@ function injectText(id, attribute, tooltip = "")
     }
 }
 
+/**
+ * Loads data from storage and saves into local variable.
+ *
+ * @param name data name
+ * @param varName variable name
+ * @returns {Promise<data>} requested data
+ */
+async function loadData(name, varName=name) {
+    return new Promise((resolve, reject) => {
+        browser.runtime.sendMessage({
+            function: "getData",
+            params: [name]
+        }).then(data => {
+            this[varName] = data.response;
+            resolve(data);
+        }, handleError);
+    });
+}
+
 /**
 * Translate a string with the i18n API.
 *

core_js/settings.js

@@ -78,45 +78,23 @@ function reset()
 */
 function save()
 {
-    browser.runtime.sendMessage({
-        function: "setData",
-        params: ["badged_color", $('input[name=badged_color]').val()]
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "setBadgedStatus",
-        params: []
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "setData",
-        params: ["ruleURL", $('input[name=rule_url]').val()]
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "setData",
-        params: ["hashURL", $('input[name=hash_url]').val()]
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "setData",
-        params: ["types", $('input[name=types]').val()]
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "setData",
-        params: ["logLimit", $('input[name=logLimit]').val()]
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "saveOnExit",
-        params: []
-    }).then(handleResponse, handleError);
-
-    browser.runtime.sendMessage({
-        function: "reload",
-        params: []
-    }).then(handleResponse, handleError);
+    saveData("badged_color", $('input[name=badged_color]').val())
+        .then(() => saveData("ruleURL", $('input[name=ruleURL]').val()))
+        .then(() => saveData("hashURL", $('input[name=hashURL]').val()))
+        .then(() => saveData("types", $('input[name=types]').val()))
+        .then(() => saveData("logLimit", $('input[name=logLimit]').val()))
+        .then(() => browser.runtime.sendMessage({
+            function: "setBadgedStatus",
+            params: []
+        }), handleError)
+        .then(() => browser.runtime.sendMessage({
+            function: "saveOnExit",
+            params: []
+        }), handleError)
+        .then(() => browser.runtime.sendMessage({
+            function: "reload",
+            params: []
+        }), handleError);
 }
 
 /**
@@ -135,66 +113,69 @@ function translate(string, ...placeholders)
 */
 function getData()
 {
-    browser.runtime.sendMessage({
-        function: "getData",
-        params: ["badged_color"]
-    }).then((data) => handleResponseData(data, "badged_color", "badged_color"), handleError);
+    loadData("badged_color")
+        .then(() => loadData("ruleURL"))
+        .then(() => loadData("hashURL"))
+        .then(() => loadData("types"))
+        .then(() => loadData("logLimit"))
+        .then(logData => {
+            if(logData.response === undefined || logData.response === -1) {
+                $('#logLimit_label').text(translate('setting_log_limit_label', "∞"));
+            } else {
+                $('#logLimit_label').text(translate('setting_log_limit_label', logData.response));
+            }
+        });
 
-    browser.runtime.sendMessage({
-        function: "getData",
-        params: ["ruleURL"]
-    }).then((data) => handleResponseData(data, "rule_url", "rule_url"), handleError);
+    loadData("contextMenuEnabled")
+        .then(() => loadData("historyListenerEnabled"))
+        .then(() => loadData("localHostsSkipping"))
+        .then(() => loadData("referralMarketing"))
+        .then(() => loadData("domainBlocking"))
+        .then(() => {
+            changeSwitchButton("localHostsSkipping", "localHostsSkipping");
+            changeSwitchButton("historyListenerEnabled", "historyListenerEnabled");
+            changeSwitchButton("contextMenuEnabled", "contextMenuEnabled");
+            changeSwitchButton("referralMarketing", "referralMarketing");
+            changeSwitchButton("domainBlocking", "domainBlocking");
+        });
+}
 
-    browser.runtime.sendMessage({
-        function: "getData",
-        params: ["hashURL"]
-    }).then((data) => handleResponseData(data, "hash_url", "hash_url"), handleError);
-
-    browser.runtime.sendMessage({
-        function: "getData",
-        params: ["types"]
-    }).then((data) => handleResponseData(data, "types", "types"), handleError);
-
-    browser.runtime.sendMessage({
-        function: "getData",
-        params: ["logLimit"]
-    }).then((data) => {
-        handleResponseData(data, "logLimit", "logLimit");
-        if(data.response === undefined || data.response === -1) {
-            $('#logLimit_label').text(translate('setting_log_limit_label', "∞"));
-        } else {
-            $('#logLimit_label').text(translate('setting_log_limit_label', data.response));
-        }
-    }, handleError);
-
-    browser.runtime.sendMessage({
-        function: "getData",
-        params: ["contextMenuEnabled"]
-    }).then((data) => {
-        handleResponseData(data, "contextMenuEnabled", "contextMenuEnabled");
+/**
+ * Loads data from storage and saves into local variable.
+ *
+ * @param name data/variable name
+ * @returns {Promise<data>} requested data
+ */
+async function loadData(name) {
+    return new Promise((resolve, reject) => {
         browser.runtime.sendMessage({
             function: "getData",
-            params: ["historyListenerEnabled"]
-        }).then((data) => {
-            handleResponseData(data, "historyListenerEnabled", "historyListenerEnabled");
-            browser.runtime.sendMessage({
-                function: "getData",
-                params: ["localHostsSkipping"]
-            }).then((data) => {
-                handleResponseData(data, "localHostsSkipping", "localHostsSkipping");
-                browser.runtime.sendMessage({
-                    function: "getData",
-                    params: ["referralMarketing"]
-                }).then((data) => {
-                    handleResponseData(data, "referralMarketing", "referralMarketing");
-                    changeSwitchButton("contextMenuEnabled", "contextMenuEnabled");
-                    changeSwitchButton("historyListenerEnabled", "historyListenerEnabled");
-                    changeSwitchButton("localHostsSkipping", "localHostsSkipping");
-                    changeSwitchButton("referralMarketing", "referralMarketing");
-                }, handleError);
-            }, handleError);
+            params: [name]
+        }).then(data => {
+            settings[name] = data.response;
+            $('input[name='+name+']').val(data.response);
+            resolve(data);
         }, handleError);
-    }, handleError);
+    });
+}
+
+/**
+ * Saves data to storage.
+ *
+ * @param key key of the data that should be saved
+ * @param data data that should be saved
+ * @returns {Promise<message>} message from background script
+ */
+async function saveData(key, data) {
+    return new Promise((resolve, reject) => {
+        browser.runtime.sendMessage({
+            function: "setData",
+            params: [key, data]
+        }).then(message => {
+            handleResponse(message);
+            resolve(message);
+        }, handleError);
+    });
 }
 
 /**
@@ -220,6 +201,7 @@ function setText()
     $('#import_settings_btn_text').text(translate('setting_html_import_button'));
     $('#import_settings_btn').prop('title', translate('setting_html_import_button_title'));
     injectText("referral_marketing_enabled", "referral_marketing_enabled");
+    injectText("domain_blocking_enabled", "domain_blocking_enabled");
 }
 
 /**
@@ -267,18 +249,6 @@ function importSettings(evt) {
     fileReader.readAsText(file);
 }
 
-/**
- * Handle the response from the storage and saves the data.
- * @param  {JSON-Object} data Data JSON-Object
- * @param varName
- * @param inputID
- */
-function handleResponseData(data, varName, inputID)
-{
-    settings[varName] = data.response;
-    $('input[name='+inputID+']').val(data.response);
-}
-
 function handleResponse(message) {
     console.log(`Message from the background script:  ${message.response}`);
 }

core_js/storage.js

@@ -106,6 +106,9 @@ function genesis() {
         // Start the clearurls.js
         start();
 
+        //Set correct icon on startup
+        changeIcon();
+
         // Start the context_menu
         contextMenuStart();
 
@@ -205,6 +208,7 @@ function initSettings() {
     storage.localHostsSkipping = true;
     storage.referralMarketing = false;
     storage.logLimit = -1;
+    storage.domainBlocking = true;
 
     if (getBrowser() === "Firefox") {
         storage.types = ["font", "image", "imageset", "main_frame", "media", "object", "object_subrequest", "other", "script", "stylesheet", "sub_frame", "websocket", "xbl", "xml_dtd", "xmlhttprequest", "xslt"];