From 8fe49ede95d12b7186d660ef7c367d49b935fba7 Mon Sep 17 00:00:00 2001
From: Vasco Santos <vasco.santos@moxy.studio>
Date: Wed, 6 Mar 2019 21:42:42 +0000
Subject: [PATCH] fix: decapsulate ipfs from dht find peer

---
 src/daemon.js            | 12 +-----------
 src/libp2p.js            | 14 ++++++++++++++
 test/daemon/core.spec.js |  2 +-
 3 files changed, 16 insertions(+), 12 deletions(-)

diff --git a/src/daemon.js b/src/daemon.js
index ad7f8693..7c452bc5 100644
--- a/src/daemon.js
+++ b/src/daemon.js
@@ -369,17 +369,7 @@ class Daemon {
           enc.write(OkResponse({
             identify: {
               id: this.libp2p.peerInfo.id.toBytes(),
-              // temporary removal of "/ipfs/..." from multiaddrs
-              // this will be solved in: https://github.com/libp2p/js-libp2p/issues/323
-              addrs: this.libp2p.peerInfo.multiaddrs.toArray().map(m => {
-                let buffer
-                try {
-                  buffer = m.decapsulate('ipfs').buffer
-                } catch (_) {
-                  buffer = m.buffer
-                }
-                return buffer
-              })
+              addrs: this.libp2p.peerInfo.multiaddrs.toArray().map(m => m.buffer)
             }
           }))
           break
diff --git a/src/libp2p.js b/src/libp2p.js
index 8b413a3b..f60c0a72 100644
--- a/src/libp2p.js
+++ b/src/libp2p.js
@@ -240,6 +240,20 @@ class DaemonLibp2p extends Libp2p {
             this.peerInfo.multiaddrs.add(addr)
           })
         }
+
+        // temporary removal of "/ipfs/..." from multiaddrs
+        // this will be solved in: https://github.com/libp2p/js-libp2p/issues/323
+        this.peerInfo.multiaddrs.toArray().forEach(m => {
+          let ma
+          try {
+            ma = m.decapsulate('ipfs')
+          } catch (_) {
+            ma = m
+          }
+
+          this.peerInfo.multiaddrs.replace(m, ma)
+        })
+
         resolve()
       })
     })
diff --git a/test/daemon/core.spec.js b/test/daemon/core.spec.js
index 780ee388..76c2851b 100644
--- a/test/daemon/core.spec.js
+++ b/test/daemon/core.spec.js
@@ -143,7 +143,7 @@ describe('core features', () => {
 
     expect(response.identify).to.eql({
       id: daemon.libp2p.peerInfo.id.toBytes(),
-      addrs: daemon.libp2p.peerInfo.multiaddrs.toArray().map(m => m.decapsulate('ipfs').buffer)
+      addrs: daemon.libp2p.peerInfo.multiaddrs.toArray().map(m => m.buffer)
     })
     stream.end()
   })