#include <types.h>
#include <keyboard.h>
#include <pic.h>
#include <io.h>
#include <keymap.h>
#include <graphics.h>
#include <libk/stdio.h>
#include <libk/math.h>
#include <libk/string.h>
#include <stdbuff.h>
#include <heap.h>

bool is_pressed[128];

#define BUFFER_SIZE 10000
stdbuff *keyboard_buffer;

void init_keyboard()
{
	/*
	outb(KEYBOARD_CMD_PORT, 0xF3);
	io_wait();
	outb(KEYBOARD_DATA_PORT, 0x00);
	io_wait();
	while (!(inb(KEYBOARD_STATUS_PORT) & 1))
		;
	if (inb(KEYBOARD_DATA_PORT) == 0xFA)
		printf("[keyboard init]\n");
	*/
}

void keyboard_handler()
{
	if (keyboard_buffer == NULL) {
		keyboard_buffer = init_buff(BUFFER_SIZE);
		memset(keyboard_buffer->data, ' ', BUFFER_SIZE);
	}

	uint8_t status = inb(KEYBOARD_STATUS_PORT);
	if (!(status & 0x1)) {
		return;
	}

	uint8_t keycode = inb(KEYBOARD_DATA_PORT);

	if (keycode < keymap_len) {
		is_pressed[keycode] = true;
		if (keycode == KEY_SPACE) {
			write_buff(keyboard_buffer, " ", 1);
		} else if (keycode == KEY_BACKSPACE) {
			if (main_fb.x != 0)
				write_buff(keyboard_buffer, "\b \b", 3);
		} else if (keycode == KEY_ENTER) {
			char *output = kalloc(sizeof(char) + 1);
			output[0] = keymap[keycode];
			write_buff(keyboard_buffer, output, 1);
			kfree(output);
		} else {
			char *output = kalloc(sizeof(char) + 1);
			if (keymap[keycode] == ' ')
				return;
			if (is_pressed[KEY_LSHIFT] || is_pressed[KEY_RSHIFT])
				output[0] = shift_keymap[keycode];
			else
				output[0] = keymap[keycode];
			write_buff(keyboard_buffer, output, 1);
			kfree(output);
		}
	} else {
		is_pressed[keycode - 128] = false;
	}

	uint32_t len =
		(uint32_t)(keyboard_buffer->head >= keyboard_buffer->tail ?
				   keyboard_buffer->head -
					   keyboard_buffer->tail :
				   BUFFER_SIZE + keyboard_buffer->head -
					   keyboard_buffer->tail);
	char *print_buff = kalloc(len + 1);
	read_buff(keyboard_buffer, print_buff, len);
	printf("%s", print_buff);
	kfree(print_buff);
}