#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <assert.h>
#include <unistd.h>
#include <event_loop.h>

static bool el_debug = true;


static task_t*
event_loop_get_next_task_to_run(event_loop_t* el) {

    task_t* task = NULL;

    for (int i = 0; i < MAX_PRIORITY_LEVEL; i++) {
        if (!el->task_array_head[i]) continue;
        task = el->task_array_head[i];
        el->task_array_head[i] = task->right;
        if (el->task_array_head[i]) {
            el->task_array_head[i]->left = NULL;
        }
        task->left = NULL;
        task->right = NULL;
        return task;
    }

    return NULL;
}

static void
event_loop_add_task_in_task_array(
    event_loop_t* el,
    task_t* new_task) {

    task_t* task, * prev_task;

    prev_task = NULL;

    task = el->task_array_head[new_task->priority];
    while (task) {
        prev_task = task;
        task = task->right;
    }
    if (prev_task) {
        prev_task->right = new_task;
        new_task->left = prev_task;
    }
    else {
        el->task_array_head[new_task->priority] = new_task;
    }
}

static bool
task_is_present_in_task_array(task_t* task) {

    return !(task->left == NULL && task->right == NULL);
}

/* one more API added later */

static void
event_loop_remove_task_from_task_array(event_loop_t* el, task_t* task) {

    if (el->task_array_head[task->priority] == task) {
        el->task_array_head[task->priority] = task->right;
    }

    if (!task->left) {
        if (task->right) {
            task->right->left = NULL;
            task->right = 0;
            return;
        }
        return;
    }
    if (!task->right) {
        task->left->right = NULL;
        task->left = NULL;
        return;
    }

    task->left->right = task->right;
    task->right->left = task->left;
    task->left = 0;
    task->right = 0;
}

void
event_loop_init(event_loop_t* el) {

    for (int i = 0; i < MAX_PRIORITY_LEVEL; i++) {
        el->task_array_head[i] = NULL;
    }
    pthread_mutex_init(&el->ev_loop_mutex, NULL);
    el->ev_loop_state = EV_LOOP_IDLE;
    pthread_cond_init(&el->ev_loop_cv, NULL);
    el->current_task = NULL;
}

static void
event_loop_schedule_task(event_loop_t* el, task_t* task) {

    pthread_mutex_lock(&el->ev_loop_mutex);

    event_loop_add_task_in_task_array(el, task);

    if (el->ev_loop_state == EV_LOOP_BUSY) {
        pthread_mutex_unlock(&el->ev_loop_mutex);
        return;
    }

    pthread_cond_signal(&el->ev_loop_cv);
    pthread_mutex_unlock(&el->ev_loop_mutex);
}


static void* event_loop_thread(void* arg) {

    task_t* task;
    EL_RES_T res;
    event_loop_t* el = (event_loop_t*)arg;
    while (1) {
        /* Lock the event Loop Mutex */
        pthread_mutex_lock(&el->ev_loop_mutex);

        while ((task = event_loop_get_next_task_to_run(el)) == NULL) {

            /* Event Loop thread do not have any task to fire
               , suspended the event loop thread*/
            if (el_debug) {
                printf("Task Array Empty, EL thread is suspending\n");
            }
            el->ev_loop_state = EV_LOOP_IDLE;
            pthread_cond_wait(&el->ev_loop_cv, &el->ev_loop_mutex);
            /* We are here when event loop thread recvs signal. On receiving the
            signal, go back, fetch the new task from task array
            again and decide if we want to block again or fire the task*/
        }
        el->ev_loop_state = EV_LOOP_BUSY;
        /* We are done with the task array, unlock it now, object of interest
        is already detached from task array*/
        pthread_mutex_unlock(&el->ev_loop_mutex);

        if (el_debug) {
            printf("EL Thread woken up, Firing the task\n");
        }

        /* Fire the task now */
        el->current_task = task;
        res = task->cbk(task->arg);
        el->current_task = NULL;

        if (res == EL_CONTINUE) {
            /* Re-schedule same task */
            event_loop_schedule_task(el, task);
        }
        else {
            free(task);
        }
    }

    return NULL;
}

task_t*
task_create_new_job(event_loop_t* el, event_cbk cbk, void* arg, TASK_PRIORITY_T priority) {

    task_t* task = (task_t*)calloc(1, sizeof(task_t));
    task->arg = arg;
    task->cbk = cbk;
    task->left = NULL;
    task->right = NULL;
    task->priority = priority;
    event_loop_schedule_task(el, task);
    return task;
}

void
event_loop_run(event_loop_t* el) {

    pthread_attr_t attr;

    assert(el->thread == NULL);

    el->thread = (pthread_t*)calloc(1, sizeof(pthread_t));

    pthread_attr_init(&attr);
    pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);

    pthread_create(el->thread, &attr,
        event_loop_thread, (void*)el);

}


void
task_cancel_job(event_loop_t* el, task_t* task) {

    /* Dont kill yourself while you are still executing */

    if (el->current_task == task) {
        return;
    }

    pthread_mutex_lock(&el->ev_loop_mutex);

    if ( task_is_present_in_task_array(task) )
        event_loop_remove_task_from_task_array(el, task);

    pthread_mutex_unlock(&el->ev_loop_mutex);

    free(task);
}