#include <assert.h>
#include <cpu/gdt.h>
#include <cpu/idt.h>
#include <cpu/spinlock.h>
#include <cpu/syscall.h>
#include <crypto/SHA1/sha1.h>
#include <drivers/ahci.h>
#include <drivers/ata.h>
#include <drivers/keyboard.h>
#include <drivers/mouse.h>
#include <drivers/pit.h>
#include <drivers/rtl8139.h>
#include <drivers/serial.h>
#include <drivers/vbe.h>
#include <fs/devfs.h>
#include <fs/ext2.h>
#include <fs/shm.h>
#include <fs/vfs.h>
#include <interrupts.h>
#include <log.h>
#include <mmu.h>
#include <multiboot.h>
#include <network/arp.h>
#include <random.h>
#include <sched/scheduler.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdio.h>
#include <string.h>
#include <typedefs.h>

#if defined(__linux__)
#error "You are not using a cross-compiler."
#endif

#if !defined(__i386__)
#error "This OS needs to be compiled with a ix86-elf compiler"
#endif

u32 inital_esp;
uintptr_t data_end;

void kernel_main(u32 kernel_end, unsigned long magic, unsigned long addr,
                 u32 inital_stack) {
  (void)kernel_end;
  data_end = 0xc0400000;
  inital_esp = inital_stack;

  disable_interrupts();
  kprintf("If you see this then the serial driver works :D.\n");
  assert(magic == MULTIBOOT_BOOTLOADER_MAGIC);

  multiboot_info_t *mb = (multiboot_info_t *)(addr + 0xc0000000);
  u32 mem_kb = mb->mem_lower;
  u32 mem_mb = (mb->mem_upper - 1000) / 1000;
  u64 memsize_kb = mem_mb * 1000 + mem_kb;
  paging_init(memsize_kb);
  klog("Paging Initalized", LOG_SUCCESS);
  mb = mmu_map_frames((multiboot_info_t *)addr, sizeof(multiboot_info_t));

  gdt_init();
  klog("GDT Initalized", LOG_SUCCESS);

  idt_init();
  klog("IDT Initalized", LOG_SUCCESS);

  syscalls_init();
  klog("Syscalls Initalized", LOG_SUCCESS);

  pit_install();
  set_pit_count(1000);
  klog("PIT driver installed", LOG_SUCCESS);

  ata_init();
  klog("ATA Initalized", LOG_SUCCESS);

  tasking_init();
  klog("Tasking Initalized", LOG_SUCCESS);

  install_mouse();
  klog("PS2 Mouse driver installed", LOG_SUCCESS);

  install_keyboard();
  klog("PS2 Keyboard driver installed", LOG_SUCCESS);

  vfs_mount("/dev", devfs_mount());
  ahci_init();
  vfs_mount("/", ext2_mount());

  add_stdout();
  add_serial();
  add_random_devices();
  shm_init();

  setup_random();

  add_keyboard();
  add_mouse();

  ////////
  rtl8139_init();
  enable_interrupts();

  /*
  u32 ip = gen_ipv4(10, 0, 2, 2);
  int err;
  u32 listen_socket = tcp_listen_ipv4(ip, 6000, &err);
  assert(!err);
  u32 socket = tcp_accept(listen_socket, &err);
  assert(!err);

  kprintf("socket: %x\n", socket);

  u64 out;
  char buffer[256];
  assert(tcp_read(socket, buffer, 256, &out));
  kprintf("got %x bytes\n", out);
  for (int i = 0; i < out; i++) {
    kprintf("%c", buffer[i]);
  }
  */

  ipv4_t gateway;
  gateway.d = gen_ipv4(10, 0, 2, 2);
  ipv4_t bitmask;
  bitmask.d = gen_ipv4(255, 255, 255, 0);
  setup_network(gateway, bitmask);

  /*
  //  u32 ip = gen_ipv4(10, 0, 2, 2);
  u32 ip = gen_ipv4(93, 184, 216, 34);
  int err;
  u32 socket = tcp_connect_ipv4(ip, 80, &err);
  assert(!err);
  kprintf("socket: %x\n", socket);
  u64 out;
  char *http_request = "GET /\r\n\r\n";
  assert(tcp_write(socket, http_request, strlen(http_request), &out));
  kprintf("sent: %x over the tcp socket\n", out);

  for (;;) {
    char buffer[256];
    assert(tcp_read(socket, buffer, 256, &out));
    kprintf("got %x bytes\n", out);
    for (int i = 0; i < out; i++) {
      kprintf("%c", buffer[i]);
    }
  }

  for (;;)
    ;
    */
  display_driver_init(mb);
  add_vbe_device();
  int pid;
  if (0 == (pid = fork())) {
    char *argv[] = {"/init", NULL};
    if (0 == exec("/init", argv)) {
      kprintf("exec() failed\n");
    }
  }
  for (;;) {
    current_task->sleep_until = pit_num_ms() + 100000000;
    //    enable_interrupts();
    switch_task();
  }
}