/*
 * The MIT License (MIT)
 *
 * Copyright © 2015 Franklin "Snaipe" Mathieu <http://snai.pe/>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
#include <stdlib.h>
#include <stdbool.h>
#include <csptr/smalloc.h>

#include "criterion/types.h"
#include "criterion/options.h"
#include "criterion/redirect.h"
#include "io/event.h"
#include "compat/posix.h"
#include "worker.h"

struct process {
    s_proc_handle *proc;
    FILE *in;
};

static s_proc_handle *g_current_proc;

void set_runner_process(void) {
    g_current_proc = get_current_process();
}

void unset_runner_process(void) {
    sfree(g_current_proc);
}

bool is_runner(void) {
    return is_current_process(g_current_proc);
}

static void close_process(void *ptr, UNUSED void *meta) {
    struct process *proc = ptr;
    fclose(proc->in);
    sfree(proc->proc);
}

struct event *worker_read_event(struct process *proc) {
    return read_event(proc->in);
}

void run_worker(struct worker_context *ctx) {
    cr_redirect_stdin();
    g_event_pipe = pipe_out(ctx->pipe, PIPE_CLOSE);

    ctx->func(ctx->test, ctx->suite);
    fclose(g_event_pipe);

    fflush(NULL); // flush all opened streams
    if (criterion_options.no_early_exit)
        return;
    _Exit(0);
}

struct process *spawn_test_worker(struct criterion_test *test,
                                  struct criterion_suite *suite,
                                  f_worker_func func,
                                  s_pipe_handle *pipe,
                                  struct test_single_param *param) {
    g_worker_context = (struct worker_context) {
        .test = test,
        .suite = suite,
        .func = func,
        .pipe = pipe,
        .param = param,
    };

    struct process *ptr = NULL;

    s_proc_handle *proc = fork_process();
    if (proc == (void *) -1) {
        abort();
    } else if (proc == NULL) {
        run_worker(&g_worker_context);
        return NULL;
    }

    ptr = smalloc(
            .size = sizeof (struct process),
            .kind = UNIQUE,
            .dtor = close_process);

    *ptr = (struct process) { .proc = proc, .in = pipe_in(pipe, PIPE_DUP) };
    return ptr;
}

struct process_status get_status(int status) {
    if (WIFEXITED(status))
        return (struct process_status) {
            .kind = EXIT_STATUS,
            .status = WEXITSTATUS(status)
        };

    if (WIFSIGNALED(status))
        return (struct process_status) {
            .kind = SIGNAL,
            .status = WTERMSIG(status)
        };

    return (struct process_status) { .kind = STOPPED };
}

struct process_status wait_proc(struct process *proc) {
    int status;
    wait_process(proc->proc, &status);

    return get_status(status);
}