import std/[asyncdispatch, asyncnet, strutils]

import modernnet

proc processClient(client: AsyncSocket) {.async.} =
  var state = 0

  while not client.isClosed():
    let
      packetLength = await client.readVarNum[:int32]()
      packet = await client.read(packetLength)
      packetId = packet.readVarNum[:int32]()

    echo "Packet ID: " & $packetId

    if packetId == 0x00:
      if state == 0:
        let
          version = packet.readVarNum[:int32]()
          address = packet.readString()
          port = packet.readNum[:uint16]

        state = packet.readVarNum[:int32]()
        continue

      elif state == 1:
        var response = newBuffer()
        response.writeVarNum(0x00)
        response.writeString($buildServerListJson("1.7.10", 5, 10, 0))

        await client.write(response)
        continue

      else:
        echo "Unimplemented state: " & $state
        continue

    elif packetId == 0x01:
      let payload = packet.readNum[:int64]()

      var b = newBuffer()
      b.writeNum(payload)

      await client.write(b)

      client.close()

      continue

    else:
      echo "Unimplemented packet: " & $packetId
      continue


proc serve() {.async.} =
  var server = newAsyncSocket()
  server.setSockOpt(OptReuseAddr, true)
  server.bindAddr(Port(12345))
  server.listen()
  
  while true:
    let client = await server.accept()
    asyncCheck processClient(client)

asyncCheck serve()
echo "Started server"
runForever()