#include "App.h"

#include "helpers.h"

#include <memory>

/* This function pushes data to the uSockets mock */
extern "C" void us_loop_read_mocked_data(struct us_loop *loop, char *data, unsigned int size);

us_listen_socket_t *listenSocket;

/* ws->getUserData returns one of these */
struct PerSocketData {
    int nothing;
    std::shared_ptr<bool> valid;
};

extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {

    /* First byte determines what compressor to use */
    if (size < 1) {
        return 0;
    }

    uWS::CompressOptions compressors[] = {
        uWS::DISABLED,
        uWS::SHARED_COMPRESSOR,
        uWS::DEDICATED_COMPRESSOR_3KB,
        uWS::DEDICATED_COMPRESSOR_4KB,
        uWS::DEDICATED_COMPRESSOR_8KB,
        uWS::DEDICATED_COMPRESSOR_16KB,
        uWS::DEDICATED_COMPRESSOR_32KB,
        uWS::DEDICATED_COMPRESSOR_64KB,
        uWS::DEDICATED_COMPRESSOR_128KB,
        uWS::DEDICATED_COMPRESSOR_256KB
    };

    uWS::CompressOptions compressor = compressors[data[0] % 10];
    data++;
    size--;

    /* Very simple WebSocket echo server */
    auto app = uWS::TemplatedApp<false>(uWS::App().ws<PerSocketData>("/*", {
        /* Settings */
        .compression = compressor,
        /* We want this to be low so that we can hit it, yet bigger than 256 */
        .maxPayloadLength = 300,
        .idleTimeout = 10,
        /* Handlers */
        .open = [](auto *ws) {

            PerSocketData *psd = (PerSocketData *) ws->getUserData();
            psd->valid.reset(new bool{true});

            //if (req->getHeader("close_me").length()) {
            //    ws->close();
            //} else if (req->getHeader("end_me").length()) {
            //    ws->end(1006);
            //}
        },
        .message = [](auto *ws, std::string_view message, uWS::OpCode opCode) {
            if (message.length() > 300) {
                /* Inform the sanitizer of the fault */
                fprintf(stderr, "Too long message passed\n");
                free((void *) -1);
            }

            if (message.length() && message[0] == 'C') {
                ws->close();
            } else if (message.length() && message[0] == 'E') {
                ws->end(1006);
            } else {
                ws->send(message, opCode, true);
            }
        },
        .drain = [](auto *ws) {
            /* Check getBufferedAmount here */
        },
        .ping = [](auto *ws) {
            /* Here we test send and end while uncorked, by having them send from deferred */
            PerSocketData *psd = (PerSocketData *) ws->getUserData();

            uWS::Loop::get()->defer([ws, valid = psd->valid]() {
                if (valid.get()) {
                    /* We haven't been closed */
                    ws->send("Hello!", uWS::TEXT, false);
                    ws->end(1000);
                }
            });
        },
        .pong = [](auto *ws) {

        },
        .close = [](auto *ws, int code, std::string_view message) {

        }
    }).listen(9001, [](us_listen_socket_t *listenSocket) {
            ::listenSocket = listenSocket;
    })).run();

    us_loop_read_mocked_data((struct us_loop *) uWS::Loop::get(), (char *) makePadded(data, size), size);

    us_listen_socket_close(0, listenSocket);
    uWS::Loop::get()->free();

    return 0;
}