53 lines
1.7 KiB
Nim
53 lines
1.7 KiB
Nim
{.used.}
|
|
|
|
import json, typetraits, tables, sequtils
|
|
|
|
type
|
|
Fleet* {.pure.} = enum
|
|
Prod = "eth.prod",
|
|
Staging = "eth.staging",
|
|
Test = "eth.test",
|
|
WakuV2Prod = "wakuv2.prod"
|
|
WakuV2Test = "wakuv2.test"
|
|
|
|
FleetNodes* {.pure.} = enum
|
|
Bootnodes = "boot",
|
|
Mailservers = "mail",
|
|
Rendezvous = "rendezvous",
|
|
Whisper = "whisper",
|
|
Waku = "waku"
|
|
|
|
FleetMeta* = object
|
|
hostname*: string
|
|
timestamp*: uint64
|
|
|
|
FleetConfig* = object
|
|
fleet*: Table[string, Table[string, Table[string, string]]]
|
|
meta*: FleetMeta
|
|
|
|
|
|
proc toFleetConfig*(jsonString: string): FleetConfig =
|
|
let fleetJson = jsonString.parseJSON
|
|
result.meta.hostname = fleetJson["meta"]["hostname"].getStr
|
|
result.meta.timestamp = fleetJson["meta"]["timestamp"].getBiggestInt.uint64
|
|
result.fleet = initTable[string, Table[string, Table[string, string]]]()
|
|
|
|
for fleet in fleetJson["fleets"].keys():
|
|
result.fleet[fleet] = initTable[string, Table[string, string]]()
|
|
for nodes in fleetJson["fleets"][fleet].keys():
|
|
result.fleet[fleet][nodes] = initTable[string, string]()
|
|
for server in fleetJson["fleets"][fleet][nodes].keys():
|
|
result.fleet[fleet][nodes][server] = fleetJson["fleets"][fleet][nodes][server].getStr
|
|
|
|
|
|
proc getNodes*(self: FleetConfig, fleet: Fleet, nodeType: FleetNodes = FleetNodes.Bootnodes): seq[string] =
|
|
if not self.fleet[$fleet].hasKey($nodeType): return
|
|
result = toSeq(self.fleet[$fleet][$nodeType].values)
|
|
|
|
proc getMailservers*(self: FleetConfig, fleet: Fleet): Table[string, string] =
|
|
if not self.fleet[$fleet].hasKey($FleetNodes.Mailservers):
|
|
result = initTable[string,string]()
|
|
return
|
|
result = self.fleet[$fleet][$FleetNodes.Mailservers]
|
|
|