#include <kernel/fs/load.h>
#include <kernel/fs/marfs/marfs.h>
#include <kernel/fs/ata_pio.h>
#include <kernel/fs/atapi_pio.h>
#include <kernel/system.h>
#include <kernel/fs/iso9660/iso9660.h>
#include <kernel/memory/alloc.h>
#include <kernel/lib/stdio.h>
#include <kernel/lib/lib.h>

void load_binaries()
{
	userspace = (uint32_t)kmalloc(10000);
	font = (struct font *)kmalloc(100000);
	; // High quality shit

	uint8_t boot_drive_id = (uint8_t)(*((uint8_t *)0x9000));
	if (boot_drive_id != 0xE0) {
		struct ata_interface *primary_master = new_ata(1, 0x1F0);
		marfs_init(primary_master);
		marfs_read_whole_file(4, (uint8_t *)userspace);
		marfs_read_whole_file(5, (uint8_t *)font);
	} else {
		char *font_p[] = { "FONT.BIN" };
		struct iso9660_entity *font_e = ISO9660_get(font_p, 1);
		if (!font_e)
			panic("Font not found!");
		ATAPI_granular_read(1 + (font_e->length / 2048), font_e->lba, (uint8_t *)font);
		kfree(font_e);

		char *user_p[] = { "USER.BIN" };
		struct iso9660_entity *user_e = ISO9660_get(user_p, 1);
		if (!user_e)
			panic("Userspace binary not found!");
		ATAPI_granular_read(1 + (user_e->length / 2048), user_e->lba, (uint8_t *)userspace);
		kfree(user_e);
	}
	vga_log("Successfully loaded binaries");
}

#define MAX_LOADERS 16

loader_t **loaders = 0;
uint8_t last_loader = 0;

uint32_t last_load_loc = 0x400000;

void loader_init()
{
	serial_printf("Setting up loader");
	loaders = (loader_t **)kmalloc(MAX_LOADERS * sizeof(uint32_t));
	memset(loaders, 0, MAX_LOADERS * sizeof(uint32_t));
	// _kill();
}

void register_loader(loader_t *load)
{
	if (last_loader + 1 > MAX_LOADERS)
		return;
	if (!load)
		return;
	loaders[last_loader] = load;
	last_loader++;
	serial_printf("Registered %s loader as id %d", load->name, last_loader - 1);
}

uint32_t loader_get_unused_load_location()
{
	last_load_loc += 0x400000;
	return last_load_loc;
}

uint8_t exec_start(uint8_t *buf)
{
	for (int i = 0; i < MAX_LOADERS; i++) {
		if (!loaders[i])
			break;
		void *priv = loaders[i]->probe(buf);
		if (priv) {
			return loaders[i]->start(buf, priv);
		}
	}
	return 0;
}