Added distinctMultiaddr method (#29)

* Added distinctMultiaddr method

* made distinctAddr filter by port and transport

* adding multi addr tests

* changing lodash require

Author: SidHarder

src/index.js

@@ -2,6 +2,7 @@
 
 const Id = require('peer-id')
 const multiaddr = require('multiaddr')
+const uniqBy = require('lodash').uniqBy
 
 exports = module.exports = PeerInfo
 
@@ -88,6 +89,13 @@ function PeerInfo (peerId) {
     })
   }
 
+  this.distinctMultiaddr = () => {
+    var result = uniqBy(this.multiaddrs, function (item) {
+      return [item.toOptions().port, item.toOptions().transport].join()
+    })
+    return result
+  }
+
   // TODO: add features to fetch multiaddr using filters
   // look at https://github.com/whyrusleeping/js-mafmt/blob/master/src/index.js
 }

test/index.spec.js

@@ -151,4 +151,81 @@ describe('peer-info', () => {
 
     expect(pi.multiaddrs.length).to.equal(4)
   })
+
+  it('get distinct multiaddr same transport multiple different ports', () => {
+    const mh1 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+    const mh2 = Multiaddr('/ip4/127.0.0.1/tcp/5002')
+    const mh3 = Multiaddr('/ip4/127.0.0.1/tcp/5003')
+    const mh4 = Multiaddr('/ip4/127.0.0.1/tcp/5004')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+    pi.multiaddr.add(mh3)
+    pi.multiaddr.add(mh4)
+
+    var distinctMultiaddr = pi.distinctMultiaddr()
+    expect(distinctMultiaddr.length).to.equal(4)
+  })
+
+  it('get distinct multiaddr same transport different port', () => {
+    const mh1 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+    const mh2 = Multiaddr('/ip4/127.0.0.1/tcp/5002')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+
+    var distinctMultiaddr = pi.distinctMultiaddr()
+    expect(distinctMultiaddr.length).to.equal(2)
+  })
+
+  it('get distinct multiaddr same transport same port', () => {
+    const mh1 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+    const mh2 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+
+    var distinctMultiaddr = pi.distinctMultiaddr()
+    expect(distinctMultiaddr.length).to.equal(1)
+  })
+
+  it('get distinct multiaddr different transport same port', () => {
+    const mh1 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+    const mh2 = Multiaddr('/ip4/127.0.0.1/udp/5001')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+
+    var distinctMultiaddr = pi.distinctMultiaddr()
+    expect(distinctMultiaddr.length).to.equal(2)
+  })
+
+  it('get distinct multiaddr different family same port same transport', () => {
+    const mh1 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+    const mh2 = Multiaddr('/ip6/::/tcp/5001')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+
+    var distinctMultiaddr = pi.distinctMultiaddr()
+    expect(distinctMultiaddr.length).to.equal(1)
+  })
+
+  it('get distinct multiaddr different family same port multiple transports', () => {
+    const mh1 = Multiaddr('/ip4/127.0.0.1/tcp/5001')
+    const mh2 = Multiaddr('/ip6/::/tcp/5001')
+    const mh3 = Multiaddr('/ip6/::/udp/5002')
+    const mh4 = Multiaddr('/ip4/127.0.0.1/udp/5002')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+    pi.multiaddr.add(mh3)
+    pi.multiaddr.add(mh4)
+
+    var distinctMultiaddr = pi.distinctMultiaddr()
+    expect(distinctMultiaddr.length).to.equal(2)
+
+    expect(distinctMultiaddr[0].toOptions().family).to.equal('ipv4')
+    expect(distinctMultiaddr[1].toOptions().family).to.equal('ipv6')
+  })
 })