//Toonz includes

#include "traster.h"

#include "timage_io.h"

//Qt includes

#include <qiodevice></qiodevice>

#include <qsharedmemory></qsharedmemory>

#include <qtemporaryfile></qtemporaryfile>

#include <qdir></qdir>

#include <qfile></qfile>

#include <qcoreapplication></qcoreapplication>

//tipc includes

#include "tipc.h"

#include "tipcmsg.h"

//---------------------------------------------------------------------

//  Local stuff

namespace

{

QHash<qstring, *="" qsharedmemory=""> sharedMemories;</qstring,>

QHash<qstring, qstring=""> temporaryFiles;</qstring,>

}

//---------------------------------------------------------------------

namespace tipc

{

//*******************************************************************************

//    Shared Memory Request

//*******************************************************************************

template <>

QString DefaultMessageParser<shmem_request>::header() const { return QString("$shmem_request"); }</shmem_request>

//------------------------------------------------------------------

template <>

void DefaultMessageParser<shmem_request>::operator()(Message &msg)</shmem_request>

{

	int size;

	QString id;

	msg >> id >> size >> clr;

	QSharedMemory *mem = new QSharedMemory(id);

	bool ok = (tipc::create(*mem, size) > 0);

	if (!ok) {

		msg << QString("err");

		delete mem;

		return;

	}

	sharedMemories.insert(id, mem);

	msg << QString("ok");

}

//*******************************************************************************

//    Shared Memory Release

//*******************************************************************************

template <>

QString DefaultMessageParser<shmem_release>::header() const { return QString("$shmem_release"); }</shmem_release>

//------------------------------------------------------------------

template <>

void DefaultMessageParser<shmem_release>::operator()(Message &msg)</shmem_release>

{

	QString id;

	msg >> id >> clr;

	QSharedMemory *mem = sharedMemories.take(id);

	if (mem)

		delete mem;

	msg << QString("ok");

}

//*******************************************************************************

//    Temporary File Request

//*******************************************************************************

template <>

QString DefaultMessageParser<tmpfile_request>::header() const { return QString("$tmpfile_request"); }</tmpfile_request>

//------------------------------------------------------------------

template <>

void DefaultMessageParser<tmpfile_request>::operator()(Message &msg)</tmpfile_request>

{

	QString id;

	msg >> id >> clr;

	//Build a temporary file with passed id group.

	//The created QTemporaryFile CANNOT be stored directly, as it internally

	//keeps the file open until the object is destroyed. Instead, we store its

	//filePath and manually remove it upon release.

	QTemporaryFile tmp(QDir::temp().filePath(id));

	tmp.setAutoRemove(false);

	if (!tmp.open()) {

		msg << QString("err");

		return;

	}

	temporaryFiles.insert(id, tmp.fileName());

	msg << QString("ok") << tmp.fileName();

}

//*******************************************************************************

//    Temporary File Release

//*******************************************************************************

template <>

QString DefaultMessageParser<tmpfile_release>::header() const { return QString("$tmpfile_release"); }</tmpfile_release>

//------------------------------------------------------------------

template <>

void DefaultMessageParser<tmpfile_release>::operator()(Message &msg)</tmpfile_release>

{

	QString id;

	msg >> id >> clr;

	QString tmpPath = temporaryFiles.take(id);

	if (!tmpPath.isEmpty()) {

		QFile file(tmpPath);

		file.remove();

	}

	msg << QString("ok");

}

//*******************************************************************************

//    Quit On Error

//*******************************************************************************

template <>

QString DefaultMessageParser<quit_on_error>::header() const { return QString("$quit_on_error"); }</quit_on_error>

//------------------------------------------------------------------

template <>

void DefaultMessageParser<quit_on_error>::operator()(Message &msg)</quit_on_error>

{

	QObject::connect(socket(), SIGNAL(error(QLocalSocket::LocalSocketError)), QCoreApplication::instance(), SLOT(quit()));

	//In Qt 5.5 originating process's termination emits 'disconnected' instead of 'error'

	QObject::connect(socket(), SIGNAL(disconnected()), QCoreApplication::instance(), SLOT(quit()));

	msg << clr << QString("ok");

}

//*******************************************************************************

//    Explicit template instantiation for all basic datatypes

//*******************************************************************************

//-------------------------- Default Messages ---------------------------

template class DefaultMessageParser<shmem_request>;</shmem_request>

template class DefaultMessageParser<shmem_release>;</shmem_release>

template class DefaultMessageParser<tmpfile_request>;</tmpfile_request>

template class DefaultMessageParser<tmpfile_release>;</tmpfile_release>

template class DefaultMessageParser<quit_on_error>;</quit_on_error>

} //namespace tipc