feat: map correct fleet according to the pubsub topic configured

waku-org · May 14, 2024 · a730d5a · a730d5a
1 parent 6f188ef
commit a730d5a
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 41 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/sdk/src/utils/discovery.ts b/packages/sdk/src/utils/discovery.ts
@@ -5,7 +5,11 @@ import {
   wakuLocalPeerCacheDiscovery,
   wakuPeerExchangeDiscovery
 } from "@waku/discovery";
-import { type Libp2pComponents, PubsubTopic } from "@waku/interfaces";
+import {
+  DefaultPubsubTopic,
+  type Libp2pComponents,
+  PubsubTopic
+} from "@waku/interfaces";
 
 const DEFAULT_NODE_REQUIREMENTS = {
   lightPush: 1,
@@ -16,8 +20,13 @@ const DEFAULT_NODE_REQUIREMENTS = {
 export function defaultPeerDiscoveries(
   pubsubTopics: PubsubTopic[]
 ): ((components: Libp2pComponents) => PeerDiscovery)[] {
+  // TODO: add a check to see if it is indeed TWN or if it is a custom network
+  const dnsFleet = pubsubTopics.includes(DefaultPubsubTopic)
+    ? enrTree["TEST"]
+    : enrTree["SANDBOX"];
+
   const discoveries = [
-    wakuDnsDiscovery([enrTree["SANDBOX"]], DEFAULT_NODE_REQUIREMENTS),
+    wakuDnsDiscovery([dnsFleet], DEFAULT_NODE_REQUIREMENTS),
     wakuLocalPeerCacheDiscovery(),
     wakuPeerExchangeDiscovery(pubsubTopics)
   ];