#include <zinnia/alloc.h>
#include <zinnia/clock.h>
#include <zinnia/cmdline.h>
#include <zinnia/compiler.h>
#include <zinnia/console.h>
#include <zinnia/elf.h>
#include <zinnia/futex.h>
#include <zinnia/init.h>
#include <zinnia/irq.h>
#include <zinnia/percpu.h>
#include <zinnia/print.h>
#include <zinnia/sched.h>
#include <zinnia/vas.h>
#include <zinnia/vmo.h>
#include <assert.h>
#include <config.h>
#include <stdint.h>

static const char zinnia_banner[] =
    "Zinnia " ZINNIA_VERSION " " CONFIG_VARIANT " (" ZINNIA_ARCH ", " ZINNIA_COMPILER_ID ", " ZINNIA_LINKER_ID ")";

static void kernel_main_task(uintptr_t arg0) {
    struct boot_info* info = (struct boot_info*)arg0;
    kprintf("%s\n", zinnia_banner); // Say hello!

    assert(info->num_files >= 1, "No init executable provided\n");

    struct vas* init_space;
    assert(vma_new(&init_space) == 0, "");

    struct boot_file* file_data = &info->files[0];
    struct paged_vmo* init_file;
    assert(vmo_new_phys(&init_file) == 0, "");
    vmo_write(&init_file->object, 0, HHDM_PTR(file_data->data), file_data->length, nullptr);

    // Load init.
    const char* init_argv[] = {"init", nullptr};
    const char* init_envp[] = {nullptr};

    struct task* init_task = nullptr;
    errno_t status = 0;
    assert(status == 0, "Failed to load init process (%i)\n", status);
    sched_add_task(&percpu_get()->sched, init_task);

    while (true) {
        arch_irq_wait();
    }
}

[[noreturn]]
void kmain(struct boot_info* info) {
    cmdline_parse(info->cmdline);

    console_init();
    mem_init(info->mem_map, info->num_mem_maps, info->virt_base, info->phys_base, info->hhdm_base);
    rsdp_addr = info->rsdp; // TODO
    percpu_init();
    sched_init(&percpu_get()->sched);
    futex_init();

    struct task* main_task;
    task_create("main", &kernel_vas, nullptr, kernel_main_task, (uintptr_t)info, &main_task);
    sched_add_task(&percpu_get()->sched, main_task);

    irq_unlock();
    sched_yield(&percpu_get()->sched);

    unreachable();
}

[[__init]]
void kernel_early_init() {
    percpu_bsp_init();
    percpu_get()->online = true;
    irq_lock();
}