#include "../include/keyboard.h"
#include <stdbool.h>

unsigned char kbdus[128] =
{
    0,  27, '1', '2', '3', '4', '5', '6', '7', '8',     /* 9 */
  '9', '0', '-', '=', '\b',     /* Backspace */
  '\t',                 /* Tab */
  'q', 'w', 'e', 'r',   /* 19 */
  't', 'y', 'u', 'i', 'o', 'p', '[', ']', '\n', /* Enter key */
    0,                  /* 29   - Control */
  'a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l', ';',     /* 39 */
 '\'', '`',   0,                /* Left shift */
 '\\', 'z', 'x', 'c', 'v', 'b', 'n',                    /* 49 */
  'm', ',', '.', '/',   0,                              /* Right shift */
  '*',
    0,  /* Alt */
  ' ',  /* Space bar */
    0,  /* Caps lock */
    0,  /* 59 - F1 key ... > */
    0,   0,   0,   0,   0,   0,   0,   0,
    0,  /* < ... F10 */
    0,  /* 69 - Num lock*/
    0,  /* Scroll Lock */
    0,  /* Home key */
    0,  /* Up Arrow */
    0,  /* Page Up */
  '-',
    0,  /* Left Arrow */
    0,
    0,  /* Right Arrow */
  '+',
    0,  /* 79 - End key*/
    0,  /* Down Arrow */
    0,  /* Page Down */
    0,  /* Insert Key */
    0,  /* Delete Key */
    0,   0,   0,
    0,  /* F11 Key */
    0,  /* F12 Key */
    0,  /* All other keys are undefined */
};

unsigned char kbdus2[128] =
{
    0,  27, '!', '@', '#', '$', '%', '^', '&', '*',     /* 9 */
  '(', ')', '_', '+', '\b',     /* Backspace */
  '\t',                 /* Tab */
  'Q', 'W', 'E', 'R',   /* 19 */
  'T', 'Y', 'U', 'I', 'O', 'P', '{', '}', '\n', /* Enter key */
    0,                  /* 29   - Control */
  'A', 'S', 'D', 'F', 'G', 'H', 'J', 'K', 'L', ':',     /* 39 */
 '\'', '`',   0,                /* Left shift */
 '\\', 'Z', 'X', 'C', 'V', 'B', 'N',                    /* 49 */
  'M', '<', '>', '?',   0,                              /* Right shift */
  '*',
    0,  /* Alt */
  ' ',  /* Space bar */
    0,  /* Caps lock */ 
    0,  /* 59 - F1 key ... > */
    0,   0,   0,   0,   0,   0,   0,   0,
    0,  /* < ... F10 */ 
    0,  /* 69 - Num lock*/
    0,  /* Scroll Lock */
    0,  /* Home key */
    0,  /* Up Arrow */
    0,  /* Page Up */
  '-',
    0,  /* Left Arrow */
    0,
    0,  /* Right Arrow */
  '+',
    0,  /* 79 - End key*/
    0,  /* Down Arrow */
    0,  /* Page Down */ 
    0,  /* Insert Key */
    0,  /* Delete Key */
    0,   0,   0,
    0,  /* F11 Key */
    0,  /* F12 Key */   
    0,  /* All other keys are undefined */
};


bool shift_pressed;
bool caps_pressed;

void keyboard_wait();

void handle_keyboard()
{//3a
	uint8_t scancode = inb(0x60);

	if(scancode & 0x80)//key released
	{
		scancode = scancode-0x80;
		if(scancode == 0x2a || scancode == 0x36)
			shift_pressed = false;
		//printf("(%X)%c released ", scancode, kbdus[scancode-0x80]);
	}
	else//key pressed
	{
		if(scancode == 0x2a || scancode == 0x36)
		{
			shift_pressed = true;
			return;
		}

		if(scancode == 0x3a)
		{
			caps_pressed = !caps_pressed;
			outb(0x60, 0xed);//set LED command
			keyboard_wait();
			outb(0x60, caps_pressed << 2);//0 is scroll lock, 1 is numlock, 2 is caps lock
			return;
		}

		if(shift_pressed || caps_pressed)
			printf("%c", kbdus2[scancode]);
		else
			printf("%c", kbdus[scancode]); //printf("(0x%X)%c", scancode, kbdus[scancode]);
	}
}

void keyboard_wait()
{
	int status;
	while(1)
	{
		status = inb(0x60);
		if(status == 0xfa || status == 0xfe)
			return;
	}
}