#pragma once

#include <netinet/in.h>

#include <memory>
#include <string>

#include "PSOEncryption.hh"
#include "PSOProtocol.hh"
#include "Version.hh"



struct Channel {
  std::unique_ptr<struct bufferevent, void (*)(struct bufferevent*)> bev;
  struct sockaddr_storage local_addr;
  struct sockaddr_storage remote_addr;
  bool is_virtual_connection;

  GameVersion version;
  std::shared_ptr<PSOEncryption> crypt_in;
  std::shared_ptr<PSOEncryption> crypt_out;

  std::string name;
  TerminalFormat terminal_send_color;
  TerminalFormat terminal_recv_color;

  struct Message {
    uint16_t command;
    uint32_t flag;
    std::string data;
  };

  typedef void (*on_command_received_t)(Channel&, uint16_t, uint32_t, std::string&);
  typedef void (*on_error_t)(Channel&, short);

  on_command_received_t on_command_received;
  on_error_t on_error;
  void* context_obj;

  // Creates an unconnected channel
  Channel(
      GameVersion version,
      on_command_received_t on_command_received,
      on_error_t on_error,
      void* context_obj,
      const std::string& name,
      TerminalFormat terminal_send_color = TerminalFormat::END,
      TerminalFormat terminal_recv_color = TerminalFormat::END);
  // Creates a connected channel
  Channel(
      struct bufferevent* bev,
      GameVersion version,
      on_command_received_t on_command_received,
      on_error_t on_error,
      void* context_obj,
      const std::string& name = "",
      TerminalFormat terminal_send_color = TerminalFormat::END,
      TerminalFormat terminal_recv_color = TerminalFormat::END);
  Channel(const Channel& other) = delete;
  Channel(Channel&& other) = delete;
  Channel& operator=(const Channel& other) = delete;
  Channel& operator=(Channel&& other) = delete;

  void replace_with(
      Channel&& other,
      on_command_received_t on_command_received,
      on_error_t on_error,
      void* context_obj,
      const std::string& name = "");

  void set_bufferevent(struct bufferevent* bev);

  inline bool connected() const {
    return this->bev.get() != nullptr;
  }
  void disconnect();

  // Receives a message. Throws std::out_of_range if no messages are available.
  Message recv(bool print_contents = true);

  // Sends a message with an automatically-constructed header.
  void send(uint16_t cmd, uint32_t flag = 0, const void* data = nullptr, size_t size = 0, bool print_contents = true);
  void send(uint16_t cmd, uint32_t flag, const std::string& data, bool print_contents = true);

  // Sends a message with a pre-existing header (as the first few bytes in the
  // data)
  void send(const void* data = nullptr, size_t size = 0, bool print_contents = true);
  void send(const std::string& data, bool print_contents = true);

private:
  static void dispatch_on_input(struct bufferevent*, void* ctx);
  static void dispatch_on_error(struct bufferevent*, short events, void* ctx);
};