#include "./prelude/kernel_prelude.hats"

#define ATS_DYNLOADFLAG 0

staload "kernel/interrupts/idt.sats"
staload "kernel/bootinfo/multiboot.sats"
staload "kernel/memory/frame.sats"
staload "lib/SATS/writer.sats"

%{^
  void breakpoint(){ 
    __asm__("int3");
  }
%}

//NOTE: how to make sure ats checks 2 contstraints in one if statement?
//TODO: put in seperate utils module (just generic size function)
fn kernel_size(kernel_start : Ptr, kernel_end : Ptr) : size_t = let
  extern castfn Ptr12size (a : Ptr1): size_t
in
  if (kernel_end > the_null_ptr) then 
    if (kernel_start > the_null_ptr) then
      Ptr12size(kernel_end) - Ptr12size(kernel_start)
    else
      i2sz(0)
  else
    i2sz(0)
end

fn test(allocptr : !allocptr, bootptr : !bootptr) : void = let
  fun loop(i : int) : void = if (i < 160) then( let val _ = allocate_frame(allocptr.0, bootptr.0 | allocptr.2, bootptr.2) in end; loop(succ(i)))
in
  loop(0)
end


extern fun breakpoint() :void  = "mac#"

extern fun main(p : Ptr1) : void = "ext#"
implement  main(p) = let
  val () = clear_screen()

  // Initialize boot info struct
  val bootptr = boot_info_init(p) 
  val (kernel_start,kernel_end) = get_kernel_ranges(bootptr)
  val (multiboot_start,multiboot_end) = get_multiboot_ranges(bootptr)
  val allocptr = frame_allocator_init(kernel_start,kernel_end,multiboot_start,multiboot_end,bootptr)
in

  print_memory_mappings(bootptr);

  print_elf_headers(bootptr);
  println!("Kernel Size: ", kernel_size(kernel_start,kernel_end));
  test(allocptr,bootptr);
  let
    val frame = allocate_frame(allocptr.0, bootptr.0 | allocptr.2, bootptr.2)
  in 
    println!(frame.num)
  end;

  // Initialize interrupt table
  idt_init();

  // Throws breakpoint exception
  breakpoint();


  println!("End");

  let
    prval () = bootptr.1(bootptr.0)
    prval () = allocptr.1(allocptr.0)
  in end


end