#include "config.h"

#include <lib/boot.h>
#include <log/log.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "disk.h"
#include "tty.h"

static void handle_debug(char *value, void *data) {
    kernel_args_t *args = data;

    if (!strcasecmp(value, "all")) {
        args->debug = DEBUG_ALL;
    } else if (!strcasecmp(value, "minimal")) {
        args->debug = DEBUG_MINIMAL;
    } else if (!strcasecmp(value, "none")) {
        args->debug = DEBUG_NONE;
    }
}

static void handle_video(char *value, void *data) {
    kernel_args_t *args = data;

    if (!strcasecmp(value, "graphics")) {
        args->video = VIDEO_GRAPHICS;
    } else if (!strcasecmp(value, "text")) {
        args->video = VIDEO_TEXT;
    } else if (!strcasecmp(value, "none")) {
        args->video = VIDEO_NONE;
    }
}

static void handle_vesa_width(char *value, void *data) {
    kernel_args_t *args = data;

    i32 width = atoll(value);
    if (width < 0) {
        return;
    }

    args->vesa_width = width;
}

static void handle_vesa_height(char *value, void *data) {
    kernel_args_t *args = data;

    i32 height = atoll(value);
    if (height < 0) {
        return;
    }

    args->vesa_height = height;
}

static void handle_vesa_bpp(char *value, void *data) {
    kernel_args_t *args = data;
    args->vesa_bpp = atol(value);
}

static void handle_console(char *value, void *data) {
    kernel_args_t *args = data;
    strncpy(args->console, value, 128);
}

static void handle_font(char *value, void *data) {
    kernel_args_t *args = data;
    strncpy(args->font, value, sizeof(args->font) - 1);
    args->font[sizeof(args->font) - 1] = '\0';
}

static bool parse_bool(const char *value, bool *out) {
    if (!value || !out) {
        return false;
    }

    if (!strcasecmp(value, "1") || !strcasecmp(value, "true")) {
        *out = true;
        return true;
    }

    if (!strcasecmp(value, "0") || !strcasecmp(value, "false")) {
        *out = false;
        return true;
    }

    return false;
}

static void handle_stage_rootfs(char *value, void *data) {
    kernel_args_t *args = data;
    bool enabled = args->stage_rootfs != 0;

    if (parse_bool(value, &enabled)) {
        if (enabled) {
            args->stage_rootfs = 1;
        } else {
            args->stage_rootfs = 0;
        }
    }
}

static const cfg_entry_t cfg_table[] = {
    { "debug", handle_debug },
    { "video", handle_video },
    { "graphics.width", handle_vesa_width },
    { "graphics.height", handle_vesa_height },
    { "graphics.bpp", handle_vesa_bpp },
    { "console", handle_console },
    { "font", handle_font },
    { "console.font", handle_font },
    { "text.font", handle_font },
    { "stage_rootfs", handle_stage_rootfs },
    { NULL, NULL },
};

void parse_config(kernel_args_t *args) {
    args->debug = BOOT_DEFAULT_DEBUG;
    args->stage_rootfs = BOOT_DEFAULT_STAGE_ROOTFS;

    args->video = BOOT_DEFAULT_VIDEO;

    args->vesa_width = BOOT_DEFAULT_VESA_WIDTH;
    args->vesa_height = BOOT_DEFAULT_VESA_HEIGHT;
    args->vesa_bpp = BOOT_DEFAULT_VESA_BPP;

    args->console[0] = '\0';

    strncpy(args->font, BOOT_DEFAULT_FONT, sizeof(args->font) - 1);
    args->font[sizeof(args->font) - 1] = '\0';

    void *config = read_rootfs("/boot/loader.conf");

    if (!config) {
        log_warn("/boot/loader.conf not found, using defaults");
        return;
    }

    parse_cfg(config, cfg_table, args);

    free(config);
}