| Index: include.preload.js |
| =================================================================== |
| --- a/include.preload.js |
| +++ b/include.preload.js |
| @@ -337,20 +337,21 @@ |
| this.observer.disconnect(); |
| clearTimeout(this.timeout); |
| } |
| }; |
| function ElemHide() |
| { |
| this.shadow = this.createShadowTree(); |
| - this.style = null; |
| + this.styles = new Map(); |
|
Manish Jethani
2017/10/14 21:35:33
Now it's a map.
|
| this.tracer = null; |
| - this.inject = true; |
| + this.inline = true; |
| this.emulatedPatterns = null; |
| + this.injectedSelectors = []; |
| this.elemHideEmulation = new ElemHideEmulation( |
| this.addSelectors.bind(this), |
| this.hideElements.bind(this) |
| ); |
| } |
| ElemHide.prototype = { |
| selectorGroupSize: 200, |
| @@ -374,34 +375,61 @@ |
| // avoid creating the shadowRoot twice. |
| let shadow = document.documentElement.shadowRoot || |
| document.documentElement.createShadowRoot(); |
| shadow.appendChild(document.createElement("shadow")); |
| return shadow; |
| }, |
| - injectSelectors(selectors, filters) |
| + injectSelectors(selectors) |
| { |
| - if (!this.style) |
| + if (selectors.length == 0 && this.injectedSelectors.length == 0) |
| + return; |
| + |
| + chrome.runtime.sendMessage({ |
|
Manish Jethani
2017/10/14 21:35:33
Keeping it simple.
|
| + type: "elemhide.injectSelectors", |
| + selectors, |
| + oldSelectors: this.injectedSelectors |
| + }); |
| + |
| + this.injectedSelectors = selectors; |
| + }, |
| + |
| + addSelectorsInline(selectors, filters, groupName) |
| + { |
| + let style = this.styles.get(groupName); |
| + |
| + if (style) |
| + { |
| + while (style.sheet.cssRules.length > 0) |
| + style.sheet.deleteRule(0); |
|
Manish Jethani
2017/10/14 21:35:33
Using CSSStyleSheet.deleteRule here.
|
| + } |
| + |
| + if (selectors.length == 0) |
| + return; |
| + |
| + if (!style) |
| { |
| // Create <style> element lazily, only if we add styles. Add it to |
| // the shadow DOM if possible. Otherwise fallback to the <head> or |
| // <html> element. If we have injected a style element before that |
| // has been removed (the sheet property is null), create a new one. |
| - this.style = document.createElement("style"); |
| + style = document.createElement("style"); |
| (this.shadow || document.head || |
| - document.documentElement).appendChild(this.style); |
| + document.documentElement).appendChild(style); |
| // It can happen that the frame already navigated to a different |
| // document while we were waiting for the background page to respond. |
| // In that case the sheet property will stay null, after addind the |
| // <style> element to the shadow DOM. |
| - if (!this.style.sheet) |
| + if (!style.sheet) |
| return; |
| + |
| + this.styles.set(groupName, style); |
| } |
| // If using shadow DOM, we have to add the ::content pseudo-element |
| // before each selector, in order to match elements within the |
| // insertion point. |
| let preparedSelectors = []; |
| if (this.shadow) |
| { |
| @@ -422,49 +450,46 @@ |
| // (Chrome also has a limit, larger... but we're not certain exactly what it |
| // is! Edge apparently has no such limit.) |
| // [1] - https://github.com/WebKit/webkit/blob/1cb2227f6b2a1035f7bdc46e5ab69debb75fc1de/Source/WebCore/css/RuleSet.h#L68 |
| for (let i = 0; i < preparedSelectors.length; i += this.selectorGroupSize) |
| { |
| let selector = preparedSelectors.slice( |
| i, i + this.selectorGroupSize |
| ).join(", "); |
| - this.style.sheet.insertRule(selector + "{display: none !important;}", |
| - this.style.sheet.cssRules.length); |
| + style.sheet.insertRule(selector + "{display: none !important;}", |
| + style.sheet.cssRules.length); |
| } |
| + |
| + if (this.tracer) |
| + this.tracer.addSelectors(selectors, filters); |
| }, |
| addSelectors(selectors, filters) |
| { |
| - if (selectors.length == 0) |
| - return; |
| - |
| - if (this.inject) |
| + if (this.inline) |
| { |
| // Insert the style rules inline if we have been instructed by the |
| // background page to do so. This is usually the case, except on platforms |
| // that do support user stylesheets via the chrome.tabs.insertCSS API |
| // (Firefox 53 onwards for now and possibly Chrome in the near future). |
| // Once all supported platforms have implemented this API, we can remove |
| // the code below. See issue #5090. |
| // Related Chrome and Firefox issues: |
| // https://bugs.chromium.org/p/chromium/issues/detail?id=632009 |
| // https://bugzilla.mozilla.org/show_bug.cgi?id=1310026 |
| - this.injectSelectors(selectors, filters); |
| + this.addSelectorsInline(selectors, filters, "emulated"); |
| } |
| else |
| { |
| - chrome.runtime.sendMessage({ |
| - type: "elemhide.injectSelectors", |
| - selectors |
| - }); |
| + this.injectSelectors(selectors); |
| + |
| + if (this.tracer && selectors.length > 0) |
| + this.tracer.addSelectors(selectors, filters); |
| } |
| - |
| - if (this.tracer) |
| - this.tracer.addSelectors(selectors, filters); |
| }, |
| hideElements(elements, filters) |
| { |
| for (let element of elements) |
| hideElement(element); |
| if (this.tracer) |
| @@ -480,27 +505,23 @@ |
| apply() |
| { |
| chrome.runtime.sendMessage({type: "elemhide.getSelectors"}, response => |
| { |
| if (this.tracer) |
| this.tracer.disconnect(); |
| this.tracer = null; |
| - if (this.style && this.style.parentElement) |
| - this.style.parentElement.removeChild(this.style); |
| - this.style = null; |
| - |
| if (response.trace) |
| this.tracer = new ElementHidingTracer(); |
| - this.inject = response.inject; |
| + this.inline = response.inline; |
| - if (this.inject) |
| - this.addSelectors(response.selectors); |
| + if (this.inline) |
| + this.addSelectorsInline(response.selectors); |
|
Manish Jethani
2017/10/14 21:35:33
Note that we don't have to pass the group name her
|
| else if (this.tracer) |
| this.tracer.addSelectors(response.selectors); |
| this.elemHideEmulation.apply(response.emulatedPatterns); |
| }); |
| } |
| }; |