#include <iostream>
#include <uwebsockets/App.h>
#include <nlohmann/json.hpp>
using namespace std;
using json = nlohmann::json;

// Структура, где мы будем хранить информацию о каждом подключившемся
struct UserData {
    int id;
    string name;
};

// Чтобы записывать типы покороче
typedef uWS::WebSocket<false, true, UserData>* websock;

// "отправить приватное сообщение" = PRIVATE_MSG
// JSON {"command": "PRIVATE_MSG", "user_to": 11, "text" : "Эй привет 11ый!"}
void process_private_msg(websock ws, json parsed_data) {
    UserData* data = ws->getUserData(); // Получаем информацию о текущему пользовале

    json payload;
    payload["command"] = "PRIVATE_MSG";
    payload["user_from"] = data->id;
    payload["text"] = parsed_data["text"];

    ws->publish("USER_" + to_string(parsed_data["user_to"]), payload.dump());
}

int main()
{
    int latest_user_id = 10; // ID последнего подключившегося
    uWS::App app = uWS::App(); // Создали приложение
    //app.get("/", )
    app.ws<UserData>("/*", {
        // "[]() {}" - Лямбда - функция
        // открытие соединения
        // Назначить ID
        // Сообщить всем что человек (такой-то) подключился
        .open = [&latest_user_id](websock ws) {
            UserData* data = ws->getUserData(); // Получаем данные пользователя
            data->id = latest_user_id++; // Каждому следующему подключив. будет назначен следующий АЙди
            cout << "New user connected: " << data->id << endl;
            ws->subscribe("USER_" + to_string(data->id)); // Каждого пользователя подписываем на личный канал
        },
        // при получении информации от клиента
        // "отправить публичное сообщение" = PUBLIC_MSG
        // "представиться" = SET_NAME

        .message = [](websock ws, string_view message, uWS::OpCode opcode) {
            // ToDO: предусмотреть обработку исключений
            json parsed_data = json::parse(message); // Распарсить JSON
            string command = parsed_data["command"];

            cout << "Got command " << command << endl;

            if (command == "PRIVATE_MSG") {
                process_private_msg(ws, parsed_data);
            }
        },
        // при закрытии соединения
        // Сообщить всем что человек (такой-то) отключился
        .close = [](websock ws, int, string_view) {},
    });

    app.listen(9001, [](auto *) {});
    app.run();
}