/* * Copyright 2010-2020 Gildas Lormeau * contact : gildas.lormeau 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 singlefile */ singlefile.extension.core.bg.business = (() => { const ERROR_CONNECTION_ERROR_CHROMIUM = "Could not establish connection. Receiving end does not exist."; const ERROR_CONNECTION_LOST_CHROMIUM = "The message port closed before a response was received."; const ERROR_CONNECTION_LOST_GECKO = "Message manager disconnected"; const ERROR_EDITOR_PAGE_CHROMIUM = "Cannot access contents of url "; const INJECT_SCRIPTS_STEP = 1; const EXECUTE_SCRIPTS_STEP = 2; const TASK_PENDING_STATE = "pending"; const TASK_PROCESSING_STATE = "processing"; const extensionScriptFiles = [ "common/index.js", "common/ui/content/content-infobar.js", "extension/lib/single-file/index.js", "extension/core/index.js", "extension/ui/index.js", "extension/core/content/content-main.js", "extension/core/content/content-download.js", "extension/ui/content/content-ui-main.js" ]; const tasks = []; let currentTaskId = 0, maxParallelWorkers; return { isSavingTab: tab => Boolean(tasks.find(taskInfo => taskInfo.tab.id == tab.id)), saveTabs, saveUrls, saveSelectedLinks, cancelTab, cancelTask: taskId => cancelTask(tasks.find(taskInfo => taskInfo.id == taskId)), cancelAllTasks: () => Array.from(tasks).forEach(cancelTask), getTasksInfo: () => tasks.map(mapTaskInfo), getTaskInfo: taskId => tasks.find(taskInfo => taskInfo.id == taskId), setCancelCallback: (taskId, cancelCallback) => { const taskInfo = tasks.find(taskInfo => taskInfo.id == taskId); if (taskInfo) { taskInfo.cancel = cancelCallback; } }, openEditor, onSaveEnd: taskId => { const taskInfo = tasks.find(taskInfo => taskInfo.id == taskId); if (taskInfo) { taskInfo.done(); } }, onInit: tab => cancelTab(tab.id), onTabRemoved: cancelTab }; async function saveSelectedLinks(tab) { const tabs = singlefile.extension.core.bg.tabs; const tabOptions = { extensionScriptFiles, tabId: tab.id, tabIndex: tab.index }; const scriptsInjected = await singlefile.extension.injectScript(tab.id, tabOptions); if (scriptsInjected) { const response = await tabs.sendMessage(tab.id, { method: "content.getSelectedLinks" }); if (response.urls && response.urls.length) { await saveUrls(response.urls); } } else { singlefile.extension.ui.bg.main.onForbiddenDomain(tab); } } async function saveUrls(urls, options = {}) { await initMaxParallelWorkers(); await Promise.all(urls.map(async url => { const tabOptions = await singlefile.extension.core.bg.config.getOptions(url); Object.keys(options).forEach(key => tabOptions[key] = options[key]); tabOptions.autoClose = true; tabOptions.extensionScriptFiles = extensionScriptFiles; addTask({ tab: { url }, status: TASK_PENDING_STATE, options: tabOptions, method: "content.save" }); })); runTasks(); } async function saveTabs(tabs, options = {}) { const config = singlefile.extension.core.bg.config; const autosave = singlefile.extension.core.bg.autosave; const ui = singlefile.extension.ui.bg.main; await initMaxParallelWorkers(); await Promise.all(tabs.map(async tab => { const tabId = tab.id; const tabOptions = await config.getOptions(tab.url); Object.keys(options).forEach(key => tabOptions[key] = options[key]); tabOptions.tabId = tabId; tabOptions.tabIndex = tab.index; tabOptions.extensionScriptFiles = extensionScriptFiles; if (options.autoSave) { if (autosave.isEnabled(tab)) { const taskInfo = addTask({ status: TASK_PROCESSING_STATE, tab, options: tabOptions, method: "content.autosave" }); runTask(taskInfo); } } else { ui.onStart(tabId, INJECT_SCRIPTS_STEP); const scriptsInjected = await singlefile.extension.injectScript(tabId, tabOptions); if (scriptsInjected || singlefile.extension.core.bg.editor.isEditor(tab)) { ui.onStart(tabId, EXECUTE_SCRIPTS_STEP); addTask({ status: TASK_PENDING_STATE, tab, options: tabOptions, method: "content.save" }); } else { ui.onForbiddenDomain(tab); } } })); runTasks(); } function addTask(info) { const taskInfo = { id: currentTaskId, status: info.status, tab: info.tab, options: info.options, method: info.method, done: function () { tasks.splice(tasks.findIndex(taskInfo => taskInfo.id == this.id), 1); runTasks(); } }; tasks.push(taskInfo); currentTaskId++; return taskInfo; } function openEditor(tab) { singlefile.extension.core.bg.tabs.sendMessage(tab.id, { method: "content.openEditor" }); } async function initMaxParallelWorkers() { if (!maxParallelWorkers) { maxParallelWorkers = (await singlefile.extension.core.bg.config.get()).maxParallelWorkers; } } function runTasks() { const processingCount = tasks.filter(taskInfo => taskInfo.status == TASK_PROCESSING_STATE).length; for (let index = 0; index < Math.min(tasks.length - processingCount, (maxParallelWorkers - processingCount)); index++) { const taskInfo = tasks.find(taskInfo => taskInfo.status == TASK_PENDING_STATE); if (taskInfo) { runTask(taskInfo); } } } async function runTask(taskInfo) { const ui = singlefile.extension.ui.bg.main; const tabs = singlefile.extension.core.bg.tabs; const taskId = taskInfo.id; taskInfo.status = TASK_PROCESSING_STATE; if (!taskInfo.tab.id) { let scriptsInjected; try { const tab = await tabs.createAndWait({ url: taskInfo.tab.url, active: false }); taskInfo.tab.id = taskInfo.options.tabId = tab.id; taskInfo.tab.index = taskInfo.options.tabIndex = tab.index; ui.onStart(taskInfo.tab.id, INJECT_SCRIPTS_STEP); scriptsInjected = await singlefile.extension.injectScript(taskInfo.tab.id, taskInfo.options); } catch (tabId) { taskInfo.tab.id = tabId; } if (scriptsInjected) { ui.onStart(taskInfo.tab.id, EXECUTE_SCRIPTS_STEP); } else { taskInfo.done(); return; } } taskInfo.options.taskId = taskId; try { await tabs.sendMessage(taskInfo.tab.id, { method: taskInfo.method, options: taskInfo.options }); if (taskInfo.options.autoClose && !taskInfo.cancelled) { tabs.remove(taskInfo.tab.id); } } catch (error) { if (error && (!error.message || !isIgnoredError(error))) { console.log(error.message ? error.message : error); // eslint-disable-line no-console ui.onError(taskInfo.tab.id); taskInfo.done(); } } } function isIgnoredError(error) { return error.message == ERROR_CONNECTION_LOST_CHROMIUM || error.message == ERROR_CONNECTION_ERROR_CHROMIUM || error.message == ERROR_CONNECTION_LOST_GECKO || error.message.startsWith(ERROR_EDITOR_PAGE_CHROMIUM + JSON.stringify(singlefile.extension.core.bg.editor.EDITOR_URL)); } function cancelTab(tabId) { Array.from(tasks).filter(taskInfo => taskInfo.tab.id == tabId && !taskInfo.options.autoSave).forEach(cancelTask); } function cancelTask(taskInfo) { const tabId = taskInfo.tab.id; taskInfo.cancelled = true; singlefile.extension.core.bg.tabs.sendMessage(tabId, { method: "content.cancelSave", options: { loadDeferredImages: taskInfo.options.loadDeferredImages, loadDeferredImagesKeepZoomLevel: taskInfo.options.loadDeferredImagesKeepZoomLevel } }); if (taskInfo.cancel) { taskInfo.cancel(); } if (taskInfo.method == "content.autosave") { singlefile.extension.ui.bg.main.onEnd(tabId, true); } singlefile.extension.ui.bg.main.onCancelled(taskInfo.tab); taskInfo.done(); } function mapTaskInfo(taskInfo) { return { id: taskInfo.id, tabId: taskInfo.tab.id, index: taskInfo.tab.index, url: taskInfo.tab.url, title: taskInfo.tab.title, cancelled: taskInfo.cancelled, status: taskInfo.status }; } })();