#include "devterm.h"
#include "keyboard.h"
#include "keys.h"

#define EMP 0XFFFF

/*
  B1 joystick up
  B2 joystick down
  B3 joystick left
  B4 joystick right

  B5 joystick A
  B6 joystick B
  B7 joystick X
  B8 joystick Y

  B9 left shift
  B10 Fn
  B11 left Ctrl
  B12 Cmd
  B13 left Alt
  B14 mouse left
  B15 mouse mid
  B16 mouse right
*/
#define _PRINT_KEY KEY_PRNT_SCRN
#define _PAUSE_KEY KEY_PAUSE

#define _LEFT_SHIFT_KEY KEY_LEFT_SHIFT
#define _LEFT_CTRL_KEY  KEY_LEFT_CTRL
#define _CMD_KEY        KEY_RIGHT_GUI
#define _LEFT_ALT       KEY_LEFT_ALT

#define _FN_KEY_UP_ARROW     KEY_PAGE_UP
#define _FN_KEY_DOWN_ARROW   KEY_PAGE_DOWN
#define _FN_KEY_LEFT_ARROW   KEY_HOME
#define _FN_KEY_RIGHT_ARROW  KEY_END


#define DEF_LAYER      0x00
#define FN_LAYER       0x01


/*
   keyboard_maps
   M11 - M18
   M21 - M28
   M31 - M38
   M41 - M48
   M51 - M58
   M61 - M68
   M71 - M78
   M81 - M88
*/
const uint16_t keyboard_maps[][MATRIX_KEYS] = {

  [DEF_LAYER] = {
    _SELECT_KEY, _START_KEY, _VOLUME_M, '`', '[', ']', '-', '=', \
    '1', '2', '3', '4', '5', '6', '7', '8', \
    '9', '0', KEY_ESC, KEY_TAB, EMP, EMP, EMP, EMP, \
    'q', 'w', 'e', 'r', 't', 'y', 'u', 'i', \
    'o', 'p', 'a', 's', 'd', 'f', 'g', 'h', \
    'j', 'k', 'l', 'z', 'x', 'c', 'v', 'b', \
    'n', 'm', ',', '.', '/', '\\', ';', '\'', \
    KEY_BACKSPACE, KEY_RETURN, _FN_KEY, _FN_KEY, ' ', EMP, EMP, EMP
  },

  [FN_LAYER] = {
    _PRINT_KEY, _PAUSE_KEY, _VOLUME_MUTE, '`', '[', ']', KEY_F11, KEY_F12, \
    KEY_F1, KEY_F2, KEY_F3, KEY_F4, KEY_F5, KEY_F6, KEY_F7, KEY_F8, \
    KEY_F9, KEY_F10, _FN_LOCK_KEYBOARD, KEY_CAPS_LOCK, EMP, EMP, EMP, EMP, \
    'q', 'w', 'e', 'r', 't', 'y', KEY_PAGE_UP, KEY_INSERT, \
    'o', 'p', 'a', 's', 'd', 'f', 'g', KEY_HOME, \
    KEY_END, KEY_PAGE_DOWN, 'l', 'z', 'x', 'c', 'v', 'b', \
    'n', 'm', _FN_BRIGHTNESS_DOWN, _FN_BRIGHTNESS_UP, '/', '\\', ';', '\'', \
    KEY_DELETE, KEY_RETURN, _FN_KEY, _FN_KEY, _FN_LIGHT_KEYBOARD, EMP, EMP, EMP
  }

};

uint16_t keyboard_pick_map[MATRIX_KEYS] = {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, 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};

static uint8_t fn_actions[MATRIX_KEYS] = {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, 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};

const uint16_t keys_maps[][KEYS_NUM] = {

  [DEF_LAYER] = {
    KEY_UP_ARROW, KEY_DOWN_ARROW, KEY_LEFT_ARROW, KEY_RIGHT_ARROW, \
    _JOYSTICK_A, _JOYSTICK_B, _JOYSTICK_X, _JOYSTICK_Y, \
    _LEFT_SHIFT_KEY, KEY_RIGHT_SHIFT, _LEFT_CTRL_KEY, KEY_RIGHT_CTRL, \
    _LEFT_ALT,  _MOUSE_LEFT,  KEY_RIGHT_ALT, _MOUSE_RIGHT,  \
    _TRACKBALL_BTN
  },

  [FN_LAYER] = {
    _FN_KEY_UP_ARROW, _FN_KEY_DOWN_ARROW, _FN_KEY_LEFT_ARROW, _FN_KEY_RIGHT_ARROW, \
    _JOYSTICK_A, _JOYSTICK_B, _JOYSTICK_X, _JOYSTICK_Y, \
    _LEFT_SHIFT_KEY, KEY_RIGHT_SHIFT, _LEFT_CTRL_KEY, KEY_RIGHT_CTRL, \
    _CMD_KEY,  _MOUSE_LEFT,  KEY_RIGHT_ALT, _MOUSE_RIGHT,  \
    _TRACKBALL_BTN
  },

};

uint16_t keys_pick_map[KEYS_NUM] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};

const uint16_t backlight_vals[3] = {0, 500, 2000};

uint8_t check_pd2() { // if swtich 2 in back is set to on(HIGH)

  return digitalRead(PD2);
}

void keyboard_release_core(DEVTERM*dv, uint16_t k) {

  switch (k) {
    case  KEY_CAPS_LOCK:
      dv->Keyboard->setAdjustForHostCapsLock(false);
      dv->Keyboard->release(k);
      break;

    case _SELECT_KEY:
      if (check_pd2() == HIGH) {
        k = ' ';
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->button(9, KEY_RELEASED);
      }
      break;
    case _START_KEY:
      if (check_pd2() == HIGH) {
        k = KEY_RETURN;
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->button(10, KEY_RELEASED);
      }
      break;

    case _FN_BRIGHTNESS_UP:
    case _FN_BRIGHTNESS_DOWN:
    case _VOLUME_P:
    case _VOLUME_M:
    case _VOLUME_MUTE:
      dv->Consumer->release();
      break;
    case  _FN_KEY:
      dv->Keyboard_state.fn_on = 0;
      dv->Keyboard_state.fn.begin = 0;
      dv->Keyboard_state.fn.time = 0;
      break;
    default:
      dv->Keyboard->release(k);
      break;
  }
}

void keyboard_release(DEVTERM*dv, uint8_t addr, uint16_t k) {
  if ( keyboard_pick_map[addr] == 0) {
    keyboard_release_core(dv, k);
  } else {
    keyboard_release_core(dv, keyboard_pick_map[addr]);
    keyboard_pick_map[addr] = 0;
  }
}

void keyboard_action(DEVTERM*dv, uint8_t row, uint8_t col, uint8_t mode) {

  uint16_t k;
  uint8_t addr;
  addr = row * MATRIX_COLS + col;
  if (dv->Keyboard_state.fn_on > 0) {
    k = keyboard_maps[dv->Keyboard_state.fn_on][addr];
    fn_actions[addr] = 1;
  } else {
    k = keyboard_maps[dv->Keyboard_state.layer][addr];
  }

  if (k == EMP) {
    return;
  }

  if (k != _FN_KEY && k != _FN_LOCK_KEYBOARD && dv->Keyboard_state.lock == 1) {
    return;
  }
  if (mode == KEY_PRESSED) {
    if ( keyboard_pick_map[addr] == 0) {
      keyboard_pick_map[addr] = k;
    }
  }

  switch (k) {
    case  KEY_CAPS_LOCK:
      if (mode == KEY_PRESSED) {

        dv->Keyboard->press(k);
        dv->Keyboard->setAdjustForHostCapsLock(true);
      } else if (mode == KEY_RELEASED) {
        keyboard_release(dv, addr, k);
      }

      break;

    case _SELECT_KEY:
      if (check_pd2() == HIGH) {
        k = ' ';
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keyboard_release(dv, addr, k);
        }
      } else {
        if (mode == KEY_PRESSED) {
          dv->Joystick->button(9, mode);
        } else {
          keyboard_release(dv, addr, k);
        }
      }
      break;
    case _START_KEY:
      if (check_pd2() == HIGH) {
        k = KEY_RETURN;
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keyboard_release(dv, addr, k);
        }
      } else {
        if (mode == KEY_PRESSED) {
          dv->Joystick->button(10, mode);
        } else {
          keyboard_release(dv, addr, k);
        }
      }
      break;

    case _FN_BRIGHTNESS_UP:
      if (mode == KEY_PRESSED) {
        dv->Consumer->press(HIDConsumer::BRIGHTNESS_UP);
      } else {
        keyboard_release(dv, addr, k);
      }
      break;
    case _FN_BRIGHTNESS_DOWN:
      if (mode == KEY_PRESSED) {
        dv->Consumer->press(HIDConsumer::BRIGHTNESS_DOWN);
      } else {
        keyboard_release(dv, addr, k);
      }
      break;

    case _VOLUME_P: {
        if (mode == KEY_PRESSED) {
          dv->Consumer->press(HIDConsumer::VOLUME_UP);
        } else {
          keyboard_release(dv, addr, k);
        }
      } break;
    case _VOLUME_M: {
        if (mode == KEY_PRESSED) {
          if (dv->Keyboard_state.sf_on > 0) {
            dv->Consumer->press(HIDConsumer::VOLUME_UP);
            uint16_t old_sf = dv->Keyboard_state.sf_on;
            keyboard_release(dv,addr,dv->Keyboard_state.sf_on);
            dv->Keyboard_state.sf_on = old_sf;
            dv->Keyboard->press(dv->Keyboard_state.sf_on);
          } else {
            dv->Consumer->press(HIDConsumer::VOLUME_DOWN);
          }
        } else {
          keyboard_release(dv, addr, k);
        }
      } break;
    case _VOLUME_MUTE: {
        if (mode == KEY_PRESSED) {
          dv->Consumer->press(HIDConsumer::MUTE);
        } else {
          keyboard_release(dv, addr, k);
        }
      } break;
    case _FN_LOCK_KEYBOARD: {
        if (mode == KEY_PRESSED) {
          dv->Keyboard_state.lock = dv->Keyboard_state.lock ^ 1;
        }
      } break;
    case _FN_LIGHT_KEYBOARD: {
        //dv->_Serial->println("light keyboard");

        if (mode == KEY_PRESSED) {
          dv->Keyboard_state.backlight ++;
          if (dv->Keyboard_state.backlight >= 3) {
            dv->Keyboard_state.backlight = 0;
          }
          timer.resume();
          pwmWrite(PA8, backlight_vals[ dv->Keyboard_state.backlight ] );
        }
      } break;
    case  _FN_KEY:
      if (mode == KEY_PRESSED) {
        if (dv->Keyboard_state.fn.lock == 0) {
          dv->Keyboard_state.fn_on = FN_LAYER;
          dv->Keyboard_state.fn.begin = k;
        }
      } else if (mode == KEY_RELEASED ) {

        dv->Keyboard_state.fn_on = 0;
        dv->Keyboard_state.fn.begin = 0;
        dv->Keyboard_state.fn.time = 0;
      }
      break;

    default:
      if (mode == KEY_PRESSED) {
        dv->Keyboard->press(k);
      } else if (mode == KEY_RELEASED) {
        keyboard_release(dv, addr, k);
      }
      break;
  }
}

void keypad_release_core(DEVTERM*dv, uint16_t k) {

  switch (k) {
    case _LEFT_SHIFT_KEY:
    case KEY_RIGHT_SHIFT:
      dv->Keyboard_state.sf_on = 0;
      if (dv->Keyboard_state.shift.lock == 0) {
        dv->Keyboard->release(k);
        dv->Keyboard_state.shift.begin = 0;
        dv->Keyboard_state.shift.time = 0;
      }
      break;

    case _JOYSTICK_UP:
      if (check_pd2() == HIGH) {
        k = KEY_UP_ARROW;
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->Y(511);
      }
      break;
    case _JOYSTICK_DOWN:
      if (check_pd2() == HIGH) {
        k = KEY_DOWN_ARROW;
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->Y(511);
      }
      break;
    case _JOYSTICK_LEFT:
      if (check_pd2() == HIGH) {
        k = KEY_LEFT_ARROW;
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->X(511);
      }
      break;
    case _JOYSTICK_RIGHT:
      if (check_pd2() == HIGH) {
        k = KEY_RIGHT_ARROW;
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->X(511);
      }
      break;
    case _JOYSTICK_A:
      if (check_pd2() == HIGH) {
        k = 'j';
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->button(2, KEY_RELEASED);
      }
      break;
    case _JOYSTICK_B:
      if (check_pd2() == HIGH) {
        k = 'k';
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->button(3, KEY_RELEASED);
      }
      break;
    case _JOYSTICK_X:
      if (check_pd2() == HIGH) {
        k = 'u';
        dv->Keyboard->release(k);
      } else {
        dv->Joystick->button(1, KEY_RELEASED );
      }
      break;
    case _JOYSTICK_Y:
      if (check_pd2() == HIGH) {
        k = 'i';
        dv->Keyboard->release(k);

      } else {
        dv->Joystick->button(4, KEY_RELEASED);
      }
      break;
    case _MOUSE_LEFT:
      dv->Mouse->release(1);
      break;
    case _MOUSE_MID:
      if (dv->state->getScrolled() == false) {
        //if no scrolling happend ,do as a normal mid mouse key click
        dv->Mouse->click(MOUSE_MIDDLE);
      }
      dv->state->releaseMiddleClick();
      break;
    case _MOUSE_RIGHT:
      dv->Mouse->release(2);
      break;
    //_LEFT_CTRL_KEY,_CMD_KEY , _LEFT_ALT
    case _LEFT_CTRL_KEY:
    case KEY_RIGHT_CTRL:
      if (dv->Keyboard_state.ctrl.lock == 0) {
        dv->Keyboard->release(k);
        dv->Keyboard_state.ctrl.begin = 0;
        dv->Keyboard_state.ctrl.time = 0;

      }
      break;

    case _LEFT_ALT:
    case KEY_RIGHT_ALT:
      if (dv->Keyboard_state.alt.lock == 0) {
        dv->Keyboard->release(k);
        dv->Keyboard_state.alt.begin = 0;
        dv->Keyboard_state.alt.time = 0;
      }
      break;
    case _CMD_KEY:
      dv->Keyboard->release(k);
      break;
    case _TRACKBALL_BTN:
      dv->Mouse->release(MOUSE_MIDDLE);
      break;
    default:
      dv->Keyboard->release(k);
      break;
  }
}

void keypad_release(DEVTERM*dv, uint8_t addr, uint16_t k) {
  if ( keys_pick_map[addr] == 0) {
    keypad_release_core(dv, k);
  } else {
    keypad_release_core(dv, keys_pick_map[addr]);
    keys_pick_map[addr] = 0;
  }
}

void keypad_action(DEVTERM*dv, uint8_t col, uint8_t mode) {

  uint16_t k;

  if (dv->Keyboard_state.fn_on > 0) {
    k = keys_maps[dv->Keyboard_state.fn_on][col];
  } else {
    k = keys_maps[dv->Keyboard_state.layer][col];
  }

  if (k == EMP) {
    return;
  }

  if (dv->Keyboard_state.lock == 1) {
    return;
  }

  if (mode == KEY_PRESSED) {
    if ( keys_pick_map[col] == 0) {
      keys_pick_map[col] = k;
    }
  }

  switch (k) {
    case _LEFT_SHIFT_KEY:
    case KEY_RIGHT_SHIFT:
      if (mode == KEY_PRESSED) {
        dv->Keyboard_state.sf_on = k;
        if (dv->Keyboard_state.shift.lock == 0) {
          dv->Keyboard->press(k);
          dv->Keyboard_state.shift.begin = k;
        }
      } else if (mode == KEY_RELEASED) {
        keypad_release(dv, col, k);
      }
      break;

    case _JOYSTICK_UP:
      if (check_pd2() == HIGH) {
        k = KEY_UP_ARROW;
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        } else {
          dv->Joystick->Y(0);
        }
      }
      break;
    case _JOYSTICK_DOWN:
      if (check_pd2() == HIGH) {
        k = KEY_DOWN_ARROW;
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        } else {
          dv->Joystick->Y(1023);
        }
      }
      break;
    case _JOYSTICK_LEFT:
      if (check_pd2() == HIGH) {
        k = KEY_LEFT_ARROW;
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        } else {
          dv->Joystick->X(0);
        }
      }
      break;
    case _JOYSTICK_RIGHT:
      if (check_pd2() == HIGH) {
        k = KEY_RIGHT_ARROW;
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        } else {
          dv->Joystick->X(1023);
        }
      }
      break;
    case _JOYSTICK_A:
      if (check_pd2() == HIGH) {
        k = 'j';
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_PRESSED) {
          dv->Joystick->button(2, mode);
        } else {
          keypad_release(dv, col, k);
        }
      }
      break;
    case _JOYSTICK_B:
      if (check_pd2() == HIGH) {
        k = 'k';
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_PRESSED) {
          dv->Joystick->button(3, mode);
        } else {
          keypad_release(dv, col, k);
        }
      }
      break;
    case _JOYSTICK_X:
      if (check_pd2() == HIGH) {
        k = 'u';
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_PRESSED) {
          dv->Joystick->button(1, mode);
        } else {
          keypad_release(dv, col, k);
        }
      }
      break;
    case _JOYSTICK_Y:
      if (check_pd2() == HIGH) {
        k = 'i';
        if (mode == KEY_PRESSED) {
          dv->Keyboard->press(k);
        } else if (mode == KEY_RELEASED) {
          keypad_release(dv, col, k);
        }
      } else {
        if (mode == KEY_PRESSED) {
          dv->Joystick->button(4, mode);
        } else {
          keypad_release(dv, col, k);
        }
      }
      break;
    case _MOUSE_LEFT:
      if (mode == KEY_PRESSED) {
        dv->Mouse->press(1);
      } else if (mode == KEY_RELEASED) {
        keypad_release(dv, col, k);
      }
      break;
    case _MOUSE_MID:
      if (mode == KEY_PRESSED) {
        dv->state->pressMiddleClick();
      } else {
        keypad_release(dv, col, k);
      }
      break;

    case _MOUSE_RIGHT:
      if (mode == KEY_PRESSED) {
        dv->Mouse->press(2);
      } else if (mode == KEY_RELEASED) {
        keypad_release(dv, col, k);
      }
      break;

    //_LEFT_CTRL_KEY,_CMD_KEY , _LEFT_ALT
    case _LEFT_CTRL_KEY:
    case KEY_RIGHT_CTRL:
      if (mode == KEY_PRESSED) {
        if (dv->Keyboard_state.ctrl.lock == 0) {
          dv->Keyboard->press(k);
          dv->Keyboard_state.ctrl.begin = k;
        }
      } else {
        keypad_release(dv, col, k);
      }
      break;

    case _LEFT_ALT:
    case KEY_RIGHT_ALT:
      if (mode == KEY_PRESSED) {
        if (dv->Keyboard_state.alt.lock == 0) {
          dv->Keyboard->press(k);
          dv->Keyboard_state.alt.begin = k;
        }
      } else {
        keypad_release(dv, col, k);
      }
      break;

    case _CMD_KEY:
      if (mode == KEY_PRESSED) {
        dv->Keyboard->press(k);
      } else {
        keypad_release(dv, col, k);
      }
      break;

    case _TRACKBALL_BTN:
      if (mode == KEY_PRESSED) {
        dv->Mouse->press(MOUSE_MIDDLE);
      } else if (mode == KEY_RELEASED) {
        keypad_release(dv, col, k);
      }
      break;
    default:
      if (mode == KEY_PRESSED) {
        dv->Keyboard->press(k);
      } else if (mode == KEY_RELEASED) {
        keypad_release(dv, col, k);
      }
      break;

  }
}