From 4dc5ed3c26596b8c30e33e721273fcd9a051170d Mon Sep 17 00:00:00 2001
From: David Dias <daviddias.p@gmail.com>
Date: Fri, 4 Mar 2016 22:40:40 +0000
Subject: [PATCH] done

---
 README.md               |  8 --------
 src/index.js            | 15 +++++++++++++++
 tests/peer-info-test.js | 27 +++++++++++++++++++++++++++
 3 files changed, 42 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index 3b40908..c4cffd6 100644
--- a/README.md
+++ b/README.md
@@ -18,11 +18,3 @@ peer-info JavaScript implementation
 ```javascript
 const PeerInfo = require('peer-info')
 ```
-
-### In the Browser through `<script>` tag
-
-Make the [peer-info.min.js](/dist/peer-info.min.js) available through your server and load it using a normal `<script>` tag, this will export the `peerId` constructor on the `window` object, such that:
-
-```JavaScript
-const PeerInfo = window.PeerInfo
-```
diff --git a/src/index.js b/src/index.js
index f070abd..f1d9990 100644
--- a/src/index.js
+++ b/src/index.js
@@ -59,6 +59,21 @@ function Peer (peerId) {
     })
   }
 
+  this.multiaddr.replace = (existing, fresh) => {
+    if (!Array.isArray(existing)) {
+      existing = [existing]
+    }
+    if (!Array.isArray(fresh)) {
+      fresh = [fresh]
+    }
+    existing.forEach((m) => {
+      this.multiaddr.rm(m)
+    })
+    fresh.forEach((m) => {
+      this.multiaddr.add(m)
+    })
+  }
+
   // TODO: add features to fetch multiaddr using filters
   // look at https://github.com/whyrusleeping/js-mafmt/blob/master/src/index.js
 }
diff --git a/tests/peer-info-test.js b/tests/peer-info-test.js
index b142493..6dbc4f1 100644
--- a/tests/peer-info-test.js
+++ b/tests/peer-info-test.js
@@ -65,4 +65,31 @@ describe('peer-info', function (done) {
     expect(pi.multiaddrs.length).to.equal(1)
     done()
   })
+
+  it('replace multiaddr', (done) => {
+    const pi = new PeerInfo()
+    expect(pi).to.exist
+    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')
+    const mh5 = Multiaddr('/ip4/127.0.0.1/tcp/5005')
+    const mh6 = Multiaddr('/ip4/127.0.0.1/tcp/5006')
+
+    pi.multiaddr.add(mh1)
+    pi.multiaddr.add(mh2)
+    pi.multiaddr.add(mh3)
+    pi.multiaddr.add(mh4)
+
+    expect(pi.multiaddrs.length).to.equal(4)
+
+    const old = [mh2, mh4]
+    const fresh = [mh5, mh6]
+
+    pi.multiaddr.replace(old, fresh)
+
+    expect(pi.multiaddrs.length).to.equal(4)
+
+    done()
+  })
 })