open.source/stylus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: use dataurl to inject page script

Browse Source
This commit is contained in:
eight 2018-10-14 02:16:48 +08:00
parent cb5cbb4d10
commit ad06551440
1 changed files with 33 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
content/apply.js
View File

@ -66,27 +66,27 @@ const APPLY = (() => {
// Since it's easy to spoof the browser version in pre-Quantum FF we're checking // Since it's easy to spoof the browser version in pre-Quantum FF we're checking
// for getPreventDefault which got removed in FF59 https://bugzil.la/691151 // for getPreventDefault which got removed in FF59 https://bugzil.la/691151
const EVENT_NAME = chrome.runtime.id; const EVENT_NAME = chrome.runtime.id;
if (CHROME || isOwnPage || Event.prototype.getPreventDefault || !injectPageScript()) { const usePageScript = CHROME || isOwnPage || Event.prototype.getPreventDefault ?
return (el, content) => { Promise.resolve(false) : injectPageScript();
// FIXME: do we have to keep el.sheet.disabled?
el.textContent = content;
};
}
return (el, content) => { return (el, content) => {
const detail = pageObject({ usePageScript.then(ok => {
method: 'setStyleContent', if (!ok) {
id: el.id, // FIXME: do we have to keep el.sheet.disabled?
content el.textContent = content;
} else {
const detail = pageObject({
method: 'setStyleContent',
id: el.id,
content
});
window.dispatchEvent(new CustomEvent(EVENT_NAME, {detail}));
}
}); });
window.dispatchEvent(new CustomEvent(EVENT_NAME, {detail}));
}; };
function injectPageScript() { function injectPageScript() {
// FIXME: does it work with XML?
const scriptContent = EVENT_NAME => { const scriptContent = EVENT_NAME => {
window.dispatchEvent(new CustomEvent(EVENT_NAME, { document.currentScript.remove();
detail: {method: 'pageScriptOK'}
}));
window.addEventListener(EVENT_NAME, function handler(e) { window.addEventListener(EVENT_NAME, function handler(e) {
const {method, id, content} = e.detail; const {method, id, content} = e.detail;
if (method === 'setStyleContent') { if (method === 'setStyleContent') {
@ -102,24 +102,27 @@ const APPLY = (() => {
} }
}, true); }, true);
}; };
let ok = false; const code = `(${scriptContent})(${JSON.stringify(EVENT_NAME)})`;
const check = e => { const src = `data:application/javascript;base64,${btoa(code)}`;
if (e.detail.method === 'pageScriptOK') { const script = document.createElement('script');
ok = true; const {resolve, promise} = deferred();
} script.src = src;
}; script.onload = () => resolve(true);
window.addEventListener(EVENT_NAME, check, true); script.onerror = () => resolve(false);
try { document.documentElement.appendChild(script);
// eslint-disable-next-line no-eval return promise;
window.eval(`(${scriptContent})(${JSON.stringify(EVENT_NAME)})`);
} catch (err) {
// csp error
}
window.removeEventListener(EVENT_NAME, check, true);
return ok;
} }
} }
function deferred() {
const o = {};
o.promise = new Promise((resolve, reject) => {
o.resolve = resolve;
o.reject = reject;
});
return o;
}
function getMatchUrl() { function getMatchUrl() {
var matchUrl = location.href; var matchUrl = location.href;
if (!matchUrl.match(/^(http|file|chrome|ftp)/)) { if (!matchUrl.match(/^(http|file|chrome|ftp)/)) {