/**
 * sesimos - secure, simple, modern web server
 * @brief Worker interface
 * @file src/workers.c
 * @author Lorenz Stechauner
 * @date 2022-12-29
 */

#include "workers.h"
#include "lib/mpmc.h"

#include "worker/func.h"
#include "async.h"

static mpmc_t tcp_acceptor_ctx, request_handler_ctx, local_handler_ctx, fastcgi_handler_ctx, proxy_handler_ctx,
              proxy_peer_handler_ctx, ws_frame_handler_ctx, chunk_handler_ctx, fastcgi_frame_handler_ctx;

int workers_init(void) {
    mpmc_init(&tcp_acceptor_ctx,          8, 64, (void (*)(void *)) tcp_acceptor_func,          "tcp");
    mpmc_init(&request_handler_ctx,       8, 64, (void (*)(void *)) request_handler_func,       "req");
    mpmc_init(&local_handler_ctx,         8, 64, (void (*)(void *)) local_handler_func,         "local");
    mpmc_init(&fastcgi_handler_ctx,       8, 64, (void (*)(void *)) fastcgi_handler_func,       "fcgi");
    mpmc_init(&proxy_handler_ctx,         8, 64, (void (*)(void *)) proxy_handler_func,         "proxy");
    mpmc_init(&proxy_peer_handler_ctx,    1,  8, (void (*)(void *)) proxy_peer_handler_func,    "prxy_p");
    mpmc_init(&ws_frame_handler_ctx,      8, 64, (void (*)(void *)) ws_frame_handler_func,      "ws");
    mpmc_init(&chunk_handler_ctx,         8, 64, (void (*)(void *)) chunk_handler_func,         "chunk");
    mpmc_init(&fastcgi_frame_handler_ctx, 8, 64, (void (*)(void *)) fastcgi_frame_handler_func, "fcgi_f");
    return -1;
}

void workers_stop(void) {
    mpmc_stop(&tcp_acceptor_ctx);
    mpmc_stop(&local_handler_ctx);
    mpmc_stop(&fastcgi_handler_ctx);
    mpmc_stop(&proxy_handler_ctx);
    mpmc_stop(&proxy_peer_handler_ctx);
    mpmc_stop(&request_handler_ctx);
    mpmc_stop(&ws_frame_handler_ctx);
    mpmc_stop(&chunk_handler_ctx);
    mpmc_stop(&fastcgi_frame_handler_ctx);
}

void workers_destroy(void) {
    mpmc_destroy(&tcp_acceptor_ctx);
    mpmc_destroy(&local_handler_ctx);
    mpmc_destroy(&fastcgi_handler_ctx);
    mpmc_destroy(&proxy_handler_ctx);
    mpmc_destroy(&proxy_peer_handler_ctx);
    mpmc_destroy(&request_handler_ctx);
    mpmc_destroy(&ws_frame_handler_ctx);
    mpmc_destroy(&chunk_handler_ctx);
    mpmc_destroy(&fastcgi_frame_handler_ctx);
}

int tcp_accept(client_ctx_t *ctx) {
    return mpmc_queue(&tcp_acceptor_ctx, ctx);
}

static int handle_request_cb(client_ctx_t *ctx) {
    return mpmc_queue(&request_handler_ctx, ctx);
}

int handle_request(client_ctx_t *ctx) {
    if (ctx->c_keep_alive && ctx->s_keep_alive) {
        return async(&ctx->socket, ASYNC_WAIT_READ, 0, ctx,
                     (void (*)(void *)) handle_request_cb,
                     (void (*)(void *)) timeout_request,
                     (void (*)(void *)) tcp_close);
    } else {
        tcp_close(ctx);
        return 0;
    }
}

int local_handle(client_ctx_t *ctx) {
    return mpmc_queue(&local_handler_ctx, ctx);
}

int fastcgi_handle(client_ctx_t *ctx) {
    return mpmc_queue(&fastcgi_handler_ctx, ctx);
}

static int fastcgi_handle_frame_cb(fastcgi_ctx_t *ctx) {
    return mpmc_queue(&fastcgi_frame_handler_ctx, ctx);
}

int fastcgi_handle_frame(fastcgi_ctx_t *ctx) {
    return async(&ctx->cnx.socket, ASYNC_WAIT_READ, 0, ctx,
                 (void (*)(void *)) fastcgi_handle_frame_cb,
                 (void (*)(void *)) fastcgi_close,
                 (void (*)(void *)) fastcgi_close);
}

int proxy_handle(client_ctx_t *ctx) {
    return mpmc_queue(&proxy_handler_ctx, ctx);
}

static int proxy_peer_handle_cb(proxy_ctx_t *ctx) {
    return mpmc_queue(&proxy_peer_handler_ctx, ctx);
}

int proxy_peer_handle(proxy_ctx_t *ctx) {
    return async(&ctx->proxy, ASYNC_WAIT_READ, ASYNC_IGNORE_PENDING, ctx,
                 (void (*)(void *)) proxy_peer_handle_cb,
                 (void (*)(void *)) proxy_peer_handle_cb,
                 (void (*)(void *)) proxy_peer_handle_cb);
}

static int ws_handle_frame_cb(ws_ctx_t *ctx) {
    return mpmc_queue(&ws_frame_handler_ctx, ctx);
}

int ws_handle_frame(ws_ctx_t *ctx) {
    return async(ctx->socket, ASYNC_WAIT_READ, 0, ctx,
                 (void (*)(void *)) ws_handle_frame_cb,
                 (void (*)(void *)) ws_close,
                 (void (*)(void *)) ws_close);
}

static int handle_chunk_cb(chunk_ctx_t *ctx) {
    return mpmc_queue(&chunk_handler_ctx, ctx);
}

int handle_chunk(chunk_ctx_t *ctx) {
    return async(ctx->socket, ASYNC_WAIT_READ, 0, ctx,
                 (void (*)(void *)) handle_chunk_cb,
                 (void (*)(void *)) ctx->err_cb,
                 (void (*)(void *)) ctx->err_cb);
}