Ported changes from Adblock Plus integration back into URL Fixer

lib/typoFixer.js

 /*
  * This file is part of the URL Fixer,
  * Copyright (C) 2006-2012 Eyeo GmbH
  *
  * URL Fixer is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * URL Fixer is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with URL Fixer.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 Cu.import("resource://gre/modules/Services.jsm");
 Cu.import("resource://gre/modules/XPCOMUtils.jsm");
 
 let {Prefs} = require("prefs");
 let {WindowObserver} = require("windowObserver");
-let {getSchemeCorrection, isKnownScheme, getDomainCorrection, getDomainReferral, onWhitelistEntryRemoved} = require("typoRules");
-let {processTypedDomain, processDomainCorrection,
-     processUserCorrection, processFalsePositive} = require("typedItCollector");
+let {getSchemeCorrection, isKnownScheme, getDomainCorrection, getDomainReferral, onWhitelistEntryAdded} = require("typoRules");
+let {processTypedDomain, processDomainCorrection, processFalsePositive} = require("typoCollector");
 let appIntegration = require("typoAppIntegration");
+let netError = require("typoNetError");
 
-// Attach our handlers to all browser windows
-new WindowObserver(
+let typoWindowObserver = null;
+
+exports.attachWindowObserver = attachWindowObserver;
+function attachWindowObserver()
 {
-  applyToWindow: function(window)
+  if (typoWindowObserver)
+    return;
+  
+  // Attach our handlers to all browser windows
+  typoWindowObserver = new WindowObserver(
   {
-    if (!appIntegration.isKnownWindow(window))
-      return;
+    applyToWindow: function(window)
+    {
+      if (!appIntegration.isKnownWindow(window))
+        return;
+      
+      netError.applyToWindow(window);
+      appIntegration.applyToWindow(window, correctURL);
+    },
 
-    let browser = appIntegration.getBrowser(window);
-    if (browser)
-      browser.addEventListener("DOMContentLoaded", handlePageLoad, false);
+    removeFromWindow: function(window)
+    {
+      if (!appIntegration.isKnownWindow(window))
+        return;
+      
+      netError.removeFromWindow(window);
+      appIntegration.removeFromWindow(window);
+    }
+  });
+}
+attachWindowObserver();
 
-    appIntegration.applyToWindow(window, correctURL);
-  },
-
-  removeFromWindow: function(window)
-  {
-    if (!appIntegration.isKnownWindow(window))
-      return;
-
-    let browser = appIntegration.getBrowser(window);
-    if (browser)
-    {
-      browser.removeEventListener("DOMContentLoaded", handlePageLoad, false);
-      if (browser.browsers)
-      {
-        for (let i = 0; i < browser.browsers.length; i++)
-        {
-          let contentWnd = browser.browsers[i].contentWindow;
-          if (contentWnd.document.documentURI.indexOf("about:neterror?") == 0)
-            removeFromNetErrorPage(contentWnd);
-        }
-      }
-    }
-
-    appIntegration.removeFromWindow(window);
-  }
-});
-
-// Load HTML code to add to network error pages
-let netErrorOverlay = null;
-let request = Cc["@mozilla.org/xmlextras/xmlhttprequest;1"].createInstance(Ci.nsIXMLHttpRequest);
-request.open("GET", "chrome://url-fixer/content/netError.xhtml");
-request.addEventListener("load", function(event)
+exports.detachWindowObserver = detachWindowObserver;
+function detachWindowObserver()
 {
-  netErrorOverlay = event.target.responseXML;
-}, false);
-request.send(null)
-
-function handlePageLoad(event)
-{
-  let document = event.target;
-  if (document.documentURI.indexOf("about:neterror?") != 0 ||
-      document.documentURI.indexOf("e=netOffline") > 0 ||
-      document.documentURI.indexOf("e=notCached") > 0)
-  {
+  if (!typoWindowObserver)
     return;
-  }
-
-  if (!netErrorOverlay || document.getElementById("url-fixer-section"))
-    return;
-
-  let container = document.getElementById("errorPageContainer");
-  if (!container)
-    return;
-
-  container.appendChild(netErrorOverlay.documentElement.cloneNode(true));
-
-  let textField = document.getElementById("url-fixer-intention");
-  textField.value = document.defaultView.location.href;
-
-  let retryButton = document.getElementById("url-fixer-retry");
-  retryButton.addEventListener("click", function()
-  {
-    let newURL = textField.value.replace(/^\s+/, "").replace(/\s+$/, "");
-    if (!newURL.length)
-      return;
-
-    let [prefix, newHost, suffix] = parseURL(newURL);
-    let oldHost = document.defaultView.location.hostname.toLowerCase();
-
-    processUserCorrection(oldHost, newHost);
-
-    if (newHost.indexOf("www.") == 0 && oldHost.indexOf("www.") == 0)
-    {
-      // Ignore www. prefix if they both start with it
-      newHost = newHost.substr(4);
-      oldHost = oldHost.substr(4);
-    }
-    if (oldHost && newHost != oldHost)
-    {
-      Prefs.custom_replace[oldHost] = newHost;
-      Prefs.custom_replace = JSON.parse(JSON.stringify(Prefs.custom_replace));
-    }
-    
-    // Remove from whitelist
-    if (oldHost in Prefs.whitelist)
-    {
-      delete Prefs.whitelist[oldHost];
-      onWhitelistEntryRemoved(oldHost);
-      Prefs.whitelist = JSON.parse(JSON.stringify(Prefs.whitelist));
-    }
-    
-    document.defaultView.location.replace(newURL);
-  }, false);
-}
-
-function removeFromNetErrorPage(window)
-{
-  let overlay = window.document.getElementById("url-fixer-section");
-  if (overlay)
-    overlay.parentNode.removeChild(overlay);
+  
+  // Detach our handlers from all browser windows
+  typoWindowObserver.shutdown();
+  typoWindowObserver = null;
 }
 
 function parseURL(url)
 {
   if (/^\s*((?:\w+:)?\/*(?:[^\/#]*@)?)([^\/:#]*)/.test(url))
     return [RegExp.$1, RegExp.$2.toLowerCase(), RegExp.rightContext];
   else
     return [url, null, null];
 }
 
@@ skipping 88 matching lines @@
           suffix += "?" + referral;
         }
 
         suffix += anchor;
       }
     }
   }
 
   if (!hasCorrection)
     return null;
+  
+  if (!appIntegration.isTypoCorrectionEnabled(window, prefix, domain, suffix))
+    return null;
 
   // Show infobar to inform and ask about correction
   let [message, yes, no] = getInfobarTexts();
   message = message.replace(/\?1\?/g, prefix+domain);
   let buttons = [
     {
       label:      yes,
       accessKey:  null,
       callback:   function()
       {
         // Yes: Do nothing
       }
     },
     {
       label:      no,
       accessKey:  null,
       callback:   function()
       {
         // No: Add to list of corrections (ignore)
-        let {onWhitelistEntryAdded} = require("typoRules");
         let entry = oldDomain.replace(/^www\./, "");
         Prefs.whitelist[entry] = true;
         onWhitelistEntryAdded(entry);
         Prefs.whitelist = JSON.parse(JSON.stringify(Prefs.whitelist));
 
         appIntegration.loadURI(window, value);
         processFalsePositive(domain, oldDomain);
       }
     }
   ];
   // We need to have persistence being set to 1 due to redirect which happens afterwards
-  appIntegration.openInfobar(window, "url-fixer-infobar-askafter", message, buttons, 1);
+  appIntegration.openInfobar(window, require("info").addonName + "-infobar-askafter", message, buttons, 1);
 
-  require("survey").incrementCorrectionsCounter();
+  require("typoSurvey").incrementCorrectionsCounter();
 
   return prefix + domain + suffix;
 }
 
 let stringBundle = null;
 
 function getInfobarTexts()
 {
   // Randomize URI to work around bug 719376
   if (!stringBundle)
-    stringBundle = Services.strings.createBundle("chrome://url-fixer/locale/typo.properties?" + Math.random());
+    stringBundle = Services.strings.createBundle("chrome://" + require("info").addonName + "/locale/typo.properties?" + Math.random());
   let result = [
     stringBundle.GetStringFromName("urlfixer.isItCorrect"),
     stringBundle.GetStringFromName("urlfixer.yes"),
     stringBundle.GetStringFromName("urlfixer.no")
   ];
 
   getInfobarTexts = function() result;
   return getInfobarTexts();
 }