mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-01-04 06:53:12 +00:00
64 lines
1.8 KiB
Nim
64 lines
1.8 KiB
Nim
import chronicles, std/[net, options, sequtils], results
|
|
import ../waku_conf
|
|
|
|
logScope:
|
|
topics = "waku conf builder discv5"
|
|
|
|
###########################
|
|
## Discv5 Config Builder ##
|
|
###########################
|
|
type Discv5ConfBuilder* = object
|
|
enabled*: Option[bool]
|
|
|
|
bootstrapNodes*: seq[string]
|
|
bitsPerHop*: Option[int]
|
|
bucketIpLimit*: Option[uint]
|
|
enrAutoUpdate*: Option[bool]
|
|
tableIpLimit*: Option[uint]
|
|
udpPort*: Option[Port]
|
|
|
|
proc init*(T: type Discv5ConfBuilder): Discv5ConfBuilder =
|
|
Discv5ConfBuilder()
|
|
|
|
proc withEnabled*(b: var Discv5ConfBuilder, enabled: bool) =
|
|
b.enabled = some(enabled)
|
|
|
|
proc withBitsPerHop*(b: var Discv5ConfBuilder, bitsPerHop: int) =
|
|
b.bitsPerHop = some(bitsPerHop)
|
|
|
|
proc withBucketIpLimit*(b: var Discv5ConfBuilder, bucketIpLimit: uint) =
|
|
b.bucketIpLimit = some(bucketIpLimit)
|
|
|
|
proc withEnrAutoUpdate*(b: var Discv5ConfBuilder, enrAutoUpdate: bool) =
|
|
b.enrAutoUpdate = some(enrAutoUpdate)
|
|
|
|
proc withTableIpLimit*(b: var Discv5ConfBuilder, tableIpLimit: uint) =
|
|
b.tableIpLimit = some(tableIpLimit)
|
|
|
|
proc withUdpPort*(b: var Discv5ConfBuilder, udpPort: Port) =
|
|
b.udpPort = some(udpPort)
|
|
|
|
proc withUdpPort*(b: var Discv5ConfBuilder, udpPort: uint) =
|
|
b.udpPort = some(Port(udpPort.uint16))
|
|
|
|
proc withBootstrapNodes*(b: var Discv5ConfBuilder, bootstrapNodes: seq[string]) =
|
|
# TODO: validate ENRs?
|
|
b.bootstrapNodes = concat(b.bootstrapNodes, bootstrapNodes)
|
|
|
|
proc build*(b: Discv5ConfBuilder): Result[Option[Discv5Conf], string] =
|
|
if not b.enabled.get(false):
|
|
return ok(none(Discv5Conf))
|
|
|
|
return ok(
|
|
some(
|
|
Discv5Conf(
|
|
bootstrapNodes: b.bootstrapNodes,
|
|
bitsPerHop: b.bitsPerHop.get(1),
|
|
bucketIpLimit: b.bucketIpLimit.get(2),
|
|
enrAutoUpdate: b.enrAutoUpdate.get(true),
|
|
tableIpLimit: b.tableIpLimit.get(10),
|
|
udpPort: b.udpPort.get(9000.Port),
|
|
)
|
|
)
|
|
)
|