refactor: convert swarm API to async/await (#2681)

* refactor: convert swarm API to async/await

* refactor: use libp2p.connections

Author: Alan Shaw

src/core/components/index.js

@@ -26,6 +26,13 @@ exports.object = {
 exports.ping = require('./ping')
 exports.start = require('./start')
 exports.stop = require('./stop')
+exports.swarm = {
+  addrs: require('./swarm/addrs'),
+  connect: require('./swarm/connect'),
+  disconnect: require('./swarm/disconnect'),
+  localAddrs: require('./swarm/localAddrs'),
+  peers: require('./swarm/peers')
+}
 exports.version = require('./version')
 
 exports.legacy = { // TODO: these will be removed as the new API is completed

src/core/components/init.js

@@ -319,6 +319,13 @@ function createApi ({
     init: () => { throw new AlreadyInitializedError() },
     object,
     start,
+    swarm: {
+      addrs: () => { throw new NotStartedError() },
+      connect: () => { throw new NotStartedError() },
+      disconnect: () => { throw new NotStartedError() },
+      localAddrs: Commands.swarm.localAddrs({ peerInfo }),
+      peers: () => { throw new NotStartedError() }
+    },
     version: Commands.version({ repo })
   }
 

src/core/components/start.js

@@ -159,6 +159,13 @@ function createApi ({
       : () => { throw new NotEnabledError('pubsub not enabled') },
     start: () => apiManager.api,
     stop,
+    swarm: {
+      addrs: () => Commands.swarm.addrs({ libp2p }),
+      connect: () => Commands.swarm.connect({ libp2p }),
+      disconnect: () => Commands.swarm.disconnect({ libp2p }),
+      localAddrs: Commands.swarm.localAddrs({ peerInfo }),
+      peers: () => Commands.swarm.peers({ libp2p })
+    },
     version: Commands.version({ repo })
   }
 

src/core/components/stop.js

@@ -117,6 +117,13 @@ function createApi ({
     init: () => { throw new AlreadyInitializedError() },
     start,
     stop: () => apiManager.api,
+    swarm: {
+      addrs: () => { throw new NotStartedError() },
+      connect: () => { throw new NotStartedError() },
+      disconnect: () => { throw new NotStartedError() },
+      localAddrs: Commands.swarm.localAddrs({ peerInfo }),
+      peers: () => { throw new NotStartedError() }
+    },
     version: Commands.version({ repo })
   }
 

src/core/components/swarm.js

@@ -0,0 +1,13 @@
+'use strict'
+
+const CID = require('cids')
+
+module.exports = ({ libp2p }) => {
+  return async function addrs () { // eslint-disable-line require-await
+    const peers = []
+    for (const [peerId, peerInfo] of libp2p.peerStore.entries()) {
+      peers.push({ id: new CID(peerId), addrs: peerInfo.multiaddrs.toArray() })
+    }
+    return peers
+  }
+}

src/core/components/swarm/addrs.js

@@ -0,0 +1,7 @@
+'use strict'
+
+module.exports = ({ libp2p }) => {
+  return function connect (addr) {
+    return libp2p.dial(addr)
+  }
+}

src/core/components/swarm/connect.js

@@ -0,0 +1,7 @@
+'use strict'
+
+module.exports = ({ libp2p }) => {
+  return function disconnect (addr) {
+    return libp2p.hangUp(addr)
+  }
+}

src/core/components/swarm/disconnect.js

@@ -0,0 +1,7 @@
+'use strict'
+
+module.exports = ({ peerInfo }) => {
+  return async function localAddrs () { // eslint-disable-line require-await
+    return peerInfo.multiaddrs.toArray()
+  }
+}

src/core/components/swarm/local-addrs.js

@@ -0,0 +1,34 @@
+'use strict'
+
+const CID = require('cids')
+
+module.exports = ({ libp2p }) => {
+  return async function peers (options) { // eslint-disable-line require-await
+    options = options || {}
+
+    const verbose = options.v || options.verbose
+    const peers = []
+
+    for (const [peerId, connections] of libp2p.connections) {
+      for (const connection of connections) {
+        const tupple = {
+          addr: connection.remoteAddr,
+          peer: new CID(peerId)
+        }
+
+        if (verbose || options.direction) {
+          tupple.direction = connection.stat.direction
+        }
+
+        if (verbose) {
+          tupple.muxer = connection.stat.multiplexer
+          tupple.latency = 'n/a'
+        }
+
+        peers.push(tupple)
+      }
+    }
+
+    return peers
+  }
+}