added "frame-tree" library

lib/frame-tree/bg/frame-tree.js

@@ -0,0 +1,45 @@
+/*
+ * Copyright 2018 Gildas Lormeau
+ * contact : gildas.lormeau <at> gmail.com
+ * 
+ * This file is part of SingleFile.
+ *
+ *   SingleFile is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU Lesser General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   SingleFile is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *   GNU Lesser General Public License for more details.
+ *
+ *   You should have received a copy of the GNU Lesser General Public License
+ *   along with SingleFile.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+/* global chrome */
+
+this.FrameTree = (() => {
+	chrome.runtime.onMessage.addListener((message, sender) => {
+		if (message.method == "FrameTree.getDataRequest") {
+			chrome.tabs.sendMessage(sender.tab.id, { method: "FrameTree.getDataRequest", windowId: message.windowId, tabId: sender.tab.id });
+		}
+		if (message.method == "FrameTree.getDataResponse") {
+			chrome.tabs.sendMessage(message.tabId, { method: "FrameTree.getDataResponse", windowId: message.windowId, content: message.content, baseURI: message.baseURI });
+		}
+	});
+
+	return {
+		async initialize(tabId) {
+			return new Promise(resolve => {
+				chrome.runtime.onMessage.addListener(message => {
+					if (message.method == "FrameTree.initResponse") {
+						resolve();
+					}
+				});
+				chrome.tabs.sendMessage(tabId, { method: "FrameTree.initRequest", windowId: "0", index: 0 });
+			});
+		}
+	};
+})();

lib/frame-tree/content/frame-tree.js

@@ -0,0 +1,208 @@
+/*
+ * Copyright 2018 Gildas Lormeau
+ * contact : gildas.lormeau <at> gmail.com
+ * 
+ * This file is part of SingleFile.
+ *
+ *   SingleFile is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU Lesser General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   SingleFile is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *   GNU Lesser General Public License for more details.
+ *
+ *   You should have received a copy of the GNU Lesser General Public License
+ *   along with SingleFile.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+/* global chrome, window, top, document, HTMLHtmlElement, addEventListener */
+
+this.FrameTree = (() => {
+
+	const MESSAGE_PREFIX = "__FrameTree__";
+	const TIMEOUT_POST_MESSAGE = 1000;
+
+	const FrameTree = {
+		getFramesData
+	};
+
+	let framesData;
+	let dataRequestCallbacks;
+
+	if (window == top) {
+		chrome.runtime.onMessage.addListener(message => {
+			if (message.method == "FrameTree.initRequest" && document.documentElement instanceof HTMLHtmlElement) {
+				dataRequestCallbacks = new Map();
+				framesData = [];
+				initRequest(message);
+			}
+			if (message.method == "FrameTree.getDataResponse") {
+				getDataResponse(message);
+			}
+		});
+	}
+	chrome.runtime.onMessage.addListener(message => {
+		if (message.method == "FrameTree.getDataRequest" && FrameTree.windowId == message.windowId) {
+			chrome.runtime.sendMessage({
+				method: "FrameTree.getDataResponse",
+				windowId: message.windowId,
+				tabId: message.tabId,
+				content: getDoctype(document) + document.documentElement.outerHTML,
+				baseURI: document.baseURI,
+				title: document.title
+			});
+		}
+	});
+	addEventListener("message", event => {
+		if (typeof event.data === "string" && event.data.startsWith(MESSAGE_PREFIX + "::")) {
+			const message = JSON.parse(event.data.substring(MESSAGE_PREFIX.length + 2));
+			if (message.initRequest) {
+				initRequest(message);
+			} else if (message.initResponse) {
+				initResponse(message);
+			} else if (message.getDataResponse) {
+				getDataResponse(message);
+			}
+		}
+	}, false);
+	return FrameTree;
+
+	async function getFramesData() {
+		await Promise.all(framesData.map(async frameData => {
+			return new Promise(resolve => {
+				dataRequestCallbacks.set(frameData.windowId, resolve);
+				if (frameData.sameDomain) {
+					top.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({
+						getDataRequest: true,
+						windowId: frameData.windowId
+					}), "*");
+				} else {
+					chrome.runtime.sendMessage({
+						method: "FrameTree.getDataRequest",
+						windowId: frameData.windowId
+					});
+				}
+			});
+		}));
+		return framesData.sort((frame1, frame2) => frame2.windowId.split(".").length - frame1.windowId.split(".").length);
+	}
+
+	function initRequest(message) {
+		FrameTree.windowId = message.windowId;
+		FrameTree.index = message.index;
+		const frameElements = document.querySelectorAll("iframe, frame");
+		if (frameElements.length) {
+			setFramesWinId(MESSAGE_PREFIX, frameElements, FrameTree.index, FrameTree.windowId, window);
+		} else {
+			top.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({ initResponse: true, framesData: [], windowId: FrameTree.windowId, index: FrameTree.index }), "*");
+		}
+	}
+
+	function initResponse(message) {
+		if (window == top) {
+			if (message.framesData) {
+				message.framesData = message.framesData instanceof Array ? message.framesData : JSON.parse(message.framesData);
+				framesData = framesData.concat(message.framesData);
+				const frameData = framesData.find(frameData => frameData.windowId == message.windowId);
+				const pendingCount = framesData.filter(frameData => !frameData.processed).length;
+				if (message.windowId != "0") {
+					frameData.processed = true;
+				}
+				if (!pendingCount || pendingCount == 1) {
+					chrome.runtime.sendMessage({ method: "FrameTree.initResponse" });
+				}
+			}
+		} else {
+			FrameTree.windowId = message.windowId;
+			FrameTree.index = message.index;
+		}
+	}
+
+	function setFramesWinId(MESSAGE_PREFIX, frameElements, index, windowId, win) {
+		const framesData = [];
+		if (win != top) {
+			win.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({ initResponse: true, windowId, index }), "*");
+		}
+		frameElements.forEach((frameElement, index) => {
+			let src, sameDomain;
+			try {
+				sameDomain = Boolean(frameElement.contentDocument && frameElement.contentWindow && top.addEventListener && top);
+				src = frameElement.src;
+			} catch (e) {
+				/* ignored */
+			}
+			framesData.push({ sameDomain, src, index, windowId: windowId + "." + index });
+		});
+		top.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({ initResponse: true, framesData, windowId, index }), "*");
+		frameElements.forEach((frameElement, index) => {
+			const frameWinId = windowId + "." + index;
+			let frameDoc, frameWindow, topWindow;
+			try {
+				frameDoc = frameElement.contentDocument;
+				frameWindow = frameElement.contentWindow;
+				topWindow = top.addEventListener && top;
+			} catch (e) {
+				/* ignored */
+			}
+			if (frameWindow && frameDoc && topWindow) {
+				setFramesWinId(MESSAGE_PREFIX, frameDoc.querySelectorAll("iframe, frame"), index, frameWinId, frameWindow);
+				topWindow.addEventListener("message", onMessage, false);
+			} else if (frameWindow) {
+				frameWindow.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({ initRequest: true, windowId: frameWinId, index }), "*");
+				setTimeout(() => {
+					top.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({ initResponse: true, framesData: [], windowId: frameWinId, index }), "*");
+				}, TIMEOUT_POST_MESSAGE);
+			}
+
+			function onMessage(event) {
+				if (typeof event.data === "string" && event.data.startsWith(MESSAGE_PREFIX + "::")) {
+					const message = JSON.parse(event.data.substring(MESSAGE_PREFIX.length + 2));
+					if (message.getDataRequest && message.windowId == frameWinId) {
+						topWindow.removeEventListener("message", onMessage, false);
+						topWindow.postMessage(MESSAGE_PREFIX + "::" + JSON.stringify({
+							getDataResponse: true,
+							windowId: message.windowId,
+							content: getDoctype(frameDoc) + frameDoc.documentElement.outerHTML,
+							baseURI: frameDoc.baseURI,
+							title: document.title
+						}), "*");
+					}
+				}
+			}
+		});
+	}
+
+	function getDataResponse(message) {
+		delete message.tabId;
+		delete message.method;
+		const frameData = framesData.find(frameData => frameData.windowId == message.windowId);
+		frameData.content = message.content;
+		frameData.baseURI = message.baseURI;
+		frameData.title = message.title;
+		dataRequestCallbacks.get(message.windowId)(message);
+	}
+
+	function getDoctype(doc) {
+		const docType = doc.doctype;
+		let docTypeStr;
+		if (docType) {
+			docTypeStr = "<!DOCTYPE " + docType.nodeName;
+			if (docType.publicId) {
+				docTypeStr += " PUBLIC \"" + docType.publicId + "\"";
+				if (docType.systemId) {
+					docTypeStr += " \"" + docType.systemId + "\"";
+				}
+			} else if (docType.systemId) {
+				docTypeStr += " SYSTEM \"" + docType.systemId + "\"";
+			} if (docType.internalSubset) {
+				docTypeStr += " [" + docType.internalSubset + "]";
+			}
+			return docTypeStr + ">\n";
+		}
+		return "";
+	}
+
+})();