#include "protocol.h"
#include "connection.h"


Connection::Connection():
	socket(), lastId() { }

Connection::~Connection()
	{ close(); }


Protocol& Connection::getProtocol() const
	{ assert(socket); return socket->protocol; }
const Address& Connection::getRemoteAddress() const
	{ assert(socket); return socket->address; }
Server* Connection::getServer() const
	{ assert(socket); return socket->server; }


void Connection::open(Socket *socket) {
	Lock lock(mutex);
	close();
	if (!socket) return;
	this->socket = socket;
	onOpen();
}


ErrorCode Connection::open(Protocol &protocol, const Address &remoteAddress)
	{ return protocol.connect(*this, remoteAddress); }


void Connection::close(ErrorCode errorCode) {
	Socket *socketCopy;
	{
		Lock lock(mutex);
		if (!socket)
			return;
		if (!errorCode && (!readQueue.empty() || !writeQueue.empty()))
			errorCode = ERR_CONNECTION_UNFINISHED_TASKS;
		
		onClose(errorCode);
		
		readQueue.clear();
		writeQueue.clear();
		socketCopy = socket;
		socket = nullptr;
	}
	socketCopy->close();
}


Connection::ReqId Connection::writeReq(const void *data, size_t size, void *userData) {
	if (!data || !size) return 0;
	Lock lock(mutex);
	if (!socket) return 0;
	writeQueue.emplace_back(++lastId, userData, data, size);
	return lastId;
}


Connection::ReqId Connection::readReq(void *data, size_t size, void *userData) {
	if (!data || !size) return 0;
	Lock lock(mutex);
	if (!socket) return 0;
	readQueue.emplace_back(++lastId, userData, data, size);
	return lastId;
}


void Connection::onOpen() { }
void Connection::onClose(ErrorCode) { }
void Connection::onReadReady(const ReadReq&) { }
void Connection::onWriteReady(const WriteReq&) { }