// Toonz includes

#include "traster.h"

#include "timage_io.h"

// Qt includes

#include <QIODevice>

#include <QSharedMemory>

#include <QTemporaryFile>

#include <QDir>

#include <QFile>

#include <QCoreApplication>

// tipc includes

#include "tipc.h"

#include "tipcmsg.h"

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

//  Local stuff

namespace {

QHash<QString, QSharedMemory *> sharedMemories;

QHash<QString, QString> temporaryFiles;

}

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

namespace tipc {

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

//    Shared Memory Request

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

template <>

QString DefaultMessageParser<SHMEM_REQUEST>::header() const {

  return QString("$shmem_request");

}

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

template <>

void DefaultMessageParser<SHMEM_REQUEST>::operator()(Message &msg) {

  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");

}

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

template <>

void DefaultMessageParser<SHMEM_RELEASE>::operator()(Message &msg) {

  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");

}

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

template <>

void DefaultMessageParser<TMPFILE_REQUEST>::operator()(Message &msg) {

  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");

}

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

template <>

void DefaultMessageParser<TMPFILE_RELEASE>::operator()(Message &msg) {

  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");

}

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

template <>

void DefaultMessageParser<QUIT_ON_ERROR>::operator()(Message &msg) {

  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>;

template class DefaultMessageParser<SHMEM_RELEASE>;

template class DefaultMessageParser<TMPFILE_REQUEST>;

template class DefaultMessageParser<TMPFILE_RELEASE>;

template class DefaultMessageParser<QUIT_ON_ERROR>;

}  // namespace tipc