#include "lokimq/lokimq.h"
#include <future>
#include <catch2/catch.hpp>

using namespace lokimq;

TEST_CASE("basic commands", "[cmd-basic]") {
    std::string listen = "tcp://127.0.0.1:4567";
    LokiMQ server{
        "", "", // generate ephemeral keys
        false, // not a service node
        {listen},
        [](auto &) { return ""; },
        [](auto /*ip*/, auto /*pk*/) { return Allow{AuthLevel::none, false}; },
    };

    std::atomic<int> hellos{0}, his{0};

    server.add_category("public", Access{AuthLevel::none});
    server.add_command("public", "hello", [&](Message& m) {
            // On every 1st, 3rd, 5th, ... hello send back a hi
            if (hellos++ % 2 == 0)
                m.send_back("public.hi");
    });
    server.start();

    LokiMQ client(
        [](LogLevel, const char* file, int line, std::string msg) { std::cerr << file << ":" << line << " --C-- " << msg << "\n"; }
        );
    //client.log_level(LogLevel::trace);

    client.add_category("public", Access{AuthLevel::none});
    client.add_command("public", "hi", [&](auto&) { his++; });
    client.start();

    std::atomic<bool> connected{false}, failed{false};
    std::string pubkey;

    client.connect_remote(listen,
            [&](std::string pk) { pubkey = std::move(pk); connected = true; },
            [&](string_view) { failed = true; },
            server.get_pubkey());

    for (int i = 0; i < 20; i++) {
        if (connected.load())
            break;
        std::this_thread::sleep_for(100ms);
    }
    REQUIRE( connected.load() );
    REQUIRE( !failed.load() );
    REQUIRE( pubkey == server.get_pubkey() );

    client.send(pubkey, "public.hello");
    std::this_thread::sleep_for(200ms);
    REQUIRE( hellos == 1 );
    REQUIRE( his == 1 );

    for (int i = 0; i < 50; i++)
        client.send(pubkey, "public.hello");

    std::this_thread::sleep_for(200ms);
    REQUIRE( hellos == 51 );
    REQUIRE( his == 26 );
}