#include <kernel/interrupts/interrupts.h>
#include <kernel/io/io.h>
#include <kernel/graphics/vesa.h>

char mouse_cycle = 0;
signed char mouse_byte[3];
signed char mouse_x = 0;
signed char mouse_y = 0;
int mouse_but_1 = 0;
int mouse_but_2 = 0;

void mouse_handler(struct regs *a_r) {
    switch (mouse_cycle) {
        case 0:
            mouse_byte[0] = receive_b(0x60);
            mouse_cycle++;
            break;
        case 1:
            mouse_byte[1] = receive_b(0x60);
            mouse_cycle++;
            break;
        case 2:
            mouse_byte[2] = receive_b(0x60);
            mouse_x = mouse_byte[1];
            mouse_y = mouse_byte[2];
            mouse_but_1 = (mouse_byte[0] % 2);
            mouse_but_2 = ((mouse_byte[0] % 4) - (mouse_byte[0] % 2)) / 2;
            mouse_cycle = 0;
            break;
        default:
            break;
    }

    if (mouse_but_1 == 1)
        vesa_draw_string("CLICK!\n");
}

void mouse_wait(unsigned char a_type) {
    unsigned int time_out = 100000;
    if (a_type == 0) {
        while (time_out--)
            if ((receive_b(0x64) & 1) == 1)
                return;
        return;
    } else {
        while (time_out--)
            if ((receive_b(0x64) & 2) == 0)
                return;
        return;
    }
}

void mouse_write(unsigned char a_write) {
    mouse_wait(1);
    send_b(0x64, 0xD4);
    mouse_wait(1);
    send_b(0x60, a_write);
}

char mouse_read() {
    mouse_wait(0);
    return receive_b(0x60);
}

void mouse_install() {
    unsigned char status;

    // Enable auxiliary mouse device
    mouse_wait(1);
    send_b(0x64, 0xA8);

    // Enable interrupts
    mouse_wait(1);
    send_b(0x64, 0x20);
    mouse_wait(0);
    status = (receive_b(0x60) | 2);
    mouse_wait(1);
    send_b(0x64, 0x60);
    mouse_wait(1);
    send_b(0x60, status);

    // Use default settings
    mouse_write(0xF6);
    mouse_read();

    // Enable mouse
    mouse_write(0xF4);
    mouse_read();

    // Setup the mouse handler
    irq_install_handler(2, mouse_handler);
}