#pragma once
#include "logger.h"

#include <QtCore/QObject>
#include <QtCore/QSet>
#include <QtNetwork/QHostAddress>

#ifdef QT6_WEBSOCKETS_AVAILABLE
#include <QtWebSockets/QWebSocketServer>
#include <QtWebSockets/qwebsocketprotocol.h>
#else
namespace LtsWebSocket { class QWebSocketServerStub; }
using QWebSocketServer = LtsWebSocket::QWebSocketServerStub;
namespace QWebSocketProtocol { enum class CloseCode { NormalClosure = 1000 }; }
#endif

namespace LtsWebSocket
{
class MessageHandler;
class ClientConnection;
class ReplyMessage;

class ConnectionHandler : public QObject
{
    Q_OBJECT
  public:
    explicit ConnectionHandler(QObject* parent = nullptr);
    ~ConnectionHandler();

#ifdef QT6_WEBSOCKETS_AVAILABLE
    bool init(const QString& serviceName, unsigned short port);

    void sendToAllClients(ReplyMessage* message);

    unsigned short webSocketPort()
    {
        return m_server->serverPort();
    }
#else
    bool init(const QString& /*serviceName*/, unsigned short /*port*/) { return false; }
    void sendToAllClients(ReplyMessage* /*message*/) {}
    unsigned short webSocketPort() { return 0; }
#endif

  signals:

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

    void clientDisconnected(ClientConnection* client);

    void clientConnected(ClientConnection* client);

  private slots:
#ifdef QT6_WEBSOCKETS_AVAILABLE
    void onServerError(QWebSocketProtocol::CloseCode closeCode);
    void onNewConnection();
#endif

  private:
    NGL::INGLLogger& m_logger;
#ifdef QT6_WEBSOCKETS_AVAILABLE
    QWebSocketServer* m_server;
#else
    void* m_server;
#endif
    QSet<ClientConnection*> m_connections;
};
} // namespace LtsWebSocket