Start moving discovery modules to waku/node/discovery_manager/

waku-org · Apr 15, 2024 · 1772a62 · 1772a62
1 parent 480a62f
commit 1772a62
Show file tree

Hide file tree

Showing 16 changed files with 68 additions and 33 deletions.
diff --git a/apps/networkmonitor/networkmonitor.nim b/apps/networkmonitor/networkmonitor.nim
@@ -25,8 +25,8 @@ import
   ../../waku/node/peer_manager,
   ../../waku/waku_node,
   ../../waku/waku_enr,
-  ../../waku/waku_discv5,
-  ../../waku/waku_dnsdisc,
+  ../../waku/node/discovery_manager/waku_discv5,
+  ../../waku/node/discovery_manager/waku_dnsdisc,
   ../../waku/waku_relay,
   ../../waku/waku_rln_relay,
   ../../waku/factory/builder,
@@ -131,12 +131,12 @@ proc setConnectedPeersMetrics(
   var analyzeFuts: seq[Future[Result[string, string]]]
 
   var (inConns, outConns) = node.peer_manager.connectedPeers(WakuRelayCodec)
-  info "connected peers", inConns=inConns.len, outConns=outConns.len
+  info "connected peers", inConns = inConns.len, outConns = outConns.len
 
   shuffle(outConns)
 
-  if outConns.len >= toInt(MaxConnectedPeers/2):
-    for p in outConns[0 ..< toInt(outConns.len/2)]:
+  if outConns.len >= toInt(MaxConnectedPeers / 2):
+    for p in outConns[0 ..< toInt(outConns.len / 2)]:
       trace "Pruning Peer", Peer = $p
       asyncSpawn(node.switch.disconnect(p))
 

diff --git a/apps/wakunode2/app.nim b/apps/wakunode2/app.nim
@@ -40,9 +40,9 @@ import
   ../../waku/waku_api/rest/health/handlers as rest_health_api,
   ../../waku/waku_api/rest/admin/handlers as rest_admin_api,
   ../../waku/waku_archive,
-  ../../waku/waku_dnsdisc,
+  ../../waku/node/discovery_manager/waku_dnsdisc,
+  ../../waku/node/discovery_manager/waku_discv5,
   ../../waku/waku_enr/sharding,
-  ../../waku/waku_discv5,
   ../../waku/waku_peer_exchange,
   ../../waku/waku_rln_relay,
   ../../waku/waku_store,
@@ -366,9 +366,7 @@ proc startRestServer(
       "/relay endpoints are not available. Please check your configuration: --relay"
 
   ## Filter REST API
-  if conf.filternode  != "" and
-     app.node.wakuFilterClient != nil:
-
+  if conf.filternode != "" and app.node.wakuFilterClient != nil:
     let filterCache = MessageCache.init()
 
     let filterDiscoHandler =

diff --git a/tests/node/test_wakunode_peer_exchange.nim b/tests/node/test_wakunode_peer_exchange.nim
@@ -15,7 +15,7 @@ import
 import
   ../../../waku/[
     waku_node,
-    waku_discv5,
+    node/discovery_manager/waku_discv5,
     waku_peer_exchange,
     node/peer_manager,
     waku_relay/protocol,

diff --git a/tests/test_waku_dnsdisc.nim b/tests/test_waku_dnsdisc.nim
@@ -13,7 +13,7 @@ import
 import
   ../../waku/node/peer_manager,
   ../../waku/waku_node,
-  ../../waku/waku_dnsdisc,
+  ../../waku/node/discovery_manager/waku_dnsdisc,
   ./testlib/common,
   ./testlib/wakucore,
   ./testlib/wakunode

diff --git a/tests/testlib/wakunode.nim b/tests/testlib/wakunode.nim
@@ -12,7 +12,7 @@ import
   ../../../waku/waku_node,
   ../../../waku/node/peer_manager,
   ../../../waku/waku_enr,
-  ../../../waku/waku_discv5,
+  ../../../waku/node/discovery_manager/waku_discv5,
   ../../../waku/factory/external_config,
   ../../../waku/factory/internal_config,
   ../../../waku/factory/builder,

diff --git a/tests/waku_peer_exchange/test_protocol.nim b/tests/waku_peer_exchange/test_protocol.nim
@@ -13,7 +13,7 @@ import
   ../../../waku/[
     waku_node,
     node/peer_manager,
-    waku_discv5,
+    node/discovery_manager/waku_discv5,
     waku_peer_exchange,
     waku_peer_exchange/rpc,
     waku_peer_exchange/rpc_codec,

diff --git a/tests/waku_peer_exchange/test_rpc_codec.nim b/tests/waku_peer_exchange/test_rpc_codec.nim
@@ -12,13 +12,12 @@ import
   eth/p2p/discoveryv5/enr
 
 import
-  ../../../waku/
-    [
-      node/peer_manager,
-      waku_discv5,
-      waku_peer_exchange/rpc,
-      waku_peer_exchange/rpc_codec,
-    ],
+  ../../../waku/[
+    node/peer_manager,
+    node/discovery_manager/waku_discv5,
+    waku_peer_exchange/rpc,
+    waku_peer_exchange/rpc_codec,
+  ],
   ../testlib/[wakucore]
 
 suite "Peer Exchange RPC":

diff --git a/tests/waku_peer_exchange/utils.nim b/tests/waku_peer_exchange/utils.nim
@@ -13,7 +13,7 @@ import
 import
   ../../../waku/[
     waku_node,
-    waku_discv5,
+    node/discovery_manager/waku_discv5,
     waku_peer_exchange,
     waku_peer_exchange/rpc,
     waku_peer_exchange/protocol,

diff --git a/waku/factory/builder.nim b/waku/factory/builder.nim
@@ -12,7 +12,8 @@ import
   libp2p/builders,
   libp2p/nameresolving/nameresolver,
   libp2p/transports/wstransport
-import ../waku_enr, ../waku_discv5, ../waku_node, ../node/peer_manager
+import
+  ../waku_enr, ../node/discovery_manager/waku_discv5, ../waku_node, ../node/peer_manager
 
 type
   WakuNodeBuilder* = object # General

diff --git a/waku/factory/node_factory.nim b/waku/factory/node_factory.nim
@@ -16,7 +16,7 @@ import
   ../waku_node,
   ../waku_core,
   ../waku_rln_relay,
-  ../waku_dnsdisc,
+  ../node/discovery_manager/waku_dnsdisc,
   ../waku_archive,
   ../waku_store,
   ../waku_filter_v2,

diff --git a/waku/node/discovery_manager/discovery_manager.nim b/waku/node/discovery_manager/discovery_manager.nim
@@ -0,0 +1,37 @@
+when (NimMajor, NimMinor) < (1, 4):
+  {.push raises: [Defect].}
+else:
+  {.push raises: [].}
+
+import waku_discv5, ../../waku_core
+
+## This module contains the logic needed to discover other peers and
+## also to make the "self" node discoverable by other peers.
+
+type DiscoveryManager* = object
+  wakuDiscv5*: Option[WakuDiscoveryV5]
+  dynamicBootstrapNodes*: seq[RemotePeerInfo]
+
+#[
+    TODO: in future PRs we will have:
+
+    WakuNode* = ref object
+        peerManager*: PeerManager
+        discManager*: DiscoveryManager  <-- we will add this
+        ...
+
+    App* = object
+        version: string
+        conf: WakuNodeConf
+        rng: ref HmacDrbgContext
+        key: crypto.PrivateKey
+
+        wakuDiscv5: Option[WakuDiscoveryV5]         <-- this will get removed
+        dynamicBootstrapNodes: seq[RemotePeerInfo]  <-- this will get removed
+
+        node: WakuNode  <-- this will contain a discManager instance
+
+        restServer: Option[WakuRestServerRef]
+        metricsServer: Option[MetricsHttpServerRef]
+
+    ]#
diff --git a/waku/waku_discv5.nim → waku/node/discovery_manager/waku_discv5.nim b/waku/waku_discv5.nim → waku/node/discovery_manager/waku_discv5.nim
@@ -14,7 +14,7 @@ import
   eth/keys as eth_keys,
   eth/p2p/discoveryv5/node,
   eth/p2p/discoveryv5/protocol
-import ./node/peer_manager/peer_manager, ./waku_core, ./waku_enr
+import ../peer_manager/peer_manager, ../../waku_core, ../../waku_enr
 
 export protocol, waku_enr
 

diff --git a/waku/waku_dnsdisc.nim → waku/node/discovery_manager/waku_dnsdisc.nim b/waku/waku_dnsdisc.nim → waku/node/discovery_manager/waku_dnsdisc.nim
@@ -21,7 +21,7 @@ import
   libp2p/multiaddress,
   libp2p/peerid,
   dnsdisc/client
-import ./waku_core
+import ../../waku_core
 
 export client
 

diff --git a/waku/node/waku_node.nim b/waku/node/waku_node.nim
@@ -41,11 +41,11 @@ import
   ../waku_lightpush/common,
   ../waku_lightpush/protocol,
   ../waku_enr,
-  ../waku_dnsdisc,
   ../waku_peer_exchange,
   ../waku_rln_relay,
   ./config,
-  ./peer_manager
+  ./peer_manager,
+  ./discovery_manager/waku_dnsdisc
 
 declarePublicCounter waku_node_messages, "number of messages received", ["type"]
 declarePublicHistogram waku_histogram_message_size,
@@ -445,8 +445,7 @@ proc filterHandleMessage*(
     node: WakuNode, pubsubTopic: PubsubTopic, message: WakuMessage
 ) {.async.} =
   if node.wakuFilter.isNil():
-    error "cannot handle filter message",
-      error = "waku filter is required"
+    error "cannot handle filter message", error = "waku filter is required"
     return
 
   await node.wakuFilter.handleMessage(pubsubTopic, message)
@@ -545,7 +544,7 @@ proc filterSubscribe*(
 proc filterUnsubscribe*(
     node: WakuNode,
     pubsubTopic: Option[PubsubTopic],
-    contentTopics: ContentTopic|seq[ContentTopic],
+    contentTopics: ContentTopic | seq[ContentTopic],
     peer: RemotePeerInfo | string,
 ): Future[FilterSubscribeResult] {.async, gcsafe, raises: [Defect, ValueError].} =
   ## Unsubscribe from a content filter V2".

diff --git a/waku/waku_api/handlers.nim b/waku/waku_api/handlers.nim
@@ -4,7 +4,8 @@ else:
   {.push raises: [].}
 
 import chronos, chronicles, std/[options, sequtils], stew/results
-import ../waku_discv5, ../waku_relay, ../waku_core, ./message_cache
+import
+  ../node/discovery_manager/waku_discv5, ../waku_relay, ../waku_core, ./message_cache
 
 ### Discovery
 

diff --git a/waku/waku_peer_exchange/protocol.nim b/waku/waku_peer_exchange/protocol.nim
@@ -11,7 +11,7 @@ import
   ../common/nimchronos,
   ../node/peer_manager,
   ../waku_core,
-  ../waku_discv5,
+  ../node/discovery_manager/waku_discv5,
   ./rpc,
   ./rpc_codec