bootstrap.js

var Cc = Components.classes, Ci = Components.interfaces, Cu = Components.utils;

var SSTART_MODULES = [
    "chrome://sstart/content/prefloader.js",
    "chrome://sstart/content/cache.js",
    "chrome://sstart/content/file.js",
    "chrome://sstart/content/url.js",
    "chrome://sstart/content/dom.js",
    "chrome://sstart/content/utils.js",
    "chrome://sstart/content/bookmark.js"
];

Cu.import("resource://gre/modules/Services.jsm");

var gWindowListener = null, linkToSStart, pageToSStart;
var sstartTabURI = "chrome://sstart/content/sstart.html";
var styleSheetService = Cc["@mozilla.org/content/style-sheet-service;1"].getService(Ci.nsIStyleSheetService);
var hideSStart = Services.io.newURI("chrome://sstart/skin/hidesstart.css", null, null);

function BrowserWindowObserver(handlers) {
	this.handlers = handlers;
}

BrowserWindowObserver.prototype = {
	observe: function (aSubject, aTopic, aData) {
		if (aTopic == "domwindowopened") {
			aSubject.QueryInterface(Ci.nsIDOMWindow).addEventListener("DOMContentLoaded", this, false);
		} else if (aTopic == "domwindowclosed") {
			if (aSubject.document.documentElement.getAttribute("windowtype") == "navigator:browser") {
				this.handlers.onShutdown(aSubject);
			}
		}
	},
	handleEvent: function (aEvent) {
		let aWindow = aEvent.currentTarget;
		aWindow.removeEventListener(aEvent.type, this, false);

		if (aWindow.document.documentElement.getAttribute("windowtype") == "navigator:browser") {
			this.handlers.onStartup(aWindow);
		}
	}
};

function browserWindowStartup (aWindow) {
	var window = aWindow.document.getElementById("main-window");
	var hbox = aWindow.document.createElement("hbox");
	hbox.id = "sstart-container";
	hbox.setAttribute("style", "overflow:hidden; height: 0;");
	var vbox = aWindow.document.createElement("vbox");
	vbox.id = "sstart-hidden-box";
	hbox.appendChild(vbox);
	window.appendChild(hbox);
	var cmenu = aWindow.document.getElementById("contentAreaContextMenu");
	var smenu = aWindow.document.createElement("menu");
	smenu.id = "sstart-add-page-menu";
	smenu.className = "menu-iconic";
	smenu.setAttribute("image", "chrome://sstart/skin/icon.png");
	var smenupopup = aWindow.document.createElement("menupopup");
	smenu.appendChild(smenupopup);
	smenu.addEventListener("click", addPage, false);
	var csp = aWindow.document.getElementById("context-savepage");
	cmenu.insertBefore(smenu, csp);
	var smitem = aWindow.document.createElement("menuitem");
	smitem.id = "sstart-add-page";
	smitem.className = "menuitem-iconic";
	smitem.setAttribute("image", "chrome://sstart/skin/icon.png");
	smitem.addEventListener("click", addPage, false);
	cmenu.insertBefore(smitem, csp);
	cmenu.addEventListener("popupshowing", cPopupShowingListener, false);
	if (Utils.isSeaMonkey) {
		aWindow.gBrowser || aWindow.getBrowser();
		aWindow.gBrowserOrigAddTab = aWindow.gBrowser.addTab;
		aWindow.gBrowser.addTab = function () {
			if (arguments[0] && arguments[0] == "about:blank" && Services.prefs.getBoolPref("extensions.sstart.overrideNewTab")) {
				arguments[0] = sstartTabURI;
			}
			return aWindow.gBrowserOrigAddTab.apply(aWindow.gBrowser, arguments);
		}
	}
	if (Object.prototype.toString.call(aWindow.gInitialPages).slice(8, -1) == "Array"
		&& aWindow.gInitialPages.indexOf(sstartTabURI) == -1) {
		aWindow.gInitialPages.push(sstartTabURI);
	} else if (Object.prototype.toString.call(aWindow.gInitialPages).slice(8, -1) == "Set"
		&& !aWindow.gInitialPages.has(sstartTabURI)) {
		aWindow.gInitialPages.add(sstartTabURI);
	}
}

function browserWindowShutdown (aWindow) {
	if (Object.prototype.toString.call(aWindow.gInitialPages).slice(8, -1) == "Array"
		&& aWindow.gInitialPages.indexOf(sstartTabURI) > -1) {
		aWindow.gInitialPages.splice(aWindow.gInitialPages.indexOf(sstartTabURI), 1);
	} else if (Object.prototype.toString.call(aWindow.gInitialPages).slice(8, -1) == "Set"
		&& aWindow.gInitialPages.has(sstartTabURI)) {
		aWindow.gInitialPages.delete(sstartTabURI);
	}
	if (Utils.isSeaMonkey) {
		aWindow.gBrowser.addTab = aWindow.gBrowserOrigAddTab;
		aWindow.gBrowserOrigAddTab = null;
	}
	var cmenu = aWindow.document.getElementById("contentAreaContextMenu");
	cmenu.removeEventListener("popupshowing", cPopupShowingListener);
	var smenu = aWindow.document.getElementById("sstart-add-page-menu");
	smenu.removeEventListener("click", addPage);
	var smitem = aWindow.document.getElementById("sstart-add-page");
	smitem.removeEventListener("click", addPage);
	cmenu.removeChild(smenu);
	cmenu.removeChild(smitem);
	var window = aWindow.document.getElementById("main-window");
	var hbox = aWindow.document.getElementById("sstart-container");
	window.removeChild(hbox);
}

function cPopupShowingListener (e) {
	var menu = Utils.getBrowserWindow().document.getElementById("sstart-add-page-menu");
	var menuitem = Utils.getBrowserWindow().document.getElementById("sstart-add-page");
	var gcmenu = Utils.getBrowserWindow().gContextMenu;
	if (gcmenu.isTextSelected || gcmenu.onTextInput) {
		menuitem.hidden = true;
		menu.hidden = true;
		Cache.setUpdateMenu(true);
		return;
	}
	var prefService = Cc["@mozilla.org/preferences-service;1"].getService(Ci.nsIPrefBranch);
	if (prefService.getBoolPref("extensions.sstart.hideContextMenu")) {
		menuitem.hidden = true;
		menu.hidden = true;
		return;
	}
	if (gcmenu.linkURL) {
		menu.setAttribute("data-url", gcmenu.linkURL);
		if (gcmenu.link) {
			menu.setAttribute("data-title", gcmenu.link.textContent.trim());
		} else {
			menu.setAttribute("data-title", gcmenu.linkURL);
		}
		menu.setAttribute("label", linkToSStart);
		menuitem.setAttribute("label", linkToSStart);
	} else {
		menu.setAttribute("data-url", Utils.getBrowserWindow().content.location.href);
		menu.setAttribute("data-title", Utils.getBrowserWindow().content.document.title);
		menu.setAttribute("label", pageToSStart);
		menuitem.setAttribute("label", pageToSStart);
	}
	if (!Cache.isUpdateMenu() && menu.firstChild.hasChildNodes()) {
		return;
	}
	menuitem.hidden = initFoldersMenu(menu.firstChild);
	menu.hidden = !menuitem.hidden;
	Cache.setUpdateMenu(false);
}
	
function initFoldersMenu (menupopup) {
	while (menupopup.firstChild) {
		menupopup.removeChild(menupopup.firstChild);
	}
	var rootId = 0;
	var bookmarks = Bookmark.getBookmarks();
	for (var i in bookmarks) {
		if (bookmarks[i].isFolder && bookmarks[i].title == "SStart") {
			rootId = bookmarks[i].id;
			break;
		}
	}
	if (rootId > 0) {
		menupopup.parentNode.setAttribute("data-fid", rootId);
		Utils.getBrowserWindow().document.getElementById("sstart-add-page").setAttribute("data-fid", rootId);
		createFoldersMenu(rootId, menupopup);
	}
	return menupopup.childNodes.length > 0;
};

function createFoldersMenu (folderId, menupopup) {
	var bookmarks = Bookmark.getBookmarks(folderId);
	for (var i in bookmarks) {
		var bookmark = bookmarks[i];
		if (!bookmark.isFolder) continue;
		var menuitem;
		var submenu = Utils.getBrowserWindow().document.createElement("menupopup");
		submenu.addEventListener("popupshowing", function(e) { e.stopPropagation(); }, false);
		createFoldersMenu(bookmark.id, submenu);
		if (submenu.childNodes.length > 0) {
			menuitem = Utils.getBrowserWindow().document.createElement("menu");
			menuitem.appendChild(submenu);
		} else {
			menuitem = Utils.getBrowserWindow().document.createElement("menuitem");
		}
		menuitem.setAttribute("label", bookmark.title);
		menuitem.setAttribute("data-fid", bookmark.id);
		menupopup.appendChild(menuitem);
	}
};

function addPage (e) {
	Utils.getBrowserWindow().document.getElementById("contentAreaContextMenu").hidePopup();
	var folderId = e.target.getAttribute("data-fid") || 0;
	if (folderId > 0) {
		var data = Utils.getBrowserWindow().document.getElementById("sstart-add-page-menu");
		var newId = Bookmark.createBookmark(data.getAttribute("data-url"), data.getAttribute("data-title"), folderId);
		var cont = Utils.getBrowserWindow().content;
		var width = Services.prefs.getIntPref("extensions.sstart.thumbnail.width");
		var height = Services.prefs.getIntPref("extensions.sstart.thumbnail.height");
		var left = Cache.alignToGrid((cont.innerWidth - width) / 2);
		var top = Cache.alignToGrid((cont.innerHeight - height) / 2);
		Bookmark.setAnnotation(newId, "bookmarkProperties/description", 
			'{"left":' + left + ',"top":' + top + ',"width":' + width + ',"height":' + height +'}');
		File.delDataFile(newId);
		if (folderId == data.getAttribute("data-fid")) {
			Cache.clearCache();
			var ssurl = sstartTabURI;
		} else {
			var ssurl = sstartTabURI + "?folder=" + folderId;
		}
		Cache.setEditOn();
		Utils.getBrowser().loadOneTab(ssurl, {inBackground: false, relatedToCurrent: true});
	}
};

var newtabAPI = {
	init: function () {
		try {
			Cc["@mozilla.org/browser/aboutnewtab-service;1"].getService(Ci.nsIAboutNewTabService);
			this.type = 1; //FF44+
		} catch (e) {
			try {
				Cu.import("resource:///modules/NewTabURL.jsm");
				this.type = 2; //FF41+
			} catch (e) {
				this.type = 0;
			}
		}
	},
	set: function () {
		switch (this.type) {
			case 1:
				var aboutNewTabService = Cc["@mozilla.org/browser/aboutnewtab-service;1"].getService(Ci.nsIAboutNewTabService);
				aboutNewTabService.newTabURL = sstartTabURI;
				break;
			case 2:
				NewTabURL.override(sstartTabURI);
				break;
		}
	},
	reset: function () {
		switch (this.type) {
			case 1:
				var aboutNewTabService = Cc["@mozilla.org/browser/aboutnewtab-service;1"].getService(Ci.nsIAboutNewTabService);
				if (aboutNewTabService.newTabURL == sstartTabURI) {
					aboutNewTabService.resetNewTabURL();
				}
				break;
			case 2:
				if (NewTabURL.get() == sstartTabURI) {
					NewTabURL.reset();
				}
				break;
		}
	}
}

function browserPref (pref, cmd) {
	var bprefs = Cc["@mozilla.org/preferences-service;1"].getService(Ci.nsIPrefService).getBranch("browser.");
	switch (cmd) {
		case "get":
			try {
				return  bprefs.getCharPref(pref);
			} catch (e) {
				return  "";
			}
			break;
		case "set":
			if (pref == "pagethumbnails") {
				try {
					bprefs.setBoolPref("pageThumbs.enabled", false);
					bprefs.setBoolPref("pagethumbnails.capturing_disabled", true);
				} catch (e) {}
				if (typeof PageThumbsStorage === "object" && typeof PageThumbsStorage.wipe === "function") {
					PageThumbsStorage.wipe();
				}
			} else {
				if (Utils.isFirefox && pref == "newtab.url") {
					newtabAPI.set();
				}
				try {
					bprefs.setCharPref(pref, sstartTabURI);
				} catch (e) {}
			}
			break;
		case "clear":
			if (pref == "pagethumbnails") {
				try {
					bprefs.clearUserPref("pageThumbs.enabled");
					bprefs.clearUserPref("pagethumbnails.capturing_disabled");
				} catch (e) {}
			} else {
				if (Utils.isFirefox && pref == "newtab.url") {
					newtabAPI.reset();
				}
				try {
					if (bprefs.getCharPref(pref) == sstartTabURI) {
						bprefs.clearUserPref(pref);
					}
				} catch (e) {}
			}
			break;
	}
}

var myPrefsWatcher = {
	observe: function (subject, topic, data) {
		if (topic != "nsPref:changed") return;
		switch (data) {
			case "bottomHeader":
			case "autoZoom":
			case "hideDecorations":
				Cache.clearCache();
				Utils.reloadEachSStartBrowser();
				break;
			case "overrideNewTab":
				if (Services.prefs.getBoolPref("extensions.sstart.overrideNewTab")) {
					browserPref("newtab.url", "set");
				} else {
					browserPref("newtab.url", "clear");
				}
				break;
			case "overrideHomePage":
				if (Services.prefs.getBoolPref("extensions.sstart.overrideHomePage")) {
					browserPref("startup.homepage", "set");
				} else {
					browserPref("startup.homepage", "clear");
				}
				break;
			case "disableSysThumbs":
				if (Services.prefs.getBoolPref("extensions.sstart.disableSysThumbs")) {
					browserPref("pagethumbnails", "set");
				} else {
					browserPref("pagethumbnails", "clear");
				}
				break;
			case "gridInterval":
				Cache.updateGridInterval(true);
				break;
			case "showGridOnUnlock":
				Cache.updateGridOnUnlock();
				break;
			case "hideContextMenu":
				Cache.setUpdateMenu(true);
				break;
			case "newtabOpen":
			case "newtabOnLockDrag":
				Cache.updateNewtabOpen();
				break;
			case "thumbnail.width":
				var twidth = Services.prefs.getIntPref("extensions.sstart.thumbnail.width");
				if (twidth < 20) {
					if (twidth <= 0) { twidth = 224; } else { twidth = 20; }
					Services.prefs.setIntPref("extensions.sstart.thumbnail.width", twidth);
				}
				break;
			case "thumbnail.height":
				var theight = Services.prefs.getIntPref("extensions.sstart.thumbnail.height");
				if (theight < 20) {
					if (theight <= 0) { theight = 128; } else { theight = 20; }
					Services.prefs.setIntPref("extensions.sstart.thumbnail.height", theight);
				}
				break;
		}
	},
	register: function () {
		var prefsService = Cc["@mozilla.org/preferences-service;1"].getService(Ci.nsIPrefService);
		this.prefBranch = prefsService.getBranch("extensions.sstart.");
		this.prefBranch.addObserver("", this, false);
	},
	unregister: function () {
		this.prefBranch.removeObserver("", this);
	}
}

var browserPrefsWatcher = {
	observe: function (subject, topic, data) {
		if (topic != "nsPref:changed") return;
		switch (data) {
			case "newtab.url":
				if (browserPref("newtab.url", "get") != sstartTabURI) {
					Services.prefs.setBoolPref("extensions.sstart.overrideNewTab", false);
				}
				break;
			case "startup.homepage":
				if (browserPref("startup.homepage", "get") != sstartTabURI) {
					Services.prefs.setBoolPref("extensions.sstart.overrideHomePage", false);
				}
				break;
		}
	},
	register: function () {
		var prefsService = Cc["@mozilla.org/preferences-service;1"].getService(Ci.nsIPrefService);
		this.prefBranch = prefsService.getBranch("browser.");
		this.prefBranch.addObserver("", this, false);
	},
	unregister: function () {
		this.prefBranch.removeObserver("", this);
	}
}

var newTabURLWatcher = {
	observe: function (subject, topic, data) {
		if (topic == "newtab-url-changed" && data != sstartTabURI) {
				Services.prefs.setBoolPref("extensions.sstart.overrideNewTab", false);
		}
	},
	register: function () {
		this.observerService = Cc["@mozilla.org/observer-service;1"].getService(Ci.nsIObserverService);
		this.observerService.addObserver(this, "newtab-url-changed", false)
	},
	unregister: function () {
		this.observerService.removeObserver(this, "newtab-url-changed");
	}
}

function install (params, reason)
{
}

function uninstall (params, reason)
{
}

function startup (params, reason)
{
	Cu.import("chrome://sstart/content/prefloader.js");
	Cu.import("chrome://sstart/content/cache.js");
    Cu.import("chrome://sstart/content/file.js");
    Cu.import("chrome://sstart/content/utils.js");
    Cu.import("chrome://sstart/content/bookmark.js");
	
	PrefLoader.loadDefaultPrefs(params.installPath, "sstart.js");

	try {
		Cu.import("resource://gre/modules/PageThumbs.jsm");
	} catch (e) {}
	
	if (!styleSheetService.sheetRegistered(hideSStart, styleSheetService.USER_SHEET)) {
		styleSheetService.loadAndRegisterSheet(hideSStart, styleSheetService.USER_SHEET);
	}

	var appInfo = Cc["@mozilla.org/xre/app-info;1"].getService(Ci.nsIXULAppInfo);
	Utils.isFirefox = (appInfo.ID == "{ec8030f7-c20a-464f-9b0e-13a3a9e97384}");
	Utils.isSeaMonkey = (appInfo.ID == "{92650c4d-4b8e-4d2a-b7eb-24ecf4f6b63a}" || appInfo.ID == "{a3210b97-8e8a-4737-9aa0-aa0e607640b9}");

	if (Utils.isFirefox) {
		newtabAPI.init();
	}

	if (Services.prefs.getBoolPref("extensions.sstart.overrideNewTab")) {
		browserPref("newtab.url", "set");
	}

	if (Services.prefs.getBoolPref("extensions.sstart.overrideHomePage")) {
		browserPref("startup.homepage", "set");
	}
	
	if (Services.prefs.getBoolPref("extensions.sstart.disableSysThumbs")) {
		browserPref("pagethumbnails", "set");
	}

	linkToSStart = Utils.translate("linkToSStart");
	pageToSStart = Utils.translate("pageToSStart");

	var ww = Cc["@mozilla.org/embedcomp/window-watcher;1"].getService(Ci.nsIWindowWatcher);
	gWindowListener = new BrowserWindowObserver({
		onStartup: browserWindowStartup,
		onShutdown: browserWindowShutdown
	});
	ww.registerNotification(gWindowListener);
	
	var wm = Cc["@mozilla.org/appshell/window-mediator;1"].getService(Ci.nsIWindowMediator);
	var winenu = wm.getEnumerator("navigator:browser");
	while (winenu.hasMoreElements()) {
		browserWindowStartup(winenu.getNext());
	}
	
	myPrefsWatcher.register();
	browserPrefsWatcher.register();
	newTabURLWatcher.register();

	Cache.updateGridInterval();
	Cache.updateNewtabOpen();
	
	if (reason != APP_STARTUP) {
		Utils.reloadEachSStartBrowser();
	} else {
		Cc["@mozilla.org/browser/search-service;1"].getService(Ci.nsIBrowserSearchService).init();
	}
}

function shutdown (params, reason)
{
	if (reason == APP_SHUTDOWN) return;

	newTabURLWatcher.unregister();
	browserPrefsWatcher.unregister();
	myPrefsWatcher.unregister();

	var ww = Cc["@mozilla.org/embedcomp/window-watcher;1"].getService(Ci.nsIWindowWatcher);
	ww.unregisterNotification(gWindowListener);
	gWindowListener = null;

	var wm = Cc["@mozilla.org/appshell/window-mediator;1"].getService(Ci.nsIWindowMediator);
	var winenu = wm.getEnumerator("navigator:browser");
	while (winenu.hasMoreElements()) {
		browserWindowShutdown(winenu.getNext());
	}
	
	browserPref("newtab.url", "clear");
	browserPref("startup.homepage", "clear");
	browserPref("pagethumbnails", "clear");
	
	PrefLoader.clearDefaultPrefs();

	if (styleSheetService.sheetRegistered(hideSStart, styleSheetService.USER_SHEET)) {
		styleSheetService.unregisterSheet(hideSStart, styleSheetService.USER_SHEET);
	}

	SSTART_MODULES.forEach(Cu.unload, Cu);
	
	Services.obs.notifyObservers(null, "chrome-flush-caches", null);
}