/**
  * The DeviceEventHandler class manages device events from TAB.
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Dassow Philipp, SLEA, 2017-09-04
  */

#include "equipment_event_handler.h"

#include "main_handler.h"
#include "ParticipantHandling/participant.h"

#include <QTimer>
#include <LtsUtilsCore/dds_vector.h>

ParticipantManagement::EquipmentEventHandling::EquipmentEventHandler::EquipmentEventHandler(MainHandler *mainHandler, QObject *parent)
  : QObject(parent)
  , Interface::Fts::EquipmentSystemEventDataRequestHandler(mainHandler->fts())
  , m_mainHandler (mainHandler)
  , m_filterId(0)
  , m_logger(getLogger("EquipmentEventHandler"))
{
}

bool ParticipantManagement::EquipmentEventHandling::EquipmentEventHandler::init()
{
  if (!m_mainHandler->tabDtc()->enableEquipmentEvent())
  {
    ERROR_LOG (m_logger) << " TAB Dtc enableEquipmentEvent() failed !";
    return false;
  }

  if (!m_mainHandler->dts()->enableEquipmentSystemEventFtcReply())
  {
    ERROR_LOG (m_logger) << " Dts enableEquipmentSystemEventDataFtcReply() failed !";
    return false;
  }
  connect(m_mainHandler->tabDtc().get(), &TrainingAreaBase::Interface::Dtc::equipmentEventChanged, this, &EquipmentEventHandler::onEquipmentEventChanged);

  return true;
}

void ParticipantManagement::EquipmentEventHandling::EquipmentEventHandler::onEquipmentSystemEventDataRequest(const rti::core::SampleIdentity &cookie,
                                                                                                             const ParticipantManagement::EquipmentSystemEventDataRequest &request)
{
  INFO_LOG(m_logger) << "IN onEquipmentSystemEventDataRequest";

  Core::FunctionReplyObject ret;

  QString stmt = QStringLiteral(" SELECT rec_id, equipment_number, sample_time, position_lon, position_lat, position_z, equipment_function_id, sub_device_id,"
                               " unique_device_id, device_specific_event_code, event_data"
                               " FROM rec_equipment_system_event WHERE sample_time > ") + QString::number(request.startTime()) + QStringLiteral(" AND "
                               " sample_time <= ") + QString::number(request.endTime());
  if (request.equipmentFunctionId() > -1)
    stmt += QStringLiteral(" AND equipment_function_id=") + QString::number(request.equipmentFunctionId());
  if (request.subDeviceId() > -1)
    stmt += QStringLiteral(" AND sub_device_id=") + QString::number(request.subDeviceId());
  if (request.deviceSpecificEventCode() > -1)
    stmt += QStringLiteral(" AND device_specific_event_code=") + QString::number(request.deviceSpecificEventCode());

  INFO_LOG(m_logger) << "stmt: " << stmt;

  NSqlQuery q(m_mainHandler->db()->database());
  if (!q.exec(stmt))
  {
    ERROR_LOG(m_logger) << "Error while selecting event-data ";
    ERROR_LOG(m_logger) << "stmt : " << stmt;
    ret.setStr("Error while selecting data");
    m_mainHandler->fts()->equipmentSystemEventDataReply(cookie, ret);
    return;
  }

  // Unique FilterId for each requestEquipmentSystemEventFtcReply
  m_filterId++;
  ret.setId(m_filterId);

  QVector<qint64> keys;
  ParticipantManagement::EquipmentSystemEventFtcReply data;
  while (q.next())
  {
    int i = 0;

    // Unique Key with filterId important for removing from dds
    qint64 id = m_filterId;
    id <<= 32;
    id |= q.value(i++).toInt();
    keys += id;

    data.id(id);

    data.equipmentNumber(q.value(i++).toInt());
    data.timeStamp(q.value(i++).toLongLong());
    double pos_lon = q.value(i++).toDouble();
    double pos_lat = q.value(i++).toDouble();
    double pos_z = q.value(i++).toDouble();
    data.position(Core::GeoPosition(pos_lon, pos_lat, pos_z));
    data.equipmentFunctionId(q.value(i++).toInt());
    data.subDeviceId(q.value(i++).toInt());
    data.uniqueDeviceId(q.value(i++).toInt());
    data.deviceSpecificEventCode(q.value(i++).toInt());
    QByteArray evtData(q.value(i++).toByteArray());
    if (evtData.size() != 0)
      data.eventData(LtsUtils::convert<std::vector<uint8_t>>(evtData));
    data.filterId(m_filterId);

    m_mainHandler->dts()->publish(data);
  }

  QTimer::singleShot(5 * 60 * 1000, this, [keys, this](){ removeFromDDS(keys); });
  m_mainHandler->fts()->equipmentSystemEventDataReply(cookie, ret);
}

void ParticipantManagement::EquipmentEventHandling::EquipmentEventHandler::onEquipmentEventChanged(const TrainingAreaBase::EquipmentEvent &data, DataModification modType)
{
  if (modType != DataModification::eAdded && modType != DataModification::eModified)
    return;

  ParticipantManagement::EquipmentSystemEvent event;

  event.id(data.equipmentNumber());
  event.equipmentNumber(data.equipmentNumber());
  event.sampleTime(data.sampleTime());
  if (!std::isnan(event.position().lon()) && !std::isnan(event.position().lat()))
    event.position(Core::GeoPosition(data.position().lon(), data.position().lat(), 0.0));
  else
    event.position(m_mainHandler->participantHandler()->participantByEquipmentNumber(data.equipmentNumber())->position().pos());
  event.equipmentFunctionId(data.source().equipmentFunctionId());
  event.subDeviceId(data.source().subDeviceId());
  event.uniqueDeviceId(data.source().unqiueId());
  event.deviceSpecificEventCode(data.code());
  event.eventData(data.data());

  if (!m_mainHandler->dts()->publish(event))
  {
    ERROR_LOG(m_logger) << "Error publishing equipment event to dds from eqnr: " << event.equipmentNumber() ;
  }
}

void ParticipantManagement::EquipmentEventHandling::EquipmentEventHandler::removeFromDDS(const QVector<qint64> &keys)
{
  for (qint64 key : keys)
    m_mainHandler->dts()->removeEquipmentSystemEventFtcReplyByKey(key);
}