refactor: separate message processing

index.js

@@ -1,2 +1,4 @@
 const Messaging = require('./lib/messaging');
-module.exports = { ClapPeer: Messaging };
+const constants = require('./lib/constants');
+
+module.exports = { ClapPeer: Messaging, ...constants };

lib/constants.js

@@ -1,3 +1,4 @@
 const HANDSHAKE = 'HANDSHAKE';
 const DM = 'MESSAGE';
+
 module.exports = { HANDSHAKE, DM };

lib/messaging.js

@@ -17,15 +17,9 @@ class Messaging extends Connection {
   }
 
   _onMessage(connectionId, messageObject) {
-    const { type, data, to } = messageObject;
-    this[type](data);
-    if (type === HANDSHAKE) {
-      this.#addNeighbor(connectionId, data);
-    }
-    if (type === DM) {
-      if (to !== this.NODE_ID) return this.send(to, data);
-      this.emit(DM, data);
-    }
+    const { type } = messageObject;
+    const handler = this[type]?.bind(this);
+    if (handler) handler(connectionId, messageObject);
   }
 
   _deleteConnection(connectionId) {
@@ -35,12 +29,10 @@ class Messaging extends Connection {
 
   send(nodeId, message) {
     const connectionId = this.neighbors.get(nodeId);
-    if (connectionId)
-      return this._send(
-        connectionId,
-        createMessage({ type: DM, data: message, to: nodeId }),
-      );
-    this.neighbors.forEach(connectionId => this._send(connectionId, message));
+    const targets = connectionId ? [connectionId] : this.neighbors;
+    targets.forEach(target =>
+      this._send(target, createMessage(DM, nodeId, message)),
+    );
   }
 
   #findNodeId(connectionId) {
@@ -51,7 +43,17 @@ class Messaging extends Connection {
 
   #addNeighbor(connectionId, neighbor) {
     this.neighbors.set(neighbor, connectionId);
-    console.log(this.neighbors.get(neighbor));
+  }
+
+  [HANDSHAKE](connectionId, messageObject) {
+    const { data } = messageObject;
+    this.#addNeighbor(connectionId, data);
+  }
+
+  [DM](_, messageObject) {
+    const { data, to } = messageObject;
+    if (to !== this.NODE_ID) return this.send(to, data);
+    this.emit(DM, data);
   }
 }