Merge pull request libp2p#70 from diasdavid/feat/hangUp

add hangup feature

Author: daviddias

README.md

@@ -102,6 +102,13 @@ dial uses the best transport (whatever works first, in the future we can have so
 - `protocol`
 - `callback`
 
+### `swarm.hangUp(pi, callback)`
+
+hangUp the muxedConn we have with the peer
+
+- `pi` - peer info project
+- `callback`
+
 ### `swarm.listen(callback)`
 
 Start listening on all added transports that are available on the current `peerInfo`.

src/dial.js

@@ -137,7 +137,7 @@ module.exports = function dial (swarm) {
 
             swarm.emit('peer-mux-established', pi)
 
-            muxedConn.on('close', () => {
+            muxedConn.once('close', () => {
               delete swarm.muxedConns[pi.id.toB58String()]
               swarm.emit('peer-mux-closed', pi)
             })

src/index.js

@@ -103,6 +103,20 @@ function Swarm (peerInfo) {
     }
   }
 
+  this.hangUp = (peerInfo, callback) => {
+    const key = peerInfo.id.toB58String()
+    if (this.muxedConns[key]) {
+      const muxer = this.muxedConns[key].muxer
+      muxer.end()
+      muxer.on('close', () => {
+        delete this.muxedConns[key]
+        callback()
+      })
+    } else {
+      callback()
+    }
+  }
+
   this.close = (callback) => {
     Object.keys(this.muxedConns).forEach((key) => {
       this.muxedConns[key].muxer.end()

test/09-swarm-with-muxing.node.js

@@ -213,10 +213,30 @@ describe('high level API - with everything mixed all together!', function () {
     })
   })
 
+  it('hangUp', (done) => {
+    let count = 0
+    const ready = () => ++count === 3 ? done() : null
+
+    swarmB.once('peer-mux-closed', (peerInfo) => {
+      expect(Object.keys(swarmA.muxedConns).length).to.equal(0)
+      ready()
+    })
+
+    swarmA.once('peer-mux-closed', (peerInfo) => {
+      expect(Object.keys(swarmA.muxedConns).length).to.equal(1)
+      ready()
+    })
+
+    swarmA.hangUp(peerB, (err) => {
+      expect(err).to.not.exist
+      ready()
+    })
+  })
+
   it('close a muxer emits event', (done) => {
     parallel([
       (cb) => swarmC.close(cb),
-      (cb) => swarmA.once('peer-mux-closed', () => cb())
+      (cb) => swarmA.once('peer-mux-closed', (peerInfo) => cb())
     ], done)
   })
 })