Issue 6433 - Use generators in getURLsFromElement

include.preload.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 17 matching lines @@
   ["input", "IMAGE"],
   ["picture", "IMAGE"],
   ["audio", "MEDIA"],
   ["video", "MEDIA"],
   ["frame", "SUBDOCUMENT"],
   ["iframe", "SUBDOCUMENT"],
   ["object", "OBJECT"],
   ["embed", "OBJECT"]
 ]);
 
-function getURLsFromObjectElement(element)
+function* getURLsFromObjectElement(element)
 {
   let url = element.getAttribute("data");
   if (url)
-    return [url];
+  {
+    yield url;
+    return;
+  }
 
   for (let child of element.children)
   {
     if (child.localName != "param")
       continue;
 
     let name = child.getAttribute("name");
     if (name != "movie" &&  // Adobe Flash
         name != "source" && // Silverlight
         name != "src" &&    // Real Media + Quicktime
         name != "FileName") // Windows Media
       continue;
 
     let value = child.getAttribute("value");
     if (!value)
       continue;
 
-    return [value];
+    yield value;
+    return;
   }
-
-  return [];
 }
 
-function getURLsFromAttributes(element)
+function* getURLsFromAttributes({src = "", srcset = ""})
 {
-  let urls = [];
+  yield src;
 
-  if (element.src)
-    urls.push(element.src);
-
-  if (element.srcset)
-  {
-    for (let candidate of element.srcset.split(","))
-    {
-      let url = candidate.trim().replace(/\s+\S+$/, "");
-      if (url)
-        urls.push(url);
-    }
-  }
-
-  return urls;
+  for (let candidate of srcset.split(","))
+    yield candidate.trim().replace(/\s+\S+$/, "");
 }
 
-function getURLsFromMediaElement(element)
+function* getURLsFromMediaElement(element)
 {
-  let urls = getURLsFromAttributes(element);
+  yield* getURLsFromAttributes(element);
 
   for (let child of element.children)
   {
     if (child.localName == "source" || child.localName == "track")
-      urls.push(...getURLsFromAttributes(child));
+      yield* getURLsFromAttributes(child);
   }
 
-  if (element.poster)
-    urls.push(element.poster);
-
-  return urls;
+  yield element.poster;
 }
 
 function getURLsFromElement(element)
 {
-  let urls;
-  switch (element.localName)
-  {
-    case "object":
-      urls = getURLsFromObjectElement(element);
-      break;
+  let urls = null;
+  if (element.localName == "object")
+    urls = getURLsFromObjectElement(element);
+  else if (["video", "audio", "picture"].includes(element.localName))
+    urls = getURLsFromMediaElement(element);
+  else
+    urls = getURLsFromAttributes(element);
 
-    case "video":
-    case "audio":
-    case "picture":
-      urls = getURLsFromMediaElement(element);
-      break;
-
-    default:
-      urls = getURLsFromAttributes(element);
-      break;
-  }
-
-  for (let i = 0; i < urls.length; i++)
-  {
-    if (/^(?!https?:)[\w-]+:/i.test(urls[i]))
-      urls.splice(i--, 1);
-  }
-
-  return urls;
+  return Array.from(urls).filter(
+    url => url && !/^(?!https?:)[\w-]+:/i.test(url)
+  );
 }
 
 function hideElement(element)
 {
   function doHide()
   {
     let propertyName = "display";
     let propertyValue = "none";
     if (element.localName == "frame")
     {
@@ skipping 398 matching lines @@
     let element = event.target;
     if (/^i?frame$/.test(element.localName))
       checkCollapse(element);
   }, true);
 }
 
 window.checkCollapse = checkCollapse;
 window.elemhide = elemhide;
 window.typeMap = typeMap;
 window.getURLsFromElement = getURLsFromElement;