#ifndef TCP_SERVER_HPP
#define TCP_SERVER_HPP

#ifdef _MSC_VER
#pragma once
#endif

#include "Defs.hpp"
#include "TcpClient.hpp"

#include <functional>

class TcpServer
{
public:
	TcpServer();
	TcpServer(uint16 port = default_server_port);
	~TcpServer();

	void Shutdown();

	uint16 AllocateID();
	void AddToClientsList(TcpClient &client);
	bool StartServer(bool accept_connections); // if accept_connections is false the user must call the funcion AcceptConnections()
	void AcceptConnections();

	void SendMessage(const NetworkMessage &message);
	void SendHandshake(const Handshake &handshake);

	void RejectConnection(TcpClient &client);
	void AcceptConnection(uint16 client);

	void CloseSocket(TcpClient &client);
	void CloseSocket(uint16 id);

	uint16 GetMaxConnections();
	void SetMaxConnections(uint16 value);

	const TcpClient &GetClientByID(uint16 id);

	std::function<void(const NetworkMessage &message)> OnMessage; // this is going to be used for plugins

private:
	static void process_client_messages(TcpServer *server, TcpClient &client);
	static void accept_connections(TcpServer *server);

	void shutdown_internal();

	bool initialize(uint16 port = default_server_port);

	bool initialized = false;
	bool running = false;

	uint16 max_connections = 0;

	std::vector<TcpClient> clients;

#ifdef _MSC_VER
	SOCKET server_tcp_socket = INVALID_SOCKET;
	struct addrinfo *result = nullptr;
	struct addrinfo hints;
#endif
};

#endif