logos-messaging-nim/tests/waku_enr/utils.nim

import
  std/options,
  sequtils,
  results,
  stew/shims/net,
  chronos,
  libp2p/crypto/crypto as libp2p_keys,
  eth/keys as eth_keys

import waku/[waku_enr, discovery/waku_discv5, waku_enr/sharding], ../testlib/wakucore

proc newTestEnrRecord*(
    privKey: libp2p_keys.PrivateKey,
    extIp: string,
    tcpPort: uint16,
    udpPort: uint16,
    indices: seq[uint64] = @[],
    flags = none(CapabilitiesBitfield),
): waku_enr.Record =
  var builder = EnrBuilder.init(privKey)
  builder.withIpAddressAndPorts(
    ipAddr = some(parseIpAddress(extIp)),
    tcpPort = some(Port(tcpPort)),
    udpPort = some(Port(udpPort)),
  )

  if indices.len > 0:
    let
      byteSeq: seq[byte] = indices.mapIt(cast[byte](it))
      relayShards = fromIndicesList(byteSeq).get()
    discard builder.withWakuRelayShardingIndicesList(relayShards)

  if flags.isSome():
    builder.withWakuCapabilities(flags.get())

  builder.build().tryGet()