#include "multiboot.h"
#include "vga.h"
#include "serial.h"

static inline void outb(unsigned short port, unsigned char val) {
  asm volatile ("outb %0, %1" : : "a"(val), "Nd"(port) : "memory");
}
static inline unsigned char inb(unsigned short port) {
  unsigned char ret;
  asm volatile("inb %1, %0" : "=a"(ret) : "Nd"(port) : "memory");
  return ret;
}

char* itoa(int value, int base) {
    char* result;

    // check that the base if valid
    if (base < 2 || base > 36) { *result = '\0'; return result; }

    char* ptr = result, *ptr1 = result, tmp_char;
    int tmp_value;

    do {
        tmp_value = value;
        value /= base;
        *ptr++ = "zyxwvutsrqponmlkjihgfedcba9876543210123456789abcdefghijklmnopqrstuvwxyz" [35 + (tmp_value - value * base)];
    } while ( value );

    // Apply negative sign
    if (tmp_value < 0) *ptr++ = '-';
    *ptr-- = '\0';
    while(ptr1 < ptr) {
        tmp_char = *ptr;
        *ptr--= *ptr1;
        *ptr1++ = tmp_char;
    }
    return result;
}

void kmain (unsigned int mbootmagick, multiboot_info_t* mbootinfo) {

    // Cursor disabling
    outb(0x3D4, 0x0A);
    outb(0x3D5, 0x20);

    //cls();
    vga_init(VGA_COLOR_BLACK, VGA_COLOR_GRAY);

    vga_write_line("=== KoalemOS ===");

    if (serial_init() > 0) {
        vga_write_line_color("Serial initialization failed", VGA_COLOR_BLACK,
                             VGA_COLOR_ORANGE);
    } else {
        vga_write_line("\nInitialized serial.");
        serial_write_string("\n=== KoalemOS ===\n");
    }
    vga_write("Checking multiboot loader: ");

    // Check multiboot header
    if (mbootmagick != MULTIBOOT_BOOTLOADER_MAGIC) {
        vga_write_color("INVALID MAGIC", VGA_COLOR_BLACK, VGA_COLOR_RED);
        return;
    }
    if (mbootinfo->flags & MULTIBOOT_INFO_BOOT_LOADER_NAME)
        vga_write((char *)mbootinfo->boot_loader_name);
    else
        vga_write_color("Unknown", VGA_COLOR_BLACK, VGA_COLOR_ORANGE);

    vga_write("\nMultiboot flags: ");
    vga_write(itoa(mbootinfo->flags, 2));

    serial_write_string("\nMultiboot flags: ");
    serial_write_string(itoa(mbootinfo->flags, 2));

    // Check videomode
    if (mbootinfo->flags & MULTIBOOT_INFO_VBE_INFO) {
        vga_write("\nVBE Mode: 0x");
        vga_write(itoa(mbootinfo->vbe_mode, 16));
        serial_write_string("\nVBE Mode: 0x");
        serial_write_string(itoa(mbootinfo->vbe_mode, 16));
    }
    if (mbootinfo->flags & MULTIBOOT_INFO_FRAMEBUFFER_INFO) {
        vga_write("\nFramebuffer: ");
        vga_write("address: 0x");
        vga_write(itoa(mbootinfo->framebuffer_addr, 16));
        serial_write_string("\nFramebuffer: ");
        serial_write_string("address: 0x");
        serial_write_string(itoa(mbootinfo->framebuffer_addr, 16));
        unsigned long *vmem = &mbootinfo->framebuffer_addr;
        *vmem = 0xff00ff;

    } else {
        vga_write_color("\nVideo info not available", VGA_COLOR_BLACK, VGA_COLOR_RED);
        serial_write_string("\nVideo info not available");
        return;
    }
    vga_write_line("\nExecution finished, halting...");
}