瀏覽代碼

fetch from main world in Firefox when `referrer` is true
fixes #1056

Gildas 3 年之前
父節點
當前提交
df65e7a409
共有 1 個文件被更改,包括 63 次插入2 次删除
  1. 63 2
      src/lib/single-file/fetch/content/content-fetch.js

+ 63 - 2
src/lib/single-file/fetch/content/content-fetch.js

@@ -21,7 +21,18 @@
  *   Source.
  */
 
-/* global browser, window */
+/* global browser, window, CustomEvent, setTimeout, clearTimeout */
+
+const FETCH_REQUEST_EVENT = "single-file-request-fetch";
+const FETCH_ACK_EVENT = "single-file-ack-fetch";
+const FETCH_RESPONSE_EVENT = "single-file-response-fetch";
+const ERR_HOST_FETCH = "Host fetch error (SingleFile)";
+const HOST_FETCH_MAX_DELAY = 2500;
+const USE_HOST_FETCH = Boolean(window.wrappedJSObject);
+
+const addEventListener = (type, listener, options) => window.addEventListener(type, listener, options);
+const dispatchEvent = event => window.dispatchEvent(event);
+const removeEventListener = (type, listener, options) => window.removeEventListener(type, listener, options);
 
 const fetch = (url, options) => window.fetch(url, options);
 
@@ -82,6 +93,55 @@ async function onFetchResponse(message) {
 	return {};
 }
 
+async function hostFetch(url, options) {
+	const result = new Promise((resolve, reject) => {
+		dispatchEvent(new CustomEvent(FETCH_REQUEST_EVENT, { detail: url }));
+		addEventListener(FETCH_ACK_EVENT, onAckFetch, false);
+		addEventListener(FETCH_RESPONSE_EVENT, onResponseFetch, false);
+		const timeout = setTimeout(() => {
+			removeListeners();
+			reject(new Error(ERR_HOST_FETCH));
+		}, HOST_FETCH_MAX_DELAY);
+
+		function onResponseFetch(event) {
+			if (event.detail) {
+				if (event.detail.url == url) {
+					removeListeners();
+					if (event.detail.response) {
+						resolve({
+							status: event.detail.status,
+							headers: new Map(event.detail.headers),
+							arrayBuffer: async () => event.detail.response
+						});
+					} else {
+						reject(event.detail.error);
+					}
+				}
+			} else {
+				reject();
+			}
+		}
+
+		function onAckFetch() {
+			clearTimeout(timeout);
+		}
+
+		function removeListeners() {
+			removeEventListener(FETCH_RESPONSE_EVENT, onResponseFetch, false);
+			removeEventListener(FETCH_ACK_EVENT, onAckFetch, false);
+		}
+	});
+	try {
+		return await result;
+	} catch (error) {
+		if (error && error.message == ERR_HOST_FETCH) {
+			return fetch(url, options);
+		} else {
+			throw error;
+		}
+	}
+}
+
 export {
 	fetchResource as fetch,
 	frameFetch
@@ -89,7 +149,8 @@ export {
 
 async function fetchResource(url, options = {}) {
 	try {
-		return await fetch(url, { cache: "force-cache", headers: options.headers });
+		const fetchOptions = { cache: "force-cache", headers: options.headers };
+		return await (options.referrer && USE_HOST_FETCH ? hostFetch(url, fetchOptions) : fetch(url, fetchOptions));
 	}
 	catch (error) {
 		requestId++;