main.pony

use "net"
use "net/ssl"
use "files"
use "encode/base64"

class MyTCPConnectionNotify is TCPConnectionNotify
  let _out: Env

  new iso create(out: Env) =>
    _out = out

    fun ref connected(conn: TCPConnection ref) =>
      _out.out.print("CONNECTED")
      var sockKey: String = Base64.encode("1234567890abcdef")
      var handshake: Handshake = Handshake("gateway.discord.gg","/",sockKey)
      conn.set_nodelay(true)
      conn.set_keepalive(10)
      conn.write(handshake.build())

    fun ref received(conn: TCPConnection ref, data: Array[U8] iso): Bool =>
      _out.out.print("GOT: " + String.from_array(consume data))
      true


actor Main
  new create(env: Env) =>

    var myFrame: Frame = Frame(true, OPTEXT, "data")
    try
      var outFile: File = File(FilePath(env.root as AmbientAuth, "out.bin"))
      outFile.writev(myFrame.build())
      outFile.sync()
    end

    var sockKey: String = Base64.encode("1234567890abcdef")
    var handshake: Handshake = Handshake("gateway.discord.gg?v=6&encoding=json","/?v=6&encoding=json",sockKey)
    env.out.print(handshake.build())

    let sslctx: (None | SSLContext) = try
      recover SSLContext
        .>set_client_verify(true)
        .>set_authority(FilePath(env.root as AmbientAuth, "cacert.pem"))
      end
    end

    try
      let ctx = sslctx as SSLContext
      let ssl = ctx.client("gateway.discord.gg")
      TCPConnection(env.root as AmbientAuth,
        SSLConnection(MyTCPConnectionNotify(env), consume ssl),
        "gateway.discord.gg",
        "443")

    end