feat: update events to conform with #74

libp2p · Mar 29, 2017 · f73c045 · f73c045
1 parent 8840c9b
commit f73c045
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/index.js b/src/index.js
@@ -41,10 +41,12 @@ class Node extends EventEmitter {
       // Received incommind dial and muxer upgrade happened, reuse this
       // muxed connection
       this.swarm.on('peer-mux-established', (peerInfo) => {
+        this.emit('peer:connect', peerInfo)
         this.peerBook.put(peerInfo)
       })
 
       this.swarm.on('peer-mux-closed', (peerInfo) => {
+        this.emit('peer:disconnect', peerInfo)
         this.peerBook.removeByB58String(peerInfo.id.toB58String())
       })
     }
@@ -62,8 +64,9 @@ class Node extends EventEmitter {
     if (this.modules.discovery) {
       let discoveries = this.modules.discovery
       discoveries = Array.isArray(discoveries) ? discoveries : [discoveries]
+
       discoveries.forEach((discovery) => {
-        discovery.on('peer', (peerInfo) => this.emit('peer', peerInfo))
+        discovery.on('peer', (peerInfo) => this.emit('peer:discovery', peerInfo))
       })
     }