/**
  * WebSocket connection to AVS to retrieve AVS events
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Christian Ehrlicher, SLSE, 2017-02-06
  */

#pragma once

#include <QAbstractSocket>
#include <QTimer>
#include <QVector>
#include <QSharedPointer>
#include "logger.h"

#include "dds_handler.h"
#include "eem_db.h"
#include <ngldb/db_service_handler.h>

#include <ExerciseEventManagement/eem.hpp>

#include <ProjectDataManagement/pdm_enums.hpp>
#include <AudioVideoManagement/avm_dtc.h>
#include <InfrastructureManagement/ism_dtc.h>

class QWebSocket;
namespace ExerciseEventManagement
{
class MainHandler;
class WebSocketEventServer;

using DbHandler = NGLDB::DbServiceHandler<DatabaseInterface>;
struct AvsData
{
  int id;
  qint64 eventTime;
  qint64 endTime;
  ProjectDataManagement::EventType eventType;
  QString streamName;
  int mainStructureElementId;
  QSharedPointer<QTimer> timer; // no plain ptr here since AvsData is copied
  AvsData()
    : id(0)
    , eventTime(0)
    , endTime(-1)
    , eventType(ProjectDataManagement::EventType::Unknown)
    , mainStructureElementId(0)
    , timer(new QTimer)
  {
    timer->setSingleShot(true);
    timer->setInterval(3000);
  }
};

class WebSocketAvs : public QObject
{
  Q_OBJECT
public:
  WebSocketAvs(MainHandler *mainHandler,
               const std::shared_ptr<DdsHandler> &ddsHandler,
               const std::shared_ptr<DbHandler> &db,
               const std::shared_ptr<WebSocketEventServer> &eventServer,
               const std::shared_ptr<AudioVideoManagement::Interface::Dtc> &avmDtc,
               const std::shared_ptr<InfrastructureManagement::Interface::Dtc> &ismDtc,
               int exerciseId,
               QObject *parent = nullptr);
  ~WebSocketAvs();

  void setWsUrl(const QString &url);
  void init();

private Q_SLOTS:
  void onTextMessageReceived(const QString &message);
  void handleAvsData(const AvsData &data, const bool &lastRangeClosed);
  int publishData(const QString &streamName);
  void openWebSocket();
  void closeWebSocket();
  void onWebSocketError(QAbstractSocket::SocketError error);
  void onWebSocketConnected();
  void onWebSocketDisconnected();
  void onSendExerciseId();

  void onCleanUpDbInternal();
  QMap<QString, QVector<AvsData>> getAllAvsData();
  QMap<QString, QVector<AvsData>> getAllInternalDbData();

private:
  AvsData jsonObjectToAvsData(const QJsonObject &obj);
  Database::EemEvent avsDataToEemEvent(const AvsData &data);
  ExerciseEventManagement::Event avsDataToEvent(const AvsData &data);

private:
  std::shared_ptr<DdsHandler> m_ddsHandler;                ///< Pointer to DDS handler
  std::shared_ptr<DbHandler>  m_dbHandler;
  std::shared_ptr<WebSocketEventServer> m_eventServer;
  std::shared_ptr<AudioVideoManagement::Interface::Dtc>     m_avmDtc;
  std::shared_ptr<InfrastructureManagement::Interface::Dtc> m_ismDtc;
  MainHandler *m_mainHandler;
  QWebSocket *m_wsSocket;
  QTimer *m_wsReconnectTimer;
  QTimer *m_cleanupTimer;
  QString m_wsUrl;
  QMap<QString, AvsData> m_avsDataMap;
  int m_exerciseId;
  qint64 m_lastUpdate;
  bool m_bWebSocketDisconnectedWarnedOnce;
  LOGGER_T m_logger;
};
} // namespace ExerciseEventManagement