Issue 521- Inject our stylesheet on per-site basis rather than globally

lib/elemHide.js

 /*
  * This file is part of Adblock Plus <http://adblockplus.org/>,
  * Copyright (C) 2006-2014 Eyeo GmbH
  *
  * Adblock Plus 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.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ skipping 10 matching lines @@
 
 Cu.import("resource://gre/modules/Services.jsm");
 
 let {Utils} = require("utils");
 let {IO} = require("io");
 let {Prefs} = require("prefs");
 let {ElemHideException} = require("filterClasses");
 let {FilterNotifier} = require("filterNotifier");
 let {AboutHandler} = require("elemHideHitRegistration");
 let {TimeLine} = require("timeline");
+let {Policy} = require("contentPolicy");
 
 /**
  * Lookup table, filters by their associated key
  * @type Object
  */
 let filterByKey = Object.create(null);
 
 /**
  * Lookup table, keys of the filters by filter text
  * @type Object
@@ skipping 12 matching lines @@
  */
 let exceptions = Object.create(null);
 
 /**
  * Currently applied stylesheet URL
  * @type nsIURI
  */
 let styleURL = null;
 
 /**
+ * Global stylesheet that should be loaded into content windows.
+ * @type nsIStyleSheet
+ */
+let styleSheet = null;
+
+let useNew = ('preloadSheet' in Utils.styleService);
+
+/**
  * Element hiding component
  * @class
  */
 let ElemHide = exports.ElemHide =
 {
+  QueryInterface: XPCOMUtils.generateQI([Ci.nsIObserver, Ci.nsISupportsWeakReference]),
+
   /**
    * Indicates whether filters have been added or removed since the last apply() call.
    * @type Boolean
    */
   isDirty: false,
 
   /**
    * Inidicates whether the element hiding stylesheet is currently applied.
    * @type Boolean
    */
   applied: false,
 
   /**
    * Called on module startup.
    */
   init: function()
   {
     TimeLine.enter("Entered ElemHide.init()");
     Prefs.addListener(function(name)
     {
       if (name == "enabled")
         ElemHide.apply();
     });
-    onShutdown.add(function()
+
+    if (useNew)
+      Services.obs.addObserver(this, "content-document-global-created", true);
+
+    onShutdown.add(() =>
     {
       ElemHide.unapply();
+      if (useNew)
+        Services.obs.removeObserver(this, "content-document-global-created");
     });
 
     TimeLine.log("done adding prefs listener");
 
     let styleFile = IO.resolveFilePath(Prefs.data_directory);
     styleFile.append("elemhide.css");
     styleURL = Services.io.newFileURI(styleFile).QueryInterface(Ci.nsIFileURL);
     TimeLine.log("done determining stylesheet URL");
 
     TimeLine.leave("ElemHide.init() done");
   },
 
+  observe: function (subject, topic, data, additional)
+  {
+    if (topic != "content-document-global-created")
+      return;
+
+    if (!Prefs.enabled)
+      return;
+
+    let process = Policy.processWindow(subject);
+    dump(subject.document.documentURIObject.spec + " is okay " +  process + "\n");
+    if (process)
+      return;
+
+    // XXX at this point the window might still be whitelisted
+    try
+    {
+      let utils = subject.QueryInterface(Ci.nsIInterfaceRequestor)
+                         .getInterface(Ci.nsIDOMWindowUtils);
+      utils.addSheet(styleSheet, Ci.nsIStyleSheetService.USER_SHEET);
+    }
+    catch (e)
+    {
+      Cu.reportError(e);
+    }
+   },
+
   /**
    * Removes all known filters
    */
   clear: function()
   {
     filterByKey = Object.create(null);
     keyByFilter = Object.create(null);
     knownExceptions = Object.create(null);
     exceptions = Object.create(null);
     ElemHide.isDirty = false;
@@ skipping 106 matching lines @@
 
     TimeLine.enter("Entered ElemHide.apply()");
 
     if (!ElemHide.isDirty || !Prefs.enabled)
     {
       // Nothing changed, looks like we merely got enabled/disabled
       if (Prefs.enabled && !ElemHide.applied)
       {
         try
         {
-          Utils.styleService.loadAndRegisterSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
+          if (!useNew)
+            Utils.styleService.loadAndRegisterSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
           ElemHide.applied = true;
         }
         catch (e)
         {
           Cu.reportError(e);
         }
         TimeLine.log("Applying existing stylesheet finished");
       }
       else if (!Prefs.enabled && ElemHide.applied)
       {
@@ skipping 31 matching lines @@
       {
         ElemHide.isDirty = false;
 
         ElemHide.unapply();
         TimeLine.log("ElemHide.unapply() finished");
 
         if (!noFilters)
         {
           try
           {
-            Utils.styleService.loadAndRegisterSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
+            if (!useNew) {
+              Utils.styleService.loadAndRegisterSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
+            } else {
+              styleSheet = Utils.styleService.preloadSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
+            }
             ElemHide.applied = true;
           }
           catch (e)
           {
             Cu.reportError(e);
           }
           TimeLine.log("Applying stylesheet finished");
         }
 
         FilterNotifier.triggerListeners("elemhideupdate");
@@ skipping 63 matching lines @@
 
   /**
    * Unapplies current stylesheet URL
    */
   unapply: function()
   {
     if (ElemHide.applied)
     {
       try
       {
-        Utils.styleService.unregisterSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
+        if (!useNew)
+          Utils.styleService.unregisterSheet(styleURL, Ci.nsIStyleSheetService.USER_SHEET);
       }
       catch (e)
       {
         Cu.reportError(e);
       }
       ElemHide.applied = false;
     }
   },
 
   /**
-   * Retrieves the currently applied stylesheet URL
-   * @type String
-   */
-  get styleURL()
-  {
-    return ElemHide.applied ? styleURL.spec : null;
-  },
-
-  /**
    * Retrieves an element hiding filter by the corresponding protocol key
    */
   getFilterByKey: function(/**String*/ key) /**Filter*/
   {
     return (key in filterByKey ? filterByKey[key] : null);
   },
 
   /**
    * Returns a list of all selectors active on a particular domain (currently
    * used only in Chrome, Opera and Safari).
@@ skipping 12 matching lines @@
 
       if (specificOnly && (!domains || domains[""]))
         continue;
 
       if (filter.isActiveOnDomain(domain) && !this.getException(filter, domain))
         result.push(filter.selector);
     }
     return result;
   }
 };