#include <irq/irq.h>
#include <libk/list.h>
#include <libk/std.h>
#include <mm/liballoc.h>
#include <sync/rw_spin_lock.h>

#if defined(__x86_64__)
#include <amd64/apic.h>
#include <amd64/intr.h>
#endif

struct irq* irq_table[0x100];

static rw_spin_lock_t irqs_lock;

bool irq_attach (void (*func) (void*, void*), void* arg, uint32_t irq_num, uint32_t flags) {
  struct irq* irq = malloc (sizeof (*irq));
  if (irq == NULL) {
    return false;
  }

  irq->func = func;
  irq->arg = arg;
  irq->irq_num = irq_num;
  irq->flags = flags;

  rw_spin_write_lock (&irqs_lock);
  irq_table[irq_num] = irq;
  rw_spin_write_unlock (&irqs_lock);

#if defined(__x86_64__)
  uint8_t resolution = amd64_resolve_irq (irq_num);
  amd64_ioapic_route_irq (irq_num, resolution, 0, amd64_lapic_id ());
#endif

  return true;
}

struct irq* irq_find (uint32_t irq_num) {
  rw_spin_read_lock (&irqs_lock);

  struct irq* irq = irq_table[irq_num];

  rw_spin_read_unlock (&irqs_lock);

  return irq;
}