Index: include.preload.js |
=================================================================== |
--- a/include.preload.js |
+++ b/include.preload.js |
@@ -338,19 +338,21 @@ |
clearTimeout(this.timeout); |
} |
}; |
function ElemHide() |
{ |
this.shadow = this.createShadowTree(); |
this.style = null; |
+ this.emulatedStyle = null; |
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 +376,67 @@ |
// 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; |
+ |
+ let message = {type: "elemhide.injectSelectors"}; |
+ |
+ if (selectors.length > 0) |
+ message.selectors = selectors; |
+ |
+ if (this.injectedSelectors.length > 0) |
+ message.replaceSelectors = this.injectedSelectors; |
Manish Jethani
2017/10/14 14:23:22
Renamed for clarity.
|
+ |
+ chrome.runtime.sendMessage(message); |
+ this.injectedSelectors = selectors; |
+ }, |
+ |
+ addSelectorsInline(selectors, filters, emulated) |
+ { |
+ if (emulated && this.emulatedStyle) |
Manish Jethani
2017/10/14 14:23:21
So there's a separate style sheet for emulated sel
|
+ { |
+ this.emulatedStyle.parentNode.removeChild(this.emulatedStyle); |
+ this.emulatedStyle = null; |
+ } |
+ |
+ if (selectors.length == 0) |
+ return; |
+ |
+ let style = emulated ? this.emulatedStyle : this.style; |
+ |
+ 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; |
+ |
+ if (emulated) |
+ this.emulatedStyle = style; |
+ else |
+ this.style = 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 +457,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, true); |
Manish Jethani
2017/10/14 14:23:21
addSelectors is now only called for emulated selec
|
} |
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 +512,27 @@ |
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); |
+ if (this.style && this.style.parentNode) |
Manish Jethani
2017/10/14 14:23:22
This needs to be parentNode, not parentElement, be
Manish Jethani
2017/10/14 21:35:33
Actually, why is this needed? The styles here will
Manish Jethani
2017/10/15 21:29:12
OK, this code is called from composer.postload.js
|
+ this.style.parentNode.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); |
else if (this.tracer) |
this.tracer.addSelectors(response.selectors); |
this.elemHideEmulation.apply(response.emulatedPatterns); |
}); |
} |
}; |