gecko/dom/base/messageWakeupService.js

97 lines
3.3 KiB
JavaScript
Raw Normal View History

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