#include <stdint.h>
#include <kernel/interrupts/interrupts.h>
#include <kernel/lib/lib.h>
#include <kernel/system.h>
#include <kernel/io/io.h>
#include <mlibc/string.h>

// Defined in isr.asm
extern void isr0();

extern void isr1();

extern void isr2();

extern void isr3();

extern void isr4();

extern void isr5();

extern void isr6();

extern void isr7();

extern void isr8();

extern void isr9();

extern void isr10();

extern void isr11();

extern void isr12();

extern void isr13();

extern void isr14();

extern void isr15();

extern void isr16();

extern void isr17();

extern void isr18();

extern void isr19();

extern void isr20();

extern void isr21();

extern void isr22();

extern void isr23();

extern void isr24();

extern void isr25();

extern void isr26();

extern void isr27();

extern void isr28();

extern void isr29();

extern void isr30();

extern void isr31();

uint32_t ignored_isr[8] = {0};

// Install ISRs in IDT
void isrs_install() {
    idt_set_gate(0, (unsigned) isr0, 0x08, 0x8E);
    idt_set_gate(1, (unsigned) isr1, 0x08, 0x8E);
    idt_set_gate(2, (unsigned) isr2, 0x08, 0x8E);
    idt_set_gate(3, (unsigned) isr3, 0x08, 0x8E);
    idt_set_gate(4, (unsigned) isr4, 0x08, 0x8E);
    idt_set_gate(5, (unsigned) isr5, 0x08, 0x8E);
    idt_set_gate(6, (unsigned) isr6, 0x08, 0x8E);
    idt_set_gate(7, (unsigned) isr7, 0x08, 0x8E);

    idt_set_gate(8, (unsigned) isr8, 0x08, 0x8E);
    idt_set_gate(9, (unsigned) isr9, 0x08, 0x8E);
    idt_set_gate(10, (unsigned) isr10, 0x08, 0x8E);
    idt_set_gate(11, (unsigned) isr11, 0x08, 0x8E);
    idt_set_gate(12, (unsigned) isr12, 0x08, 0x8E);
    idt_set_gate(13, (unsigned) isr13, 0x08, 0x8E);
    idt_set_gate(14, (unsigned) isr14, 0x08, 0x8E);
    idt_set_gate(15, (unsigned) isr15, 0x08, 0x8E);

    idt_set_gate(16, (unsigned) isr16, 0x08, 0x8E);
    idt_set_gate(17, (unsigned) isr17, 0x08, 0x8E);
    idt_set_gate(18, (unsigned) isr18, 0x08, 0x8E);
    idt_set_gate(19, (unsigned) isr19, 0x08, 0x8E);
    idt_set_gate(20, (unsigned) isr20, 0x08, 0x8E);
    idt_set_gate(21, (unsigned) isr21, 0x08, 0x8E);
    idt_set_gate(22, (unsigned) isr22, 0x08, 0x8E);
    idt_set_gate(23, (unsigned) isr23, 0x08, 0x8E);

    idt_set_gate(24, (unsigned) isr24, 0x08, 0x8E);
    idt_set_gate(25, (unsigned) isr25, 0x08, 0x8E);
    idt_set_gate(26, (unsigned) isr26, 0x08, 0x8E);
    idt_set_gate(27, (unsigned) isr27, 0x08, 0x8E);
    idt_set_gate(28, (unsigned) isr28, 0x08, 0x8E);
    idt_set_gate(29, (unsigned) isr29, 0x08, 0x8E);
    idt_set_gate(30, (unsigned) isr30, 0x08, 0x8E);
    idt_set_gate(31, (unsigned) isr31, 0x08, 0x8E);

    vga_log("Installed Interrupt Service Routines", 7);
}

// Error exception messages
const char *exception_messages[] = {
        "Division By Zero",
        "Debug",
        "Non Maskable Interrupt",
        "Breakpoint",
        "Into Detected Overflow",
        "Out of Bounds",
        "Invalid Opcode",
        "No Coprocessor",

        "Double Fault",
        "Coprocessor Segment Overrun",
        "Bad TSS",
        "Segment Not Present",
        "Stack Fault",
        "General Protection Fault",
        "Page Fault",
        "Unknown Interrupt",

        "Coprocessor Fault",
        "Alignment Check",
        "Machine Check",
        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved",

        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved",
        "Reserved"
};

// Master exception/interrupt/fault handler - halt via panic
void fault_handler(struct regs *r) {
    if (r->int_no < 32 && !(ignored_isr[r->int_no / 32] & (1 << (r->int_no % 32)))) {
        uint32_t faulting_address;
        asm volatile("mov %%cr2, %0" : "=r" (faulting_address));

        serial_write("\n[DEBUG]\nEIP: ");
        serial_write_hex(r->eip);
        serial_write("\nEAX: ");
        serial_write_hex(r->eax);
        serial_write("\nEBX: ");
        serial_write_hex(r->ebx);
        serial_write("\nECX: ");
        serial_write_hex(r->ecx);
        serial_write("\nEDX: ");
        serial_write_hex(r->edx);
        serial_write("\nFaulting address: ");
        serial_write_hex(faulting_address);
        serial_write("\nError flags: ");
        serial_write_hex(r->eflags);
        serial_write("\nError code: ");
        serial_write_hex(r->err_code);
        char *message = (char *) exception_messages[r->int_no];
        strcat(message, " Exception");
        if (r->err_code == 2) halt_loop(); // Idk loop?
        panic(message);
    }
}


void isr_ignore(uint8_t int_no) { ignored_isr[int_no / 32] |= 1 << (int_no % 32); }

void isr_remember(uint8_t int_no) { ignored_isr[int_no / 32] &= ~(1 << (int_no % 32)); }