SingleFile/extension/ui/bg/ui.js

/*
 * 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 browser, singlefile */

singlefile.ui = (() => {

	const DEFAULT_ICON_PATH = "/extension/ui/resources/icon_16.png";
	const WAIT_ICON_PATH_PREFIX = "/extension/ui/resources/icon_16_wait";
	const DEFAULT_TITLE = "Save page with SingleFile";
	const DEFAULT_COLOR = [2, 147, 20, 255];
	const BADGE_PROPERTIES = [{ name: "text", browserActionMethod: "setBadgeText" }, { name: "color", browserActionMethod: "setBadgeBackgroundColor" }, { name: "title", browserActionMethod: "setTitle" }, { name: "path", browserActionMethod: "setIcon" }];
	const STORE_URLS = ["https://chrome.google.com", "https://addons.mozilla.org"];
	const MENU_ID_SAVE_PAGE = "save-page";
	const MENU_ID_SAVE_SELECTED = "save-selected";
	const MENU_ID_SAVE_FRAME = "save-frame";

	const tabs = {};
	const badgeTabs = {};
	let badgeRefreshPending = [];

	browser.runtime.onInstalled.addListener(refreshContextMenu);
	if (browser.menus && browser.menus.onClicked) {
		browser.menus.onClicked.addListener((event, tab) => {
			if (event.menuItemId == MENU_ID_SAVE_PAGE) {
				processTab(tab);
			}
			if (event.menuItemId == MENU_ID_SAVE_SELECTED) {
				processTab(tab, { selected: true });
			}
			if (event.menuItemId == MENU_ID_SAVE_FRAME) {
				processTab(tab, { frameId: event.frameId });
			}
		});
	}
	browser.browserAction.onClicked.addListener(async tab => {
		if (isAllowedURL(tab.url)) {
			const tabs = await browser.tabs.query({ currentWindow: true, highlighted: true });
			if (!tabs.length) {
				processTab(tab);
			} else {
				tabs.forEach(processTab);
			}
		}
	});
	browser.tabs.onActivated.addListener(async activeInfo => {
		const tab = browser.tabs.get(activeInfo.tabId);
		onTabActivated(tab.id, isAllowedURL(tab.url));
	});
	browser.tabs.onCreated.addListener(tab => onTabActivated(tab.id, isAllowedURL(tab.url)));
	browser.tabs.onUpdated.addListener((tabId, changeInfo, tab) => onTabActivated(tab.id, isAllowedURL(tab.url)));
	browser.tabs.onRemoved.addListener(tabId => onTabRemoved(tabId));
	browser.runtime.onMessage.addListener((request, sender) => {
		if (request.processProgress && request.maxIndex) {
			onTabProgress(sender.tab.id, request.index, request.maxIndex);
		}
		if (request.processEnd) {
			onTabEnd(sender.tab.id);
		}
		if (request.processError) {
			if (request.error) {
				console.error("Initialization error", request.error); // eslint-disable-line no-console
			}
			onTabError(sender.tab.id);
		}
	});
	return { update: refreshContextMenu };

	async function refreshContextMenu() {
		const config = await singlefile.config.get();
		if (browser.menus && browser.menus.removeAll && browser.menus.create) {
			if (config.contextMenuEnabled) {
				await browser.menus.removeAll();
				browser.menus.create({
					id: MENU_ID_SAVE_PAGE,
					contexts: ["page"],
					title: DEFAULT_TITLE
				});
				browser.menus.create({
					id: MENU_ID_SAVE_SELECTED,
					contexts: ["selection"],
					title: "Save selection with SingleFile"
				});
				browser.menus.create({
					id: MENU_ID_SAVE_FRAME,
					contexts: ["frame"],
					title: "Save frame with SingleFile"
				});
			} else {
				await browser.menus.removeAll();
			}
		}
	}

	async function processTab(tab, options) {
		const tabId = tab.id;
		try {
			await singlefile.core.processTab(tab, options);
			tabs[tabId] = {
				id: tabId,
				text: "...",
				color: DEFAULT_COLOR,
				title: "Initializing...",
				path: DEFAULT_ICON_PATH,
				progress: -1,
				barProgress: -1
			};
			refreshBadge(tabId);
		} catch (error) {
			if (error) {
				onTabError(tabId);
			} else {
				tabs[tabId] = {
					id: tabId,
					text: "↻",
					color: [255, 141, 1, 255],
					title: "Reload the page",
					path: DEFAULT_ICON_PATH,
					progress: -1,
					barProgress: -1
				};
				refreshBadge(tabId);
			}
		}
	}

	function onTabError(tabId) {
		const tabData = tabs[tabId];
		tabData.text = "ERR";
		tabData.color = [229, 4, 12, 255];
		tabData.title = DEFAULT_TITLE;
		tabData.path = DEFAULT_ICON_PATH;
		tabData.progress = -1;
		tabData.barProgress = -1;
		refreshBadge(tabId);
	}

	function onTabEnd(tabId) {
		const tabData = tabs[tabId];
		tabData.text = "OK";
		tabData.color = [4, 229, 36, 255];
		tabData.title = DEFAULT_TITLE;
		tabData.path = DEFAULT_ICON_PATH;
		tabData.progress = -1;
		tabData.barProgress = -1;
		refreshBadge(tabId);
	}

	function onTabProgress(tabId, index, maxIndex) {
		const tabData = tabs[tabId];
		const progress = Math.max(Math.min(20, Math.floor((index / maxIndex) * 20)), 0);
		if (tabData.progress != progress) {
			tabData.progress = progress;
			tabData.text = "";
			tabData.title = "Save progress: " + (progress * 5) + "%";
			tabData.color = [4, 229, 36, 255];
			const barProgress = Math.floor((index / maxIndex) * 8);
			if (tabData.barProgress != barProgress) {
				tabData.barProgress = barProgress;
				tabData.path = WAIT_ICON_PATH_PREFIX + barProgress + ".png";
			}
			refreshBadge(tabId);
		}
	}

	function onTabRemoved(tabId) {
		delete tabs[tabId];
	}

	function onTabActivated(tabId, isActive) {
		if (browser.browserAction && browser.browserAction.enable && browser.browserAction.disable) {
			if (isActive) {
				browser.browserAction.enable(tabId);
			} else {
				browser.browserAction.disable(tabId);
			}
		}
	}

	function isAllowedURL(url) {
		return url && (url.startsWith("http://") || url.startsWith("https://") || url.startsWith("file://")) && !STORE_URLS.find(storeUrl => url.startsWith(storeUrl));
	}

	async function refreshBadge(tabId) {
		await Promise.all(badgeRefreshPending);
		const promise = refreshBadgeAsync(tabId);
		badgeRefreshPending.push(promise);
		await promise;
		badgeRefreshPending = badgeRefreshPending.filter(pending => pending != promise);
	}

	async function refreshBadgeAsync(tabId) {
		for (let property of BADGE_PROPERTIES) {
			await refreshBadgeProperty(tabId, property.name, property.browserActionMethod);
		}
	}

	async function refreshBadgeProperty(tabId, property, browserActionMethod) {
		const tabData = tabs[tabId];
		const value = tabData[property];
		if (!badgeTabs[tabId]) {
			badgeTabs[tabId] = {};
		}
		if (JSON.stringify(badgeTabs[tabId][property]) != JSON.stringify(value)) {
			const browserActionParameter = { tabId };
			badgeTabs[tabId][property] = browserActionParameter[property] = value;
			if (browser.browserAction && browser.browserAction[browserActionMethod]) {
				await browser.browserAction[browserActionMethod](browserActionParameter);
			}
		}
	}

})();