use lazy_static::lazy_static;
use x86_64::structures::idt::{InterruptStackFrame, InterruptDescriptorTable};
use crate::println;
use pic8259_simple::ChainedPics;
use spin;
use crate::gdt;

const PIC_1_OFFSET: u8 = 32;
const PIC_2_OFFSET: u8 = PIC_1_OFFSET + 8;
static PICS: spin::Mutex<ChainedPics> =
    spin::Mutex::new(unsafe { ChainedPics::new(PIC_1_OFFSET, PIC_2_OFFSET) });

lazy_static! {
  static ref IDT: InterruptDescriptorTable = {
    let mut idt = InterruptDescriptorTable::new();
    idt.breakpoint.set_handler_fn(breakpoint_handler);
    unsafe {
    idt.double_fault.set_handler_fn(double_fault_handler)
      .set_stack_index(gdt::DOUBLE_FAULT_IST_INDEX);
    }
    idt
  };
}

pub fn init_idt() {
  IDT.load();
}

pub fn initalize_pics() {
  unsafe {
    PICS.lock().initialize();
  }
}

extern "x86-interrupt" fn breakpoint_handler(stack_frame: &mut InterruptStackFrame) {

  println!("EXCEPTION - BREAKPOINT\n{:#?}", stack_frame);
}

extern "x86-interrupt" fn double_fault_handler(stack_frame: &mut InterruptStackFrame, code: u64) {
  panic!("EXCEPTION - DOUBLE FAULT (code {})\n{:#?}", code, stack_frame);
}