mirror of
https://github.com/logos-messaging/logos-delivery.git
synced 2026-06-06 05:59:33 +00:00
72 lines
2.2 KiB
Nim
72 lines
2.2 KiB
Nim
import chronicles, std/[net, options, sequtils], results
|
|
import ../waku_conf
|
|
|
|
logScope:
|
|
topics = "waku conf builder discv5"
|
|
|
|
const
|
|
DefaultDiscv5Enabled*: bool = false
|
|
DefaultDiscv5BitsPerHop: int = 1
|
|
DefaultDiscv5BucketIpLimit: uint = 2
|
|
DefaultDiscv5EnrAutoUpdate: bool = true
|
|
DefaultDiscv5TableIpLimit: uint = 10
|
|
DefaultDiscv5UdpPort: Port = Port(9000)
|
|
|
|
###########################
|
|
## 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: uint16) =
|
|
b.udpPort = some(Port(udpPort))
|
|
|
|
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(DefaultDiscv5Enabled):
|
|
return ok(none(Discv5Conf))
|
|
|
|
return ok(
|
|
some(
|
|
Discv5Conf(
|
|
bootstrapNodes: b.bootstrapNodes,
|
|
bitsPerHop: b.bitsPerHop.get(DefaultDiscv5BitsPerHop),
|
|
bucketIpLimit: b.bucketIpLimit.get(DefaultDiscv5BucketIpLimit),
|
|
enrAutoUpdate: b.enrAutoUpdate.get(DefaultDiscv5EnrAutoUpdate),
|
|
tableIpLimit: b.tableIpLimit.get(DefaultDiscv5TableIpLimit),
|
|
udpPort: b.udpPort.get(DefaultDiscv5UdpPort),
|
|
)
|
|
)
|
|
)
|