#include <kernel/graphics/vesa.h>
#include <kernel/fs/ata_pio.h>
#include <kernel/fs/marfs/marfs.h>
#include <kernel/lib/alloc.h>
#include <kernel/fs/iso9660/iso9660.h>
#include "atapi_pio.h"

void install_melvix() {
    info("You're booting from a CD, Melvix will only run after an install");
    asm volatile ("cli");
    struct ATA_INTERFACE *primary_master = new_ATA(1, 0x1F0);
    if (marfs_init(primary_master) != 0) {
        panic("No HDD found!");
    }

    struct marfs_SUPERBLOCK *currentSB = marfs_read_superblock();
    if (currentSB->signature == 0x1083B99F34B59645) { // WEEEOOOWEEEOOO
        panic("Melvix seems to be already installed!");
    }
    kfree(currentSB);

    info("Installing...\n");

    // Copy MBR
    info("Copying MBR... ");
    char *stage1_p[] = {"BOOT", "HDD1.BIN"};
    struct ISO9660_entity *stage1_e = ISO9660_get(stage1_p, 2);
    if (!stage1_e)
        panic("Couldn't find the first HDD bootloader!");
    uint8_t *stage1 = ISO9660_read(stage1_e);
    kfree(stage1_e);
    marfs_write_mbr(stage1);

    // Format disk
    info("Formatting disk...");
    marfs_format();

    // Copy second stage
    info("Copying second stage...");
    char *stage2_p[] = {"BOOT", "HDD2.BIN"};
    struct ISO9660_entity *stage2_e = ISO9660_get(stage2_p, 2);
    if (!stage2_e)
        panic("Couldn't find the second HDD bootloader!");
    uint8_t *stage2 = ISO9660_read(stage2_e);
    marfs_new_file(stage2_e->length, stage2, 0, 0, 0);
    kfree(stage2_e);

    // Copy the kernel
    info("Copying the kernel... ");
    char *kernel_p[] = {"BOOT", "KERNEL.BIN"};
    struct ISO9660_entity *kernel_e = ISO9660_get(kernel_p, 2);
    if (!kernel_e)
        panic("WTH Kernel not found!?");
    uint8_t *kernel = kmalloc(kernel_e->length + 2048);
    ATAPI_granular_read(1 + (kernel_e->length / 2048), kernel_e->LBA, kernel);
    marfs_new_file(kernel_e->length, kernel, 0, 0, 0);
    kfree(kernel);
    kfree(kernel_e);

    info("Installation successful!");
}