SingleFile/extension/core/bg/companion.js

/*
 * Copyright 2010-2020 Gildas Lormeau
 * contact : gildas.lormeau <at> gmail.com
 * 
 * This file is part of SingleFile.
 *
 *   The code in this file is free software: you can redistribute it and/or 
 *   modify it under the terms of the GNU Affero General Public License 
 *   (GNU AGPL) as published by the Free Software Foundation, either version 3
 *   of the License, or (at your option) any later version.
 * 
 *   The code in this file 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 Affero 
 *   General Public License for more details.
 *
 *   As additional permission under GNU AGPL version 3 section 7, you may 
 *   distribute UNMODIFIED VERSIONS OF THIS file without the copy of the GNU 
 *   AGPL normally required by section 4, provided you include this license 
 *   notice and a URL through which recipients can access the Corresponding 
 *   Source.
 */

/* global browser */

import * as ui from "./../../ui/bg/index.js";

let enabled = true;

export {
	enabled,
	onMessage,
	externalSave,
	save
};

async function onMessage(message) {
	if (message.method.endsWith(".state")) {
		return { enabled };
	}
}

async function externalSave(options) {
	try {
		options.autoSaveExternalSave = false;
		const port = browser.runtime.connectNative("singlefile_companion");
		port.postMessage({
			method: "externalSave",
			pageData: options
		});
		await new Promise((resolve, reject) => {
			port.onDisconnect.addListener(() => {
				if (port.error) {
					reject(port.error.message);
				} else if (!browser.runtime.lastError || browser.runtime.lastError.message.includes("Native host has exited")) {
					resolve();
				}
			});
		});
		ui.onEnd(options.tabId, options.autoSave);
	} catch (error) {
		console.error(error); // eslint-disable-line no-console			
		ui.onError(options.tabId);
	}
}

async function save(pageData) {
	const port = browser.runtime.connectNative("singlefile_companion");
	port.postMessage({
		method: "save",
		pageData
	});
	await new Promise((resolve, reject) => {
		port.onDisconnect.addListener(() => {
			if (port.error) {
				reject(port.error.message);
			} else if (!browser.runtime.lastError || browser.runtime.lastError.message.includes("Native host has exited")) {
				resolve();
			}
		});
	});
}