#include <base/attributes.h>
#include <base/types.h>
#include <log/log.h>
#include <stdbool.h>
#include <x86/asm.h>
#include <x86/boot.h>
#include <x86/regs.h>
#include <x86/serial.h>

#include "config.h"
#include "disk.h"
#include "elf.h"
#include "memory.h"
#include "stdlib.h"
#include "tty.h"
#include "vesa.h"

ALIGNED(8)
static boot_info_t info = { 0 };

static void log_sink(const char *msg) {
    puts(msg);
}

#ifndef BOOT_LOG_COLOR
#define BOOT_LOG_COLOR true
#endif

static unsigned int log_opts(void) {
#if BOOT_LOG_COLOR
    return LOG_OPT_LOCATION | LOG_OPT_COLOR;
#else
    return LOG_OPT_LOCATION;
#endif
}

NORETURN void _load_entry(u16 boot_disk) {
    init_serial(SERIAL_COM1, SERIAL_DEFAULT_LINE, SERIAL_DEFAULT_BAUD);
    log_init(log_sink);
    log_set_lvl(LOG_DEBUG);
    log_set_options(log_opts());

    log_info("BIOS boot started");

    get_e820(&info.memory_map);
    arch_init_alloc();

    void *smp_trampoline = mmap_try_alloc_top(0x1000, E820_KERNEL, 0x1000, 0x000fffffULL);

    if (smp_trampoline) {
        info.smp_trampoline_paddr = (u64)(uintptr_t)smp_trampoline;
    } else {
        log_warn("no low page for SMP trampoline, using UP mode");
    }

    u64 acpi_root_ptr = 0;
    get_rsdp(&acpi_root_ptr);
    info.acpi_root_ptr = acpi_root_ptr;

    disk_init(boot_disk);
    bios_boot_root_hint(&info.boot_root_hint);

    log_debug("reading loader config");
    parse_config(&info.args);

    u64 rootfs_paddr = 0;
    u64 rootfs_size = 0;

    if (info.args.stage_rootfs && stage_rootfs_image(&rootfs_paddr, &rootfs_size)) {
        info.boot_rootfs_paddr = rootfs_paddr;
        info.boot_rootfs_size = rootfs_size;

        log_debug("rootfs staged at %#llx size=%llu", rootfs_paddr, rootfs_size);
    }

    log_debug("setting up video");
    init_graphics(&info);

    if (info.video.mode == VIDEO_GRAPHICS && info.video.framebuffer) {
        u64 pitch = info.video.bytes_per_line;
        if (!pitch) {
            pitch = (u64)info.video.width * info.video.bytes_per_pixel;
        }

        u64 fb_size = pitch * info.video.height;
        if (fb_size) {
            mmap_add_entry(&info.memory_map, info.video.framebuffer, fb_size, E820_RESERVED);
            clean_mmap(&info.memory_map);
        }
    }

    log_info("entering kernel");

    load_kernel(&info);

    halt();
    __builtin_unreachable();
}