#pragma once
#include "client_message_factory.h"
#include "logger.h"
#include "ltswebsocketserver_export.h"

#include <QtCore/QObject>
#include <QtCore/QQueue>
#include <QtCore/QObject>
#include <memory>

class QTimer;

#ifdef QT6_WEBSOCKETS_AVAILABLE
namespace LtsWebSocket
{
class ConnectionHandler;
class ClientConnection;

class LTSWEBSOCKETSERVER_EXPORT LtsWebSocketServer : public QObject
{
    Q_OBJECT
  public:
    explicit LtsWebSocketServer(AbstractClientMessageFactory* messageFactory, QObject* parent = nullptr);
    virtual ~LtsWebSocketServer();

    bool init(const QString& serviceName, const unsigned short port = 0);

    bool init(const QString& serviceName, const QPair<quint16, quint16>& portRange);

    void sendToAllClients(ReplyMessage* message);

    unsigned short webSocketPort() const;

    void setMessageDelay(qint32 delay);
  signals:

    void clientDisconnected(ClientConnection* client);

    void clientConnected(ClientConnection* client);

  private slots:

    void onNewTextMessage(ReplyHandler* client, const QString& message);

    void onQueueTimerTimeout();

  private:
    struct MessageCredentials
    {
        ReplyHandler* client{nullptr};
        QString message;

        MessageCredentials(ReplyHandler* client, const QString& message) : client(client), message(message) {}
    };

    QQueue<MessageCredentials> m_messageQueue;
    QTimer* m_queueTimer{nullptr};
    AbstractClientMessageFactory* m_messageFactory{nullptr};
    std::unique_ptr<ConnectionHandler> m_connectionHandler;
    qint32 m_messageDelay{0};
    LOGGER_T m_logger;
};
} // namespace LtsWebSocket
#else
#include <QtCore/QTimer>

namespace LtsWebSocket
{
class LTSWEBSOCKETSERVER_EXPORT LtsWebSocketServer : public QObject
{
    Q_OBJECT
  public:
    explicit LtsWebSocketServer(AbstractClientMessageFactory* /*messageFactory*/, QObject* parent = nullptr);
    virtual ~LtsWebSocketServer();

    bool init(const QString& /*serviceName*/, const unsigned short /*port*/ = 0) { return false; }
    bool init(const QString& /*serviceName*/, const QPair<quint16, quint16>& /*portRange*/) { return false; }
    void sendToAllClients(ReplyMessage* /*message*/) {}
    unsigned short webSocketPort() const { return 0; }
    void setMessageDelay(qint32 /*delay*/) {}

  signals:
    void clientDisconnected(void* /*client*/);
    void clientConnected(void* /*client*/);

  private slots:
    void onNewTextMessage(void* /*client*/, const QString& /*message*/) {}
    void onQueueTimerTimeout() {}

  private:
    QTimer* m_queueTimer{nullptr};
    LOGGER_T m_logger;
};
} // namespace LtsWebSocket
#endif