import { bootstrap } from "@libp2p/bootstrap";
import { enrTree, wakuDnsDiscovery } from "@waku/dns-discovery";
import { LightNode } from "@waku/interfaces";
import { createLightNode } from "@waku/sdk";
import { expect } from "chai";

import { makeLogFileName, NimGoNode } from "../src/index.js";

describe("Use static and several ENR trees for bootstrap", function () {
  let waku: LightNode;
  let nwaku: NimGoNode;

  afterEach(async function () {
    !!nwaku && (await nwaku.stop());
    !!waku && waku.stop().catch((e) => console.log("Waku failed to stop", e));
  });

  it("", async function () {
    this.timeout(10_000);

    nwaku = new NimGoNode(makeLogFileName(this));
    await nwaku.start();
    const multiAddrWithId = await nwaku.getMultiaddrWithId();

    const NODE_REQUIREMENTS = {
      store: 3,
      lightPush: 3,
      filter: 3
    };

    waku = await createLightNode({
      libp2p: {
        peerDiscovery: [
          bootstrap({ list: [multiAddrWithId.toString()] }),
          wakuDnsDiscovery(
            [enrTree["PROD"], enrTree["TEST"]],
            NODE_REQUIREMENTS
          )
        ]
      }
    });
    await waku.start();

    const peersDiscovered = await waku.libp2p.peerStore.all();

    // 3 from DNS Disc, 1 from bootstrap
    expect(peersDiscovered.length).to.eq(3 + 1);
    // should also have the bootstrap peer
    expect(
      peersDiscovered.find(
        (p) => p.id.toString() === multiAddrWithId.getPeerId()?.toString()
      )
    ).to.not.be.undefined;
  });
});