nim-codex/tests/dagger/helpers/nodeutils.nim

import std/sequtils

import pkg/chronos
import pkg/libp2p

import pkg/dagger/stores
import pkg/dagger/blocktype as bt

import ../examples

proc generateNodes*(
  num: Natural,
  blocks: openArray[bt.Block] = [],
  secureManagers: openarray[SecureProtocol] = [
    SecureProtocol.Noise,
  ]): seq[tuple[switch: Switch, blockexc: NetworkStore]] =
  for i in 0..<num:
    let
      switch = newStandardSwitch(transportFlags = {ServerFlags.ReuseAddr})
      wallet = WalletRef.example
      network = BlockExcNetwork.new(switch)
      localStore = MemoryStore.new(blocks.mapIt( it ))
      engine = BlockExcEngine.new(localStore, wallet, network)
      networkStore = NetworkStore.new(engine, localStore)

    switch.mount(network)

    # initialize our want lists
    engine.wantList = blocks.mapIt( it.cid )
    switch.mount(network)
    result.add((switch, networkStore))

proc connectNodes*(nodes: seq[Switch]) {.async.} =
  for dialer in nodes:
    for node in nodes:
      if dialer.peerInfo.peerId != node.peerInfo.peerId:
        await dialer.connect(node.peerInfo.peerId, node.peerInfo.addrs)