#include <uefi/boot_services.h>
#include <uefi/types.h>

#include "bootconfig.h"
#include "error.h"
#include "fs.h"
#include "status.h"

namespace boot {

constexpr uint64_t jsixboot = 0x746f6f627869736a; // "jsixboot"

static const wchar_t *
read_string(util::buffer &data)
{
    uint16_t size = *util::read<uint16_t>(data);
    const wchar_t *string = reinterpret_cast<const wchar_t*>(data.pointer);
    data += size;
    return string;
}

static void
read_descriptor(descriptor &e, util::buffer &data)
{
    e.flags = static_cast<desc_flags>(*util::read<uint16_t>(data));
    e.path = read_string(data);
}

bootconfig::bootconfig(util::buffer data, uefi::boot_services *bs)
{
    status_line status {L"Loading boot config"};

    if (*util::read<uint64_t>(data) != jsixboot)
        error::raise(uefi::status::load_error, L"Bad header in boot config");

    const uint8_t version = *util::read<uint8_t>(data);
    if (version != 1)
        error::raise(uefi::status::incompatible_version, L"Bad version in boot config");

    uint8_t num_panics = *util::read<uint8_t>(data);
    m_flags = *util::read<uint16_t>(data);

    read_descriptor(m_kernel, data);
    read_descriptor(m_init, data);
    m_initrd = read_string(data);
    m_symbols = read_string(data);

    m_panics.count = num_panics;
    m_panics.pointer = new descriptor [num_panics];

    for (unsigned i = 0; i < num_panics; ++i)
        read_descriptor(m_panics[i], data);
}

} // namespace boot