/**
 * Copyright (c) 2016 rxi
 *
 * This library is free software; you can redistribute it and/or modify it
 * under the terms of the MIT license. See LICENSE for details.
 */

#include <string.h>
#include <pc.h>
#include <dpmi.h>
#include "luaobj.h"
#include "keyboard.h"

#define KEYBOARD_KEY_MAX 256
volatile char keyboard_keyStates[KEYBOARD_KEY_MAX];

volatile struct {
  struct { unsigned char type; unsigned char code; } data[32];
  int readi, writei;
} keyboard_events;

_go32_dpmi_seginfo old_keyb_handler_seginfo, new_keyb_handler_seginfo;


void keyboard_handler() {
  static unsigned char code, extended;
  code = inportb(0x60);

  if (code == 224) {
    /* Handle extended code */
    extended = 1 << 7;

  } else {
    /* Handle key up / down */
    if (code & (1 << 7)) {
      /* Key up */
      code &= ~(1 << 7);
      code |= extended;
      keyboard_keyStates[code] = 0;
      keyboard_events.data[keyboard_events.writei & 31].code = code;
      keyboard_events.data[keyboard_events.writei & 31].type = 0;
      keyboard_events.writei++;

    } else {
      /* Key down */
      code |= extended;
      if (!keyboard_keyStates[code]) { /* Ignore key repeat */
        keyboard_keyStates[code] = 1;
        keyboard_events.data[keyboard_events.writei & 31].code = code;
        keyboard_events.data[keyboard_events.writei & 31].type = 1;
        keyboard_events.writei++;
      }
    }
    extended = 0x0;
  }

  outportb(0x20, 0x20);
}


void keyboard_handler_end() {}


int keyboard_init(void) {
  _go32_dpmi_lock_data((char*)&keyboard_keyStates, KEYBOARD_KEY_MAX);
  _go32_dpmi_lock_data((char*)&keyboard_events, sizeof(keyboard_events));
  _go32_dpmi_lock_code(keyboard_handler,(unsigned long)keyboard_handler_end -
                       (unsigned long)keyboard_handler);
  _go32_dpmi_get_protected_mode_interrupt_vector(9, &old_keyb_handler_seginfo);
  new_keyb_handler_seginfo.pm_offset = (int)keyboard_handler;
  if (_go32_dpmi_allocate_iret_wrapper(&new_keyb_handler_seginfo) != 0) {
    return 1;
  }
  if (_go32_dpmi_set_protected_mode_interrupt_vector(
        9, &new_keyb_handler_seginfo) != 0
  ) {
    _go32_dpmi_free_iret_wrapper(&new_keyb_handler_seginfo);
    return 1;
  }
  return 0;
}


void keyboard_deinit(void) {
  _go32_dpmi_set_protected_mode_interrupt_vector(9, &old_keyb_handler_seginfo);
  _go32_dpmi_free_iret_wrapper(&new_keyb_handler_seginfo);
}


/**
 * Lua binds
 */

const char *scancodeMap[] = {
  [  1] = "escape",
  [  2] = "1",
  [  3] = "2",
  [  4] = "3",
  [  5] = "4",
  [  6] = "5",
  [  7] = "6",
  [  8] = "7",
  [  9] = "8",
  [ 10] = "9",
  [ 11] = "0",
  [ 12] = "-",
  [ 13] = "=",
  [ 14] = "backspace",
  [ 15] = "tab",
  [ 16] = "q",
  [ 17] = "w",
  [ 18] = "e",
  [ 19] = "r",
  [ 20] = "t",
  [ 21] = "y",
  [ 22] = "u",
  [ 23] = "i",
  [ 24] = "o",
  [ 25] = "p",
  [ 26] = "[",
  [ 27] = "]",
  [ 28] = "enter",
  [ 29] = "lctrl",
  [ 30] = "a",
  [ 31] = "s",
  [ 32] = "d",
  [ 33] = "f",
  [ 34] = "g",
  [ 35] = "h",
  [ 36] = "j",
  [ 37] = "k",
  [ 38] = "l",
  [ 39] = ";",
  [ 40] = "\"",
  [ 41] = "`",
  [ 42] = "lshift",
  [ 43] = "\\",
  [ 44] = "z",
  [ 45] = "x",
  [ 46] = "c",
  [ 47] = "v",
  [ 48] = "b",
  [ 49] = "n",
  [ 50] = "m",
  [ 51] = ",",
  [ 52] = ".",
  [ 53] = "/",
  [ 54] = "rshift",
  [ 55] = "*",
  [ 56] = "lalt",
  [ 57] = " ",
  [ 58] = "capslock",
  [ 59] = "f1",
  [ 60] = "f2",
  [ 61] = "f3",
  [ 62] = "f4",
  [ 63] = "f5",
  [ 64] = "f6",
  [ 65] = "f7",
  [ 66] = "f8",
  [ 67] = "f9",
  [ 68] = "f10",
  [ 69] = "numlock",
  [ 70] = "scrolllock",
  [ 71] = "home",
  [ 72] = "up",
  [ 73] = "pageup",
  [ 74] = "-",
  [ 75] = "left",
  [ 76] = "5",
  [ 77] = "right",
  [ 79] = "end",
  [ 78] = "+",
  [ 80] = "down",
  [ 81] = "pagedown",
  [ 82] = "insert",
  [ 83] = "delete",
  [ 84] = "?",
  [ 85] = "?",
  [ 86] = "?",
  [ 87] = "?",
  [ 88] = "?",
  [ 89] = "?",
  [ 90] = "?",
  [ 91] = "?",
  [ 92] = "?",
  [ 93] = "?",
  [ 94] = "?",
  [ 95] = "?",
  [ 96] = "enter",
  [ 97] = "lctrl",
  [ 98] = "/",
  [ 99] = "f12",
  [100] = "rctrl",
  [101] = "pause",
  [102] = "home",
  [103] = "up",
  [104] = "pageup",
  [105] = "left",
  [106] = "right",
  [107] = "end",
  [108] = "down",
  [109] = "pagedown",
  [110] = "insert",
  [111] = "?",
  [112] = "?",
  [113] = "?",
  [114] = "?",
  [115] = "?",
  [116] = "?",
  [117] = "?",
  [118] = "?",
  [119] = "pause",
  [120] = "?",
  [121] = "?",
  [122] = "?",
  [123] = "?",
  [124] = "?",
  [125] = "?",
  [126] = "?",
  [127] = "?",
  [128] = NULL,
};


int l_keyboard_isDown(lua_State *L) {
  int code = luaL_checknumber(L, 1);
  if (code < 0 || code >= KEYBOARD_KEY_MAX) return 0;
  lua_pushboolean(L, keyboard_keyStates[code]);
  return 1;
}


int l_keyboard_poll(lua_State *L) {
  lua_newtable(L);
  int idx = 1;

  while (keyboard_events.readi != keyboard_events.writei) {
    lua_newtable(L);

    int type = keyboard_events.data[keyboard_events.readi & 31].type;
    int code = keyboard_events.data[keyboard_events.readi & 31].code;
    code &= 127;

    lua_pushstring(L, type ? "down" : "up");
    lua_setfield(L, -2, "type");
    lua_pushnumber(L, code);
    lua_setfield(L, -2, "code");
    lua_pushstring(L, scancodeMap[code]);
    lua_setfield(L, -2, "key");

    lua_rawseti(L, -2, idx);
    keyboard_events.readi++;
    idx++;
  }
  return 1;
}


int luaopen_keyboard(lua_State *L) {
  luaL_Reg reg[] = {
    { "poll",       l_keyboard_poll       },
    { "isDown",     l_keyboard_isDown     },
    { 0, 0 },
  };
  luaL_newlib(L, reg);
  return 1;
}