fixup! fix: Remove event listener once done

waku-org · Jul 20, 2022 · be48882 · be48882
1 parent 9b071e6
commit be48882
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/lib/waku.ts b/src/lib/waku.ts
@@ -324,10 +324,16 @@ export class Waku {
 
         await new Promise<void>((resolve) => {
           const cb = (evt: CustomEvent<PeerProtocolsChangeData>): void => {
-            if (evt.detail.protocols.includes(StoreCodecs)) {
-              log("Resolving for", StoreCodecs, evt.detail.protocols);
-              this.libp2p.peerStore.removeEventListener("change:protocols", cb);
-              resolve();
+            for (const codec of Object.values(StoreCodecs)) {
+              if (evt.detail.protocols.includes(codec)) {
+                log("Resolving for", StoreCodecs, evt.detail.protocols);
+                this.libp2p.peerStore.removeEventListener(
+                  "change:protocols",
+                  cb
+                );
+                resolve();
+                break;
+              }
             }
           };
           this.libp2p.peerStore.addEventListener("change:protocols", cb);