nwaku/tools/simulation/quicksim2.nim

import
  std/[os, strutils, times, options], #options as what # TODO: Huh? Redefinition?
  chronicles, 
  eth/common as eth_common, 
  eth/keys,
  json_rpc/[rpcclient, rpcserver],
  libp2p/protobuf/minprotobuf
import
  ../../waku/v2/protocol/waku_filter/rpc,
  ../../waku/v2/protocol/waku_store/rpc,
  ../../waku/v2/protocol/waku_message,
  ../../waku/v2/utils/time,
  ../../waku/v2/node/waku_node, 
  ../../waku/v2/node/waku_payload,
  ../../waku/v2/node/jsonrpc/[jsonrpc_types,jsonrpc_utils]
  

from strutils import rsplit
template sourceDir: string = currentSourcePath.rsplit(DirSep, 1)[0]

const sigWakuPath = sourceDir / ".." / ".." / "waku" / "v2" / "node" / "jsonrpc" / "jsonrpc_callsigs.nim"
createRpcSigs(RpcHttpClient, sigWakuPath)

const topicAmount = 10 #100

proc message(i: int): ProtoBuffer =
  let value = "hello " & $(i)

  var result = initProtoBuffer()
  result.write(1, value)
  result.finish()

proc handler(topic: string, data: seq[byte]) {.async, gcsafe.} =
  debug "Hit handler", topic=topic, data=data

# Scenario xx1 - 16 full nodes
#########################################
let amount = 16
var nodes: seq[RPCHttpClient]
for i in 0..<amount:
  var node = newRpcHttpClient()
  nodes.add(node)
  waitFor nodes[i].connect("localhost", Port(8547+i), false)
  var res = waitFor nodes[i].post_waku_v2_relay_v1_subscriptions(@[DefaultPubsubTopic])

os.sleep(2000)

# # TODO: Show plaintext message in log
# for i in 0..<topicAmount:
#   os.sleep(50)
#   # TODO: This would then publish on a subtopic here
#   var s = "hello " & $2
#   var res3 = waitFor nodes[0].wakuPublish(DefaultPbsubTopic, s)

# Scenario xx3 - same as xx1 but publish from multiple nodes
# To compare FloodSub and GossipSub factor
for i in 0..<topicAmount:
  os.sleep(50)
  # TODO: This would then publish on a subtopic here
  var res3 = waitFor nodes[0].post_waku_v2_relay_v1_message(DefaultPubsubTopic, WakuRelayMessage(payload: message(0).buffer, contentTopic: some(DefaultContentTopic), timestamp: some(getNanosecondTime(epochTime()))))
  res3 = waitFor nodes[1].post_waku_v2_relay_v1_message(DefaultPubsubTopic, WakuRelayMessage(payload: message(1).buffer, contentTopic: some(DefaultContentTopic), timestamp: some(getNanosecondTime(epochTime()))))
  res3 = waitFor nodes[2].post_waku_v2_relay_v1_message(DefaultPubsubTopic, WakuRelayMessage(payload: message(2).buffer, contentTopic: some(DefaultContentTopic), timestamp: some(getNanosecondTime(epochTime()))))
  res3 = waitFor nodes[3].post_waku_v2_relay_v1_message(DefaultPubsubTopic, WakuRelayMessage(payload: message(3).buffer, contentTopic: some(DefaultContentTopic), timestamp: some(getNanosecondTime(epochTime()))))
  res3 = waitFor nodes[4].post_waku_v2_relay_v1_message(DefaultPubsubTopic, WakuRelayMessage(payload: message(4).buffer, contentTopic: some(DefaultContentTopic), timestamp: some(getNanosecondTime(epochTime()))))

# Scenario xx2 - 14 full nodes, two edge nodes
# Assume one full topic
#########################################
#let nodea = newRpcHttpClient()
#let nodeb = newRpcHttpClient()
#
#waitFor nodea.connect("localhost", Port(8545))
#waitFor nodeb.connect("localhost", Port(8546))
#
#let version = waitFor nodea.wakuVersion()
#info "Version is", version
#
#let res1 = waitFor nodea.wakuSubscribe(DefaultPbsubTopic)
#let res2 = waitFor nodeb.wakuSubscribe(DefaultPbsubTopic)
#
#let amount = 14
#var nodes: seq[RPCHttpClient]
#for i in 0..<amount:
#  var node = newRpcHttpClient()
#  nodes.add(node)
#  waitFor nodes[i].connect("localhost", Port(8547+i))
#  var res = waitFor nodes[i].wakuSubscribe(DefaultPbsubTopic)
#
#os.sleep(2000)
#
## TODO: Show plaintext message in log
#for i in 0..<topicAmount:
#  os.sleep(50)
#  # TODO: This would then publish on a subtopic here
#  var s = "hello " & $2
#  var res3 = waitFor nodea.wakuPublish(DefaultPbsubTopic, s)

# Misc old scenarios
#########################################

# All full nodes connected etc
#
# let node1 = newRpcHttpClient()
# let node2 = newRpcHttpClient()
# let node3 = newRpcHttpClient()
# let node4 = newRpcHttpClient()
# let node5 = newRpcHttpClient()
# let node6 = newRpcHttpClient()

# waitFor node1.connect("localhost", Port(8547))
# waitFor node2.connect("localhost", Port(8548))
# waitFor node3.connect("localhost", Port(8549))
# waitFor node4.connect("localhost", Port(8550))
# waitFor node5.connect("localhost", Port(8551))
# waitFor node6.connect("localhost", Port(8552))

# let version = waitFor node6.wakuVersion()
# info "Version is", version

# # TODO: Implement handler logic
# # All subscribing to foobar topic
# let res2 = waitFor node2.wakuSubscribe("foobar")
# let res3 = waitFor node3.wakuSubscribe("foobar")
# let res4 = waitFor node4.wakuSubscribe("foobar")
# let res5 = waitFor node5.wakuSubscribe("foobar")
# let res6 = waitFor node6.wakuSubscribe("foobar")
# os.sleep(2000)

# # info "Posting envelopes on all subscribed topics"
# for i in 0..<topicAmount:
#   os.sleep(50)
#   let res2 = waitFor node1.wakuPublish("foobar", "hello world")

# os.sleep(2000)

# for i in 0..<topicAmount:
#   os.sleep(50)
#   let res2 = waitFor node1.wakuPublish("foobar", "hello world2")

# Node 00 and 05 also subscribe
# XXX I confirm this works. As in - with this we have A-B
#  Now to tweak it!
# let node0 = newRpcHttpClient()
# let node5 = newRpcHttpClient()
# waitFor node0.connect("localhost", Port(8547))
# waitFor node5.connect("localhost", Port(8552))
# let res4 = waitFor node0.wakuSubscribe("foobar")
# let res5 = waitFor node5.wakuSubscribe("foobar")