#ifndef CONNECTION_H
#define CONNECTION_H

#include <boost/asio.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/function.hpp>

#include "../common/connectionbase.h"

class Connection : public ConnectionBase, public boost::enable_shared_from_this<Connection> {
	private:
		friend class TCPServer;
		friend class Client;
		
		boost::asio::ip::tcp::socket socket;
		
		boost::function<void (Message::p)> recv_callback;
		boost::function<void (const std::string&)> error_callback;
		
		Connection(boost::asio::io_service& io_service);
		
		//! Callback for when data is read.
		void handle_read(uint8_t* data, std::size_t bytes, const boost::system::error_code& error_code);
		
		//! Callback for when data is written.
		void handle_write();
	
	protected:
		//! Implements request_data().
		virtual void request_data(std::size_t bytes);
		
		//! Implements write_data().
		virtual void write_data(uint8_t* data, std::size_t bytes);
		
		//! Implements got_message().
		virtual void got_message(const Message::p& msg);
		
		//! Implements error().
		virtual void error(const std::string& msg);
	
	public:
		typedef boost::shared_ptr<Connection> p;
		
		//! Constructs a new instance and returns a shared pointer.
		static p create(boost::asio::io_service& io_service);
		
		//! Initiates an asynchronous message receive.
		//! \param callback Callback for received message.
		//! \param error Callback for error.
		void recv(boost::function<void (Message::p)> callback, boost::function<void (const std::string&)> error = 0);
};

#endif