#pragma once

#include <cstdint>
#include <string>
#include <queue>
#include <boost/asio.hpp>
#include <boost/thread.hpp>
#include <boost/thread/shared_mutex.hpp>

#include "sdk/smsdk_ext.h"
#include "Define.h"

class SocketHandler;

template <class SocketType>
class Socket {
public:
	Socket(SM_SocketType st, typename SocketType::socket *asioSocket = nullptr);
	~Socket();

	bool IsOpen();

	bool Bind(const char *hostname, uint16_t port, bool async = true);
	bool Connect(const char *hostname, uint16_t port, bool async = true);
	bool Disconnect();
	bool Listen();
	bool Send(const std::string &data, bool async = true);
	bool SendTo(const std::string &data, const char *hostname, uint16_t port, bool async = true);
	bool SetOption(SM_SocketOption so, int value, bool lock = true);

	IPluginFunction *connectCallback;
	IPluginFunction *incomingCallback;
	IPluginFunction *receiveCallback;
	IPluginFunction *sendqueueEmptyCallback;
	IPluginFunction *disconnectCallback;
	IPluginFunction *errorCallback;

	int32_t smHandle;
	int32_t smCallbackArg;
	volatile unsigned int sendQueueLength;

private:
	void ReceiveHandler(char *buf, size_t bufferSize, size_t bytes, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);

	void BindPostResolveHandler(typename SocketType::resolver *, typename SocketType::resolver::iterator, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);

	void ConnectPostResolveHandler(typename SocketType::resolver *, typename SocketType::resolver::iterator, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);
	void ConnectPostConnectHandler(typename SocketType::resolver *, typename SocketType::resolver::iterator, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);

	void ListenIncomingHandler(boost::asio::ip::tcp::socket *newAsioSocket, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);

	void SendPostSendHandler(char *buf, size_t bytes, const boost::system::error_code &err, boost::shared_lock<boost::shared_mutex> *);

	void SendToPostResolveHandler(typename SocketType::resolver *, typename SocketType::resolver::iterator, char *buf, size_t bufLen, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);
	void SendToPostSendHandler(typename SocketType::resolver *, typename SocketType::resolver::iterator, char *buf, size_t bufLen, size_t bytesTransferred, const boost::system::error_code &, boost::shared_lock<boost::shared_mutex> *);

	//void InitializeResolver();
	void InitializeSocket();

	SM_SocketType sm_sockettype;
	std::queue<SocketOption *> socketOptionQueue;

	typename SocketType::socket *socket;
	boost::mutex socketMutex;
	//typename SocketType::resolver* resolver;
	typename SocketType::endpoint *localEndpoint;
	boost::mutex *localEndpointMutex;
	boost::asio::ip::tcp::acceptor *tcpAcceptor;
	boost::mutex *tcpAcceptorMutex;

	boost::shared_mutex handlerMutex;
};