SingleFile/WebContent/core/scripts/bg/bgcore.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() {

	singlefile.PageData = PageData;
	singlefile.DocData = DocData;

	function PageData(tabId, pageId, senderId, config, processSelection, callback) {
		var timeoutError, that = this;
		this.pageId = pageId;
		this.docs = [];
		this.processedDocs = 0;
		this.initializedDocs = 0;
		this.processableDocs = 0;
		this.senderId = senderId;
		this.config = config;
		this.processSelection = processSelection;
		this.processing = true;
		this.tabId = tabId;
		this.requestManager = new singlefile.nio.RequestManager();
		this.progressIndex = 0;
		this.progressMax = 0;
		this.title = null;
		this.url = null;
		this.top = null;
		this.timeoutPageInit = null;
		this.portsId = [];
		timeoutError = setTimeout(function() {
			that.processing = false;
			chrome.extension.sendRequest(that.senderId, {
				processError : true,
				tabId : tabId
			});
		}, 15000);
		wininfo.init(tabId, function(processableDocs) {
			clearTimeout(timeoutError);
			that.processableDocs = processableDocs;
			callback();
		});
	}

	PageData.prototype = {
		initProcess : function() {
			var that = this;
			this.docs.forEach(function(docData) {
				if (that.config.processInBackground) {
					if (docData.processDocCallback)
						docData.processDocCallback();
				} else
					docData.process();
			});
		},
		processDoc : function(port, topWindow, winId, index, content, title, url, baseURI, characterSet, canvasData, callbacks) {
			var that = this, docData;
			docData = new DocData(port, winId, index, content, baseURI, characterSet, canvasData);
			if (topWindow) {
				this.top = docData;
				this.title = title || "";
				this.url = url;
			}
			this.docs.push(docData);
			if (this.config.processInBackground && docData.content) {
				docData.parseContent();
				docData.processDocCallback = singlefile.initProcess(docData.doc, docData.doc.documentElement, topWindow, baseURI, characterSet, this.config,
						canvasData, this.requestManager, function(maxIndex) {
							callbacks.init(that, docData, maxIndex);
						}, function(index, maxIndex) {
							callbacks.progress(that, docData, index);
						}, function() {
							callbacks.end(that, docData);
						});
			}
		},
		processDocFragment : function(docData, mutationEventId, content) {
			var doc = document.implementation.createHTMLDocument();
			doc.body.innerHTML = content;
			docData.processDocCallback = singlefile.initProcess(doc, doc.documentElement, this.top == docData, docData.baseURI, docData.characterSet,
					this.config, null, this.requestManager, function() {
						docData.processDocCallback();
					}, null, function() {
						docData.setDocFragment(doc.body.innerHTML, mutationEventId);
					});
		},
		setDocContent : function(docData, content, callback) {
			var that = this;

			function buildPage(docData, setFrameContent, getContent, callback) {
				function setContent(docData) {
					var parent = docData.parent;
					if (parent)
						setFrameContent(docData, function() {
							parent.processedChildren++;
							if (parent.processedChildren == parent.childrenLength)
								getContent(parent, function() {
									setContent(parent);
								});
						});
					else if (callback)
						callback(docData);
				}

				if (docData.childrenLength)
					docData.children.forEach(function(data) {
						buildPage(data, setFrameContent, getContent, callback);
					});
				else
					setContent(docData);
			}

			function bgPageEnd(pageData, docData, callback) {
				var content = singlefile.util.getDocContent(docData.doc);
				if (pageData.config.displayProcessedPage)
					pageData.top.setContent(content);
				else
					callback(pageData.tabId, pageData.pageId, pageData.top, content);
			}

			if (content)
				docData.content = content;
			this.processedDocs++;
			if (this.processSelection)
				if (this.config.processInBackground)
					bgPageEnd(this, docData, callback);
				else
					docData.getContent(function() {
						that.top.setContent(docData.content);
					});
			else if (this.processedDocs == this.docs.length) {
				this.docs.forEach(function(docData) {
					var parentWinId = docData.winId.match(/((?:\d*\.?)*)\.\d*/);
					parentWinId = parentWinId ? parentWinId[1] : null;
					if (parentWinId)
						that.docs.forEach(function(data) {
							if (data.winId && data.winId == parentWinId)
								docData.parent = data;
						});
					if (docData.parent)
						docData.parent.setChild(docData);
				});
				if (this.config.processInBackground)
					buildPage(this.top, function(docData, callback) {
						docData.parent.docFrames[docData.index].setAttribute("src", "data:text/html;charset=utf-8,"
								+ encodeURI(singlefile.util.getDocContent(docData.doc)));
						delete docData.doc;
						callback();
					}, function(docData, callback) {
						callback();
					}, function(docData) {
						bgPageEnd(that, docData, callback);
					});
				else
					buildPage(this.top, function(docData, callback) {
						docData.parent.setFrameContent(docData, callback);
					}, function(docData, callback) {
						docData.getContent(callback);
					}, function(docData) {
						docData.setContent();
					});
			}
		},
		computeProgress : function() {
			var that = this;
			this.progressIndex = 0;
			this.progressMax = 0;
			this.docs.forEach(function(docData) {
				that.progressIndex += docData.progressIndex || 0;
				that.progressMax += docData.progressMax || 0;
			});
		},
		endProcess : function(content) {
			var that = this;
			if (this.config.savePage)
				singlefile.storage.addContent(this.title ? this.title.replace(/[\\\/:\*\?\"><|]/gi, "").trim() || "Unnamed page" : "Unnamed page", content
						.replace(/<meta[^>]*http-equiv\s*=\s*["']?content-type[^>]*>/gi, "").replace(/<meta[^>]*charset\s*=[^>]*>/gi, ""),
						this.config.filenameMaxLength, function(processed, filename) {
							chrome.extension.sendRequest(that.senderId, {
								pageSaved : true,
								tabId : that.tabId,
								pageId : that.pageId,
								processed : processed,
								filename : filename
							});
						});
		},
		getResourceContentRequest : function(url, requestId, winId, characterSet, mediaTypeParam, docData) {
			this.requestManager.send(url, function(content) {
				docData.getResourceContentResponse(content, requestId);
			}, characterSet, mediaTypeParam);
		},
		getDocData : function(winId) {
			var found;
			this.docs.forEach(function(docData) {
				if (docData.winId == winId)
					found = docData;
			});
			return found;
		}
	};

	function DocData(port, winId, index, content, baseURI, characterSet, canvasData) {
		this.port = port;
		this.content = content;
		this.baseURI = baseURI;
		this.characterSet = characterSet;
		this.canvasData = canvasData;
		this.winId = winId;
		this.index = index;
		this.children = [];
		this.doc = null;
		this.docFrames = null;
		this.processDocCallback = null;
		this.getContentCallback = null;
		this.setFrameContentCallback = null;
		this.processedChildren = 0;
		this.childrenLength = 0;
	}

	DocData.prototype = {
		parseContent : function() {
			var doc = document.implementation.createHTMLDocument();
			doc.open();
			doc.write(this.content);
			doc.close();
			this.doc = doc;
			this.docFrames = doc.querySelectorAll("iframe, frame");
			delete this.content;
		},
		setChild : function(childDoc) {
			this.children[childDoc.index] = childDoc;
			this.childrenLength++;
		},
		process : function() {
			this.port.postMessage({
				processDoc : true,
				winId : this.winId
			});
		},
		setDocFragment : function(content, mutationEventId) {
			this.port.postMessage({
				setDocFragment : true,
				content : content,
				mutationEventId : mutationEventId
			});
		},
		getResourceContentResponse : function(content, requestId) {
			this.port.postMessage({
				getResourceContentResponse : true,
				requestId : requestId,
				winId : this.winId,
				content : content
			});
		},
		setContent : function(content) {
			this.port.postMessage({
				setContentRequest : true,
				content : content,
				winProperties : singlefile.winProperties
			});
		},
		getContent : function(callback) {
			this.getContentCallback = callback;
			this.port.postMessage({
				getContentRequest : true,
				winId : this.winId
			});
		},
		setFrameContent : function(docData, callback) {
			docData.setFrameContentCallback = callback;
			this.port.postMessage({
				setFrameContentRequest : true,
				winId : this.winId,
				index : docData.index,
				content : docData.content
			});
		}
	};

	(function() {
		var property, winProperties = {};
		for (property in window)
			winProperties[property] = true;
		singlefile.winProperties = winProperties;
	})();

})();