feat: emit event on peer connected

Author: vasco-santos

src/index.js

@@ -1,5 +1,6 @@
 'use strict'
 
+const { EventEmitter } = require('events')
 const libp2pRecord = require('libp2p-record')
 const MemoryStore = require('interface-datastore').MemoryDatastore
 const waterfall = require('async/waterfall')
@@ -27,7 +28,7 @@ const assert = require('assert')
  *
  * Original implementation in go: https://github.com/libp2p/go-libp2p-kad-dht.
  */
-class KadDHT {
+class KadDHT extends EventEmitter {
   /**
    * Create a new KadDHT.
    *
@@ -40,6 +41,7 @@ class KadDHT {
    * @param {object} options.selectors selectors object with namespace as keys and function(key, records)
    */
   constructor (sw, options) {
+    super()
     assert(sw, 'libp2p-kad-dht requires a instance of Switch')
     options = options || {}
     options.validators = options.validators
@@ -608,6 +610,10 @@ class KadDHT {
       ], callback)
     })
   }
+
+  _peerConnected (peerInfo) {
+    this.emit('peer', peerInfo)
+  }
 }
 
 module.exports = KadDHT

src/network.js

@@ -122,6 +122,8 @@ class Network {
           return this._log.error('Failed to add to the routing table', err)
         }
 
+        this.dht._peerConnected(peer)
+
         this._log('added to the routing table: %s', peer.id.toB58String())
       })
     })

src/query.js

@@ -205,6 +205,7 @@ function execQuery (next, query, path, callback) {
         if (query.dht._isSelf(closer.id)) {
           return cb()
         }
+        query.dht._peerConnected(closer)
         closer = query.dht.peerBook.put(closer)
         addPeerToQuery(closer.id, query.dht, path, cb)
       }, callback)