#include<source/keyboard.h>
#include<types.h>
#include<asm.h>
#include<source/stdio.h>

#define BUFFER_SIZE 200
#define BUFFER_LOG 200
char buffer[BUFFER_LOG][BUFFER_SIZE];
size_t buffer_size[BUFFER_LOG];
size_t buffer_current=0;
size_t buffer_all=0;
size_t buffer_index=0;

#define PIC1_COMMAND_PORT 0x20
#define PIC1_DATA_PORT 0x21
#define PIC2_COMMAND_PORT 0xA0
#define PIC2_DATA_PORT 0xA1
// IO Ports for Keyboard
#define KEYBOARD_DATA_PORT 0x60
#define KEYBOARD_STATUS_PORT 0x64

void previous_field(void);
void tty(char *buffer);
void prompt(void);
void clear(void);
void us_en(char keymap[]);
void us_en_shift(char keymap[]);

char charcode[256];
char shift_charcode[256];
bool ispressed[128];
#define lshift 0x2A
#define rshift 0x36
#define lctrl 0x1D
#define rctrl 0x1D

void init_keyboard()
{
    us_en(charcode);
    us_en_shift(shift_charcode);
}

void deletelast()
{
    previous_field();
    printf(" ");
    previous_field();
}

void backspace()
{
    if(buffer_index<=0) return;

    deletelast();
    buffer[buffer_current][--buffer_index]='\0';
    return;
}

void enter()
{
    printf("\n");
    if(buffer_index>0)
    {
        tty(buffer[buffer_current]);
        buffer_size[buffer_current]=buffer_index;
        if(buffer_current==buffer_all) buffer_current=(++buffer_all);
        else
        {
            for(size_t i=0;i<BUFFER_SIZE;i++) buffer[buffer_all][i]='\0';
            buffer_current=buffer_all;
        }
        buffer_index=0;
    }
    prompt();
    return;
}

void space()
{
    buffer[buffer_current][buffer_index++]=' ';
    printf(" ");
}

void keyup()
{
    if(buffer_current>0)
    {
        buffer_size[buffer_current]=buffer_index;
        for(size_t i=0;i<buffer_index;i++) deletelast();
        buffer_current--;
        buffer_index=buffer_size[buffer_current];
        printf("%s",buffer[buffer_current]);
    }
}

void keydown()
{
    if(buffer_current<buffer_all)
    {
        buffer_size[buffer_current]=buffer_index;
        for(size_t i=0;i<buffer_index;i++) deletelast();
        buffer_current++;
        buffer_index=buffer_size[buffer_current];
        printf("%s",buffer[buffer_current]);
    }
}

void keyleft()
{
}

void keyright()
{
}

void keyboard_handler()
{
    ioport_out(PIC1_COMMAND_PORT, 0x20);
    uint8_t status = ioport_in(KEYBOARD_STATUS_PORT);

    if (status & 0x1)
    {
        uint8_t keycode = ioport_in(KEYBOARD_DATA_PORT);
        if(keycode<0x80)
        {
            char c=charcode[keycode];
            ispressed[keycode]=1;
            // printf("%d ",&keycode);

            switch(keycode)
            {
                case 0x0E:
                    backspace();
                    break;
                case 0x1C:
                    enter();
                    break;
                case 0x39:
                    space();
                    break;
                case 72:
                    keyup();
                    break;
                case 80:
                    keydown();
                    break;
                case 75:
                    keyleft();
                    break;
                case 77:
                    keyright();
                    break;

                default:
                    if(c!=' ')
                    {
                        if(ispressed[lctrl]||ispressed[rctrl])
                        {
                            if(c=='l')
                            {
                                clear();
                                prompt();
                                printf("%s",buffer[buffer_current]);
                                return;
                            }
                        }
                        if(ispressed[lshift]||ispressed[rshift])
                        {
                            c=shift_charcode[keycode];
                        }
                        buffer[buffer_current][buffer_index++]=c;
                        printf("%c",c);
                    }
                    break;
            }

        }
        else
        {
            ispressed[keycode-0x80]=0;
        }
    }
}