/**
 * mod.c
 * Функции загрузки модулей
 *
 * Файл с функциями для загрузки модулей в формате ELF файлов полученных от
 * загрузчика
 *
 */

#include <fb.h>
#include <limine.h>
#include <mod.h>
#include <sys.h>
#include <tool.h>

static volatile struct limine_module_request module_request = {
	.id = LIMINE_MODULE_REQUEST,
	.revision = 0,
	.response = (struct limine_module_response *)0
};

static struct limine_module_response *module_response;
uint64_t modules_count = 0;
module_info_t module_list[MOD_MAX];
static char *graphics_module_message = "Графический модуль-объект";
static char *other_module_message = "Неизвестный тип модуля";

static env_t main_env;

void *bootpng_ptr;
uint64_t bootpng_size;

static void *elf_entry(elf64_header_t *module_bin) {
	// Приводим заголовок ELF файла к типу elf64_header_t
	elf64_header_t *elf_header = (elf64_header_t *)module_bin;

	// Возвращаем указатель на точку входа
	return (void *)((uint64_t)elf_header->e_entry + (uint64_t)module_bin);
}

void mod_list_show( ) {
	for (uint64_t i = 0; i < modules_count; i++) {
		fb_printf("Имя: %s\n", module_list[i].name);
		fb_printf("Описание модуля: %s\n", module_list[i].message);
		fb_printf("Тип модуля: %u\n", module_list[i].type);
		fb_printf("Код ошибки модуля: %u\n", module_list[i].err_code);
		fb_printf("Размер данных: %u\n", module_list[i].data_size);
		fb_printf("Адрес данных: 0x%x\n", module_list[i].data);
	}
}

module_info_t *mod_find(char *tag) {
	for (uint64_t i = 0; i < modules_count; i++) {
		if (!tool_starts_with(module_list[i].name, tag)) {
			return &module_list[i];
		}
	}
	return (module_info_t *)NULL;
}

void mod_init( ) {
	module_response = module_request.response;
	uint64_t module_count = module_response->module_count;
	struct limine_file *module_ptr = (struct limine_file *)0;

	for (uint64_t i = 0; i < module_count; i++) {
		module_ptr = module_response->modules[i];

		LOG("[%d] %s [%s] 0x%x Размер: %u\n", i, module_ptr->path,
		    module_ptr->cmdline, module_ptr->address, module_ptr->size);

		if (modules_count >= MOD_MAX) {
			LOG("Модуль не обработан. Максимум %u модулей!\n", MOD_MAX);
			break;
		}

		module_list[modules_count].name = module_ptr->cmdline;
		module_list[modules_count].message = other_module_message;

		if (tool_starts_with(module_ptr->cmdline, "[BOOTIMG]")) {
			module_list[modules_count].data_size = module_ptr->size;
			module_list[modules_count].data = module_ptr->address;
			bootpng_ptr = module_ptr->address;
			bootpng_size = module_ptr->size;
			module_list[modules_count].type = 1; // Графика
			module_list[modules_count].message = graphics_module_message;
			modules_count++;
			continue;
		}

		if (!tool_starts_with(module_ptr->cmdline, "[MOD]")) {
			module_list[modules_count].type = 255; // Неизвестный тип модуля
			modules_count++;
			continue;
		}

		module_info_t (*module_init)(env_t * env) =
		    (module_info_t(*)(env_t * env))
		        elf_entry((elf64_header_t *)module_ptr->address);

		LOG("\t->Точка входа: 0x%x\n", module_init);

		main_env.offset = (uint64_t)module_ptr->address;
		main_env.info = (module_info_t *)0;

		sys_install(main_env);

		main_env.fb_printf = &fb_printf;

		module_info_t ret = module_init(&main_env);

		module_list[modules_count].message = ret.message;
		module_list[modules_count].data_size = ret.data_size;

		if (ret.data_size != 0) {
			module_list[modules_count].data =
			    (&(ret.data) + (uint64_t)module_ptr->address);
		}

		modules_count++;
	}
	LOG("Модулей обработано: %u\n", modules_count);
}