Issue 5593 - Use messaging in popup for prefs, whitelisting, and stats

lib/whitelisting.js

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-present 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
  * 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 Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 /** @module whitelisting */
 
 "use strict";
 
 const {defaultMatcher} = require("matcher");
-const {RegExpFilter} = require("filterClasses");
+const {Filter, RegExpFilter} = require("filterClasses");
 const {FilterNotifier} = require("filterNotifier");
+const {FilterStorage} = require("filterStorage");
 const {stringifyURL, getDecodedHostname,
        extractHostFromFrame, isThirdParty} = require("url");
 const {port} = require("messaging");
 const devtools = require("devtools");
 const {verifySignature} = require("rsa");
 
 let sitekeys = new ext.PageMap();
 
 function match(page, url, typeMask, docDomain, sitekey)
 {
@@ skipping 45 matching lines @@
       filter = match(page, frame.url, typeMask, docDomain, sitekey);
       frame = parent;
     }
 
     return filter;
   }
 
   return match(page, page.url, typeMask);
 };
 
+port.on("filters.isWhitelisted", message =>
+{
+  return !!checkWhitelisted(new ext.Page(message.tab));
+});
+
+port.on("filters.whitelist", message =>
+{
+  let page = new ext.Page(message.tab);
+  let host = getDecodedHostname(page.url).replace(/^www\./, "");
+  let filter = Filter.fromText("@@||" + host + "^$document");
+  if (filter.subscriptions.length && filter.disabled)
+  {
+    filter.disabled = false;
+  }
+  else
+  {
+    filter.disabled = false;
+    FilterStorage.addFilter(filter);
+  }
+});
+
+port.on("filters.unwhitelist", message =>
+{
+  let page = new ext.Page(message.tab);
+  // Remove any exception rules applying to this URL
+  let filter = checkWhitelisted(page);
+  while (filter)
+  {
+    FilterStorage.removeFilter(filter);
+    if (filter.subscriptions.length)
+      filter.disabled = true;
+    filter = checkWhitelisted(page);
+  }
+});
+
 function revalidateWhitelistingState(page)
 {
   FilterNotifier.emit(
     "page.WhitelistingStateRevalidate",
     page, checkWhitelisted(page)
   );
 }
 
 FilterNotifier.on("filter.behaviorChanged", () =>
 {
@@ skipping 89 matching lines @@
 {
   chrome.webRequest.onHeadersReceived.addListener(
     onHeadersReceived,
     {
       urls: ["http://*/*", "https://*/*"],
       types: ["main_frame", "sub_frame"]
     },
     ["responseHeaders"]
   );
 }