*Update dist

dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;AAAA,OAAO,YAAY,MAAM,QAAQ,CAAC;AAkBlC,eAAO,MAAM,YAAY,eAAuB,CAAC;AAEjD,MAAM,CAAC,OAAO,OAAO,QAAS,SAAQ,YAAY;IAChD,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,GAAG,CAAS;IACpB,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,GAAG,CAAM;IACjB,OAAO,CAAC,KAAK,CAAM;IACnB,OAAO,CAAC,QAAQ,CAAS;gBAEb,EACV,OAAkB,EAClB,GAAS,EACT,aAAiB,EACjB,EAA2B,EAC3B,KAAY,EACZ,QAAmB,GACpB;;;;;;;KAAK;IAmBN,OAAO,CAAC,aAAa;IAkCrB,OAAO,CAAC,SAAS;IA+CjB,SAAS,CAAC,IAAI,EAAE,GAAG,EAAE,GAAG,SAAW;IAenC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,SAAW;CAY1C"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;AAAA,OAAO,YAAY,MAAM,QAAQ,CAAC;AAkBlC,eAAO,MAAM,YAAY,eAAuB,CAAC;AAIjD,MAAM,CAAC,OAAO,OAAO,QAAS,SAAQ,YAAY;IAChD,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,GAAG,CAAS;IACpB,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,GAAG,CAAM;IACjB,OAAO,CAAC,KAAK,CAAM;IACnB,OAAO,CAAC,QAAQ,CAAS;gBAEb,EACV,OAAkB,EAClB,GAAS,EACT,aAAiB,EACjB,EAA2B,EAC3B,KAAY,EACZ,QAAmB,GACpB;;;;;;;KAAK;IAmBN,OAAO,CAAC,aAAa;IAkCrB,OAAO,CAAC,SAAS;IAqDjB,SAAS,CAAC,IAAI,EAAE,GAAG,EAAE,GAAG,SAAW;IAenC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,SAAW;CAY1C"}

dist/index.js

@@ -20,6 +20,7 @@ const LRU_SIZE = 255;
 const TTL = 255;
 const PROTOCOL = "lumeweb.flood";
 exports.FLOOD_SYMBOL = Symbol.for(PROTOCOL);
+const closedMap = new Map();
 class DHTFlood extends events_1.default {
     id;
     ttl;
@@ -74,14 +75,19 @@ class DHTFlood extends events_1.default {
                 async onopen() {
                     self.emit("peer-open", peer);
                 },
-                async ondestroy() {
-                    self.emit("peer-remove", peer);
-                },
             });
             if (chan) {
                 peer[exports.FLOOD_SYMBOL] = chan;
             }
         }
+        if (!closedMap.has(peer)) {
+            const close = () => {
+                self.emit("peer-remove", peer);
+                peer.removeListener("close", close);
+                closedMap.delete(peer);
+            };
+            peer.on("close", close);
+        }
         chan = peer[exports.FLOOD_SYMBOL];
         if (!chan) {
             throw new Error("could not find channel");

src/index.ts

@@ -18,6 +18,8 @@ const PROTOCOL = "lumeweb.flood";
 
 export const FLOOD_SYMBOL = Symbol.for(PROTOCOL);
 
+const closedMap = new Map();
+
 export default class DHTFlood extends EventEmitter {
   private id: Buffer;
   private ttl: number;
@@ -91,22 +93,28 @@ export default class DHTFlood extends EventEmitter {
     let chan: any;
 
     const self = this;
-
     if (!mux.opened({ protocol: this.protocol })) {
       chan = mux.createChannel({
         protocol: this.protocol,
         async onopen() {
           self.emit("peer-open", peer);
         },
-        async ondestroy() {
-          self.emit("peer-remove", peer);
-        },
       });
       if (chan) {
         peer[FLOOD_SYMBOL] = chan;
       }
     }
 
+    if (!closedMap.has(peer)) {
+      const close = () => {
+        self.emit("peer-remove", peer);
+        peer.removeListener("close", close);
+        closedMap.delete(peer);
+      };
+
+      peer.on("close", close);
+    }
+
     chan = peer[FLOOD_SYMBOL];
 
     if (!chan) {

test.js

@@ -1,62 +1,61 @@
-const test = require('tape')
-const Hyperswarm = require('hyperswarm')
-const sodium = require('sodium-universal')
-const b4a = require('b4a')
-const { default: DHTFlood } = require('./')
-const crypto = require('crypto')
+const test = require("tape");
+const Hyperswarm = require("hyperswarm");
+const sodium = require("sodium-universal");
+const b4a = require("b4a");
+const { default: DHTFlood } = require("./");
+const crypto = require("crypto");
 
-const topicName = crypto.randomBytes(10)
+const topicName = crypto.randomBytes(10);
 
-test('Broadcast through several peers', (t) => {
-    const peer1 = createPeer()
-    const peer2 = createPeer()
-    const peer3 = createPeer()
-    t.plan(2)
+test("Broadcast through several peers", (t) => {
+  const peer1 = createPeer();
+  const peer2 = createPeer();
+  const peer3 = createPeer();
+  t.plan(2);
 
   Promise.all([peer1, peer2, peer3]).then((peers) => {
-        const peer1 = peers.shift()
-        const peer2 = peers.shift()
-        const peer3 = peers.shift()
+    const peer1 = peers.shift();
+    const peer2 = peers.shift();
+    const peer3 = peers.shift();
 
-        const flood1 = new DHTFlood({ swarm: peer1 })
-        const flood2 = new DHTFlood({ swarm: peer2 })
-        const flood3 = new DHTFlood({ swarm: peer3 })
-        const data = Buffer.from('Hello World')
+    const flood1 = new DHTFlood({ swarm: peer1 });
+    const flood2 = new DHTFlood({ swarm: peer2 });
+    const flood3 = new DHTFlood({ swarm: peer3 });
+    const data = Buffer.from("Hello World");
 
-        flood1.on('message', () => t.error('Got own message'))
+    flood1.on("message", () => t.error("Got own message"));
 
-        flood2.on('message', (message) => {
-            t.deepEquals(message, data, 'Data got  broadcast')
-        })
-        flood3.on('message', (message) => {
-            t.deepEquals(message, data, 'Data got  broadcast')
-        })
+    flood2.on("message", (message) => {
+      t.deepEquals(message, data, "Data got  broadcast");
+    });
+    flood3.on("message", (message) => {
+      t.deepEquals(message, data, "Data got  broadcast");
+    });
 
-        function maybeFlood () {
+    function maybeFlood() {
       if (peer1.peers.size === 2) {
-                flood1.broadcast(data)
+        flood1.broadcast(data);
+        peer1.removeListener("connection", maybeFlood);
       }
     }
 
-        peer1.on('connection', () => {
-            maybeFlood()
-        })
+    peer1.on("connection", maybeFlood);
 
     t.teardown(() => {
-            [peer1, peer2, peer3].forEach((item) => item.destroy())
-        })
-    })
-})
+      [peer1, peer2, peer3].forEach((item) => item.destroy());
+    });
+  });
+});
 
-async function createPeer () {
-    const swarm = new Hyperswarm()
-    await swarm.dht.ready()
-    await swarm.listen()
+async function createPeer() {
+  const swarm = new Hyperswarm();
+  await swarm.dht.ready();
+  await swarm.listen();
 
-    const topic = b4a.allocUnsafe(32)
-    sodium.crypto_generichash(topic, b4a.from(topicName))
+  const topic = b4a.allocUnsafe(32);
+  sodium.crypto_generichash(topic, b4a.from(topicName));
 
-    swarm.join(topic)
+  swarm.join(topic);
 
-    return swarm
+  return swarm;
 }