nim-codex/tests/dagger/bitswap/utils.nim

import std/sequtils

import pkg/chronos
import pkg/libp2p

import pkg/dagger/bitswap
import pkg/dagger/stores/memorystore
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, bitswap: Bitswap]] =

  for i in 0..<num:
    let
      switch = newStandardSwitch(transportFlags = {ServerFlags.ReuseAddr})
      wallet = WalletRef.example
      network = BitswapNetwork.new(switch)
      bitswap = Bitswap.new(MemoryStore.new(blocks), wallet, network)

    switch.mount(network)

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

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)