#include "main_handler.h"
#include <NGLLogger/NGLLoggerBootstrap.h>
#include <DDS/DDSConnector/dds_connector.h>
#include <QtCore/QThread>
#include <QtCore/QCoreApplication>
#include <QtCore/QDateTime>
#include <QtCore/QElapsedTimer>
#include "proxy_emulator.h"

#include <QTimer>

VulnerabilityStateGenerator::MainHandler::MainHandler(QObject *parent)
  : QObject(parent)
  , m_domainId(0)
  , m_exerciseId(0) 
  , m_proxyEmulator(nullptr)  
  , m_genTimer(new QTimer())
  , m_logger(NGL::NGLLoggerBootstrap::getNGLRootLogger().getNGLLogger("MainHandler"))
{
}

VulnerabilityStateGenerator::MainHandler::~MainHandler()
{  
}

bool VulnerabilityStateGenerator::MainHandler::init(int32_t domainId, int32_t exerciseId)
{
  m_domainId        = domainId;
  m_exerciseId      = exerciseId;

  delete m_proxyEmulator;
  m_proxyEmulator   = nullptr;

  // DdsConnector
  DEBUG_LOG(m_logger) << "Creating and initializing DdsConnector ...";
  m_ddsConnector.reset(new DDSConnector::DdsConnector(m_domainId, IDL::PartitionType::Exercise, m_exerciseId));
  if (!m_ddsConnector->init())
  {
    ERROR_LOG(m_logger) << "Could not init DdsConnector";
    return false;
  }
  m_ddsConnector->initPublisherSubscriber();
  DEBUG_LOG(m_logger) << "... done";

  // dtc
  DEBUG_LOG(m_logger) << "Creating and initializing PPM DTC ...";
  m_ppmDtc = ParticipantManagement::Interface::Dtc::create(m_ddsConnector);
  QThread::msleep(100);
  if (!m_ppmDtc->enableAll())
  {
    ERROR_LOG(m_logger) << "Could not init PPM DTC";
    return false;
  }
  DEBUG_LOG(m_logger) << "... done";  

  fillMapFromEnum();

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

  // ProxyEmulator
  DEBUG_LOG(m_logger) << "Creating and initializing ProxyEmulator ...";
  m_proxyEmulator = new ProxyEmulator(m_ddsConnector, this);
  QThread::msleep(100);
  if (!m_proxyEmulator->init())
  {
    ERROR_LOG(m_logger) << "Could not init ProxyEmulator";
    return false;
  }

  m_partMap = m_ppmDtc->participants();

  m_proxyEmulator->giveInitState(m_partMap);

  connect(m_genTimer, &QTimer::timeout, this, &MainHandler::setNewState);
  m_genTimer->start(1000);

  return true;
}


void VulnerabilityStateGenerator::MainHandler::quit()
{
  ERROR_LOG(m_logger) << "Are the following procs running?\n"
                      << "svcParticipantManagement -d " << m_domainId << " -e 0\n"
                      << "svcParticipantManagement -d " << m_domainId << " -e " << m_exerciseId << "\n"
                      << "... shutting down";
  QCoreApplication::exit(-1);
}


ProjectDataManagement::VulnerabilityState VulnerabilityStateGenerator::MainHandler::randomVulnerabilityState()
{
  ProjectDataManagement::VulnerabilityState result;

  qsrand(QDateTime::currentMSecsSinceEpoch());
  unsigned int index = 0;
  if (m_vsMap.size() > 0)
    index = qrand() % m_vsMap.size();


  return m_vsMap.value(index);

}

void VulnerabilityStateGenerator::MainHandler::setNewState()
{
  for (auto it = m_partMap.begin(); it != m_partMap.end(); ++it)
  {
    if (it.value().equipmentNumber().is_set() && it.value().equipmentNumber().get() > 0)
      m_proxyEmulator->publishNewState(randomVulnerabilityState(), it.value());
  }
}

void VulnerabilityStateGenerator::MainHandler::fillMapFromEnum()
{
  m_vsMap.insert(0, ProjectDataManagement::VulnerabilityState_def::CombatReady);
  m_vsMap.insert(1, ProjectDataManagement::VulnerabilityState_def::LightDamage);
  m_vsMap.insert(2, ProjectDataManagement::VulnerabilityState_def::MediumDamage);
  m_vsMap.insert(3, ProjectDataManagement::VulnerabilityState_def::HeavyDamage);
  m_vsMap.insert(4, ProjectDataManagement::VulnerabilityState_def::MobilityKill);
  m_vsMap.insert(5, ProjectDataManagement::VulnerabilityState_def::VisibleMobilityKill);
  m_vsMap.insert(6, ProjectDataManagement::VulnerabilityState_def::WeaponSystemDamage);
  m_vsMap.insert(7, ProjectDataManagement::VulnerabilityState_def::VisibleWeaponSystemDamage);
  m_vsMap.insert(8, ProjectDataManagement::VulnerabilityState_def::CatastrophicKill);
  m_vsMap.insert(9, ProjectDataManagement::VulnerabilityState_def::TemporarySeverance);
  m_vsMap.insert(10, ProjectDataManagement::VulnerabilityState_def::NearMiss);
  m_vsMap.insert(11, ProjectDataManagement::VulnerabilityState_def::HitWithNoEffect);
}