import pkg/[
  chronos,
  chronicles,
  stew/byteutils]

import ../ws/ws

proc main() {.async.} =
  let ws = when defined tls:
    await WebSocket.connect(
        "127.0.0.1",
        Port(8888),
        path = "/wss",
        flags = {TLSFlags.NoVerifyHost, TLSFlags.NoVerifyServerName})
    else:
      await WebSocket.connect(
        "127.0.0.1",
        Port(8888),
        path = "/ws")

  trace "Websocket client: ", State = ws.readyState

  let reqData = "Hello Server"
  while true:
    try:
      await ws.send(reqData)
      let buff = await ws.recv()
      if buff.len <= 0:
        break

      let dataStr = string.fromBytes(buff)
      trace "Server Response: ", data = dataStr

      assert dataStr == reqData
      break
    except WebSocketError as exc:
      error "WebSocket error:", exception = exc.msg
      raise exc

    await sleepAsync(100.millis)

  # close the websocket
  await ws.close()

waitFor(main())