Build 5.6.1

src/drivers/npm/package.json

@@ -2,7 +2,7 @@
   "name": "wappalyzer",
   "description": "Uncovers the technologies used on websites",
   "homepage": "https://github.com/AliasIO/Wappalyzer",
-  "version": "5.6.0",
+  "version": "5.6.1",
   "author": "Elbert Alias",
   "license": "GPL-3.0",
   "repository": {
@@ -12,6 +12,7 @@
   "main": "driver.js",
   "files": [
     "apps.json",
+    "browsers/zombie.js",
     "index.js",
     "driver.js",
     "wappalyzer.js"

src/drivers/webextension/js/driver.js

@@ -24,20 +24,22 @@ browser.tabs.onRemoved.addListener((tabId) => {
  * Get a value from localStorage
  */
 function getOption(name, defaultValue = null) {
-  return new Promise(async (resolve) => {
+  return new Promise(async (resolve, reject) => {
     let value = defaultValue;
 
     try {
       const option = await browser.storage.local.get(name);
 
-      if (option[name]) {
+      if (option[name] !== undefined) {
         value = option[name];
       }
     } catch (error) {
-      wappalyzer.log(error, 'driver', 'error');
+      wappalyzer.log(error.message, 'driver', 'error');
+
+      return reject(error.message);
     }
 
-    resolve(value);
+    return resolve(value);
   });
 }
 
@@ -45,14 +47,16 @@ function getOption(name, defaultValue = null) {
  * Set a value in localStorage
  */
 function setOption(name, value) {
-  return new Promise(async (resolve) => {
+  return new Promise(async (resolve, reject) => {
     try {
       await browser.storage.local.set({ [name]: value });
     } catch (error) {
-      wappalyzer.log(error, 'driver', 'error');
+      wappalyzer.log(error.message, 'driver', 'error');
+
+      return reject(error.message);
     }
 
-    resolve();
+    return resolve();
   });
 }
 
@@ -334,10 +338,10 @@ wappalyzer.driver.ping = async (hostnameCache = {}, adCache = []) => {
       url: `${wappalyzer.config.websiteURL}installed`,
     });
   } else if (version !== previousVersion && upgradeMessage) {
-    //openTab({
-    //  url: `${wappalyzer.config.websiteURL}upgraded?v${version}`,
-    //  background: true,
-    //});
+    openTab({
+     url: `${wappalyzer.config.websiteURL}upgraded?v${version}`,
+     background: true,
+    });
   }
 
   await setOption('version', version);

src/drivers/webextension/js/options.js

@@ -1,26 +1,51 @@
 /** global: browser */
 /** global: Wappalyzer */
-/* globals browser chrome Wappalyzer */
+/* globals browser Wappalyzer */
 /* eslint-env browser */
 
 const wappalyzer = new Wappalyzer();
 
-function getOption(name, defaultValue, callback) {
-  browser.storage.local.get(name)
-    .then((item) => {
-      callback(item.hasOwnProperty(name) ? item[name] : defaultValue);
+/**
+ * Get a value from localStorage
+ */
+function getOption(name, defaultValue = null) {
+  return new Promise(async (resolve, reject) => {
+    let value = defaultValue;
+
+    try {
+      const option = await browser.storage.local.get(name);
+
+      if (option[name] !== undefined) {
+        value = option[name];
+      }
+    } catch (error) {
+      wappalyzer.log(error.message, 'driver', 'error');
+
+      return reject(error.message);
+    }
+
+    return resolve(value);
   });
 }
 
+/**
+ * Set a value in localStorage
+ */
 function setOption(name, value) {
-  (chrome || browser).runtime.sendMessage({
-    id: 'set_option',
-    key: name,
-    value,
+  return new Promise(async (resolve, reject) => {
+    try {
+      await browser.storage.local.set({ [name]: value });
+    } catch (error) {
+      wappalyzer.log(error.message, 'driver', 'error');
+
+      return reject(error.message);
+    }
+
+    return resolve();
   });
 }
 
-document.addEventListener('DOMContentLoaded', () => {
+document.addEventListener('DOMContentLoaded', async () => {
   const nodes = document.querySelectorAll('[data-i18n]');
 
   Array.prototype.forEach.call(nodes, (node) => {
@@ -39,33 +64,33 @@ document.addEventListener('DOMContentLoaded', () => {
     window.open(wappalyzer.config.websiteURL);
   });
 
-  getOption('upgradeMessage', true, (value) => {
-    const el = document.querySelector('#option-upgrade-message');
+  let el;
+  let value;
+
+  // Upgrade message
+  value = await getOption('upgradeMessage', true);
+
+  el = document.querySelector('#option-upgrade-message');
 
   el.checked = value;
 
-    el.addEventListener('change', () => {
-      setOption('upgradeMessage', el.checked);
-    });
-  });
+  el.addEventListener('change', e => setOption('upgradeMessage', e.target.checked));
 
-  getOption('dynamicIcon', true, (value) => {
-    const el = document.querySelector('#option-dynamic-icon');
+  // Dynamic icon
+  value = await getOption('dynamicIcon', true);
+
+  el = document.querySelector('#option-dynamic-icon');
 
   el.checked = value;
 
-    el.addEventListener('change', () => {
-      setOption('dynamicIcon', el.checked);
-    });
-  });
+  el.addEventListener('change', e => setOption('dynamicIcon', e.target.checked));
+
+  // Tracking
+  value = await getOption('tracking', true);
 
-  getOption('tracking', true, (value) => {
-    const el = document.querySelector('#option-tracking');
+  el = document.querySelector('#option-tracking');
 
   el.checked = value;
 
-    el.addEventListener('change', () => {
-      setOption('tracking', el.checked);
-    });
-  });
+  el.addEventListener('change', e => setOption('tracking', e.target.checked));
 });

src/drivers/webextension/manifest.json

@@ -4,7 +4,7 @@
   "author": "Elbert Alias",
   "homepage_url": "https://www.wappalyzer.com",
   "description": "Identify web technologies",
-  "version": "5.6.0",
+  "version": "5.6.1",
   "default_locale": "en",
   "manifest_version": 2,
   "icons": {