michael@0: /* -*- Mode: javascript; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: let Cc = Components.classes;
michael@0: let Ci = Components.interfaces;
michael@0: let Cu = Components.utils;
michael@0: 
michael@0: Cu.import("resource://gre/modules/XPCOMUtils.jsm");
michael@0: Cu.import("resource://gre/modules/Services.jsm");
michael@0: 
michael@0: XPCOMUtils.defineLazyModuleGetter(this, "ContentLinkHandler",
michael@0:   "resource:///modules/ContentLinkHandler.jsm");
michael@0: XPCOMUtils.defineLazyModuleGetter(this, "LanguageDetector",
michael@0:   "resource:///modules/translation/LanguageDetector.jsm");
michael@0: XPCOMUtils.defineLazyModuleGetter(this, "LoginManagerContent",
michael@0:   "resource://gre/modules/LoginManagerContent.jsm");
michael@0: XPCOMUtils.defineLazyModuleGetter(this, "InsecurePasswordUtils",
michael@0:   "resource://gre/modules/InsecurePasswordUtils.jsm");
michael@0: XPCOMUtils.defineLazyModuleGetter(this, "PrivateBrowsingUtils",
michael@0:   "resource://gre/modules/PrivateBrowsingUtils.jsm");
michael@0: XPCOMUtils.defineLazyModuleGetter(this, "UITour",
michael@0:   "resource:///modules/UITour.jsm");
michael@0: 
michael@0: // Creates a new nsIURI object.
michael@0: function makeURI(uri, originCharset, baseURI) {
michael@0:   return Services.io.newURI(uri, originCharset, baseURI);
michael@0: }
michael@0: 
michael@0: addMessageListener("Browser:HideSessionRestoreButton", function (message) {
michael@0:   // Hide session restore button on about:home
michael@0:   let doc = content.document;
michael@0:   let container;
michael@0:   if (doc.documentURI.toLowerCase() == "about:home" &&
michael@0:       (container = doc.getElementById("sessionRestoreContainer"))){
michael@0:     container.hidden = true;
michael@0:   }
michael@0: });
michael@0: 
michael@0: if (Services.appinfo.processType == Services.appinfo.PROCESS_TYPE_CONTENT) {
michael@0:   addEventListener("contextmenu", function (event) {
michael@0:     sendAsyncMessage("contextmenu", {}, { event: event });
michael@0:   }, false);
michael@0: } else {
michael@0:   addEventListener("DOMFormHasPassword", function(event) {
michael@0:     InsecurePasswordUtils.checkForInsecurePasswords(event.target);
michael@0:     LoginManagerContent.onFormPassword(event);
michael@0:   });
michael@0:   addEventListener("DOMAutoComplete", function(event) {
michael@0:     LoginManagerContent.onUsernameInput(event);
michael@0:   });
michael@0:   addEventListener("blur", function(event) {
michael@0:     LoginManagerContent.onUsernameInput(event);
michael@0:   });
michael@0: 
michael@0:   addEventListener("mozUITour", function(event) {
michael@0:     if (!Services.prefs.getBoolPref("browser.uitour.enabled"))
michael@0:       return;
michael@0: 
michael@0:     let handled = UITour.onPageEvent(event);
michael@0:     if (handled)
michael@0:       addEventListener("pagehide", UITour);
michael@0:   }, false, true);
michael@0: }
michael@0: 
michael@0: let AboutHomeListener = {
michael@0:   init: function(chromeGlobal) {
michael@0:     chromeGlobal.addEventListener('AboutHomeLoad', () => this.onPageLoad(), false, true);
michael@0:   },
michael@0: 
michael@0:   handleEvent: function(aEvent) {
michael@0:     switch (aEvent.type) {
michael@0:       case "AboutHomeLoad":
michael@0:         this.onPageLoad();
michael@0:         break;
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   receiveMessage: function(aMessage) {
michael@0:     switch (aMessage.name) {
michael@0:       case "AboutHome:Update":
michael@0:         this.onUpdate(aMessage.data);
michael@0:         break;
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   onUpdate: function(aData) {
michael@0:     let doc = content.document;
michael@0:     if (doc.documentURI.toLowerCase() != "about:home")
michael@0:       return;
michael@0: 
michael@0:     if (aData.showRestoreLastSession && !PrivateBrowsingUtils.isWindowPrivate(content))
michael@0:       doc.getElementById("launcher").setAttribute("session", "true");
michael@0: 
michael@0:     // Inject search engine and snippets URL.
michael@0:     let docElt = doc.documentElement;
michael@0:     // set the following attributes BEFORE searchEngineName, which triggers to
michael@0:     // show the snippets when it's set.
michael@0:     docElt.setAttribute("snippetsURL", aData.snippetsURL);
michael@0:     if (aData.showKnowYourRights)
michael@0:       docElt.setAttribute("showKnowYourRights", "true");
michael@0:     docElt.setAttribute("snippetsVersion", aData.snippetsVersion);
michael@0:     docElt.setAttribute("searchEngineName", aData.defaultEngineName);
michael@0:   },
michael@0: 
michael@0:   onPageLoad: function() {
michael@0:     let doc = content.document;
michael@0:     if (doc.documentURI.toLowerCase() != "about:home" ||
michael@0:         doc.documentElement.hasAttribute("hasBrowserHandlers")) {
michael@0:       return;
michael@0:     }
michael@0: 
michael@0:     doc.documentElement.setAttribute("hasBrowserHandlers", "true");
michael@0:     let self = this;
michael@0:     addMessageListener("AboutHome:Update", self);
michael@0:     addEventListener("click", this.onClick, true);
michael@0:     addEventListener("pagehide", function onPageHide(event) {
michael@0:       if (event.target.defaultView.frameElement)
michael@0:         return;
michael@0:       removeMessageListener("AboutHome:Update", self);
michael@0:       removeEventListener("click", self.onClick, true);
michael@0:       removeEventListener("pagehide", onPageHide, true);
michael@0:       if (event.target.documentElement)
michael@0:         event.target.documentElement.removeAttribute("hasBrowserHandlers");
michael@0:     }, true);
michael@0: 
michael@0:     // XXX bug 738646 - when Marketplace is launched, remove this statement and
michael@0:     // the hidden attribute set on the apps button in aboutHome.xhtml
michael@0:     if (Services.prefs.getPrefType("browser.aboutHome.apps") == Services.prefs.PREF_BOOL &&
michael@0:         Services.prefs.getBoolPref("browser.aboutHome.apps"))
michael@0:       doc.getElementById("apps").removeAttribute("hidden");
michael@0: 
michael@0:     sendAsyncMessage("AboutHome:RequestUpdate");
michael@0: 
michael@0:     doc.addEventListener("AboutHomeSearchEvent", function onSearch(e) {
michael@0:       sendAsyncMessage("AboutHome:Search", { searchData: e.detail });
michael@0:     }, true, true);
michael@0:   },
michael@0: 
michael@0:   onClick: function(aEvent) {
michael@0:     if (!aEvent.isTrusted || // Don't trust synthetic events
michael@0:         aEvent.button == 2 || aEvent.target.localName != "button") {
michael@0:       return;
michael@0:     }
michael@0: 
michael@0:     let originalTarget = aEvent.originalTarget;
michael@0:     let ownerDoc = originalTarget.ownerDocument;
michael@0:     if (ownerDoc.documentURI != "about:home") {
michael@0:       // This shouldn't happen, but we're being defensive.
michael@0:       return;
michael@0:     }
michael@0: 
michael@0:     let elmId = originalTarget.getAttribute("id");
michael@0: 
michael@0:     switch (elmId) {
michael@0:       case "restorePreviousSession":
michael@0:         sendAsyncMessage("AboutHome:RestorePreviousSession");
michael@0:         ownerDoc.getElementById("launcher").removeAttribute("session");
michael@0:         break;
michael@0: 
michael@0:       case "downloads":
michael@0:         sendAsyncMessage("AboutHome:Downloads");
michael@0:         break;
michael@0: 
michael@0:       case "bookmarks":
michael@0:         sendAsyncMessage("AboutHome:Bookmarks");
michael@0:         break;
michael@0: 
michael@0:       case "history":
michael@0:         sendAsyncMessage("AboutHome:History");
michael@0:         break;
michael@0: 
michael@0:       case "apps":
michael@0:         sendAsyncMessage("AboutHome:Apps");
michael@0:         break;
michael@0: 
michael@0:       case "addons":
michael@0:         sendAsyncMessage("AboutHome:Addons");
michael@0:         break;
michael@0: 
michael@0:       case "sync":
michael@0:         sendAsyncMessage("AboutHome:Sync");
michael@0:         break;
michael@0: 
michael@0:       case "settings":
michael@0:         sendAsyncMessage("AboutHome:Settings");
michael@0:         break;
michael@0:     }
michael@0:   },
michael@0: };
michael@0: AboutHomeListener.init(this);
michael@0: 
michael@0: 
michael@0: let ContentSearchMediator = {
michael@0: 
michael@0:   whitelist: new Set([
michael@0:     "about:newtab",
michael@0:   ]),
michael@0: 
michael@0:   init: function (chromeGlobal) {
michael@0:     chromeGlobal.addEventListener("ContentSearchClient", this, true, true);
michael@0:     addMessageListener("ContentSearch", this);
michael@0:   },
michael@0: 
michael@0:   handleEvent: function (event) {
michael@0:     if (this._contentWhitelisted) {
michael@0:       this._sendMsg(event.detail.type, event.detail.data);
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   receiveMessage: function (msg) {
michael@0:     if (msg.data.type == "AddToWhitelist") {
michael@0:       for (let uri of msg.data.data) {
michael@0:         this.whitelist.add(uri);
michael@0:       }
michael@0:       this._sendMsg("AddToWhitelistAck");
michael@0:       return;
michael@0:     }
michael@0:     if (this._contentWhitelisted) {
michael@0:       this._fireEvent(msg.data.type, msg.data.data);
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   get _contentWhitelisted() {
michael@0:     return this.whitelist.has(content.document.documentURI.toLowerCase());
michael@0:   },
michael@0: 
michael@0:   _sendMsg: function (type, data=null) {
michael@0:     sendAsyncMessage("ContentSearch", {
michael@0:       type: type,
michael@0:       data: data,
michael@0:     });
michael@0:   },
michael@0: 
michael@0:   _fireEvent: function (type, data=null) {
michael@0:     content.dispatchEvent(new content.CustomEvent("ContentSearchService", {
michael@0:       detail: {
michael@0:         type: type,
michael@0:         data: data,
michael@0:       },
michael@0:     }));
michael@0:   },
michael@0: };
michael@0: ContentSearchMediator.init(this);
michael@0: 
michael@0: 
michael@0: var global = this;
michael@0: 
michael@0: // Lazily load the finder code
michael@0: addMessageListener("Finder:Initialize", function () {
michael@0:   let {RemoteFinderListener} = Cu.import("resource://gre/modules/RemoteFinder.jsm", {});
michael@0:   new RemoteFinderListener(global);
michael@0: });
michael@0: 
michael@0: 
michael@0: let ClickEventHandler = {
michael@0:   init: function init() {
michael@0:     Cc["@mozilla.org/eventlistenerservice;1"]
michael@0:       .getService(Ci.nsIEventListenerService)
michael@0:       .addSystemEventListener(global, "click", this, true);
michael@0:   },
michael@0: 
michael@0:   handleEvent: function(event) {
michael@0:     // Bug 903016: Most of this code is an unfortunate duplication from
michael@0:     // contentAreaClick in browser.js.
michael@0:     if (!event.isTrusted || event.defaultPrevented || event.button == 2)
michael@0:       return;
michael@0: 
michael@0:     let [href, node] = this._hrefAndLinkNodeForClickEvent(event);
michael@0: 
michael@0:     let json = { button: event.button, shiftKey: event.shiftKey,
michael@0:                  ctrlKey: event.ctrlKey, metaKey: event.metaKey,
michael@0:                  altKey: event.altKey, href: null, title: null,
michael@0:                  bookmark: false };
michael@0: 
michael@0:     if (href) {
michael@0:       json.href = href;
michael@0:       if (node) {
michael@0:         json.title = node.getAttribute("title");
michael@0: 
michael@0:         if (event.button == 0 && !event.ctrlKey && !event.shiftKey &&
michael@0:             !event.altKey && !event.metaKey) {
michael@0:           json.bookmark = node.getAttribute("rel") == "sidebar";
michael@0:           if (json.bookmark)
michael@0:             event.preventDefault(); // Need to prevent the pageload.
michael@0:         }
michael@0:       }
michael@0: 
michael@0:       sendAsyncMessage("Content:Click", json);
michael@0:       return;
michael@0:     }
michael@0: 
michael@0:     // This might be middle mouse navigation.
michael@0:     if (event.button == 1)
michael@0:       sendAsyncMessage("Content:Click", json);
michael@0:   },
michael@0: 
michael@0:   /**
michael@0:    * Extracts linkNode and href for the current click target.
michael@0:    *
michael@0:    * @param event
michael@0:    *        The click event.
michael@0:    * @return [href, linkNode].
michael@0:    *
michael@0:    * @note linkNode will be null if the click wasn't on an anchor
michael@0:    *       element (or XLink).
michael@0:    */
michael@0:   _hrefAndLinkNodeForClickEvent: function(event) {
michael@0:     function isHTMLLink(aNode) {
michael@0:       // Be consistent with what nsContextMenu.js does.
michael@0:       return ((aNode instanceof content.HTMLAnchorElement && aNode.href) ||
michael@0:               (aNode instanceof content.HTMLAreaElement && aNode.href) ||
michael@0:               aNode instanceof content.HTMLLinkElement);
michael@0:     }
michael@0: 
michael@0:     let node = event.target;
michael@0:     while (node && !isHTMLLink(node)) {
michael@0:       node = node.parentNode;
michael@0:     }
michael@0: 
michael@0:     if (node)
michael@0:       return [node.href, node];
michael@0: 
michael@0:     // If there is no linkNode, try simple XLink.
michael@0:     let href, baseURI;
michael@0:     node = event.target;
michael@0:     while (node && !href) {
michael@0:       if (node.nodeType == content.Node.ELEMENT_NODE) {
michael@0:         href = node.getAttributeNS("http://www.w3.org/1999/xlink", "href");
michael@0:         if (href)
michael@0:           baseURI = node.ownerDocument.baseURIObject;
michael@0:       }
michael@0:       node = node.parentNode;
michael@0:     }
michael@0: 
michael@0:     // In case of XLink, we don't return the node we got href from since
michael@0:     // callers expect <a>-like elements.
michael@0:     // Note: makeURI() will throw if aUri is not a valid URI.
michael@0:     return [href ? makeURI(href, null, baseURI).spec : null, null];
michael@0:   }
michael@0: };
michael@0: ClickEventHandler.init();
michael@0: 
michael@0: ContentLinkHandler.init(this);
michael@0: 
michael@0: addEventListener("DOMWebNotificationClicked", function(event) {
michael@0:   sendAsyncMessage("DOMWebNotificationClicked", {});
michael@0: }, false);
michael@0: 
michael@0: let PageStyleHandler = {
michael@0:   init: function() {
michael@0:     addMessageListener("PageStyle:Switch", this);
michael@0:     addMessageListener("PageStyle:Disable", this);
michael@0: 
michael@0:     // Send a CPOW to the parent so that it can synchronously request
michael@0:     // the list of style sheets.
michael@0:     sendSyncMessage("PageStyle:SetSyncHandler", {}, {syncHandler: this});
michael@0:   },
michael@0: 
michael@0:   get markupDocumentViewer() {
michael@0:     return docShell.contentViewer.QueryInterface(Ci.nsIMarkupDocumentViewer);
michael@0:   },
michael@0: 
michael@0:   // Called synchronously via CPOW from the parent.
michael@0:   getStyleSheetInfo: function() {
michael@0:     let styleSheets = this._filterStyleSheets(this.getAllStyleSheets());
michael@0:     return {
michael@0:       styleSheets: styleSheets,
michael@0:       authorStyleDisabled: this.markupDocumentViewer.authorStyleDisabled,
michael@0:       preferredStyleSheetSet: content.document.preferredStyleSheetSet
michael@0:     };
michael@0:   },
michael@0: 
michael@0:   // Called synchronously via CPOW from the parent.
michael@0:   getAllStyleSheets: function(frameset = content) {
michael@0:     let selfSheets = Array.slice(frameset.document.styleSheets);
michael@0:     let subSheets = Array.map(frameset.frames, frame => this.getAllStyleSheets(frame));
michael@0:     return selfSheets.concat(...subSheets);
michael@0:   },
michael@0: 
michael@0:   receiveMessage: function(msg) {
michael@0:     switch (msg.name) {
michael@0:       case "PageStyle:Switch":
michael@0:         this.markupDocumentViewer.authorStyleDisabled = false;
michael@0:         this._stylesheetSwitchAll(content, msg.data.title);
michael@0:         break;
michael@0: 
michael@0:       case "PageStyle:Disable":
michael@0:         this.markupDocumentViewer.authorStyleDisabled = true;
michael@0:         break;
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   _stylesheetSwitchAll: function (frameset, title) {
michael@0:     if (!title || this._stylesheetInFrame(frameset, title)) {
michael@0:       this._stylesheetSwitchFrame(frameset, title);
michael@0:     }
michael@0: 
michael@0:     for (let i = 0; i < frameset.frames.length; i++) {
michael@0:       // Recurse into sub-frames.
michael@0:       this._stylesheetSwitchAll(frameset.frames[i], title);
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   _stylesheetSwitchFrame: function (frame, title) {
michael@0:     var docStyleSheets = frame.document.styleSheets;
michael@0: 
michael@0:     for (let i = 0; i < docStyleSheets.length; ++i) {
michael@0:       let docStyleSheet = docStyleSheets[i];
michael@0:       if (docStyleSheet.title) {
michael@0:         docStyleSheet.disabled = (docStyleSheet.title != title);
michael@0:       } else if (docStyleSheet.disabled) {
michael@0:         docStyleSheet.disabled = false;
michael@0:       }
michael@0:     }
michael@0:   },
michael@0: 
michael@0:   _stylesheetInFrame: function (frame, title) {
michael@0:     return Array.some(frame.document.styleSheets, (styleSheet) => styleSheet.title == title);
michael@0:   },
michael@0: 
michael@0:   _filterStyleSheets: function(styleSheets) {
michael@0:     let result = [];
michael@0: 
michael@0:     for (let currentStyleSheet of styleSheets) {
michael@0:       if (!currentStyleSheet.title)
michael@0:         continue;
michael@0: 
michael@0:       // Skip any stylesheets that don't match the screen media type.
michael@0:       if (currentStyleSheet.media.length > 0) {
michael@0:         let mediaQueryList = currentStyleSheet.media.mediaText;
michael@0:         if (!content.matchMedia(mediaQueryList).matches) {
michael@0:           continue;
michael@0:         }
michael@0:       }
michael@0: 
michael@0:       result.push({title: currentStyleSheet.title,
michael@0:                    disabled: currentStyleSheet.disabled});
michael@0:     }
michael@0: 
michael@0:     return result;
michael@0:   },
michael@0: };
michael@0: PageStyleHandler.init();
michael@0: 
michael@0: let TranslationHandler = {
michael@0:   init: function() {
michael@0:     let webProgress = docShell.QueryInterface(Ci.nsIInterfaceRequestor)
michael@0:                               .getInterface(Ci.nsIWebProgress);
michael@0:     webProgress.addProgressListener(this, Ci.nsIWebProgress.NOTIFY_STATE_DOCUMENT);
michael@0:   },
michael@0: 
michael@0:   /* nsIWebProgressListener implementation */
michael@0:   onStateChange: function(aWebProgress, aRequest, aStateFlags, aStatus) {
michael@0:     if (!aWebProgress.isTopLevel ||
michael@0:         !(aStateFlags & Ci.nsIWebProgressListener.STATE_STOP))
michael@0:       return;
michael@0: 
michael@0:     let url = aRequest.name;
michael@0:     if (!url.startsWith("http://") && !url.startsWith("https://"))
michael@0:       return;
michael@0: 
michael@0:     // Grab a 60k sample of text from the page.
michael@0:     let encoder = Cc["@mozilla.org/layout/documentEncoder;1?type=text/plain"]
michael@0:                     .createInstance(Ci.nsIDocumentEncoder);
michael@0:     encoder.init(content.document, "text/plain", encoder.SkipInvisibleContent);
michael@0:     let string = encoder.encodeToStringWithMaxLength(60 * 1024);
michael@0: 
michael@0:     // Language detection isn't reliable on very short strings.
michael@0:     if (string.length < 100)
michael@0:       return;
michael@0: 
michael@0:     LanguageDetector.detectLanguage(string).then(result => {
michael@0:       if (result.confident)
michael@0:         sendAsyncMessage("LanguageDetection:Result", result.language);
michael@0:     });
michael@0:   },
michael@0: 
michael@0:   // Unused methods.
michael@0:   onProgressChange: function() {},
michael@0:   onLocationChange: function() {},
michael@0:   onStatusChange:   function() {},
michael@0:   onSecurityChange: function() {},
michael@0: 
michael@0:   QueryInterface: XPCOMUtils.generateQI([Ci.nsIWebProgressListener,
michael@0:                                          Ci.nsISupportsWeakReference])
michael@0: };
michael@0: 
michael@0: if (Services.prefs.getBoolPref("browser.translation.detectLanguage"))
michael@0:   TranslationHandler.init();