// 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 {</shmem_request>

  return QString("$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 {</shmem_release>

  return QString("$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 {</tmpfile_request>

  return QString("$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 {</tmpfile_release>

  return QString("$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 {</quit_on_error>

  return QString("$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