'use strict'; { let NULL = new Uint8Array(); let xmlFeedOrImage = /^(?:(?:application|text)\/(?:(?:r(?:ss|df)|atom)\+)xml(;|$))|image\//i; let rawXml = /^(?:application|text)\/xml;/i; let brokenOnLoad; let pendingRequests = new Map(); let reloadingTabs = new Map(); let tabKey = (tabId, url) => `${tabId}:${url}`; let cleanup = r => { pendingRequests.delete(r.requestId); let key = tabKey(r.tabId, r.url); if (reloadingTabs.get(key) === false) { reloadingTabs.delete(key); } }; let filter = { urls: [""], types: ["main_frame", "sub_frame", "object"] }; for (let event of ["onCompleted", "onErrorOccurred"]) browser.webRequest[event].addListener(cleanup, filter); let executeAll = async (scripts, where) => { let {url, tabId, frameId} = where; let count = 0; for (let details of scripts.values()) { details = Object.assign({ runAt: "document_start", matchAboutBlank: true, frameId, }, details); try { await browser.tabs.executeScript(tabId, details); count++; debug("Execute on start OK", url, details); } catch (e) { error(e, "Execute on start failed", url, details); } } return count; }; var RequestUtil = { getContentMetaData(request) { return request.content || (request.content = new ContentMetaData(request)); }, async executeOnStart(request, details) { let {requestId, url, tabId, frameId, statusCode} = request; if (statusCode >= 300 && statusCode < 400) return; let scripts = pendingRequests.get(requestId); let scriptKey = JSON.stringify(details); if (!scripts) { pendingRequests.set(requestId, scripts = new Map()); scripts.set(scriptKey, details); } else { scripts.set(scriptKey, details); return; } if (frameId === 0) { let key = tabKey(tabId, url); debug("Checking whether %s is a reloading tab...", key); if (reloadingTabs.get(key)) { reloadingTabs.set(key, false); // doom it for removal in cleanup return; } } let content = this.getContentMetaData(request); debug(url, content.type, content.charset); if (xmlFeedOrImage.test(content.type) && !/\/svg\b/i.test(content.type)) return; if (typeof brokenOnLoad === "undefined") { brokenOnLoad = await (async () => parseInt((await browser.runtime.getBrowserInfo()).version) < 61)(); } let mustCheckFeed = brokenOnLoad && frameId === 0 && rawXml.test(content.type); debug("mustCheckFeed = %s, brokenOnLoad = %s", mustCheckFeed, brokenOnLoad); let filter = browser.webRequest.filterResponseData(requestId); let buffer = []; let first = true; let done = false; let mustReload = false; let runAndFlush = async () => { let scriptsRan = await executeAll(scripts, request); if (mustCheckFeed && !scriptsRan) { mustReload = true; debug(`Marking as "must reload"`, tabId, url); reloadingTabs.set(tabKey(tabId, url), true); } if (buffer && buffer.length) { debug("Flushing %s buffer chunks", buffer.length); for (let chunk of buffer) { filter.write(chunk); } filter.disconnect(); buffer = null; } if (done) { filter.onstop(null); } }; if (brokenOnLoad) { filter.onstart = event => { filter.write(NULL); debug("onstart", url); } } filter.ondata = event => { if (first) { runAndFlush(); first = false; } if (buffer) { buffer.push(event.data); return; } debug("ondata", url); filter.write(event.data); filter.disconnect(); }; filter.onstop = event => { done = true; if (mustReload && !buffer) { mustReload = false; browser.tabs.update(tabId, {url}); } } } } }