#include "network/network.h"
#include <asio.hpp>

namespace network::tcp
{
    const uint32_t NETWORK_MAGIC = 0x37305545;
	const uint32_t MAX_MSG_SIZE = 10000;

	class connection : public network::connection
	{
		friend class server;
		friend class client;

	public:
		connection(asio::io_context &io_ctx, bool client = false);

		virtual void connected() override;
		virtual void disconnect() override;
		virtual void send_data(std::shared_ptr<std::string> buffer) override;
		virtual void send_message(const message &msg) override;

		asio::ip::tcp::socket m_socket;

	private:
		std::string m_header_buffer;
		std::string m_body_buffer;

		void read_header();
		void handle_send(std::shared_ptr<std::string> buf, const asio::error_code &err, size_t bytes_transferred);
		void handle_header(const asio::error_code &err, size_t bytes_transferred);
		void handle_data(const asio::error_code &err, size_t bytes_transferred);
	};

	class server : public network::server
	{
	private:
		void accept_conn();
		void handle_accept(std::shared_ptr<connection> conn, const asio::error_code &err);

		asio::ip::tcp::acceptor m_acceptor;

	public:
		server(asio::io_context &io_ctx, const std::string &host, uint32_t port);
	};

	class client : public network::client
	{
	private:
		void handle_accept(const asio::error_code &err);

	public:
		client(asio::io_context &io_ctx, const std::string &host, uint32_t port);
	};
}