/**
  * Tlm Function Topic Module Test
  * $Id$
  * $HeadURL$
  * $Author$
  */

#include "eem_tests.h"

#include <nglapplication/nglcoreapplication.h>
#include <LtsCommandLineParser/LtsCommandLineParser.h>
#include <QTimer>
#include <QDateTime>
#include <QTest>
#include <QJsonObject>
#include <QJsonDocument>
#include <LtsUtilsCore/json_object.h>

/* Module tests for ExerciseEventManagement
 * 1. Dynamic data
 * 2. IED Events
 * 3. Fire events
 * 4. Bite messages
 * 5. Hit Building
 * 6. Sanitation events
 */


//-----------------------------------------------------------------------------------------------------------------------
namespace ExerciseEventManagement
{
EemTests::EemTests(QObject *parent)
  : QObject(parent)
  , m_logger(getLogger("Test"))
{}

//-----------------------------------------------------------------------------------------------------------------------
bool EemTests::init(int domainId, int exerciseId, int testId)
{
  bool bRet = true;
  DEBUG_LOG(m_logger) << "init begin";

  m_testId = testId;

  m_ddsConnector42 = std::make_shared<DDSConnector::DdsConnector>(domainId, IDL::PartitionType::Exercise, exerciseId);
  m_ddsConnector42->init();
  m_ddsConnector42->initPublisherSubscriber();

  m_ppmDts = ParticipantManagement::Interface::Dts::create(m_ddsConnector42);
  bRet &= m_ppmDts->enableDynamicParticipantData();

  m_tabDts = TrainingAreaBase::Interface::Dts::create(m_ddsConnector42);
  bRet &= m_tabDts->enableEquipmentGearEvent();
  bRet &= m_tabDts->enableEquipmentGearChargeState();
  bRet &= m_tabDts->enableEquipmentGearAssignment();
  bRet &= m_tabDts->enableEquipmentBite();

  m_gimDts = GuezIedManagement::Interface::Dts::create(m_ddsConnector42);
  bRet &= m_gimDts->enableIed();

  m_dtc = ExerciseEventManagement::Interface::Dtc::create(m_ddsConnector42);
  bRet &= m_dtc->enableEvent();

  m_bldDts = Buildings::Interface::Dts::create(m_ddsConnector42);
  bRet &= m_bldDts->enableBuildingState();

  DEBUG_LOG(m_logger) << "bRet: " << bRet;

  return bRet;
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::startTest()
{
  INFO_LOG(m_logger) << "startTest m_testId = " << m_testId;

  if (m_testId == 1) // publish dynamicParticipantData
  {
    const qint64 time = QDateTime::currentMSecsSinceEpoch();
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::MediumDamage, time - 15000 * 60, 38);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(2, ProjectDataManagement::VulnerabilityState::MediumDamage, time - 13000 * 30, 38);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::HeavyDamage, time - 10000 * 60, 38);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::MobilityKill, time - 8000 * 30, 38);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(2, ProjectDataManagement::VulnerabilityState::MobilityKill, time - 6000 * 60, 38);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::CatastrophicKill, time - 4000 * 30, 38);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(2, ProjectDataManagement::VulnerabilityState::CatastrophicKill, time - 2000, 38);
    QTest::qWait(500);
  }

  if (m_testId == 2)
  {
    //ied
    const qint64 time = QDateTime::currentMSecsSinceEpoch();
    tryCreateIed(time - 1000 * 60);
    QTest::qWait(500);

    // dynamic participant data for ied
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::HeavyDamage, time + 1000 * 60, 49, ProjectDataManagement::StateChangeOriginator::ImprovisedExplosiveDevice);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::MobilityKill, time + 2000 * 60, 49, ProjectDataManagement::StateChangeOriginator::ImprovisedExplosiveDevice);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(1, ProjectDataManagement::VulnerabilityState::CatastrophicKill, time + 3000 * 60, 70, ProjectDataManagement::StateChangeOriginator::ImprovisedExplosiveDevice);
    QTest::qWait(5000);

    checkEvents(ProjectDataManagement::EventType::HitEffectSimulation, 3);
  }

  if (m_testId == 3)
    trySendFireEvent();

  if (m_testId == 4)
    trySendBiteMessage();

  if (m_testId == 5)
    trySendBuildingState();

  if (m_testId == 6)
  {
    const qint64 time = QDateTime::currentMSecsSinceEpoch();
    tryCreateDynamicParticipantData(2, ProjectDataManagement::VulnerabilityState::HeavyDamage, time - 10000 * 60, 0, ProjectDataManagement::StateChangeOriginator::Sanitation);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(2, ProjectDataManagement::VulnerabilityState::MediumDamage, time - 8000 * 30, 0, ProjectDataManagement::StateChangeOriginator::Sanitation);
    QTest::qWait(500);
    tryCreateDynamicParticipantData(2, ProjectDataManagement::VulnerabilityState::LightDamage, time - 5000 * 30, 0, ProjectDataManagement::StateChangeOriginator::Sanitation);
    QTest::qWait(500);

    checkEvents(ProjectDataManagement::EventType::Sanitation, 3);
  }

  if (m_testId <= 0 || m_testId > 6)
  {
    ERROR_LOG(m_logger) << "Unknown test case: " << m_testId;
    failExit();
  }

  QTimer::singleShot(0, qApp, &QCoreApplication::quit);
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::tryCreateDynamicParticipantData(int participantId, ProjectDataManagement::VulnerabilityState vulnState, qint64 time, int chargeTypeId,
                                               ProjectDataManagement::StateChangeOriginator originator)
{
  ParticipantManagement::DynamicParticipantData dp;
  const rti::core::int64 esTime = static_cast<rti::core::int64>(time);
  ParticipantManagement::StateInformation si;
  si.stateTime(esTime);
  si.vulnerabilityState(vulnState);
  if (chargeTypeId > 0)
    si.originatorChargeTypeId(chargeTypeId);
  si.originatorId(participantId + 1);
  si.originatorType(originator);
  si.vitality(100);
  if (originator == ProjectDataManagement::StateChangeOriginator::Sanitation)
  {
    si.sanitationResult(ParticipantManagement::SanitationResult::Sufficient);
    si.sanitationProceduresTaken(1);
  }

  dp.participantId(participantId);
  //dp.isManipulated(true);
  dp.stateHistory().push_back(si);
  dp.sampleTime(esTime);

  DEBUG_LOG(m_logger) << "Publishing DynamicParticipantData for participant " << participantId ;
  if (!m_ppmDts->publish(dp))
  {
    ERROR_LOG(m_logger) << "Error requesting create DynamicParticipantData";
    failExit();
  }
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::trySendFireEvent()
{
  TrainingAreaBase::EquipmentGearAssignment gearAssigment;

  gearAssigment.equipmentNumber(1002);
  gearAssigment.sampleTime(QDateTime::currentMSecsSinceEpoch());
  std::vector<TrainingAreaBase::Gear> gears;

  gears.resize(2);
  gears[0].gearId(132456);
  gears[0].gearTypeId(1003);  // Sniper
  gears[1].gearId(45778);
  gears[1].gearTypeId(1006);  // HMG

  gearAssigment.gears(gears);

  DEBUG_LOG(m_logger) << "Publish EquipmentGearAssigment";
  if (!m_tabDts->publish(gearAssigment))
  {
    ERROR_LOG(m_logger) << "Error while publishing EquipmentGearAssigment";
    failExit();
  }

  TrainingAreaBase::EquipmentGearChargeState gearChargeState;

  gearChargeState.equipmentNumber(1002);
  gearChargeState.sampleTime(QDateTime::currentMSecsSinceEpoch());

  std::vector<TrainingAreaBase::GearCharge> vGearCharge;
  vGearCharge.resize(1);
  vGearCharge[0].gearId(132456);
  vGearCharge[0].gearTypeId(1003);
  vGearCharge[0].chargeTypeId(44);
  vGearCharge[0].remain(250);

  gearChargeState.charge(vGearCharge);

  DEBUG_LOG(m_logger) << "Publish EquipmentGearChargeState";
  if (!m_tabDts->publish(gearChargeState))
  {
    ERROR_LOG(m_logger) << "Error while publishing EquipmentGearChargeState";
    failExit();
  }

  TrainingAreaBase::EquipmentGearEvent gearEvent;
  gearEvent.equipmentNumber(1002);
  gearEvent.sampleTime(QDateTime::currentMSecsSinceEpoch()+1000);

  TrainingAreaBase::GearCharge gc;
  gc.gearId(132456);
  gc.gearTypeId(1003);
  gc.chargeTypeId(44);
  gc.remain(200);
  gearEvent.chargeAmount(200);
  gearEvent.charge(gc);

  DEBUG_LOG(m_logger) << "Publish EquipmentGearEvent";
  if (!m_tabDts->publish(gearEvent))
  {
    ERROR_LOG(m_logger) << "Error while publishing EquipmentGearEvent";
    failExit();
  }

  gearEvent.equipmentNumber(1003);
  gc.gearId(45678);
  gc.gearTypeId(1009);
  gc.chargeTypeId(9);
  gc.remain(350);
  gearEvent.chargeAmount(100);
  gearEvent.charge(gc);

  DEBUG_LOG(m_logger) << "Publish EquipmentGearEvent";
  if (!m_tabDts->publish(gearEvent))
  {
    ERROR_LOG(m_logger) << "Error while publishing EquipmentGearEvent";
    failExit();
  }

  QTest::qWait(2000);

  const QMap<qint64, ExerciseEventManagement::Event> events = m_dtc->events();

  DEBUG_LOG(m_logger) << "events: " << events.size();

  int noEvents = 0;
  for(const ExerciseEventManagement::Event &e : events)
  {
    INFO_LOG(m_logger) << "eventId: " << e.id() << ", eventType: " << e.eventTypeId().underlying();
    if (e.eventTypeId() == ProjectDataManagement::EventType_def::WeaponAssignment ||
        e.eventTypeId() == ProjectDataManagement::EventType_def::FireEvent)
      noEvents ++;
  }

  if (noEvents != 5)
  {
    ERROR_LOG(m_logger) << "Error number of events (" << QString::number(noEvents) << ") not as expected (5)";
    failExit();
  }

  QTimer::singleShot(500, qApp, &QCoreApplication::quit);
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::trySendBiteMessage()
{
  TrainingAreaBase::EquipmentBite bite;

  bite.batteryState(2);
  bite.equipmentNumber(1002);
  bite.biteState(16);
  bite.configError(true);
  bite.configState(8);
  bite.sampleTime(QDateTime::currentMSecsSinceEpoch());

  std::vector<TrainingAreaBase::EquipmentBiteError> beVec;
  beVec.resize(2);

  beVec[0].canDeviceNumber(1000);
  beVec[0].errorNumber(23);
  beVec[0].subdeviceNumber(110);
  beVec[0].uniqueId(198765);

  beVec[1].canDeviceNumber(1001);
  beVec[1].errorNumber(24);
  beVec[1].subdeviceNumber(111);
  beVec[1].uniqueId(9876);

  bite.biteErrors(beVec);

  if (!m_tabDts->publish(bite))
  {
    ERROR_LOG(m_logger) << "Error while publishing EquipmentBite";
    failExit();
  }

}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::trySendBuildingState()
{
  Buildings::BuildingState bState;

  bState.id(301);
  bState.damagePercent(80);
  bState.damagePercentOld(0);
  bState.sampleTime(QDateTime::currentMSecsSinceEpoch());
  bState.vulnerabilityState(ProjectDataManagement::VulnerabilityState::CatastrophicKill);
  bState.structureElementId(26);
  //  bState.aesId(1);
  //  bState.assetId(10);
  //  bState.chargeTypeId(44);
  //  bState.originatorType(ProjectDataManagement::StateChangeOriginator::AreaEffectSimulation);
  //  bState.originatorParticipantId(6);

  if (!m_bldDts->publish(bState))
  {
    ERROR_LOG(m_logger) << "Error while publishing building state";
    failExit();
  }

  QTest::qWait(2000); // wait until it is really published

}


//-----------------------------------------------------------------------------------------------------------------------
void EemTests::tryCreateIed(qint64 time)
{
  GuezIedManagement::Ied ied;

  ied.id(100);
  ied.name("Test IED");
  ied.partyId(4);
  ied.equipmentNumber(1003);
  ied.type(GuezIedManagement::IedType::IedBigPyro);
  ied.explosiveForce(100);
  ied.pairingState(GuezIedManagement::IedPairingState_def::NotPaired);
  ied.tacticalState(GuezIedManagement::IedTacticalState_def::TriggeredByControlDevice);
  ied.pyrotechnics(true);
  ied.symbol("AP6D|CMPL|10014000001103000000");
  const rti::core::int64 esTime = static_cast<rti::core::int64>(time);
  ied.sampleTime(esTime);

  DEBUG_LOG(m_logger) << "Publishing Ied " ;
  if (!m_gimDts->publish(ied))
  {
    ERROR_LOG(m_logger) << "Error requesting publish Ied";
    failExit();
  }
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::checkEvents(const ProjectDataManagement::EventType &et, int expected)
{
  const QMap<qint64, ExerciseEventManagement::Event> events = m_dtc->events();

  DEBUG_LOG(m_logger) << "All published events: " << events.size();
  int noEvents = 0;

  for (const ExerciseEventManagement::Event &e : events)
  {
    INFO_LOG(m_logger) << "eventId: " << e.id() << ", eventType: " << e.eventTypeId().underlying();
    if (e.eventTypeId() == et)
    {
      QJsonObject prop = LtsUtils::JsonObject::fromJson(e.properties());

      int ot = prop[QStringLiteral("OriginatorType")].toInt();
      if (et == ProjectDataManagement::EventType::HitEffectSimulation && ot == 20) //ImprovisedExplosiveDevice
        noEvents ++;
      if (et == ProjectDataManagement::EventType::Sanitation && ot == 10) // Sanitation
        noEvents ++;
    }
  }

  DEBUG_LOG(m_logger) << "No events: " << noEvents;

  if (noEvents != expected)
  {
    ERROR_LOG(m_logger) << "Error number of events (" << QString::number(noEvents) << ") not as expected (" << QString::number(expected) << ")";
    failExit();
  }

  QTimer::singleShot(500, qApp, &QCoreApplication::quit);
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::failExit()
{
  m_bTestsSuccessful = false;
  QTimer::singleShot(50, qApp, QCoreApplication::quit);
}

//-----------------------------------------------------------------------------------------------------------------------
void EemTests::ExitSuccess()
{
  m_bTestsSuccessful = true;
  QTimer::singleShot(50, qApp, QCoreApplication::quit);
}

}

// ------------------------------------------------------------------
class CmdLineParser : public LtsCommandLineParser::LtsCommandLineParser
{
public:
  CmdLineParser()
    : LtsCommandLineParser()
  {
    addOption({ QStringList() << QLatin1String("t") << QLatin1String("testId")
                , QLatin1String("TestId.")
                , QLatin1String("Default: -1 (Unknown)")
                , QLatin1String("0")
              });
  }
  int testId() const { return isSet(QLatin1String("t")) ? value(QLatin1String("t")).toInt() : -1; }
};

// ------------------------------------------------------------------
int main(int argc, char **argv)
{
  NGLCoreApplication app(argc, argv);

  int iRet = -1;
  CmdLineParser   options;
  if (!options.init())
    return iRet;

  app.createLogger("EEM", options.exerciseId());

  std::unique_ptr<ExerciseEventManagement::EemTests> EemTests(new ExerciseEventManagement::EemTests);

  if (!EemTests->init(options.domainId(), options.exerciseId(), options.testId()))
    return iRet;

  EemTests->startTest();

  STARTUP_LOG(app.baseLogger());

  app.exec();

  iRet = EemTests->testsSuccessful() ? 0 : 1;
  fprintf(stderr, "%d\n", iRet);

  DEBUG_LOG(app.baseLogger()) << "** TEST result = " << iRet;

  SHUTDOWN_LOG(app.baseLogger());

  return iRet;
}