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: Components.utils.import("resource://gre/modules/XPCOMUtils.jsm");
michael@0: 
michael@0: const Cc = Components.classes;
michael@0: const Ci = Components.interfaces;
michael@0: 
michael@0: const CATEGORY_WAKEUP_REQUEST = "wakeup-request";
michael@0: 
michael@0: function MessageWakeupService() { };
michael@0: 
michael@0: MessageWakeupService.prototype =
michael@0: {
michael@0:   classID:          Components.ID("{f9798742-4f7b-4188-86ba-48b116412b29}"),
michael@0:   QueryInterface:   XPCOMUtils.generateQI([Ci.nsIObserver]),
michael@0: 
michael@0:   messagesData: [],
michael@0: 
michael@0:   get messageManager() {
michael@0:     if (!this._messageManager)
michael@0:       this._messageManager = Cc["@mozilla.org/parentprocessmessagemanager;1"].
michael@0:                              getService(Ci.nsIMessageListenerManager);
michael@0:     return this._messageManager;
michael@0:   },
michael@0: 
michael@0:   requestWakeup: function(aMessageName, aCid, aIid, aMethod) {
michael@0:     this.messagesData[aMessageName] = {
michael@0:       cid: aCid,
michael@0:       iid: aIid,
michael@0:       method: aMethod,
michael@0:     };
michael@0: 
michael@0:     this.messageManager.addMessageListener(aMessageName, this);
michael@0:   },
michael@0: 
michael@0:   receiveMessage: function(aMessage) {
michael@0:     let data = this.messagesData[aMessage.name];
michael@0:     // TODO: When bug 593407 is ready, stop doing the wrappedJSObject hack
michael@0:     //       and use this line instead:
michael@0:     //                  QueryInterface(Ci.nsIMessageListener);
michael@0:     let service = Cc[data.cid][data.method](Ci[data.iid]).
michael@0:                   wrappedJSObject;
michael@0: 
michael@0:     // The receiveMessage() call itself may spin an event loop, and we
michael@0:     // do not want to swap listeners in that - it would cause the current
michael@0:     // message to be answered by two listeners. So, we call that first,
michael@0:     // then queue the swap for the next event loop
michael@0:     let ret = service.receiveMessage(aMessage);
michael@0: 
michael@0:     if (data.timer) {
michael@0:       // Handle the case of two such messages happening in quick succession
michael@0:       data.timer.cancel();
michael@0:       data.timer = null;
michael@0:     }
michael@0: 
michael@0:     data.timer = Cc["@mozilla.org/timer;1"].createInstance(Ci.nsITimer);
michael@0:     let self = this;
michael@0:     data.timer.initWithCallback(function() {
michael@0:       self.messageManager.addMessageListener(aMessage.name, service);
michael@0:       self.messageManager.removeMessageListener(aMessage.name, self);
michael@0:       delete self.messagesData[aMessage.name];
michael@0:     }, 0, Ci.nsITimer.TYPE_ONE_SHOT);
michael@0: 
michael@0:     return ret;
michael@0:   },
michael@0: 
michael@0:   observe: function TM_observe(aSubject, aTopic, aData) {
michael@0:     switch (aTopic) {
michael@0:       case "profile-after-change":
michael@0:         {
michael@0:           var catMan = Cc["@mozilla.org/categorymanager;1"].
michael@0:                            getService(Ci.nsICategoryManager);
michael@0:           var entries = catMan.enumerateCategory(CATEGORY_WAKEUP_REQUEST);
michael@0:           while (entries.hasMoreElements()) {
michael@0:             var entry = entries.getNext().QueryInterface(Ci.nsISupportsCString).data;
michael@0:             var value = catMan.getCategoryEntry(CATEGORY_WAKEUP_REQUEST, entry);
michael@0:             var parts = value.split(",");
michael@0:             var cid = parts[0];
michael@0:             var iid = parts[1];
michael@0:             var method = parts[2];
michael@0:             var messages = parts.slice(3);
michael@0:             messages.forEach(function(messageName) {
michael@0:               this.requestWakeup(messageName, cid, iid, method);
michael@0:             }, this);
michael@0:           }
michael@0:         }
michael@0:         break;
michael@0:     }
michael@0:   },
michael@0: };
michael@0: 
michael@0: var components = [MessageWakeupService];
michael@0: this.NSGetFactory = XPCOMUtils.generateNSGetFactory(components);
michael@0: