added hook function to appIntegration to handle function-overwrites from other extensions

lib/appIntegration.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/. */
 
-// Extensions like Tab Mix Plus will try to recompile our handlers and fail
-// badly - they manage to replace our handler but the recompiled version
-// won't work because the closure is missing its variables. We replace
-// toString() function to prevent this kind of recompiling on earlier stages
-// (produce a syntax error but still leave the source code viewable).
-function doNotRecompile()
-{
-  let result = Function.prototype.toString.apply(this);
-  return result + "\n$%&!/DO_NOT_RECOMPILE"
-}
-
+let functionHooks = new WeakMap();
+let {hook} = require("hooks");
 let {application} = require("info");
 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;
 
     exports.applyToWindow = function(window, corrector)
     {
       let urlbar = exports.getURLBar(window);
-      if (urlbar && urlbar.handleCommand && !("urlfixerOldHandler" in urlbar.handleCommand))
+      if (urlbar && urlbar.handleCommand && !functionHooks.has(window))
       {
         // Handle new URLs being entered
-        let oldHandler = urlbar.handleCommand;
-        urlbar.handleCommand = function()
-        {
-          try
-          {
-            let correction = corrector(window, urlbar.value);
-            if (correction)
-              urlbar.value = correction;
-          }
-          catch(e)
-          {
-            if (e == Cr.NS_BINDING_ABORTED)
-              return;
-            else
-              Cu.reportError(e);
-          }
-          oldHandler.apply(this, arguments);
-        }
-        urlbar.handleCommand.urlfixerOldHandler = oldHandler;
-        urlbar.handleCommand.toString = doNotRecompile;
+        let unhook = hook(urlbar, "handleCommand", function() {
+          let correction = corrector(window, urlbar.value);
+          if (correction)
+            urlbar.value = correction;
+        });
+        functionHooks.set(window, unhook);
       }
     };
 
     exports.removeFromWindow = function(window)
     {
-      let urlbar = exports.getURLBar(window);
-      if (urlbar && urlbar.handleCommand && "urlfixerOldHandler" in urlbar.handleCommand)
-        urlbar.handleCommand = urlbar.handleCommand.urlfixerOldHandler;
+      if (functionHooks.has(window))
+      {
+        let unhook = functionHooks.get(window);
+        unhook();
+        functionHooks.delete(window);
+      }
     };
-    
+
     exports.openInfobar = function(window, id, message, buttons, persistence)
     {
       let browser = exports.getBrowser(window);
       let infobar = browser.getNotificationBox();
       let notif = infobar.getNotificationWithValue(id);
-      
+
       if (notif)
       {
         infobar.removeNotification(notif);
       }
-      
+
       notif = infobar.appendNotification(
         message,
         id,
         require("info").addonRoot + "icon64.png",
         infobar.PRIORITY_INFO_HIGH,
         buttons
       );
       notif.persistence = persistence;
     };
-    
+
     exports.loadURI = function(uri)
     {
       exports.getBrowser(Services.wm.getMostRecentWindow("navigator:browser")).loadURI(uri);
     };
 
     break;
   }
   case "seamonkey":
   {
+    let eventListeners = new WeakMap();
+    
     // SeaMonkey
     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;
 
     exports.applyToWindow = function(window, corrector)
     {
       let urlbar = exports.getURLBar(window);
-      if (urlbar && window.handleURLBarCommand && !("urlfixerOldHandler" in window.handleURLBarCommand))
+      let goButton = window.document.getElementById("go-button-container");
+      
+      if (urlbar && urlbar._fireEvent && !functionHooks.has(window))
       {
-        // Handle new URLs being entered
-        let oldHandler = window.handleURLBarCommand;
-        window.handleURLBarCommand = function()
+        function correctURL()
         {
-          try
+          let correction = corrector(window, urlbar.value);
+          if (correction)
+            urlbar.value = correction;
+        }
+        
+        let unhook = hook(urlbar, "_fireEvent", function(eventType) {

[Wladimir Palant, 2012/09/27 14:36:01]

This bracket should be on next line.

+          if (eventType == "textentered")
           {
-            let correction = corrector(window, urlbar.value);
-            if (correction)
-              urlbar.value = correction;
+            correctURL();
           }
-          catch(e)
-          {
-            if (e == Cr.NS_BINDING_ABORTED)
-              return;
-            else
-              Cu.reportError(e);
-          }
-          oldHandler.apply(this, arguments);
+        });
+        functionHooks.set(window, unhook);
+        
+        if (goButton)
+        {
+          goButton.addEventListener("command", correctURL, true);
+          eventListeners.set(window, {
+            "listener": correctURL,
+            "element": goButton
+          });
         }
-        window.handleURLBarCommand.urlfixerOldHandler = oldHandler;
-        window.handleURLBarCommand.toString = doNotRecompile;
       }
     };
 
     exports.removeFromWindow = function(window)
     {
-      if (window.handleURLBarCommand && "urlfixerOldHandler" in window.handleURLBarCommand)
-        window.handleURLBarCommand = window.handleURLBarCommand.urlfixerOldHandler;
+      if (functionHooks.has(window))
+      {
+        let unhook = functionHooks.get(window);
+        unhook();
+        functionHooks.delete(window);
+      }
+      
+      if (eventListeners.has(window))
+      {
+        let eventListener = eventListeners.get(window);
+        eventListener.element.removeEventListener("command", eventListener.listener, true);

[Wladimir Palant, 2012/09/27 14:36:01]

eventListeners.delete(window)?

+      }
     };
-    
+
     exports.openInfobar = function(window, id, message, buttons, persistence)
     {
       let browser = exports.getBrowser(window);
       let infobar = browser.getNotificationBox();
       let notif = infobar.getNotificationWithValue(id);
-      
+
       if (notif)
       {
         infobar.removeNotification(notif);
       }
-      
+
       notif = infobar.appendNotification(
         message,
         id,
         require("info").addonRoot + "icon64.png",
         infobar.PRIORITY_INFO_HIGH,
         buttons
       );
       notif.persistence = persistence;
     };
-    
+
     exports.loadURI = function(uri)
     {
       exports.getBrowser(Services.wm.getMostRecentWindow("navigator:browser")).loadURI(uri);
     };
 
     break;
   }
   case "fennec":
   {
     // XUL Fennec
@@ skipping 13 matching lines @@
         {
           url = url || this._edit.value;
           try
           {
             let correction = corrector(window, url);
             if (correction)
               url = correction;
           }
           catch(e)
           {
-            if (e == Cr.NS_BINDING_ABORTED)
-              return;
-            else
-              Cu.reportError(e);
+            Cu.reportError(e);
           }
           oldHandler.call(this, url);
         }
         window.BrowserUI.goToURI.urlfixerOldHandler = oldHandler;
         window.BrowserUI.goToURI.toString = doNotRecompile;
       }
     };
 
     exports.removeFromWindow = function(window)
     {
       if ("BrowserUI" in window && window.BrowserUI.goToURI && "urlfixerOldHandler" in window.BrowserUI.goToURI)
         window.BrowserUI.goToURI = window.BrowserUI.goToURI.urlfixerOldHandler;
     };
-    
+
     exports.openInfobar = function()
     {
       // TODO: Implement infobar
     };
-    
+
     exports.loadURI = function(window, uri)
     {
       // TODO: Implement infobar
     };
 
     break;
   }
   case "fennec2":
   {
     // Native Fennec
@@ skipping 13 matching lines @@
         {
           let params = Array.prototype.slice.apply(arguments);
           try
           {
             let correction = corrector(window, params[0]);
             if (correction)
               params[0] = correction;
           }
           catch(e)
           {
-            if (e == Cr.NS_BINDING_ABORTED)
-              return null;
-            else
-              Cu.reportError(e);
+            Cu.reportError(e);
           }
           return oldFunc.apply(this, params);
         };
 
         window.BrowserApp.observe = function(subject, topic, data)
         {
           // Huge hack: we replace addTab/loadURI when the observer is
           // triggered. This seems to be the only way to know that the calls
           // originate from user input.
           let method = null;
@@ skipping 21 matching lines @@
         window.BrowserApp.observe.urlfixerOldHandler = oldHandler;
         window.BrowserApp.observe.toString = doNotRecompile;
       }
     };
 
     exports.removeFromWindow = function(window)
     {
       if ("BrowserApp" in window && window.BrowserApp.observe && "urlfixerOldHandler" in window.BrowserApp.observe)
         window.BrowserApp.observe = window.BrowserApp.observe.urlfixerOldHandler;
     };
-    
+
     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,
           {
             persistence: persistence
           }
         );
       }
     };
-    
+
     exports.loadURI = function(uri)
     {
       let window = Services.wm.getMostRecentWindow("navigator:browser");
       if ("BrowserApp" in window && "loadURI" in window.BrowserApp)
         window.BrowserApp.loadURI(uri);
     };
 
     break;
   }
   default:
   {
     exports.isKnownWindow = function(window) false;
     break;
   }
 }