#pragma once

#include "message_broker.h"
#include "connection_pool.h"
#include <uWebSockets/App.h>
#include <unordered_map>
#include <string>
#include <atomic>
#include <thread>
#include <shared_mutex>
#include <nlohmann/json.hpp>
#include <queue>
#include <condition_variable>

struct WebSocketUserData {
    std::string userId;
    bool pongReceived = true;
};

class WebSocketServer {
public:
    WebSocketServer(int port, ConnectionPool &pool, MessageBroker &broker);
    void run();
    void stop();

private:
    void startServer();
    void processMessageQueue();
    void pingClients();
    void handleBrokerMessage(const std::string &message);
    void handleWebSocketMessage(uWS::WebSocket<false, true, WebSocketUserData> *ws, std::string_view message, uWS::OpCode opCode);
    void handleWebSocketClose(uWS::WebSocket<false, true, WebSocketUserData> *ws);
    std::string getUserIdFromFalukantUserId(int &userId);

    int port;
    ConnectionPool &pool;
    MessageBroker &broker;

    std::atomic<bool> running{false};
    std::thread serverThread;
    std::thread messageProcessingThread;
    std::thread pingThread;

    std::unordered_map<std::string, uWS::WebSocket<false, true, WebSocketUserData> *> connections;
    std::shared_mutex connectionsMutex;

    std::queue<std::string> messageQueue;
    std::mutex queueMutex;
    std::condition_variable queueCV;
};