Issue 2351 - Add a custom menu item for whitelisting the current site

adblockplus/Api.jsm

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2015 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 15 matching lines @@
 Cu.import("resource://gre/modules/Messaging.jsm");
 
 function require(module)
 {
   let result = {};
   result.wrappedJSObject = result;
   Services.obs.notifyObservers(result, "adblockplus-require", module);
   return result.exports;
 }
 
+let {Filter} = require("filterClasses");
 let {FilterStorage} = require("filterStorage");
+let {defaultMatcher} = require("matcher");
 let {Prefs} = require("prefs");
 let {UI} = require("ui");
 
 var AdblockPlusApi =
 {
   get filtersLoaded()
   {
     return !FilterStorage._loading;
   },
   get acceptableAdsEnabled()
@@ skipping 26 matching lines @@
         case "setAcceptableAdsEnabled":
           if ("enable" in data)
           {
             this.acceptableAdsEnabled = !!data["enable"];
             return {"success" : true};
           }
           return {"success": false, "error": "malformed request"};
       }
       return {"success": false, "error": "malformed request"};
     }).bind(this), "AdblockPlus:Api");
+  },
+  isLocal: function(location)
+  {
+    return !location.schemeIs("http") && !location.schemeIs("https");
+  },
+  isPageWhitelisted: function(location)
+  {
+    return defaultMatcher.whitelist.matchesAny(
+      location.spec, "DOCUMENT", location.host, false, null);
+  },
+  whitelistSite: function(location, whitelisted)
+  {
+    if (whitelisted)
+    {
+      var host = location.host.replace(/^www\./, "");
+      var filter = Filter.fromText("@@||" + host + "^$document");
+      if (filter.subscriptions.length && filter.disabled)
+        filter.disabled = false;
+      else
+      {
+        filter.disabled = false;
+        FilterStorage.addFilter(filter);
+      }
+    }
+    else
+    {
+      // Remove any exception rules applying to this URL
+      var filter = AdblockPlusApi.isPageWhitelisted(location);
+      while (filter)
+      {
+        FilterStorage.removeFilter(filter);
+        if (filter.subscriptions.length)
+          filter.disabled = true;
+        filter = AdblockPlusApi.isPageWhitelisted(location);
+      }
+    }
   }
 };