import std/options, stew/results, stew/shims/net, chronos, libp2p/switch, libp2p/builders, libp2p/nameresolving/nameresolver, libp2p/crypto/crypto as libp2p_keys, eth/keys as eth_keys import ../../../waku/waku_node, ../../../waku/node/peer_manager, ../../../waku/waku_enr, ../../../waku/waku_discv5, ../../apps/wakunode2/external_config, ../../apps/wakunode2/internal_config, ./common # Waku node proc defaultTestWakuNodeConf*(): WakuNodeConf = WakuNodeConf( cmd: noCommand, tcpPort: Port(60000), websocketPort: Port(8000), listenAddress: ValidIpAddress.init("0.0.0.0"), rpcAddress: ValidIpAddress.init("127.0.0.1"), restAddress: ValidIpAddress.init("127.0.0.1"), metricsServerAddress: ValidIpAddress.init("127.0.0.1"), dnsAddrsNameServers: @[ValidIpAddress.init("1.1.1.1"), ValidIpAddress.init("1.0.0.1")], nat: "any", maxConnections: 50, topics: @[], relay: true ) proc newTestWakuNode*(nodeKey: crypto.PrivateKey, bindIp: ValidIpAddress, bindPort: Port, extIp = none(ValidIpAddress), extPort = none(Port), extMultiAddrs = newSeq[MultiAddress](), peerStorage: PeerStorage = nil, maxConnections = builders.MaxConnections, wsBindPort: Port = (Port)8000, wsEnabled: bool = false, wssEnabled: bool = false, secureKey: string = "", secureCert: string = "", wakuFlags = none(CapabilitiesBitfield), nameResolver: NameResolver = nil, sendSignedPeerRecord = false, dns4DomainName = none(string), discv5UdpPort = none(Port), agentString = none(string), clusterId: uint32 = 2.uint32, topics: seq[string] = @["/waku/2/rs/2/0"], peerStoreCapacity = none(int)): WakuNode = var resolvedExtIp = extIp # Update extPort to default value if it's missing and there's an extIp or a DNS domain let extPort = if (extIp.isSome() or dns4DomainName.isSome()) and extPort.isNone(): some(Port(60000)) else: extPort let conf = defaultTestWakuNodeConf() if dns4DomainName.isSome() and extIp.isNone(): # If there's an error resolving the IP, an exception is thrown and test fails let dns = (waitFor dnsResolve(dns4DomainName.get(), conf)).valueOr: raise newException(Defect, error) resolvedExtIp = some(ValidIpAddress.init(dns)) let netConf = NetConfig.init( bindIp = bindIp, clusterId = clusterId, bindPort = bindPort, extIp = resolvedExtIp, extPort = extPort, extMultiAddrs = extMultiAddrs, wsBindPort = wsBindPort, wsEnabled = wsEnabled, wssEnabled = wssEnabled, wakuFlags = wakuFlags, dns4DomainName = dns4DomainName, discv5UdpPort = discv5UdpPort, ).valueOr: raise newException(Defect, "Invalid network configuration: " & error) var enrBuilder = EnrBuilder.init(nodeKey) enrBuilder.withShardedTopics(topics).isOkOr: raise newException(Defect, "Invalid record: " & error) enrBuilder.withIpAddressAndPorts( ipAddr = netConf.enrIp, tcpPort = netConf.enrPort, udpPort = netConf.discv5UdpPort, ) enrBuilder.withMultiaddrs(netConf.enrMultiaddrs) if netConf.wakuFlags.isSome(): enrBuilder.withWakuCapabilities(netConf.wakuFlags.get()) let record = enrBuilder.build().valueOr: raise newException(Defect, "Invalid record: " & $error) var builder = WakuNodeBuilder.init() builder.withRng(rng()) builder.withNodeKey(nodeKey) builder.withRecord(record) builder.withNetworkConfiguration(netConf) builder.withPeerStorage(peerStorage, capacity = peerStoreCapacity) builder.withSwitchConfiguration( maxConnections = some(maxConnections), nameResolver = nameResolver, sendSignedPeerRecord = sendSignedPeerRecord, secureKey = if secureKey != "": some(secureKey) else: none(string), secureCert = if secureCert != "": some(secureCert) else: none(string), agentString = agentString, ) return builder.build().get()