feat: libp2p: allows passing constructor for the libp2p modules

pgte · Jun 10, 2018 · bab0910 · bab0910
1 parent cd449ff
commit bab0910
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/core/components/libp2p.js b/src/core/components/libp2p.js
@@ -19,7 +19,7 @@ module.exports = function libp2p (self) {
           mdns: get(config, 'Discovery.MDNS.Enabled'),
           webRTCStar: get(config, 'Discovery.webRTCStar.Enabled'),
           bootstrap: get(config, 'Bootstrap'),
-          modules: self._libp2pModules,
+          modules: prepareModules(self, self._libp2pModules),
           // EXPERIMENTAL
           pubsub: get(self._options, 'EXPERIMENTAL.pubsub', false),
           dht: get(self._options, 'EXPERIMENTAL.dht', false),
@@ -69,3 +69,10 @@ module.exports = function libp2p (self) {
     })
   }
 }
+
+function prepareModules (self, modules) {
+  if (typeof modules === 'function') {
+    return modules(self._peerInfo)
+  }
+  return modules
+}
diff --git a/src/core/config.js b/src/core/config.js
@@ -36,7 +36,7 @@ const schema = Joi.object().keys({
     Bootstrap: Joi.array().items(Joi.multiaddr().IPFS().options({ convert: false }))
   }).allow(null),
   libp2p: Joi.object().keys({
-    modules: Joi.object().allow(null) // TODO: schemas for libp2p modules?
+    modules: Joi.any().allow(null) // TODO: schemas for libp2p modules?
   }).allow(null)
 }).options({ allowUnknown: true })