/** \file ServerCatchcopy.cpp
\brief Define the server of catchcopy
\author alpha_one_x86
\version 0002
\date 2010 */

#include "ServerCatchcopy.h"
#include "VariablesCatchcopy.h"
#include "ExtraSocketCatchcopy.h"

#include <QFile>

ServerCatchcopy::ServerCatchcopy()
{
	name="Default avanced copier";
	autoReply=true;
	idNextClient=0;
	error_string="Unknown error";
	connect(&server, SIGNAL(newConnection()), this, SLOT(newConnection()));
}

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

bool ServerCatchcopy::isListening()
{
	return server.isListening();
}

void ServerCatchcopy::setName(const QString & name)
{
	this->name=name;
}

QString ServerCatchcopy::getName()
{
	return name;
}

bool ServerCatchcopy::listen()
{
	QLocalSocket socketTestConnection;
	pathSocket=ExtraSocketCatchcopy::pathSocket();
	socketTestConnection.connectToServer(pathSocket);
	if(socketTestConnection.waitForConnected(CATCHCOPY_COMMUNICATION_TIMEOUT))
	{
		error_string="Other server is listening";
		emit error(error_string);
		return false;
	}
	else
	{
		if(!server.removeServer(pathSocket))
		{
			error_string="Unable to remove the old server";
			emit error(error_string);
		}
		if(server.listen(pathSocket))
			return true;
		else
		{
			error_string=QString("Unable to listen %1: %2").arg(pathSocket).arg(server.errorString());
			emit error(error_string);
			return false;
		}
	}
}

void ServerCatchcopy::close()
{
	if(server.isListening())
	{
		int index=0;
		while(index<ClientList.size())
		{
			ClientList.at(index).socket->disconnectFromServer();
			index++;
		}
		server.close();
		if(!server.removeServer(pathSocket))
		{
			error_string="Unable to remove the old server";
			emit error(error_string);
		}
	}
}

const QString ServerCatchcopy::errorStringServer()
{
	return server.errorString();
}

const QString ServerCatchcopy::errorString()
{
	return error_string;
}

/// \brief New connexion
void ServerCatchcopy::newConnection()
{
	while(server.hasPendingConnections())
	{
		QLocalSocket *clientSocket = server.nextPendingConnection();
		if(clientSocket!=NULL)
		{
			do
			{
				idNextClient++;
				if(idNextClient>2000000000)
					idNextClient=0;
			} while(clientIdFound(idNextClient));
			Client newClient;
			newClient.id			= idNextClient;
			newClient.socket		= clientSocket;
			newClient.haveData		= false;
			newClient.firstProtocolReplied	= false;
			newClient.detectTimeOut		= new QTimer(this);
			newClient.detectTimeOut->setSingleShot(true);
			newClient.detectTimeOut->setInterval(CATCHCOPY_COMMUNICATION_TIMEOUT);
			connect(newClient.socket,	SIGNAL(error(QLocalSocket::LocalSocketError)),	this, SLOT(connectionError(QLocalSocket::LocalSocketError)));
			connect(newClient.socket,	SIGNAL(readyRead()),				this, SLOT(readyRead()));
			connect(newClient.socket,	SIGNAL(disconnected()),				this, SLOT(disconnected()));
			connect(newClient.detectTimeOut,SIGNAL(timeout()),				this, SLOT(checkTimeOut()));
			ClientList << newClient;
			emit connectedClient(newClient.id);
		}
	}
}

bool ServerCatchcopy::clientIdFound(quint32 id)
{
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).id==id)
			return true;
		index++;
	}
	return false;
}

/// \brief new error at connexion
void ServerCatchcopy::connectionError(QLocalSocket::LocalSocketError error)
{
	QLocalSocket *socket=qobject_cast<QLocalSocket *>(QObject::sender());
	if(socket==NULL)
	{
		qWarning() << "Unlocated client socket!";
		return;
	}
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).socket==socket)
		{
			if(error!=QLocalSocket::PeerClosedError)
				qWarning() << "error detected for the client: " << index << ", type: " << error;
			ClientList.at(index).socket->disconnectFromServer();
			return;
		}
		index++;
	}
}

void ServerCatchcopy::disconnected()
{
	QLocalSocket *socket=qobject_cast<QLocalSocket *>(QObject::sender());
	if(socket==NULL)
	{
		qWarning() << "Unlocated client socket!";
		return;
	}
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).socket==socket)
		{
			emit disconnectedClient(ClientList.at(index).id);
			disconnect(ClientList.at(index).socket);
			disconnect(ClientList.at(index).detectTimeOut);
			delete ClientList.at(index).detectTimeOut;
			ClientList.at(index).socket->abort();
			ClientList.at(index).socket->disconnectFromServer();
			ClientList.at(index).socket->deleteLater();
			ClientList.removeAt(index);
			return;
		}
		index++;
	}
	qWarning() << "Unlocated client!";
}

void ServerCatchcopy::disconnectClient(quint32 id)
{
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).id==id)
		{
			ClientList.at(index).socket->disconnectFromServer();
			return;
		}
		index++;
	}
	qWarning() << "Unlocated client!";
}

void ServerCatchcopy::readyRead()
{
	QLocalSocket *socket=qobject_cast<QLocalSocket *>(QObject::sender());
	if(socket==NULL)
	{
		qWarning() << "Unlocated client socket!";
		return;
	}
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).socket==socket)
		{
			while(socket->bytesAvailable()>0)
			{
				if(!ClientList.at(index).haveData)
				{
					if(socket->bytesAvailable()<(int)sizeof(int))//ignore because first int is cuted!
					{
						/*error_string="Bytes available is not sufficient to do a int";
						emit error(error_string);
						disconnectClient(ClientList.at(index).id);*/
						return;
					}
					QDataStream in(socket);
					in.setVersion(QDataStream::Qt_4_4);
					in >> ClientList[index].dataSize;
					ClientList[index].dataSize-=sizeof(int);
					if(ClientList.at(index).dataSize>64*1024*1024) // 64MB
					{
						error_string="Reply size is >64MB, seam corrupted";
						emit error(error_string);
						disconnectClient(ClientList.at(index).id);
						return;
					}
					if(ClientList.at(index).dataSize<(int)(sizeof(int) //orderId
						     + sizeof(quint32) //returnCode
						     + sizeof(quint32) //string list size
								))
					{
						error_string="Reply size is too small to have correct code";
						emit error(error_string);
						disconnectClient(ClientList.at(index).id);
						return;
					}
					ClientList[index].haveData=true;
				}
				if(ClientList.at(index).dataSize<(ClientList.at(index).data.size()+socket->bytesAvailable()))
					ClientList[index].data.append(socket->read(ClientList.at(index).dataSize-ClientList.at(index).data.size()));
				else
					ClientList[index].data.append(socket->readAll());
				if(ClientList.at(index).dataSize==(quint32)ClientList.at(index).data.size())
				{
					if(!checkDataIntegrity(ClientList.at(index).data))
					{
						emit error("Data integrity wrong: "+QString(ClientList.at(index).data.toHex()));
						ClientList[index].data.clear();
						ClientList[index].haveData=false;
						qWarning() << "Data integrity wrong";
						return;
					}
					QStringList returnList;
					quint32 orderId;
					QDataStream in(ClientList.at(index).data);
					in.setVersion(QDataStream::Qt_4_4);
					in >> orderId;
					in >> returnList;
					ClientList[index].data.clear();
					ClientList[index].haveData=false;
					if(ClientList.at(index).queryNoReplied.contains(orderId))
					{
						emit error("Duplicate query id");
						qWarning() << "Duplicate query id";
						return;
					}
					ClientList[index].queryNoReplied << orderId;
					if(!ClientList.at(index).firstProtocolReplied && returnList.size()==2 && returnList.first()=="protocol" && autoReply)
					{
						ClientList[index].firstProtocolReplied=true;
						protocolSupported(ClientList.at(index).id,orderId,(returnList.last()==CATCHCOPY_PROTOCOL_VERSION));
					}
					else
						parseInput(ClientList.at(index).id,orderId,returnList);
				}
			}
			if(ClientList.at(index).haveData)
				ClientList.at(index).detectTimeOut->start();
			else
				ClientList.at(index).detectTimeOut->stop();
			return;
		}
		index++;
	}
	emit error("Unallocated client!");
	qWarning() << "Unallocated client!";
}

bool ServerCatchcopy::checkDataIntegrity(QByteArray data)
{
	quint32 orderId;
	qint32 listSize;
	QDataStream in(data);
	in.setVersion(QDataStream::Qt_4_4);
	in >> orderId;
	in >> listSize;
	if(listSize>65535)
	{
		emit error("List size is wrong");
		qWarning() << "List size is wrong";
		return false;
	}
	int index=0;
	while(index<listSize)
	{
		qint32 stringSize;
		in >> stringSize;
		if(stringSize>65535)
		{
			emit error("String size is wrong");
			return false;
		}
		if(stringSize>(in.device()->size()-in.device()->pos()))
		{
			emit error(QString("String size is greater than the data: %1>(%2-%3)").arg(stringSize).arg(in.device()->size()).arg(in.device()->pos()));
			return false;
		}
		in.device()->seek(in.device()->pos()+stringSize);
		index++;
	}
	if(in.device()->size()!=in.device()->pos())
	{
		emit error("Remaining data after string list parsing");
		return false;
	}
	return true;
}

void ServerCatchcopy::parseInput(quint32 client,quint32 orderId,QStringList returnList)
{
	switch(parseInputCurrentProtocol(client,orderId,returnList))
	{
		case Ok:
			emit newQuery(client,orderId,returnList);
		break;
		case Replied:
		break;
		case ExtensionWrong:
			if(autoReply)
				protocolExtensionSupported(client,orderId,false);
			else
				emit newQuery(client,orderId,returnList);
		break;
		case WrongArgument:
			if(autoReply)
				incorrectArgument(client,orderId);
			else
				emit newQuery(client,orderId,returnList);
		break;
		case WrongArgumentListSize:
			if(autoReply)
				incorrectArgumentListSize(client,orderId);
			else
				emit newQuery(client,orderId,returnList);
		break;
		case UnknowOrder:
			emit error("Unknown query");
			qWarning() << "Unknown query";
			if(autoReply)
				unknowOrder(client,orderId);
			else
				emit newQuery(client,orderId,returnList);
		break;
	}
}

ServerCatchcopy::inputReturnType ServerCatchcopy::parseInputCurrentProtocol(quint32 client,quint32 orderId,QStringList returnList)
{
	if(returnList.size()==0)
		return WrongArgumentListSize;
	//if is supported
	QString firstArgument=returnList.first();
	if(firstArgument=="protocol")
	{
		if(returnList.size()!=2)
			return WrongArgumentListSize;
		emit askProtocolCompatibility(client,orderId,returnList.last());
		return Ok;
	}
	else if(firstArgument=="protocol extension")
	{
		if(returnList.size()>3 || returnList.size()<2)
			return WrongArgumentListSize;
		if(!autoReply)
		{
			if(returnList.size()==2)
				emit askProtocolExtension(client,orderId,returnList.last());
			else
				emit askProtocolExtension(client,orderId,returnList.at(1),returnList.last());
		}
		return ExtensionWrong;
	}
	else if(firstArgument=="client")
	{
		if(returnList.size()!=2)
			return WrongArgumentListSize;
		emit clientName(client,returnList.last());
		if(autoReply)
		{
			clientRegistered(client,orderId);
			return Replied;
		}
		else
		{
			emit clientName(client,orderId,returnList.last());
			return Ok;
		}
	}
	else if(firstArgument=="server")
	{
		if(returnList.size()!=2)
			return WrongArgumentListSize;
		if(returnList.last()!="name?")
			return WrongArgument;
		if(autoReply)
		{
			serverName(client,orderId,name);
			return Replied;
		}
		else
		{
			askServerName(client,orderId);
			return Ok;
		}
	}
	else if(firstArgument=="cp")
	{
		if(returnList.size()<3)
			return WrongArgumentListSize;
		QStringList sourceList=returnList;
		sourceList.removeFirst();
		sourceList.removeLast();
		emitNewCopy(client,orderId,sourceList,returnList.last());
		return Ok;
	}
	else if(firstArgument=="cp-?")
	{
		if(returnList.size()<2)
			return WrongArgumentListSize;
		QStringList sourceList=returnList;
		sourceList.removeFirst();
		emitNewCopy(client,orderId,sourceList);
		return Ok;
	}
	else if(firstArgument=="mv")
	{
		if(returnList.size()<3)
			return WrongArgumentListSize;
		QStringList sourceList=returnList;
		sourceList.removeFirst();
		sourceList.removeLast();
		emitNewMove(client,orderId,sourceList,returnList.last());
		return Ok;
	}
	else if(firstArgument=="mv-?")
	{
		if(returnList.size()<2)
			return WrongArgumentListSize;
		QStringList sourceList=returnList;
		sourceList.removeFirst();
		emitNewMove(client,orderId,sourceList);
		return Ok;
	}
	else //if is not supported
		return UnknowOrder;
}

void ServerCatchcopy::emitNewCopy(quint32 client,quint32 orderId,QStringList sources)
{
	emit newCopy(client,orderId,sources);
	LinkGlobalToLocalClient newAssociation;
	newAssociation.idClient=client;
	newAssociation.orderId=orderId;
	newAssociation.globalOrderId=incrementOrderId();
	LinkGlobalToLocalClientList << newAssociation;
	emit newCopy(newAssociation.globalOrderId,sources);
}

void ServerCatchcopy::emitNewCopy(quint32 client,quint32 orderId,QStringList sources,QString destination)
{
	emit newCopy(client,orderId,sources,destination);
	LinkGlobalToLocalClient newAssociation;
	newAssociation.idClient=client;
	newAssociation.orderId=orderId;
	newAssociation.globalOrderId=incrementOrderId();
	LinkGlobalToLocalClientList << newAssociation;
	emit newCopy(newAssociation.globalOrderId,sources,destination);
}

void ServerCatchcopy::emitNewMove(quint32 client,quint32 orderId,QStringList sources)
{
	emit newMove(client,orderId,sources);
	LinkGlobalToLocalClient newAssociation;
	newAssociation.idClient=client;
	newAssociation.orderId=orderId;
	newAssociation.globalOrderId=incrementOrderId();
	LinkGlobalToLocalClientList << newAssociation;
	emit newMove(newAssociation.globalOrderId,sources);
}

void ServerCatchcopy::emitNewMove(quint32 client,quint32 orderId,QStringList sources,QString destination)
{
	emit newMove(client,orderId,sources,destination);
	LinkGlobalToLocalClient newAssociation;
	newAssociation.idClient=client;
	newAssociation.orderId=orderId;
	newAssociation.globalOrderId=incrementOrderId();
	LinkGlobalToLocalClientList << newAssociation;
	emit newMove(newAssociation.globalOrderId,sources,destination);
}

void ServerCatchcopy::copyFinished(quint32 globalOrderId,bool withError)
{
	int index=0;
	while(index<LinkGlobalToLocalClientList.size())
	{
		if(LinkGlobalToLocalClientList.at(index).globalOrderId==globalOrderId)
		{
			copyFinished(LinkGlobalToLocalClientList.at(index).idClient,LinkGlobalToLocalClientList.at(index).orderId,withError);
			LinkGlobalToLocalClientList.removeAt(index);
			orderList.removeOne(globalOrderId);
			return;
		}
		index++;
	}
}

void ServerCatchcopy::copyCanceled(quint32 globalOrderId)
{
	int index=0;
	while(index<LinkGlobalToLocalClientList.size())
	{
		if(LinkGlobalToLocalClientList.at(index).globalOrderId==globalOrderId)
		{
			copyCanceled(LinkGlobalToLocalClientList.at(index).idClient,LinkGlobalToLocalClientList.at(index).orderId);
			LinkGlobalToLocalClientList.removeAt(index);
			orderList.removeOne(globalOrderId);
			return;
		}
		index++;
	}
}

void ServerCatchcopy::setAutoReply(bool value)
{
	autoReply=value;
}

bool ServerCatchcopy::getAutoReply()
{
	return autoReply;
}

void ServerCatchcopy::reply(quint32 client,quint32 orderId,quint32 returnCode,QString returnString)
{
	reply(client,orderId,returnCode,QStringList() << returnString);
}

void ServerCatchcopy::reply(quint32 client,quint32 orderId,quint32 returnCode,QStringList returnList)
{
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).id==client)
		{
			if(ClientList.at(index).socket->isValid() && ClientList.at(index).socket->state()==QLocalSocket::ConnectedState)
			{
				if(!ClientList.at(index).queryNoReplied.contains(orderId))
				{
					qWarning() << "Reply to missing query or previously replied";
					return;
				}
				ClientList[index].queryNoReplied.removeOne(orderId);
				//cut string list and send it as block of 32KB
				QByteArray block;
				QDataStream out(&block, QIODevice::WriteOnly);
				out.setVersion(QDataStream::Qt_4_4);
				out << int(0);
				out << orderId;
				out << returnCode;
				out << returnList;
				out.device()->seek(0);
				out << block.size();
				emit dataSend(client,orderId,returnCode,block);
				emit dataSend(client,orderId,returnCode,returnList);
				do
				{
					QByteArray blockToSend;
					int byteWriten;
					blockToSend=block.left(32*1024);//32KB
					block.remove(0,blockToSend.size());
					byteWriten = ClientList[index].socket->write(blockToSend);
					if(!ClientList[index].socket->isValid())
					{
						error_string="Socket is not valid";
						emit error(error_string);
						return;
					}
					if(ClientList[index].socket->errorString()!="Unknown error" && ClientList[index].socket->errorString()!="")
					{
						error_string="Error with socket: "+ClientList[index].socket->errorString();
						emit error(error_string);
						return;
					}
					if(blockToSend.size()!=byteWriten)
					{
						error_string="All the bytes have not be written";
						emit error(error_string);
						return;
					}
				}
				while(block.size());
			}
			else
			{
				error_string="Socket is not valid or not connected";
				emit error(error_string);
			}
			return;
		}
		index++;
	}
	qWarning() << "Client id not found:" << client;
}

void ServerCatchcopy::protocolSupported(quint32 client,quint32 orderId,bool value)
{
	if(value)
		reply(client,orderId,1000,"protocol supported");
	else
		reply(client,orderId,5003,"protocol not supported");
}

void ServerCatchcopy::incorrectArgumentListSize(quint32 client,quint32 orderId)
{
	reply(client,orderId,5000,"incorrect argument list size");
}

void ServerCatchcopy::incorrectArgument(quint32 client,quint32 orderId)
{
	reply(client,orderId,5001,"incorrect argument");
}

void ServerCatchcopy::protocolExtensionSupported(quint32 client,quint32 orderId,bool value)
{
	if(value)
		reply(client,orderId,1001,"protocol extension supported");
	else
		reply(client,orderId,1002,"protocol extension not supported");
}

void ServerCatchcopy::clientRegistered(quint32 client,quint32 orderId)
{
	reply(client,orderId,1003,"client registered");
}

void ServerCatchcopy::serverName(quint32 client,quint32 orderId,QString name)
{
	reply(client,orderId,1004,name);
}

void ServerCatchcopy::copyFinished(quint32 client,quint32 orderId,bool withError)
{
	if(!withError)
		reply(client,orderId,1005,"finished");
	else
		reply(client,orderId,1006,"finished with error(s)");
}

void ServerCatchcopy::copyCanceled(quint32 client,quint32 orderId)
{
	reply(client,orderId,1007,"canceled");
}

void ServerCatchcopy::unknowOrder(quint32 client,quint32 orderId)
{
	reply(client,orderId,5002,"unknown order");
}

void ServerCatchcopy::checkTimeOut()
{
	QTimer *timer=qobject_cast<QTimer *>(QObject::sender());
	if(timer==NULL)
	{
		qWarning() << "Unallocated client timer!";
		return;
	}
	int index=0;
	while(index<ClientList.size())
	{
		if(ClientList.at(index).detectTimeOut==timer)
		{
			ClientList.at(index).detectTimeOut->stop();
			if(ClientList.at(index).haveData)
			{
				error_string="The client is too long to send the next part of the reply: "+ClientList.at(index).data;
				ClientList[index].haveData=false;
				ClientList[index].data.clear();
				ClientList.at(index).socket->disconnectFromServer();
				emit error(error_string);
			}
			return;
		}
		index++;
	}
}

quint32 ServerCatchcopy::incrementOrderId()
{
	do
	{
		nextOrderId++;
		if(nextOrderId>2000000)
			nextOrderId=0;
	} while(orderList.contains(nextOrderId));
	return nextOrderId;
}
