#include "App.h"

#include "helpers.h"

/* 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);

uWS::TemplatedApp<false> *app;
us_listen_socket_t *listenSocket;

extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {

    app = new uWS::TemplatedApp<false>(uWS::App().get("/*", [](auto *res, auto *req) {
        if (req->getHeader("write").length()) {
            res->writeStatus("200 OK")->writeHeader("write", "true")->write("Hello");
            res->write(" world!");
            res->end();
        } else if (req->getQuery().length()) {
            res->close();
        } else {
            res->end("Hello world!");
        }
    }).post("/*", [](auto *res, auto *req) {
        res->onAborted([]() {
            /* We might as well use this opportunity to stress the loop a bit */
            uWS::Loop::get()->defer([]() {

            });
        });
        res->onData([res](std::string_view chunk, bool isEnd) {
            if (isEnd) {
                res->cork([res, chunk]() {
                    res->write("something ahead");
                    res->end(chunk);
                });
            }
        });
    }).any("/:candy/*", [](auto *res, auto *req) {
        if (req->getParameter(0).length() == 0) {
            free((void *) -1);
        }
        /* Some invalid queries */
        req->getParameter(30000);
        req->getParameter(-34234);
        req->getHeader("yhello");
        req->getQuery();

        /*req->onAborted([]() {

        });*/

        /* As of now, this will be called immediately, but changes could make it properly deferred */
        uWS::Loop::get()->defer([res]() {
            res->end("Deferred answer here");
        });
    }).listen(9001, [](us_listen_socket_t *listenSocket) {
        if (listenSocket) {
            std::cout << "Listening on port " << 9001 << std::endl;
            ::listenSocket = listenSocket;
        }
    }));

    return 0;
}

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

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

    return 0;
}