feat: filter IPFS addrs correctly

libp2p · Aug 3, 2017 · a49b546 · a49b546
1 parent 0d1f849
commit a49b546
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/index.js b/src/index.js
@@ -2,7 +2,6 @@
 
 const connect = require('pull-ws/client')
 const mafmt = require('mafmt')
-const includes = require('lodash.includes')
 const Connection = require('interface-connection').Connection
 
 const maToUrl = require('./ma-to-url')
@@ -49,10 +48,10 @@ class WebSockets {
     }
 
     return multiaddrs.filter((ma) => {
-      if (includes(ma.protoNames(), 'ipfs')) {
-        ma = ma.decapsulate('ipfs')
-      }
-      return mafmt.WebSockets.matches(ma) || mafmt.WebSocketsSecure.matches(ma)
+      return mafmt.WebSockets.matches(ma) ||
+        mafmt.WebSocketsSecure.matches(ma) ||
+        mafmt.WebSocketsIPFS.matches(ma) ||
+        mafmt.WebSocketsSecureIPFS.matches(ma)
     })
   }
 }