{.used.}

import
  std/[sequtils, tables],
  testutils/unittests,
  stew/shims/net,
  stew/[base32, results],
  chronicles,
  chronos,
  libp2p/crypto/crypto,
  eth/keys,
  discovery/dnsdisc/builder
import
  ../../waku/v2/node/peer_manager/peer_manager,
  ../../waku/v2/node/dnsdisc/waku_dnsdisc,
  ../../waku/v2/node/wakunode2,
  ../test_helpers

procSuite "Waku DNS Discovery":
  asyncTest "Waku DNS Discovery end-to-end":
    ## Tests integrated DNS discovery, from building
    ## the tree to connecting to discovered nodes
    
    # Create nodes and ENR. These will be added to the discoverable list
    let
      bindIp = ValidIpAddress.init("0.0.0.0")
      nodeKey1 = crypto.PrivateKey.random(Secp256k1, rng[])[]
      node1 = WakuNode.new(nodeKey1, bindIp, Port(60000))
      enr1 = node1.enr
      nodeKey2 = crypto.PrivateKey.random(Secp256k1, rng[])[]
      node2 = WakuNode.new(nodeKey2, bindIp, Port(60002))
      enr2 = node2.enr
      nodeKey3 = crypto.PrivateKey.random(Secp256k1, rng[])[]
      node3 = WakuNode.new(nodeKey3, bindIp, Port(60003))
      enr3 = node3.enr
    
    node1.mountRelay()
    node2.mountRelay()
    node3.mountRelay()
    await allFutures([node1.start(), node2.start(), node3.start()])
    
    # Build and sign tree
    var tree = buildTree(1,                   # Seq no
                         @[enr1, enr2, enr3], # ENR entries
                         @[]).get()           # No link entries

    let treeKeys = keys.KeyPair.random(rng[])

    # Sign tree
    check:
      tree.signTree(treeKeys.seckey()).isOk()

    # Create TXT records at domain
    let
      domain = "testnodes.aq"
      zoneTxts = tree.buildTXT(domain).get()
      username = Base32.encode(treeKeys.pubkey().toRawCompressed())
      location = LinkPrefix & username & "@" & domain # See EIP-1459: https://eips.ethereum.org/EIPS/eip-1459
    
    # Create a resolver for the domain

    proc resolver(domain: string): Future[string] {.async, gcsafe.} =
      return zoneTxts[domain]
    
    # Create Waku DNS discovery client on a new Waku v2 node using the resolver

    let
      nodeKey4 = crypto.PrivateKey.random(Secp256k1, rng[])[]
      node4 = WakuNode.new(nodeKey4, bindIp, Port(60004))
    
    node4.mountRelay()
    await node4.start()
    
    var wakuDnsDisc = WakuDnsDiscovery.init(location, resolver).get()

    let res = wakuDnsDisc.findPeers()

    check:
      # We have discovered all three nodes
      res.isOk()
      res[].len == 3
      res[].mapIt(it.peerId).contains(node1.switch.peerInfo.peerId)
      res[].mapIt(it.peerId).contains(node2.switch.peerInfo.peerId)
      res[].mapIt(it.peerId).contains(node3.switch.peerInfo.peerId)
    
    # Connect to discovered nodes
    await node4.connectToNodes(res[])

    check:
      # We have successfully connected to all discovered nodes
      node4.peerManager.peers().anyIt(it.peerId == node1.switch.peerInfo.peerId)
      node4.peerManager.connectedness(node1.switch.peerInfo.peerId) == Connected
      node4.peerManager.peers().anyIt(it.peerId == node2.switch.peerInfo.peerId)
      node4.peerManager.connectedness(node2.switch.peerInfo.peerId) == Connected
      node4.peerManager.peers().anyIt(it.peerId == node3.switch.peerInfo.peerId)
      node4.peerManager.connectedness(node3.switch.peerInfo.peerId) == Connected

    await allFutures([node1.stop(), node2.stop(), node3.stop(), node4.stop()])