import DHTOnlineBase from "./dhtOnlineBase.js"; import DHTFlood from "@lumeweb/dht-flood"; const DISCONNECT_SMOOTH = 500; export default class DHTOnline extends DHTOnlineBase { private flood: DHTFlood; private swarm: any; constructor( swarm: any, { id = swarm.keyPair.publicKey, data = {}, ...opts } = {} ) { super(id, opts); this.flood = new DHTFlood({ id, swarm, ...opts }); this.swarm = swarm; this.flood.on("peer-open", (peer) => this.handlePeerAdd(peer)); this.flood.on("peer-remove", (peer) => this.handlePeerRemove(peer)); this.flood.on("message", (message, id) => this.onGetBroadcast(message, id)); this.swarm.on("connection", (peer: any) => this.flood.send(peer, Buffer.from("hello"), 0) ); this.data = data; [...this.swarm.peers.values()].forEach(this.handlePeerAdd.bind(this)); } handlePeerAdd(peer: any) { const id = peer.remotePublicKey; this.onAddPeer(id); } handlePeerRemove(peer: any) { const id = peer.remotePublicKey; // Wait for a bit and check if we're still disconnected before removing the peer setTimeout(() => { if (this.swarm._allConnections.has(id)) { return; } this.onRemovePeer(id); }, DISCONNECT_SMOOTH); } broadcast(message: any, ttl?: number) { this.flood.broadcast(message, ttl); } }