#pragma once

#include "cdef.h"
#include "cpu.h"

#define READ_IRQ()  read_cr8()
#define WRITE_IRQ(x)  write_cr8(x)

#define NUM_IDT_DESC (256)
#define NUM_GDT_DESC (7)
#define IDT_DESC_SIZE (16)
#define GDT_DESC_SIZE (8)
// + 8 because TSS descriptor is 16 bytes instead
#define GDT_SIZE ((NUM_GDT_DESC) * (GDT_DESC_SIZE))
#define IDT_SIZE ((NUM_IDT_DESC) * (IDT_DESC_SIZE))

#define GDT_K_CODE (1)
#define GDT_NULL (0)
#define GDT_K_DATA (2)
#define GDT_U_CODE (3)
#define GDT_U_DATA (4)
#define GDT_U_TSS (5)
#define SEL(idx, TI, RPL) ((((uint16)idx) << 3) | (((uint16) TI) << 2) | ((uint16) RPL))

struct PACKED gdtr
{
    uint16 size;
    uint64 offset;
};

struct PACKED idtr
{
    uint16 size;
    uint64 offset;
};

struct PACKED gdt_desc
{
    uint16 seg_l;
    uint16 base_l;
    uint8 base_m;
    uint8 attr1;
    uint8 attr2;
    uint8 base_h;
};

struct PACKED idt_desc
{
    uint16 offset_l;
    uint16 seg_sel;
    uint16 attr;
    uint16 offset_m;
    uint32 offset_h;
    uint32 reserved;
};

struct PACKED intr_frame
{
    uint64 gs;
    uint64 fs;
    uint64 es;
    uint64 ds;
    uint64 r15;
    uint64 r14;
    uint64 r13;
    uint64 r12;
    uint64 r11;
    uint64 r10;
    uint64 r9;
    uint64 r8;
    uint64 rsi;
    uint64 rdi;
    uint64 rbp;
    uint64 rdx;
    uint64 rcx;
    uint64 rbx;
    uint64 rax;
    uint64 error_code;
    uint64 rip;
    uint64 cs;
    uint64 rflags;
    uint64 rsp;
    uint64 ss;
};

struct PACKED tss
{
    uint32 unused0;
    uint64 rsp0;
    uint64 rsp1;
    uint64 rsp2;
    uint32 unused1[19];
};

#define INTR_VEC_TIMER    (50)
#define INTR_VEC_SPURIOUS (255)
#define INTR_VEC_SYSCALL  (51)
#define INTR_LIMIT_INTEL (31)


// returns the new exception frame pointer
// interrupt handlers should EOI
typedef void* (*intr_handler)(struct intr_frame *frame);

int32 intr_init();

void* ASM_F intr_dispatcher(uint32 vec, struct intr_frame *frame);

void set_intr_handler(uint32 vec, intr_handler handler);

void send_ipi(uint32 vec);

void stop_cpu();