#include "keyboard.h"
#include "serial.h"
#include "vga.h"
#include "../kernel/io.h" // Отсюда заберём inb
#include "../kernel/logger.h"

// Карта скан-кодов (US Layout, первый сет). 0 — это непечатаемые символы
static const char kbdus[128] = {
    0,  27, '1', '2', '3', '4', '5', '6', '7', '8', '9', '0', '-', '=', '\b',
  '\t', 'q', 'w', 'e', 'r', 't', 'y', 'u', 'i', 'o', 'p', '[', ']', '\n',
    0,  'a', 's', 'd', 'f', 'g', 'h', 'j', 'k', 'l', ';', '\'', '`',   0,
 '\\', 'z', 'x', 'c', 'v', 'b', 'n', 'm', ',', '.', '/',   0, '*',   0,
  ' ',   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
    0,   0,   0,   0, '-',   0,   0,   0, '+',   0,   0,   0,   0,   0,
    0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0 // Остальное нули
};

void keyboard_init(void) {
    // Чисто для логов, unmask мы уже сделали в kmain
    log_info(LOG_NORMAL, "Keyboard driver ready.\n");
}

void keyboard_handler(void) {
    // Читаем байт из порта контроллера клавиатуры (8042)
    uint8_t scancode = inb(0x60);

    // Если старший бит (0x80) установлен, это отпускание клавиши (Key Release). Игнорируем пока.
    if (scancode & 0x80) {
        // Здесь можно отслеживать отпускание Shift/Ctrl
    } else {
        // Клавиша нажата (Key Press). Проверяем, входит ли в нашу карту
        if (scancode < 128 && kbdus[scancode] != 0) {
            char c = kbdus[scancode];
            
            // Выводим символ на экран через VGA драйвер
            vga_putchar(c);
            
            // И дублируем в сериал для надежной отладки
            char str[2] = {c, 0};
            kprint_serial(str);
        }
    }
}