import
  chronicles, chronicles/[topics_registry, timings],
  confutils, confutils/std/net,
  ../beacon_chain/spec/eth2_apis/rest_beacon_client

type
  Config = object
    serverIpAddress {.
      defaultValue: ValidIpAddress.init("127.0.0.1")
      defaultValueDesc: "127.0.0.1"
      desc: "IP address of the beacon node's REST server"
      abbr: "a"
      name: "address" }: ValidIpAddress

    serverPort {.
      defaultValue: 5052
      desc: "Listening port of the beacon node's REST server"
      abbr: "p"
      name: "port" }: Port

    startSlot {.
      defaultValue: 0
      desc: "The starting slot from which to start history traversal"
      abbr: "s"
      name: "start-slot" }: uint

    requestsCount {.
      desc: "Number of requests to send to the beacon node's REST server"
      abbr: "n"
      name: "count" }: uint

proc main =
  let config = Config.load
  let serverAddress = initTAddress(config.serverIpAddress, config.serverPort)
  let client = RestClientRef.new(serverAddress)

  setLogLevel(LogLevel.INFO)

  template benchmark(apiNameIdent: untyped): untyped {.dirty.} =
    block:
      const apiName = astToStr apiNameIdent
      info "Benchmarking ...", apiName
      info.logTime(apiName):
        for slot in config.startSlot ..< (config.startSlot + config.requestsCount):
          let ident = StateIdent(kind: StateQueryKind.Slot, slot: slot.Slot)
          discard waitFor client.`apiNameIdent`(ident)

  benchmark(getStateRoot)
  benchmark(getStateFork)
  benchmark(getStateFinalityCheckpoints)
  benchmark(getStateValidatorBalances)

when isMainModule:
  main()