SingleFile/extension/ui/bg/ui-editor.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, singlefile, window, document, prompt, matchMedia */

singlefile.extension.ui.bg.editor = (() => {

	const editorElement = document.querySelector(".editor");
	const toolbarElement = document.querySelector(".toolbar");
	const highlightYellowButton = document.querySelector(".highlight-yellow-button");
	const highlightPinkButton = document.querySelector(".highlight-pink-button");
	const highlightBlueButton = document.querySelector(".highlight-blue-button");
	const highlightGreenButton = document.querySelector(".highlight-green-button");
	const highlightButtons = Array.from(document.querySelectorAll(".highlight-button"));
	const toggleNotesButton = document.querySelector(".toggle-notes-button");
	const toggleHighlightsButton = document.querySelector(".toggle-highlights-button");
	const removeHighlightButton = document.querySelector(".remove-highlight-button");
	const addYellowNoteButton = document.querySelector(".add-note-yellow-button");
	const addPinkNoteButton = document.querySelector(".add-note-pink-button");
	const addBlueNoteButton = document.querySelector(".add-note-blue-button");
	const addGreenNoteButton = document.querySelector(".add-note-green-button");
	const editPageButton = document.querySelector(".edit-page-button");
	const formatPageButton = document.querySelector(".format-page-button");
	const cutInnerPageButton = document.querySelector(".cut-inner-page-button");
	const cutOuterPageButton = document.querySelector(".cut-outer-page-button");
	const undoCutPageButton = document.querySelector(".undo-cut-page-button");
	const undoAllCutPageButton = document.querySelector(".undo-all-cut-page-button");
	const redoCutPageButton = document.querySelector(".redo-cut-page-button");
	const savePageButton = document.querySelector(".save-page-button");
	const printPageButton = document.querySelector(".print-page-button");
	const lastButton = toolbarElement.querySelector(".buttons:last-of-type [type=button]:last-of-type");

	let tabData, tabDataContents = [];

	addYellowNoteButton.title = browser.i18n.getMessage("editorAddYellowNote");
	addPinkNoteButton.title = browser.i18n.getMessage("editorAddPinkNote");
	addBlueNoteButton.title = browser.i18n.getMessage("editorAddBlueNote");
	addGreenNoteButton.title = browser.i18n.getMessage("editorAddGreenNote");
	highlightYellowButton.title = browser.i18n.getMessage("editorHighlightYellow");
	highlightPinkButton.title = browser.i18n.getMessage("editorHighlightPink");
	highlightBlueButton.title = browser.i18n.getMessage("editorHighlightBlue");
	highlightGreenButton.title = browser.i18n.getMessage("editorHighlightGreen");
	toggleNotesButton.title = browser.i18n.getMessage("editorToggleNotes");
	toggleHighlightsButton.title = browser.i18n.getMessage("editorToggleHighlights");
	removeHighlightButton.title = browser.i18n.getMessage("editorRemoveHighlight");
	editPageButton.title = browser.i18n.getMessage("editorEditPage");
	formatPageButton.title = browser.i18n.getMessage("editorFormatPage");
	cutInnerPageButton.title = browser.i18n.getMessage("editorCutInnerPage");
	cutOuterPageButton.title = browser.i18n.getMessage("editorCutOuterPage");
	undoCutPageButton.title = browser.i18n.getMessage("editorUndoCutPage");
	undoAllCutPageButton.title = browser.i18n.getMessage("editorUndoAllCutPage");
	redoCutPageButton.title = browser.i18n.getMessage("editorRedoCutPage");
	savePageButton.title = browser.i18n.getMessage("editorSavePage");
	printPageButton.title = browser.i18n.getMessage("editorPrintPage");

	addYellowNoteButton.onmouseup = () => editorElement.contentWindow.postMessage(JSON.stringify({ method: "addNote", color: "note-yellow" }), "*");
	addPinkNoteButton.onmouseup = () => editorElement.contentWindow.postMessage(JSON.stringify({ method: "addNote", color: "note-pink" }), "*");
	addBlueNoteButton.onmouseup = () => editorElement.contentWindow.postMessage(JSON.stringify({ method: "addNote", color: "note-blue" }), "*");
	addGreenNoteButton.onmouseup = () => editorElement.contentWindow.postMessage(JSON.stringify({ method: "addNote", color: "note-green" }), "*");
	document.addEventListener("mouseup", event => {
		editorElement.contentWindow.focus();
		toolbarOnTouchEnd(event);
	}, true);
	document.onmousemove = toolbarOnTouchMove;
	highlightButtons.forEach(highlightButton => {
		highlightButton.onmouseup = () => {
			if (toolbarElement.classList.contains("cut-inner-mode")) {
				disableCutInnerPage();
			}
			if (toolbarElement.classList.contains("cut-outer-mode")) {
				disableCutOuterPage();
			}
			if (toolbarElement.classList.contains("remove-highlight-mode")) {
				disableRemoveHighlights();
			}
			const disabled = highlightButton.classList.contains("highlight-disabled");
			resetHighlightButtons();
			if (disabled) {
				highlightButton.classList.remove("highlight-disabled");
				editorElement.contentWindow.postMessage(JSON.stringify({ method: "enableHighlight", color: "single-file-highlight-" + highlightButton.dataset.color }), "*");
			} else {
				highlightButton.classList.add("highlight-disabled");
			}
		};
	});
	toggleNotesButton.onmouseup = () => {
		if (toggleNotesButton.getAttribute("src") == "/extension/ui/resources/button_note_visible.png") {
			toggleNotesButton.src = "/extension/ui/resources/button_note_hidden.png";
			editorElement.contentWindow.postMessage(JSON.stringify({ method: "hideNotes" }), "*");
		} else {
			toggleNotesButton.src = "/extension/ui/resources/button_note_visible.png";
			editorElement.contentWindow.postMessage(JSON.stringify({ method: "displayNotes" }), "*");
		}
	};
	toggleHighlightsButton.onmouseup = () => {
		if (toggleHighlightsButton.getAttribute("src") == "/extension/ui/resources/button_highlighter_visible.png") {
			toggleHighlightsButton.src = "/extension/ui/resources/button_highlighter_hidden.png";
			editorElement.contentWindow.postMessage(JSON.stringify({ method: "hideHighlights" }), "*");
		} else {
			displayHighlights();
		}
	};
	removeHighlightButton.onmouseup = () => {
		if (toolbarElement.classList.contains("cut-inner-mode")) {
			disableCutInnerPage();
		}
		if (toolbarElement.classList.contains("cut-outer-mode")) {
			disableCutOuterPage();
		}
		if (removeHighlightButton.classList.contains("remove-highlight-disabled")) {
			removeHighlightButton.classList.remove("remove-highlight-disabled");
			toolbarElement.classList.add("remove-highlight-mode");
			resetHighlightButtons();
			displayHighlights();
			editorElement.contentWindow.postMessage(JSON.stringify({ method: "enableRemoveHighlights" }), "*");
			editorElement.contentWindow.postMessage(JSON.stringify({ method: "displayHighlights" }), "*");
		} else {
			disableRemoveHighlights();
		}
	};
	editPageButton.onmouseup = () => {
		if (toolbarElement.classList.contains("cut-inner-mode")) {
			disableCutInnerPage();
		}
		if (toolbarElement.classList.contains("cut-outer-mode")) {
			disableCutOuterPage();
		}
		if (editPageButton.classList.contains("edit-disabled")) {
			enableEditPage();
		} else {
			disableEditPage();
		}
	};
	formatPageButton.onmouseup = () => {
		if (formatPageButton.classList.contains("format-disabled")) {
			formatPage();
		} else {
			cancelFormatPage();
		}
	};
	cutInnerPageButton.onmouseup = () => {
		if (toolbarElement.classList.contains("edit-mode")) {
			disableEditPage();
		}
		if (toolbarElement.classList.contains("cut-outer-mode")) {
			disableCutOuterPage();
		}
		if (cutInnerPageButton.classList.contains("cut-disabled")) {
			enableCutInnerPage();

		} else {
			disableCutInnerPage();
		}
	};
	cutOuterPageButton.onmouseup = () => {
		if (toolbarElement.classList.contains("edit-mode")) {
			disableEditPage();
		}
		if (toolbarElement.classList.contains("cut-inner-mode")) {
			disableCutInnerPage();
		}
		if (cutOuterPageButton.classList.contains("cut-disabled")) {
			enableCutOuterPage();
		} else {
			disableCutOuterPage();
		}
	};
	undoCutPageButton.onmouseup = () => {
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "undoCutPage" }), "*");
	};
	undoAllCutPageButton.onmouseup = () => {
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "undoAllCutPage" }), "*");
	};
	redoCutPageButton.onmouseup = () => {
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "redoCutPage" }), "*");
	};
	savePageButton.onmouseup = () => {
		savePage();
	};
	if (window.print) {
		printPageButton.onmouseup = () => {
			editorElement.contentWindow.postMessage(JSON.stringify({ method: "printPage" }), "*");
		};
	} else {
		printPageButton.remove();
	}

	let toolbarPositionPointer, toolbarMoving, toolbarTranslateMax;
	let orientationPortrait = matchMedia("(orientation: portrait)").matches;
	let toolbarTranslate = 0;
	toolbarElement.ondragstart = event => event.preventDefault();
	toolbarElement.ontouchstart = toolbarOnTouchStart;
	toolbarElement.onmousedown = toolbarOnTouchStart;
	toolbarElement.ontouchmove = toolbarOnTouchMove;
	toolbarElement.ontouchend = toolbarOnTouchEnd;

	function viewportSizeChange() {
		orientationPortrait = matchMedia("(orientation: portrait)").matches;
		toolbarElement.style.setProperty("transform", orientationPortrait ? `translate(0, ${toolbarTranslate}px)` : `translate(${toolbarTranslate}px, 0)`);
	}

	function toolbarOnTouchStart(event) {
		const position = getPosition(event);
		toolbarPositionPointer = (orientationPortrait ? position.screenY : position.screenX) - toolbarTranslate;
		toolbarTranslateMax = ((orientationPortrait ? -lastButton.getBoundingClientRect().top + toolbarTranslate + window.innerHeight : -lastButton.getBoundingClientRect().left + toolbarTranslate + window.innerWidth)) - 35;
	}

	function toolbarOnTouchMove(event) {
		if (toolbarPositionPointer != null) {
			const position = getPosition(event);
			toolbarTranslate = Math.min(Math.max((orientationPortrait ? position.screenY : position.screenX) - toolbarPositionPointer, toolbarTranslateMax), 0);
			if (toolbarTranslate) {
				toolbarMoving = true;
				toolbarElement.style.setProperty("transform", orientationPortrait ? `translate(0, ${toolbarTranslate}px)` : `translate(${toolbarTranslate}px, 0)`);
				editorElement.style.setProperty("pointer-events", "none");
				event.preventDefault();
			}
		}
	}

	function toolbarOnTouchEnd(event) {
		if (toolbarMoving) {
			editorElement.style.removeProperty("pointer-events");
			event.preventDefault();
			event.stopPropagation();
		}
		toolbarPositionPointer = null;
		toolbarMoving = false;
	}

	let updatedResources = {};

	window.onresize = viewportSizeChange;
	window.onmessage = event => {
		const message = JSON.parse(event.data);
		if (message.method == "setMetadata") {
			document.title = "[SingleFile] " + message.title;
			if (message.filename) {
				tabData.filename = message.filename;
			}
			if (message.icon) {
				const linkElement = document.createElement("link");
				linkElement.rel = "icon";
				linkElement.href = message.icon;
				document.head.appendChild(linkElement);
			}
		}
		if (message.method == "setContent") {
			const pageData = {
				content: message.content,
				filename: tabData.filename
			};
			tabData.options.openEditor = false;
			singlefile.extension.core.content.download.downloadPage(pageData, tabData.options);
		}
		if (message.method == "disableFormatPage") {
			tabData.options.disableFormatPage = true;
			formatPageButton.remove();
		}
		if (message.method == "onUpdate") {
			tabData.docSaved = message.saved;
		}
		if (message.method == "onInit") {
			const defaultEditorMode = tabData.options.defaultEditorMode;
			if (defaultEditorMode == "edit") {
				enableEditPage();
			} else if (defaultEditorMode == "format" && !tabData.options.disableFormatPage) {
				formatPage();
			} else if (defaultEditorMode == "cut") {
				enableCutInnerPage();
			} else if (defaultEditorMode == "cut-external") {
				enableCutOuterPage();
			}
		}
		if (message.method == "savePage") {
			savePage();
		}
	};

	window.onload = () => {
		browser.runtime.sendMessage({ method: "editor.getTabData" });
		browser.runtime.onMessage.addListener(message => {
			if (message.method == "devtools.resourceCommitted") {
				updatedResources[message.url] = { content: message.content, type: message.type, encoding: message.encoding };
				return Promise.resolve({});
			}
			if (message.method == "content.save") {
				tabData.options = message.options;
				savePage();
				browser.runtime.sendMessage({ method: "ui.processInit" });
				return Promise.resolve({});
			}
			if (message.method == "common.promptValueRequest") {
				browser.runtime.sendMessage({ method: "tabs.promptValueResponse", value: prompt(message.promptMessage) });
				return Promise.resolve({});
			}
			if (message.method == "editor.setTabData") {
				if (message.truncated) {
					tabDataContents.push(message.content);
				} else {
					tabDataContents = [message.content];
				}
				if (!message.truncated || message.finished) {
					tabData = JSON.parse(tabDataContents.join(""));
					tabData.docSaved = true;
					tabDataContents = [];
					editorElement.contentWindow.postMessage(JSON.stringify({ method: "init", content: tabData.content }), "*");
					editorElement.contentWindow.focus();
					delete tabData.content;
				}
				return Promise.resolve({});
			}
		});
	};

	window.onbeforeunload = event => {
		if (tabData.options.warnUnsavedPage && !tabData.docSaved) {
			event.preventDefault();
			event.returnValue = "";
		}
	};

	function disableEditPage() {
		editPageButton.classList.add("edit-disabled");
		toolbarElement.classList.remove("edit-mode");
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "disableEditPage" }), "*");
	}

	function disableCutInnerPage() {
		cutInnerPageButton.classList.add("cut-disabled");
		toolbarElement.classList.remove("cut-inner-mode");
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "disableCutInnerPage" }), "*");
	}

	function disableCutOuterPage() {
		cutOuterPageButton.classList.add("cut-disabled");
		toolbarElement.classList.remove("cut-outer-mode");
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "disableCutOuterPage" }), "*");
	}

	function resetHighlightButtons() {
		highlightButtons.forEach(highlightButton => highlightButton.classList.add("highlight-disabled"));
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "disableHighlight" }), "*");
	}

	function disableRemoveHighlights() {
		toolbarElement.classList.remove("remove-highlight-mode");
		removeHighlightButton.classList.add("remove-highlight-disabled");
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "disableRemoveHighlights" }), "*");
	}

	function displayHighlights() {
		toggleHighlightsButton.src = "/extension/ui/resources/button_highlighter_visible.png";
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "displayHighlights" }), "*");
	}

	function enableEditPage() {
		editPageButton.classList.remove("edit-disabled");
		toolbarElement.classList.add("edit-mode");
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "enableEditPage" }), "*");
	}

	function formatPage() {
		formatPageButton.classList.remove("format-disabled");
		updatedResources = {};
		editorElement.contentWindow.postMessage(JSON.stringify({ method: tabData.options.applySystemTheme ? "formatPage" : "formatPageNoTheme" }), "*");
	}

	function cancelFormatPage() {
		formatPageButton.classList.add("format-disabled");
		updatedResources = {};
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "cancelFormatPage" }), "*");
	}

	function enableCutInnerPage() {
		cutInnerPageButton.classList.remove("cut-disabled");
		toolbarElement.classList.add("cut-inner-mode");
		resetHighlightButtons();
		disableRemoveHighlights();
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "enableCutInnerPage" }), "*");
	}

	function enableCutOuterPage() {
		cutOuterPageButton.classList.remove("cut-disabled");
		toolbarElement.classList.add("cut-outer-mode");
		resetHighlightButtons();
		disableRemoveHighlights();
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "enableCutOuterPage" }), "*");
	}

	function savePage() {
		editorElement.contentWindow.postMessage(JSON.stringify({ method: "getContent", compressHTML: tabData.options.compressHTML, updatedResources }), "*");
	}

	function getPosition(event) {
		if (event.touches && event.touches.length) {
			const touch = event.touches[0];
			return touch;
		} else {
			return event;
		}
	}

	return {};

})();