mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-01-02 14:03:06 +00:00
bc8acf7611
* introduce createNode # Conflicts: # apps/wakunode2/cli_args.nim * remove confutils dependency on the library * test: remove websocket in default test config * update to latest specs * test: cli_args * align to spec changes (sovereign, message conf, entrypoints * accept enr, entree and multiaddr as entry points * post rebase * format * change from "sovereign" to "core" * add example * get example to continue running * nitpicks * idiomatic constructors * fix enum naming * replace procs with consts * remove messageConfirmation * use pure enum * rename example file
78 lines
2.5 KiB
Nim
78 lines
2.5 KiB
Nim
import std/strutils
|
|
|
|
import results, eth/p2p/discoveryv5/enr
|
|
|
|
import waku/waku_core/peers
|
|
|
|
type EntryNodeType {.pure.} = enum
|
|
EnrTree
|
|
Enr
|
|
Multiaddr
|
|
|
|
proc classifyEntryNode(address: string): Result[EntryNodeType, string] =
|
|
## Classifies an entry node address by its type
|
|
## Returns the type as EntryNodeType enum
|
|
if address.len == 0:
|
|
return err("Empty entry node address")
|
|
|
|
let lowerAddress = address.toLowerAscii()
|
|
if lowerAddress.startsWith("enrtree:"):
|
|
return ok(EnrTree)
|
|
elif lowerAddress.startsWith("enr:"):
|
|
return ok(Enr)
|
|
elif address[0] == '/':
|
|
return ok(Multiaddr)
|
|
else:
|
|
return
|
|
err("Unrecognized entry node format. Must start with 'enrtree:', 'enr:', or '/'")
|
|
|
|
proc parseEnrToMultiaddrs(enrStr: string): Result[seq[string], string] =
|
|
## Parses an ENR string and extracts multiaddresses from it
|
|
let enrRec = enr.Record.fromURI(enrStr).valueOr:
|
|
return err("Invalid ENR record")
|
|
|
|
let remotePeerInfo = toRemotePeerInfo(enrRec).valueOr:
|
|
return err("Failed to convert ENR to peer info: " & $error)
|
|
|
|
# Convert RemotePeerInfo addresses to multiaddr strings
|
|
var multiaddrs: seq[string]
|
|
for addr in remotePeerInfo.addrs:
|
|
multiaddrs.add($addr & "/p2p/" & $remotePeerInfo.peerId)
|
|
|
|
if multiaddrs.len == 0:
|
|
return err("No valid addresses found in ENR")
|
|
|
|
return ok(multiaddrs)
|
|
|
|
proc processEntryNodes*(
|
|
entryNodes: seq[string]
|
|
): Result[(seq[string], seq[string], seq[string]), string] =
|
|
## Processes entry nodes and returns (enrTreeUrls, bootstrapEnrs, staticNodes)
|
|
## ENRTree URLs for DNS discovery, ENR records for bootstrap, multiaddrs for static nodes
|
|
var enrTreeUrls: seq[string]
|
|
var bootstrapEnrs: seq[string]
|
|
var staticNodes: seq[string]
|
|
|
|
for node in entryNodes:
|
|
let nodeType = classifyEntryNode(node).valueOr:
|
|
return err("Entry node error: " & error)
|
|
|
|
case nodeType
|
|
of EnrTree:
|
|
# ENRTree URLs go to DNS discovery configuration
|
|
enrTreeUrls.add(node)
|
|
of Enr:
|
|
# ENR records go to bootstrap nodes for discv5
|
|
bootstrapEnrs.add(node)
|
|
# Additionally, extract multiaddrs for static connections
|
|
let multiaddrsRes = parseEnrToMultiaddrs(node)
|
|
if multiaddrsRes.isOk():
|
|
for maddr in multiaddrsRes.get():
|
|
staticNodes.add(maddr)
|
|
# If we can't extract multiaddrs, just use it as bootstrap (already added above)
|
|
of Multiaddr:
|
|
# Multiaddresses go to static nodes
|
|
staticNodes.add(node)
|
|
|
|
return ok((enrTreeUrls, bootstrapEnrs, staticNodes))
|