Issue 4129 - Reuse EventEmitter class in messaging module

lib/messaging.js

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2016 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus 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 General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
  */
 
 "use strict";
 
+let {EventEmitter} = require("events");
+
 const MESSAGE_NAME = "AdblockPlus:Message";
 const RESPONSE_NAME = "AdblockPlus:Response";
 
 function isPromise(value)
 {
   // value instanceof Promise won't work - there can be different Promise
   // classes (e.g. in different contexts) and there can also be promise-like
   // classes (e.g. Task).
   return (value && typeof value.then == "function");
 }
@@ skipping 84 matching lines @@
 
 /**
  * Communication port wrapping the message manager API to send and receive
  * messages.
  * @param {nsIMessageManager} messageManager
  * @constructor
  */
 function Port(messageManager)
 {
   this._messageManager = messageManager;
+  this._eventEmitter = new EventEmitter();
 
-  this._callbacks = new Map();
   this._responseCallbacks = new Map();
   this._responseCallbackCounter = 0;
 
   this._handleRequest = this._handleRequest.bind(this);
   this._handleResponse = this._handleResponse.bind(this);
   this._messageManager.addMessageListener(MESSAGE_NAME, this._handleRequest);
   this._messageManager.addMessageListener(RESPONSE_NAME, this._handleResponse);
 }
 Port.prototype = {
   /**
@@ skipping 65 matching lines @@
     callbackData[2] = --expectedResponses;
     if (expectedResponses <= 0)
     {
       this._responseCallbacks.delete(callbackID);
       callback(processor.value);
     }
   },
 
   _dispatch: function(messageName, payload, sender)
   {
-    let callbacks = this._callbacks.get(messageName);
-    if (!callbacks)
-      return undefined;
-
-    callbacks = callbacks.slice();
+    let callbacks = this._eventEmitter.listeners(messageName);
     let processor = new ResponseProcessor(messageName);
     for (let callback of callbacks)
     {
       try
       {
         processor.add(callback(payload, sender));
       }
       catch (e)
       {
         Cu.reportError(e);
@@ skipping 13 matching lines @@
    *      response).
    */
 
   /**
    * Adds a handler for the specified message.
    * @param {string} messageName message that would trigger the callback
    * @param {Port~messageHandler} callback
    */
   on: function(messageName, callback)
   {
-    let callbacks = this._callbacks.get(messageName);
-    if (callbacks)
-      callbacks.push(callback);
-    else
-      this._callbacks.set(messageName, [callback]);
+    this._eventEmitter.on(messageName, callback);
   },
 
   /**
    * Removes a handler for the specified message.
    * @param {string} messageName message that would trigger the callback
    * @param {Port~messageHandler} callback
    */
   off: function(messageName, callback)
   {
-    let callbacks = this._callbacks.get(messageName);
-    if (!callbacks)
-      return;
-
-    let index = callbacks.indexOf(callback);
-    if (index >= 0)
-      callbacks.splice(index, 1);
+    this._eventEmitter.off(messageName, callback);
   },
 
   /**
    * Sends a message.
    * @param {string} messageName message identifier
    * @param [payload] data to attach to the message
    */
   emit: function(messageName, payload)
   {
     sendMessage(this._messageManager, messageName, payload, undefined);
@@ skipping 39 matching lines @@
 catch (e)
 {
   // Parent
   messageManager = Cc["@mozilla.org/parentprocessmessagemanager;1"]
                      .getService(Ci.nsIMessageListenerManager);
 }
 
 let port = new Port(messageManager);
 onShutdown.add(() => port.disconnect());
 exports.port = port;