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

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

#ifdef QT6_WEBSOCKETS_AVAILABLE
class QWebSocket;
#else
namespace LtsWebSocket { class QWebSocketStub; }
using QWebSocket = LtsWebSocket::QWebSocketStub;
namespace LtsWebSocket
{
class QWebSocketStub
{
  public:
    QWebSocketStub(QObject* parent = nullptr) { Q_UNUSED(parent); }
    virtual ~QWebSocketStub() = default;
    void sendTextMessage(const QString& /*message*/) {}
    void close() {}
    bool isValid() const { return false; }
    QString errorString() const { return QString(); }
};
} // namespace LtsWebSocket
#endif

namespace LtsWebSocket
{

class LTSWEBSOCKETSERVER_EXPORT ClientConnection : public QThread, public ReplyHandler
{
    Q_OBJECT
  public:
    explicit ClientConnection(QWebSocket* socket, QObject* parent = nullptr);

    ~ClientConnection();

#ifdef QT6_WEBSOCKETS_AVAILABLE
    bool init();

    void addReply(ReplyMessage* message) override;

    QWebSocket* webSocket()
    {
        return m_socket;
    }
#else
    bool init() { return false; }
    void addReply(ReplyMessage* /*message*/) override {}
    QWebSocket* webSocket() { return nullptr; }
#endif

  signals:

    void newTextMessage(ClientConnection* client, const QString& message);

    void disconnected(ClientConnection* client);

  private slots:
#ifdef QT6_WEBSOCKETS_AVAILABLE
    void onNewTextMessageFromSocket(const QString& message);
    void sendMessagesFromQueue();
    void onSocketDisconnect();
#endif

  private:
    NGL::INGLLogger& m_logger;
#ifdef QT6_WEBSOCKETS_AVAILABLE
    void sendMessages();
#endif
    QWebSocket* m_socket;
    QQueue<ReplyMessage*> m_messageQueue;
};
} // namespace LtsWebSocket