SingleFile/WebContent/core/scripts/bg/background.js

/*
 * Copyright 2011 Gildas Lormeau
 * contact : gildas.lormeau <at> gmail.com
 * 
 * This file is part of SingleFile Core.
 *
 *   SingleFile Core 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 Core 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 Core.  If not, see <http://www.gnu.org/licenses/>.
 */

(function() {

	var DEFAULT_CONFIG = {
		removeFrames : false,
		removeScripts : true,
		removeObjects : true,
		removeHidden : false,
		removeUnusedCSSRules : false,
		displayProcessedPage : false,
		savePage : false,
		filenameMaxLength : 90,
		processInBackground : true,
		getContent : true,
		getRawDoc : false
	};

	var tabs = singlefile.tabs = [], processingPagesCount = 0, pageId = 0;

	function executeScripts(tabId, scripts, callback, index) {
		if (!index)
			index = 0;
		if (index < scripts.length)
			chrome.tabs.executeScript(tabId, {
				file : scripts[index].file,
				code : scripts[index].code,
				allFrames : true
			}, function() {
				executeScripts(tabId, scripts, callback, index + 1);
			});
		else if (callback)
			callback();
	}

	function processInit(tabId, port, message) {
		var pageData = tabs[tabId][message.pageId];
		pageData.portsId.push(port.portId_);
		if (!pageData.getDocData(message.winId))
			pageData.processDoc(port, message.topWindow, message.winId, message.index, message.content, message.title, message.url, message.baseURI,
					message.characterSet, message.canvasData, {
						init : docInit,
						progress : docProgress,
						end : docEnd
					});
	}

	function setContentResponse(tabId, pageId, docData, content) {
		var pageData = tabs[tabId][pageId];
		pageData.endProcess(content);
		processingPagesCount--;
		chrome.extension.sendRequest(pageData.senderId, {
			processEnd : true,
			tabId : tabId,
			pageId : pageId,
			blockingProcess : !pageData.config.processInBackground || pageData.config.displayProcessedPage,
			processingPagesCount : processingPagesCount,
			content : pageData.config.getContent ? content : null,
			url : pageData.url,
			title : pageData.title
		});
		if (!pageData.config.processInBackground || pageData.config.displayProcessedPage) {
			pageData.processing = false;
			tabs[tabId].processing = false;
		}
		if (pageData.pendingDelete)
			deletePageData(pageData);
	}

	function docInit(pageData, docData, maxIndex) {
		function pageInit() {
			pageData.timeoutPageInit = null;
			pageData.processableDocs = pageData.initializedDocs;
			pageData.initProcess();
			processingPagesCount++;
			chrome.extension.sendRequest(pageData.senderId, {
				processStart : true,
				tabId : pageData.tabId,
				pageId : pageData.pageId,
				blockingProcess : !pageData.config.processInBackground || pageData.config.displayProcessedPage,
				processingPagesCount : processingPagesCount
			});
			if (pageData.config.processInBackground && !pageData.config.displayProcessedPage)
				tabs[pageData.tabId].processing = false;
		}

		if (!docData.initialized) {
			docData.initialized = true;
			if (pageData.initializedDocs != pageData.processableDocs) {
				docData.progressMax = maxIndex;
				pageData.initializedDocs++;
				if (pageData.timeoutPageInit)
					clearTimeout(pageData.timeoutPageInit);
				pageData.timeoutPageInit = setTimeout(pageInit, 5000);
				if (pageData.initializedDocs == pageData.processableDocs || pageData.processSelection || pageData.config.removeFrames
						|| pageData.config.getRawDoc) {
					clearTimeout(pageData.timeoutPageInit);
					pageInit();
				}
			}
		}
	}

	function docProgress(pageData, docData, index) {
		var progressIndex = 0, progressMax = 0;
		docData.progressIndex = index;
		tabs.forEach(function(tabData) {
			if (tabData) {
				tabData.progressIndex = 0;
				tabData.progressMax = 0;
				tabData.forEach(function(pageData) {
					if (pageData) {
						pageData.computeProgress();
						tabData.progressIndex += pageData.progressIndex;
						tabData.progressMax += pageData.progressMax;
					}
				});
				progressIndex += tabData.progressIndex;
				progressMax += tabData.progressMax;
			}
		});
		chrome.extension.sendRequest(pageData.senderId, {
			processProgress : true,
			tabId : pageData.tabId,
			pageId : pageData.pageId,
			pageIndex : pageData.progressIndex,
			pageMaxIndex : pageData.progressMax,
			tabIndex : tabs[pageData.tabId].progressIndex,
			tabMaxIndex : tabs[pageData.tabId].progressMax,
			index : progressIndex,
			maxIndex : progressMax
		});
	}

	function docEnd(pageData, docData, content) {
		pageData.setDocContent(docData, content, setContentResponse);
	}

	function process(tabId, senderId, config, processSelection) {
		var pageData, configScript = "singlefile.config = " + JSON.stringify(config) + "; singlefile.pageId = " + pageId + ";"
				+ (processSelection ? "singlefile.processSelection = " + processSelection : "");
		if (tabs[tabId] && tabs[tabId].processing)
			return;
		tabs[tabId] = tabs[tabId] || [];
		tabs[tabId].processing = true;
		pageData = new singlefile.PageData(tabId, pageId, senderId, config, processSelection, function() {
			executeScripts(tabId, [ {
				code : "var singlefile = {};"
			}, {
				file : "scripts/common/util.js"
			}, {
				file : "scripts/common/docprocessor.js"
			}, {
				code : configScript
			}, {
				file : "scripts/content/content.js"
			} ]);
		});
		tabs[tabId][pageId] = pageData;
		pageId++;
	}

	function deletePageData(pageData) {
		tabs[pageData.tabId][pageData.pageId] = null;
		tabs[pageData.tabId] = tabs[pageData.tabId].filter(function(pageData) {
			return pageData;
		});
		if (!tabs[pageData.tabId].length)
			tabs[pageData.tabId] = null;
	}

	chrome.extension.onConnect
			.addListener(function(port) {
				var tabId = port.sender.tab.id, portPageId = [];

				function onDisconnect() {
					var pageData = tabs[tabId][portPageId[port.portId_]];
					if (!pageData)
						return;
					pageData.portsId = pageData.portsId.filter(function(id) {
						return id != port.portId_;
					});
					if (!pageData.portsId.length)
						if (pageData.processing)
							pageData.pendingDelete = true;
						else
							deletePageData(pageData);
				}

				function onMessage(message) {
					var pageData, docData;
					// if (!message.getResourceContentRequest && !message.docProgress)
					// console.log("onMessage", message, port.portId_);
					if (message.winId) {
						portPageId[port.portId_] = message.pageId;
						if (message.processInit)
							processInit(tabId, port, message);
						else {
							pageData = tabs[tabId][message.pageId];
							docData = pageData.getDocData(message.winId);
							if (message.processDocFragment)
								pageData.processDocFragment(docData, message.mutationEventId, message.content);
							if (message.getResourceContentRequest)
								pageData.getResourceContentRequest(message.url, message.requestId, message.winId, message.characterSet, message.mediaTypeParam,
										docData);
							if (message.docInit)
								docInit(pageData, docData, message.maxIndex);
							if (message.docProgress)
								docProgress(pageData, docData, message.index);
							if (message.docEnd)
								docEnd(pageData, docData, message.content);
							if (message.setFrameContentResponse)
								docData.children[message.index].setFrameContentCallback();
							if (message.getContentResponse) {
								docData.content = message.content;
								docData.getContentCallback();
							}
							if (message.setContentResponse)
								setContentResponse(tabId, message.pageId, docData, message.content);
						}
					}
				}

				if (port.name == "singlefile") {
					port.onMessage.addListener(onMessage);
					port.onDisconnect.addListener(onDisconnect);
				}
			});

	chrome.extension.onRequestExternal.addListener(function(request, sender, sendResponse) {
		// console.log("onRequestExternal", request);
		var property, config = JSON.parse(JSON.stringify(DEFAULT_CONFIG));
		if (request.config)
			for (property in request.config)
				config[property] = request.config[property];
		if (request.processSelection)
			process(request.id, sender.id, config, true);
		else if (request.tabIds)
			request.tabIds.forEach(function(tabId) {
				process(tabId, sender.id, config);
			});
		else
			process(request.id, sender.id, config);
		sendResponse({});
	});

})();