fix: emit peer discovery for dht discovery

libp2p · Feb 5, 2019 · 9e7a080 · 9e7a080
1 parent 558e598
commit 9e7a080
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -72,7 +72,7 @@
     "libp2p-circuit": "~0.3.0",
     "libp2p-delegated-content-routing": "~0.2.2",
     "libp2p-delegated-peer-routing": "~0.2.2",
-    "libp2p-kad-dht": "~0.14.2",
+    "libp2p-kad-dht": "~0.14.5",
     "libp2p-mdns": "~0.12.0",
     "libp2p-mplex": "~0.8.4",
     "libp2p-secio": "~0.11.0",

diff --git a/src/index.js b/src/index.js
@@ -400,7 +400,10 @@ class Node extends EventEmitter {
       },
       (cb) => {
         if (this._dht) {
-          this._dht.start(cb)
+          this._dht.start(() => {
+            this._dht.on('peer', (peerInfo) => this.emit('peer:discovery', peerInfo))
+            cb()
+          })
         } else {
           cb()
         }

diff --git a/test/peer-discovery.node.js b/test/peer-discovery.node.js
@@ -9,6 +9,9 @@ const signalling = require('libp2p-webrtc-star/src/sig-server')
 const parallel = require('async/parallel')
 const crypto = require('crypto')
 
+const PeerId = require('peer-id')
+const PeerInfo = require('peer-info')
+
 const createNode = require('./utils/create-node')
 const echo = require('./utils/echo')
 
@@ -241,6 +244,9 @@ describe('peer discovery', () => {
   describe('MulticastDNS', () => {
     setup({
       config: {
+        dht: {
+          enabled: false
+        },
         peerDiscovery: {
           mdns: {
             enabled: true,
@@ -266,6 +272,9 @@ describe('peer discovery', () => {
   describe.skip('WebRTCStar', () => {
     setup({
       config: {
+        dht: {
+          enabled: false
+        },
         peerDiscovery: {
           webRTCStar: {
             enabled: true
@@ -287,6 +296,9 @@ describe('peer discovery', () => {
   describe('MulticastDNS + WebRTCStar', () => {
     setup({
       config: {
+        dht: {
+          enabled: false
+        },
         peerDiscovery: {
           mdns: {
             enabled: true,
@@ -309,4 +321,37 @@ describe('peer discovery', () => {
       })
     })
   })
+
+  describe('dht', () => {
+    setup({
+      config: {
+        peerDiscovery: {
+          mdns: {
+            enabled: false
+          },
+          webRTCStar: {
+            enabled: false
+          }
+        }
+      }
+    })
+
+    it('find a peer', function (done) {
+      this.timeout(15 * 1000)
+
+      nodeA.once('peer:discovery', (peerInfo) => {
+        expect(nodeB.peerInfo.id.toB58String())
+          .to.eql(peerInfo.id.toB58String())
+        done()
+      })
+
+      // connect two dhts
+      const publicPeerId = new PeerId(nodeB._dht.peerInfo.id.id, null, nodeB._dht.peerInfo.id.pubKey)
+      const target = new PeerInfo(publicPeerId)
+      target.multiaddrs = nodeB._dht.peerInfo.multiaddrs
+      nodeA._dht.switch.dial(target, (err) => {
+        expect(err).to.not.exist()
+      })
+    })
+  })
 })