import
  json, stew/byteutils, unittest, eth/rlp

proc append(output: var RlpWriter, js: JsonNode) =
  case js.kind
  of JNull, JFloat, JObject:
    raise newException(ValueError, "Unsupported JSON value type " & $js.kind)
  of JBool:
    output.append js.bval.int
  of JInt:
    output.append int(js.num)
  of JString:
    output.append js.str
  of JArray:
    output.append js.elems

proc `==`(lhs: JsonNode, rhs: string): bool =
  lhs.kind == JString and lhs.str == rhs

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

  suite filename:
    for testname, testdata in js:
      test testname:
        let
          input = testdata{"in"}
          output = testdata{"out"}

        if input.isNil or output.isNil or output.kind != JString:
          skip()
          continue

        if input == "VALID":
          var rlp = rlpFromHex(output.str)
          discard rlp.inspect
        elif input == "INVALID":
          var success = true
          var inspectOutput = ""
          expect MalformedRlpError, UnsupportedRlpError, ValueError:
            var rlp = rlpFromHex(output.str)
            inspectOutput = rlp.inspect(1)
            discard rlp.getType
            while rlp.hasData: discard rlp.toNodes
            success = false
          if not success:
            echo "  ACCEPTED MALFORMED BYTES: ", output.str
            echo "  INTERPRETATION:\n", inspectOutput
        else:
          if input.kind == JString and input.str.len != 0 and input.str[0] == '#':
            continue

          var outRlp = initRlpWriter()
          outRlp.append input
          let
            actual = outRlp.finish.toHex
            expected = output.str
          check actual == expected