SingleFile/extension/core/content/content-bootstrap.js

/*
 * Copyright 2010-2019 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 singlefile, frameTree, browser, window, addEventListener, removeEventListener, document, location, docHelper, setTimeout */

this.singlefile.bootstrap = this.singlefile.bootstrap || (async () => {

	let unloadListenerAdded, options, autoSaveEnabled, autoSaveTimeout, autoSavingPage;
	browser.runtime.sendMessage({ initAutoSave: true }).then(message => {
		options = message.options;
		autoSaveEnabled = message.autoSaveEnabled;
		refresh();
	});
	browser.runtime.onMessage.addListener(message => onMessage(message));
	browser.runtime.sendMessage({ loadURL: true });
	return {};

	async function onMessage(message) {
		if (message.autoSavePage) {
			autoSavingPage = false;
			singlefile.pageAutoSaved = false;
			options = message.options;
			await autoSavePage();
			if (options.autoSaveRepeat) {
				setTimeout(() => onMessage(message), options.autoSaveRepeatDelay * 1000);
			}
		}
		if (message.initAutoSave) {
			options = message.options;
			autoSaveEnabled = message.autoSaveEnabled;
			refresh();
		}
	}

	async function autoSavePage() {
		if ((!autoSavingPage || autoSaveTimeout) && !singlefile.pageAutoSaved) {
			autoSavingPage = true;
			if (options.autoSaveDelay && !autoSaveTimeout) {
				autoSaveTimeout = setTimeout(() => {
					autoSavePage();
				}, options.autoSaveDelay * 1000);
			} else {
				const docData = docHelper.preProcessDoc(document, window, options);
				let framesData = [];
				autoSaveTimeout = null;
				if (!options.removeFrames && this.frameTree) {
					framesData = await frameTree.getAsync(options);
				}
				browser.runtime.sendMessage({
					autoSaveContent: true,
					content: docHelper.serialize(document, false),
					canvasData: docData.canvasData,
					fontsData: docData.fontsData,
					stylesheetContents: docData.stylesheetContents,
					imageData: docData.imageData,
					postersData: docData.postersData,
					usedFonts: docData.usedFonts,
					shadowRootContents: docData.shadowRootContents,
					referrer: docData.referrer,
					framesData,
					url: location.href
				});
				docHelper.postProcessDoc(document, options);
				singlefile.pageAutoSaved = true;
				autoSavingPage = false;
			}
		}
	}

	async function refresh() {
		if (autoSaveEnabled && options && (options.autoSaveUnload || options.autoSaveLoadOrUnload)) {
			if (!unloadListenerAdded) {
				addEventListener("unload", onUnload);
				addEventListener("single-file-push-state", onUnload);
				unloadListenerAdded = true;
			}
		} else {
			removeEventListener("unload", onUnload);
			removeEventListener("single-file-push-state", onUnload);
			unloadListenerAdded = false;
		}
	}

	function onUnload() {
		if (!singlefile.pageAutoSaved || options.autoSaveUnload) {
			const docData = docHelper.preProcessDoc(document, window, options);
			if (this.frameTree && !options.removeFrames) {
				browser.runtime.sendMessage({
					autoSaveContent: true,
					content: docHelper.serialize(document),
					canvasData: docData.canvasData,
					fontsData: docData.fontsData,
					stylesheetContents: docData.stylesheetContents,
					imageData: docData.imageData,
					postersData: docData.postersData,
					usedFonts: docData.usedFonts,
					shadowRootContents: docData.shadowRootContents,
					referrer: docData.referrer,
					framesData: frameTree.getSync(options),
					url: location.href
				});
			}
		}
	}
})();