mirror of
https://github.com/logos-messaging/logos-delivery.git
synced 2026-04-18 06:03:07 +00:00
56 lines
1.8 KiB
Nim
56 lines
1.8 KiB
Nim
import chronicles, std/options, results
|
|
import libp2p/crypto/crypto, libp2p/crypto/curve25519, libp2p/protocols/mix/curve25519
|
|
import ../waku_conf, waku/waku_mix
|
|
|
|
logScope:
|
|
topics = "waku conf builder mix"
|
|
|
|
##################################
|
|
## Mix Config Builder ##
|
|
##################################
|
|
type MixConfBuilder* = object
|
|
enabled: Option[bool]
|
|
mixKey: Option[string]
|
|
mixNodes: seq[MixNodePubInfo]
|
|
enableSpamProtection: bool
|
|
userMessageLimit: Option[int]
|
|
|
|
proc init*(T: type MixConfBuilder): MixConfBuilder =
|
|
MixConfBuilder()
|
|
|
|
proc withEnabled*(b: var MixConfBuilder, enabled: bool) =
|
|
b.enabled = some(enabled)
|
|
|
|
proc withMixKey*(b: var MixConfBuilder, mixKey: string) =
|
|
b.mixKey = some(mixKey)
|
|
|
|
proc withMixNodes*(b: var MixConfBuilder, mixNodes: seq[MixNodePubInfo]) =
|
|
b.mixNodes = mixNodes
|
|
|
|
proc withEnableSpamProtection*(b: var MixConfBuilder, enable: bool) =
|
|
b.enableSpamProtection = enable
|
|
|
|
proc withUserMessageLimit*(b: var MixConfBuilder, limit: int) =
|
|
b.userMessageLimit = some(limit)
|
|
|
|
proc build*(b: MixConfBuilder): Result[Option[MixConf], string] =
|
|
if not b.enabled.get(false):
|
|
return ok(none[MixConf]())
|
|
else:
|
|
if b.mixKey.isSome():
|
|
let mixPrivKey = intoCurve25519Key(ncrutils.fromHex(b.mixKey.get()))
|
|
let mixPubKey = public(mixPrivKey)
|
|
return ok(
|
|
some(MixConf(mixKey: mixPrivKey, mixPubKey: mixPubKey, mixnodes: b.mixNodes,
|
|
enableSpamProtection: b.enableSpamProtection,
|
|
userMessageLimit: b.userMessageLimit))
|
|
)
|
|
else:
|
|
let (mixPrivKey, mixPubKey) = generateKeyPair().valueOr:
|
|
return err("Generate key pair error: " & $error)
|
|
return ok(
|
|
some(MixConf(mixKey: mixPrivKey, mixPubKey: mixPubKey, mixnodes: b.mixNodes,
|
|
enableSpamProtection: b.enableSpamProtection,
|
|
userMessageLimit: b.userMessageLimit))
|
|
)
|