Issue 6826 - Use contentScripts API for snippets if available

lib/contentFiltering.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
@@ skipping 24 matching lines @@
 // See https://crbug.com/608854
 const styleSheetRemovalSupported = info.platform == "gecko";
 
 const selectorGroupSize = 1024;
 
 let userStyleSheetsSupported = true;
 
 let snippetsLibrarySource = "";
 let executableCode = new Map();
 
+let registeredContentScripts = new Map();
+
 function* splitSelectors(selectors)
 {
   // Chromium's Blink engine supports only up to 8,192 simple selectors, and
   // even fewer compound selectors, in a rule. The exact number of selectors
   // that would work depends on their sizes (e.g. "#foo .bar" has a size of 2).
   // Since we don't know the sizes of the selectors here, we simply split them
   // into groups of 1,024, based on the reasonable assumption that the average
   // selector won't have a size greater than 8. The alternative would be to
   // calculate the sizes of the selectors and divide them up accordingly, but
   // this approach is more efficient and has worked well in practice. In theory
@@ skipping 167 matching lines @@
   let inline = !userStyleSheetsSupported;
 
   let {elemhide, snippets} = message.filterTypes ||
                              {elemhide: true, snippets: true};
 
   if (!checkWhitelisted(sender.page, sender.frame, null,
                         RegExpFilter.typeMap.DOCUMENT))
   {
     let hostname = extractHostFromFrame(sender.frame);
 
-    if (snippets)
+    if (snippets && !browser.contentScripts)
     {
       for (let script of Snippets.getScriptsForDomain(hostname))
         executeScript(script, sender.page.id, sender.frame.id);
     }
 
     if (elemhide && !checkWhitelisted(sender.page, sender.frame, null,
                                       RegExpFilter.typeMap.ELEMHIDE))
     {
       let specificOnly = checkWhitelisted(sender.page, sender.frame, null,
                                           RegExpFilter.typeMap.GENERICHIDE);
@@ skipping 29 matching lines @@
   updateFrameStyles(sender.page.id, sender.frame.id, message.selectors,
                     message.groupName, message.appendOnly);
 });
 
 fetch(browser.extension.getURL("/snippets.js"), {cache: "no-cache"})
 .then(response => response.ok ? response.text() : "")
 .then(text =>
 {
   snippetsLibrarySource = text;
 });
+
+if (browser.contentScripts)
+{
+  Snippets.on("snippets.filterAdded", ({script, domains, text}) =>
+  {
+    let details = {
+      js: [{code: getExecutableCode(script)}],
+      allFrames: true,
+      matchAboutBlank: true,
+      runAt: "document_start",
+      matches: []
+    };
+
+    for (let [domain, include] of domains)
+    {
+      if (domain == "")
+        continue;
+
+      if (!include && !details.excludeMatches)
+        details.excludeMatches = [];
+
+      let matches = include ? details.matches : details.excludeMatches;
+
+      matches.push(`http://*.${domain}/*`);
+      matches.push(`https://*.${domain}/*`);
+    }
+
+    browser.contentScripts.register(details).then(contentScript =>
+    {
+      registeredContentScripts.set(text, contentScript);
+    });
+  });
+
+  Snippets.on("snippets.filterRemoved", ({text}) =>
+  {
+    let contentScript = registeredContentScripts.get(text);
+    if (contentScript)
+    {
+      contentScript.unregister();
+      registeredContentScripts.delete(text);
+    }
+  });
+
+  Snippets.on("snippets.filtersCleared", () =>
+  {
+    for (let contentScript of registeredContentScripts.values())
+      contentScript.unregister();
+
+    registeredContentScripts.clear();
+  });
+}