#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());
}

// JSON {"command": "PUBLIC_MSG", "text" : "Всем приветы в этом чате!"}
void process_public_msg(websock ws, json parsed_data) {
    UserData* data = ws->getUserData();

    json payload = {
        {"command", "PUBLIC_MSG"},
        {"text", parsed_data["text"]},
        {"user_from", data->id}
    };

    ws->publish("PUBLIC_CHANNEL", payload.dump());
}

// Кто в онлайне, а кто в оффлайне?
// {"command": "STATUS", "online": True/False, "user_id": 10}
string user_status(UserData* data, bool online) {
    json payload = {
        {"command", "STATUS"},
        {"online", online},
        {"user_id", data->id},
        {"name", data->name}
    };
    return payload.dump();
}

// Карта, ключ = айди пользователя, значения = данные пользователя
map<int, UserData*> all_users;

int main()
{
    int latest_user_id = 10; // ID последнего подключившегося
    uWS::App app = uWS::App(); // Создали приложение

    // Создали раздел "/" на сайте
    app.get("/", [](auto *response, auto *request) {
        response->writeHeader("Content-Type", "text/html; charset=ut8"); // Отдаем ХТМЛ
        response->end("Hello, this is <b>SkillChat</b> Server C++ 5000"); // <- Вот этот ХТМЛ
    });

    app.ws<UserData>("/*", {
        // "[]() {}" - Лямбда - функция
        // открытие соединения
        // Назначить ID
        // Сообщить всем что человек (такой-то) подключился
        .open = [&latest_user_id](websock ws) {
            UserData* data = ws->getUserData(); // Получаем данные пользователя
            data->id = latest_user_id++; // Каждому следующему подключив. будет назначен следующий АЙди

            data->name = "UnnamedUser#" + to_string(data->id); // Даем временное имя

            all_users[data->id] = data; // Положить информацию о пользователе в карту

            cout << "New user connected: " << data->id << endl;

            ws->subscribe("USER_" + to_string(data->id)); // Каждого пользователя подписываем на личный канал

            ws->publish("PUBLIC_CHANNEL", user_status(data, true)); // Уведомляем всех что "новичок" подключился

            ws->subscribe("PUBLIC_CHANNEL"); // Всех-всех-всех подписываем на публичный канал

            for (auto item : all_users) {
                //item.second UserData*
                //item.first int == id
                ws->send(user_status(item.second, true), uWS::OpCode::TEXT);
            }
            // ToDo: уведомим новичка о всех "старичках" в чате
        },
        // при получении информации от клиента
        // "отправить публичное сообщение" = 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);
            }

            if (command == "PUBLIC_MSG") {
                process_public_msg(ws, parsed_data);
            }

            // ToDo: command "SET_NAME"
        },
        // при закрытии соединения
        // Сообщить всем что человек (такой-то) отключился
        .close = [](websock ws, int, string_view) {
            UserData* data = ws->getUserData();
            // ToDo: Всем сообщить что пользователь отключился
            ws->publish("PUBLIC_CHANNEL", user_status(data, false));
            all_users.erase(data->id); // Стираем пользователя
        },
    });

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