#pragma once

#include "server.hpp"
#include "receiver.hpp"
#include "senderreceiver.hpp"
#include "sender.hpp"

#include "api.h"

#include "cmix.pb.h"

#include <boost/asio/io_service.hpp>

#include <list>

/*!
 * \file
 */

/*!
 * \brief The NodeNetworkSettings struct
 */
struct NodeNetworkSettings {
	bool is_first; ///< Are we the first node in the network.
	std::string next_host; ///< Next nodes host.
	std::string next_port; ///< Next nodes port.
};

/*!
 * \brief The Node class
 */
class Node
{
	struct CMixClientData {
		SharedKey shared_value;
	};

	boost::asio::io_service io_service;
	Server server;
	
	typedef std::list<Receiver> Purgatory;
	Purgatory purgatory;
	
	typedef std::map<std::string, SenderReceiver> ClientConnections;
	ClientConnections clients;
	typedef std::map<std::string, CMixClientData> ClientData;
	ClientData data;
	
	NodeNetworkSettings network_settings;
	
	Receiver prev_node;
	Sender next_node;
	
	Api api;
	KeyPair keypair;
	std::vector<uint8_t> network_pub_key;
	
	bool shutting_down;
	
	void send_bye();
	
	void accept_handler(boost::asio::ip::tcp::socket&& socket);

	void start_precomputation();
	void start_initialisation();
	
	void handle_node_initialization(cmix_proto::Initialization const& init);
	void handle_node_bye(cmix_proto::Bye);
	void handle_node_message(cmix_proto::CMixMessage message);
	
	void handle_client_keyexchange(ClientConnections::key_type handle, cmix_proto::KeyExchange ke);
	void handle_client_bye(ClientConnections::key_type handle, cmix_proto::Bye);
	void handle_client_message(ClientConnections::key_type handle, cmix_proto::CMixMessage message);
	
	void handle_imanode(Purgatory::iterator handle);
	void handle_imaclient(Purgatory::iterator handle, cmix_proto::ImAClient c);
	void handle_message(Purgatory::iterator handle, cmix_proto::CMixMessage message);

public:
	/*!
	 * \brief Node
	 * \param listen_settings The listen settings for the accepter.
	 * \param network_settings The network settings containing if we are first and who is the next node.
	 */
	Node(ListenSettings const& listen_settings, NodeNetworkSettings network_settings);
	~Node();
	
	/*!
	 * \brief run Starts the underlying io_service
	 */
	void run();
};