{.used.}

import
  std/[os, json, tables, strutils, algorithm],
  stew/byteutils,
  ../../eth/trie/[db, hexary]

type
  TestOp = object
    idx: int
    key: seq[byte]
    value: seq[byte]

proc cmp(lhs, rhs: TestOp): int = cmp(lhs.idx, rhs.idx)
proc `<=`(lhs, rhs: TestOp): bool = lhs.idx <= rhs.idx

proc runSingleTest(testSequence: openarray[TestOp],
                   secureMode: bool,
                   expectedRootHash: string): bool =
  var
    db = newMemoryDB()
    t = initHexaryTrie(db)

  for op in testSequence:
    let
      k = op.key
      v = op.value

    if v.len > 0:
      if secureMode:
        t.SecureHexaryTrie.put k, v
      else:
        t.put k, v
    else:
      if secureMode:
        t.SecureHexaryTrie.del k
      else:
        t.del k

  return t.rootHashHex == expectedRootHash

proc runTests*(filename: string) =
  let js = json.parseFile(filename)

  for testname, testdata in js:
    template testStatus(status: string) =
      echo status, " ", filename, " :: ", testname

    template invalidTest =
      testStatus "IGNORED"
      continue

    let
      input = testdata{"in"}
      root = testdata{"root"}
      secureMode = "secure" in filename
      permuteOrder = "anyorder" in filename

    if input.isNil or root.isNil or root.kind != JString:
      invalidTest()

    var inputs = newSeq[TestOp](0)

    case input.kind
    of JArray:
      for pair in input.elems:
        if pair.kind != JArray or pair.elems.len != 2:
          invalidTest()

        let
          k = pair.elems[0]
          v = pair.elems[1]

        if k.kind == JString:
          case v.kind
          of JString:
            inputs.add(TestOp(idx: inputs.len,
                              key: k.str.toBytes,
                              value: v.str.toBytes))
          of JNull:
            inputs.add(TestOp(idx: inputs.len,
                              key: k.str.toBytes,
                              value: @[]))

          else: invalidTest()
        else: invalidTest()

    of JObject:
      for k, v in input.fields:
        case v.kind
        of JString:
          inputs.add(TestOp(idx: inputs.len,
                            key: k.toBytes,
                            value: v.str.toBytes))
        of JNull:
          inputs.add(TestOp(idx: inputs.len,
                            key: k.toBytes,
                            value: @[]))

        else: invalidTest()
    else: invalidTest()

    let expectedRootHash = root.str.substr(2).toUpperAscii

    if permuteOrder:
      sort(inputs, cmp)
      while true:
        if not runSingleTest(inputs, secureMode, expectedRootHash):
          testStatus "FAILED"
          break

        if not nextPermutation(inputs):
          testStatus "OK"
          break

    else:
      if runSingleTest(inputs, secureMode, expectedRootHash):
        testStatus "OK"
      else:
        testStatus "FAILED"

for file in walkDirRec("tests/cases"):
  if file.endsWith("json"):
    runTests(file)