#include <arch/x86/interrupts.h>
#include <hal/acpi/acpi.h>
#include <hal/ioapic.h>
#include <lunaix/common.h>
#include <lunaix/mm/mmio.h>

#define IOAPIC_REG_SEL *((volatile uint32_t*)(_ioapic_base + IOAPIC_IOREGSEL))
#define IOAPIC_REG_WIN *((volatile uint32_t*)(_ioapic_base + IOAPIC_IOWIN))

static volatile uintptr_t _ioapic_base;

void
ioapic_init()
{
    // Remapping the IRQs

    acpi_context* acpi_ctx = acpi_get_context();

    _ioapic_base = ioremap(acpi_ctx->madt.ioapic->ioapic_addr & ~0xfff, 4096);
}

void
ioapic_write(uint8_t sel, uint32_t val)
{
    IOAPIC_REG_SEL = sel;
    IOAPIC_REG_WIN = val;
}

uint32_t
ioapic_read(uint8_t sel)
{
    IOAPIC_REG_SEL = sel;
    return IOAPIC_REG_WIN;
}

void
ioapic_redirect(uint8_t irq, uint8_t vector, uint8_t dest, uint32_t flags)
{
    uint8_t reg_sel = IOAPIC_IOREDTBL_BASE + irq * 2;

    // Write low 32 bits
    ioapic_write(reg_sel, (vector | flags) & 0x1FFFF);

    // Write high 32 bits
    ioapic_write(reg_sel + 1, (dest << 24));
}