#include <kapi/boot_modules.hpp>

#include <arch/boot/boot.hpp>
#include <arch/boot/ld.hpp>

#include <kapi/boot.hpp>
#include <kapi/boot_modules/boot_module.hpp>
#include <kapi/boot_modules/registry.hpp>
#include <kapi/memory.hpp>
#include <kapi/system.hpp>

#include <kstd/print.hpp>

#include <multiboot2/information.hpp>

#include <algorithm>
#include <bit>
#include <cstdint>

namespace kapi::boot_modules
{

  auto init() -> void
  {
    registry::init();

    kstd::println("[x86_64:BOOT_MODULES] Registering boot modules.");

    auto modules = boot::bootstrap_information.mbi->modules();
    std::ranges::for_each(modules, [](auto const & module) {
      registry::get().add(kapi::boot_modules::boot_module{
        .name = module.string(),
        .start_address =
            memory::linear_address{module.start_address + std::bit_cast<std::uintptr_t>(&arch::boot::TEACHOS_VMA)},
        .size = module.end_address - module.start_address,
      });
    });
  }
}  // namespace kapi::boot_modules