#ifndef RAGNAROK_SOURCE_NETWORK_CONNECTION_CONNECTION_H
#define RAGNAROK_SOURCE_NETWORK_CONNECTION_CONNECTION_H

#include "Client/ClientOptions.h"
#include "Network/Connection/ConnectionBuffer.h"

#include <winsock2.h>

class CRenderSystemBase;
struct PacketLengths;

struct ConnectionEndpoint {
  const char* address;
  int port;
  const char* overrideAddress;
};

struct ConnectionAddressCache {
  unsigned address = 0;
  unsigned session = 0xFFFFFFFFu;
};

struct ConnectionContext {
  ClientOptions& options;
  const unsigned& session;
  ConnectionAddressCache& addressCache;
  CRenderSystemBase* renderSystem;
};

class CConnection {
public:
  explicit CConnection(ConnectionContext& context);
  virtual ~CConnection();
  virtual bool Connect(const ConnectionEndpoint& endpoint);
  void Close();
  bool Receive();
  bool Flush();
  bool Update();
  void UpdateStatus();
  int Status() const;
  void Send(int bytes, const void* data);
  int Read(void* output, int bytes, unsigned char exact);
  bool ReadPacket(void* output, unsigned short* bytes, const PacketLengths& lengths);
  void Prepend(int bytes, const void* data);
  void ResetDeferred();
  void ResetBuffers();
  void SetDeferred(unsigned char value);

  SOCKET NativeSocket() const {
    return this->m_socket;
  }
  ConnectionBuffer& Input() {
    return this->m_input;
  }
  unsigned& SendDeadline() {
    return this->m_sendDeadline;
  }

protected:
  /* 0004 */ SOCKET m_socket;
  /* 0008 */ sockaddr_in m_address;
  /* 0018 */ unsigned char m_deferred;
  /* 001C */ unsigned m_sendDeadline;
  /* 0020 */ int m_status;
  /* 0024 */ ConnectionBuffer m_output;
  /* 003C */ ConnectionBuffer m_input;
  /* 0054 */ ConnectionBuffer m_pending;
  ConnectionContext* m_context;
};

class CRagConnection : public CConnection {
public:
  explicit CRagConnection(ConnectionContext& context);
  ~CRagConnection() override;

private:
  /* 006C */ unsigned char unknown_006C;
};

CRagConnection& GetRagConnection(ConnectionContext& context);

#if defined(_WIN32) && !defined(_WIN64)
static_assert(sizeof(CConnection) == 112);
static_assert(sizeof(CRagConnection) == 116);
#endif

#endif