import
  unittest, options,
  ../confutils/envvar/envvar_serialization,
  ../confutils/envvar/utils

const
  commonPrefix = "Nimbus"

template readWrite(key: string, val: typed) =
  test key:
    setValue(key, val)
    var outVal: type val
    getValue(key, outVal)
    check outVal == val

proc testUtils() =
  type
    Fruit = enum
      Apple

  suite "envvar utils test suite":
    readWrite("some number", 123'u32)
    readWrite("some number 64", 123'u64)
    readWrite("some bytes", @[1.byte, 2.byte])
    readWrite("some int list", @[4,5,6])
    readWrite("some array", [1.byte, 2.byte, 4.byte])
    readWrite("some string", "hello world")
    readWrite("some enum", Apple)
    readWrite("some boolean", true)
    readWrite("some float32", 1.234'f32)
    readWrite("some float64", 1.234'f64)

proc testEncoder() =
  type
    Class = enum
      Truck
      MPV
      SUV

    Fuel = enum
      Gasoline
      Diesel

    Engine = object
      cylinder: int
      valve: int16
      fuel: Fuel

    Suspension = object
      dist: int
      length: int

    Vehicle = object
      name: string
      color: int
      class: Class
      engine: Engine
      wheel: int
      suspension: array[3, Suspension]
      door: array[4, int]
      antennae: Option[int]
      bumper: Option[string]

  suite "envvar encoder test suite":
    test "basic encoder and decoder":
      let v = Vehicle(
        name: "buggy",
        color: 213,
        class: MPV,
        engine: Engine(
          cylinder: 3,
          valve: 2,
          fuel: Diesel
        ),
        wheel: 6,
        door: [1,2,3,4],
        suspension: [
          Suspension(dist: 1, length: 5),
          Suspension(dist: 2, length: 6),
          Suspension(dist: 3, length: 7)
        ],
        bumper: some("Chromium")
      )

      Envvar.encode(commonPrefix, v)
      let x = Envvar.decode(commonPrefix, Vehicle)
      check x == v
      check x.antennae.isNone
      check x.bumper.get() == "Chromium"

type
  ValidIpAddress {.requiresInit.} = object
    value: string

  TestObject = object
    address: Option[ValidIpAddress]

proc readValue(r: var EnvvarReader, value: var ValidIpAddress) =
  r.readValue(value.value)

proc writeValue(w: var EnvvarWriter, value: ValidIpAddress) =
  w.writeValue(value.value)

proc testOptionalFields() =
  suite "optional fields test suite":
    test "optional field with requiresInit pragma":

      var z = TestObject(address: some(ValidIpAddress(value: "1.2.3.4")))
      Envvar.saveFile(commonPrefix, z)
      var x = Envvar.loadFile(commonPrefix, TestObject)
      check x.address.isSome
      check x.address.get().value == "1.2.3.4"
      
testUtils()
testEncoder()
testOptionalFields()