Integrated URL Fixer into Adblock Plus

lib/typoAppIntegration.js

 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
  * You can obtain one at http://mozilla.org/MPL/2.0/. */
 
 let {hook} = require("hooks");
+let {application, addonName} = require("info");
+
 let functionHooks = new WeakMap();
 
 exports.removeFromWindow = function(window)
 {
   if (functionHooks.has(window))
   {
     let unhook = functionHooks.get(window);
     unhook();
     functionHooks.delete(window);
   }
 };
 
-let {application} = require("info");
+switch (addonName)
+{
+  case "url-fixer":
+  {
+    // URL Fixer
+    exports.isTypoCorrectionEnabled = function(window, prefix, domain, suffix) true;
+    
+    break;
+  }
+  case "adblockplus":
+  {
+    // Adblock Plus
+    let {Prefs} = require("prefs");
+    
+    // Do not ask to opt-in if user found setting
+    if (!Prefs.correctTyposAsked)
+    {
+      let onPrefChange = function(name)
+      {
+        if (name == "correctTypos")
+        {
+          Prefs.correctTyposAsked = true;
+          Prefs.removeListener(onPrefChange);
+        }
+      }
+      
+      Prefs.addListener(onPrefChange);
+    }
+    
+    exports.isTypoCorrectionEnabled = function(window, prefix, domain, suffix)
+    {
+      if (!Prefs.correctTyposAsked && !Prefs.correctTypos)
+      {
+        let {Utils} = require("utils");
+        let message = Utils.getString("typo_optin_message").replace(/\?1\?/, domain);
+        let yes = Utils.getString("typo_optin_yes");
+        let no = Utils.getString("typo_optin_no");
+        let buttons = [
+          {
+            label:      yes,
+            accessKey:  null,
+            callback:   function()
+            {
+              // Yes: Enable typo correction
+              Prefs.correctTypos = true;
+              exports.loadURI(window, prefix + domain + suffix);
+              Prefs.correctTyposAsked = true;
+            }
+          },
+          {
+            label:      no,
+            accessKey:  null,
+            callback:   function()
+            {
+              // No: Do nothing
+              Prefs.correctTyposAsked = true;
+            }
+          }
+        ];
+        // We need to have persistence being set to 1 due to redirect which happens afterwards
+        exports.openInfobar(window, "adblockplus-infobar-correct-typos-ask", message, buttons, 1);
+      }
+      
+      return Prefs.correctTypos;
+    };
+    
+    break;
+  }
+}
+
 switch (application)
 {
   case "firefox":
   {
     // Firefox
     exports.isKnownWindow = function(window) window.document.documentElement.getAttribute("windowtype") == "navigator:browser";
 
     exports.getURLBar = function(window) "gURLBar" in window ? window.gURLBar : null;
 
     exports.getBrowser = function(window) "gBrowser" in window ? window.gBrowser : null;
@@ skipping 17 matching lines @@
     exports.openInfobar = function(window, id, message, buttons, persistence)
     {
       let browser = exports.getBrowser(window);
       let infobar = browser.getNotificationBox();
       let notification = infobar.getNotificationWithValue(id);
 
       if (notification)
       {
         infobar.removeNotification(notification);
       }
-
       notification = infobar.appendNotification(
         message,
         id,
-        require("info").addonRoot + "icon64.png",
+        "chrome://" + addonName + "/skin/icon16.png",
         infobar.PRIORITY_INFO_HIGH,
         buttons
       );
       notification.persistence = persistence;
     };
 
     exports.loadURI = function(window, uri)
     {
       exports.getBrowser(window).loadURI(uri);
     };
@@ skipping 65 matching lines @@
       let notification = infobar.getNotificationWithValue(id);
 
       if (notification)
       {
         infobar.removeNotification(notification);
       }
 
       notification = infobar.appendNotification(
         message,
         id,
-        require("info").addonRoot + "icon64.png",
+        "chrome://" + addonName + "/skin/icon16.png",
         infobar.PRIORITY_INFO_HIGH,
         buttons
       );
       notification.persistence = persistence;
     };
 
     exports.loadURI = function(window, uri)
     {
       exports.getBrowser(window).loadURI(uri);
     };
@@ skipping 36 matching lines @@
         let notification = infobar.getNotificationWithValue(id);
 
         if (notification)
         {
           infobar.removeNotification(notification);
         }
 
         notification = infobar.appendNotification(
           message,
           id,
-          require("info").addonRoot + "icon64.png",
+          "chrome://" + addonName + "/skin/icon16.png",
           infobar.PRIORITY_INFO_HIGH,
           buttons
         );
         notification.persistence = persistence;
       }
     };
 
     exports.loadURI = function(window, uri)
     {
       if ("BrowserUI" in window && "goToURI" in window.BrowserUI)
@@ skipping 52 matching lines @@
         functionHooks.set(window, unhook);
       }
     };
 
     exports.openInfobar = function(window, id, message, buttons, persistence)
     {
       if ("BrowserApp" in window && "selectedTab" in window.BrowserApp)
       {
         window.NativeWindow.doorhanger.show(message, id, buttons, window.BrowserApp.selectedTab.id,
           {
-            // No navigation is happening after doorhanger is shown
-            // so persistence needs to be reduced by one
-            persistence: persistence - 1
+            persistence: persistence
           }
         );
       }
     };
 
     exports.loadURI = function(window, uri)
     {
       if ("BrowserApp" in window && "loadURI" in window.BrowserApp)
         window.BrowserApp.loadURI(uri);
     };
 
     break;
   }
   default:
   {
     exports.isKnownWindow = function(window) false;
     break;
   }
 }