#include "App.h"
#include <thread>
#include <memory>
#include <chrono>

struct us_listen_socket_t* listen_socket;

bool g_exit = false;

class MessageProducer
{
public:

    MessageProducer(uWS::App * app)
        : mApp(app)
        , mThread(nullptr)
    {
        if (mThread)
        {
            mThread->join();
        }
    }

    void start()
    {
        mThread = std::make_shared<std::thread>(&MessageProducer::broadcastLoop, this);
    }
    bool stop()
    {
        g_exit = true;
    }
    void broadcastLoop()
    {
        auto loop = uWS::Loop::get();
        while (!g_exit)
        {
            std::this_thread::sleep_for(std::chrono::seconds(1));
			loop->defer([this]()
				{
					this->mApp->publish("Test", "Test Message.", uWS::OpCode::TEXT);
				});

        }
    }

    uWS::App * mApp;
    std::shared_ptr<std::thread> mThread;
};




int main() 
{
    /* ws->getUserData returns one of these */
    struct PerSocketData 
    {

    };

    uWS::App* app = new uWS::App();
    MessageProducer msgProducer(app);
    msgProducer.start();

    app->ws<PerSocketData>("/*",
        {
            /* Settings */
            .compression = uWS::DEDICATED_COMPRESSOR_3KB,
            .maxPayloadLength = 16 * 1024 * 1024,
            .idleTimeout = 10,
            .maxBackpressure = 1 * 1024 * 1024,
            /* Handlers */
        .open = [](auto* ws)
        {
            /* Let's make every connection subscribe to the "broadcast" topic */
            ws->subscribe("Test");
        },
        .message = [](auto* ws, std::string_view message, uWS::OpCode opCode)
        {
            /* Exit gracefully if we get a closedown message (ASAN debug) */
            if (message == "closedown")
            {
                /* Bye bye */
                us_listen_socket_close(0, listen_socket);
                ws->close();
             }
        },
        .drain = [](auto* ws)
        {
            /* Check getBufferedAmount here */
        },
        .ping = [](auto* ws)
        {

        },
        .pong = [](auto* ws)
        {

        },
        .close = [](auto* ws, int code, std::string_view message)
        {
            /* We automatically unsubscribe from any topic here */
        }
        });

    app->listen(9001, [](auto* token) 
            {
            listen_socket = token;
            if (token)
            {
                std::cout << "Listening on port " << 9001 << std::endl;
            }
            }).run();

}