#include "irq.h"
#include "kthread.h"
#include "list.h"
#include "wait.h"

static struct wait_queue *waitQueues = NULL;

int waitQueueInit(struct wait_queue *wq)
{
    uint32_t flags;
    disable_IRQs(flags);
    list_add_tail(waitQueues, wq);
    restore_IRQs(flags);
    return 0;
}

int waitQueueFree(struct wait_queue *wq)
{
    uint32_t flags;
    disable_IRQs(flags);
    list_delete(waitQueues, wq);
    restore_IRQs(flags);
    return 0;
}

int wakeUp(struct wait_queue *wq)
{
    struct kthread *th;
    uint32_t flags;

    disable_IRQs(flags);
    list_collapse(wq->thread, th)
    {
        kthreadAddThread(th);
    }

    restore_IRQs(flags);

    return 0;
}

int wait(struct wait_queue *wq)
{
    return waitTimeout(wq, 0);
}

int waitTimeout(struct wait_queue *wq, unsigned long msec)
{
    struct kthread *current, *next;
    uint32_t flags;
    int ret;

    disable_IRQs(flags);

    current        = getCurrentThread();
    current->state = WAITING;
    next           = kthreadSelectNext();
    kthreadUnsched(current);

    list_add_tail(wq->thread, current);
    ret = kthreadWait(current, next, msec);

    restore_IRQs(flags);
    return ret;
}