/* Copyright 2020 funatake
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include QMK_KEYBOARD_H
#include "bmp.h"
#include "bmp_custom_keycode.h"
#include "keycode_str_converter.h"

#define JP_ZHTG KC_GRV  // hankaku/zenkaku|kanzi
#define JP_YEN  KC_INT3 // yen, |
#define JP_CIRC KC_EQL  // ^, ~
#define JP_AT   KC_LBRC // @, `
#define JP_LBRC KC_RBRC // [, {
#define JP_COLN KC_QUOT // :, *
#define JP_RBRC KC_NUHS // ], }
#define JP_BSLS KC_INT1 // \, _

extern keymap_config_t keymap_config;

#ifdef RGBLIGHT_ENABLE
//Following line allows macro to read current RGB settings
extern rgblight_config_t rgblight_config;
rgblight_config_t RGB_current_config;
#endif

// Layer shorthand
enum layuer_shorthand {
  _JPB = 0,
  _MJB,
  __CS,
  __FN,
  _FN2,
  _FN3,
};

// #define CK_WMH LT(__CS, KC_MHEN)
// #define CK_WHN LT(__FN, KC_HENK)
// #define CK_MMH LT(__CS, KC_LANG2)
// #define CK_MHN LT(__FN, KC_LANG1)

#define KEYS_MAX 256
#define PHYS 0
#define LOGI 1
#define TRAN 2
bool keys[KEYS_MAX][3] = {};

#define MODE_NORMAL 0
#define MODE_MAC 1
uint8_t mode = MODE_NORMAL;
uint8_t layer = _JPB;
uint16_t last_press_keycode = KC_NO;

// Defines the keycodes used by our macros in process_record_user
enum custom_keycodes {
  CUSTOM_KEYCODE_START = BMP_SAFE_RANGE,
  CK_WMH,
  CK_WHN,
  CK_MMH,
  CK_MHN,
  CUSTOM_KEYCODE_END
};

const key_string_map_t custom_keys_user =
{
    .start_kc = CUSTOM_KEYCODE_START,
    .end_kc= CUSTOM_KEYCODE_END,
    .key_strings = "CUSTOM_KEYCODE_START\0CK_WMH\0CK_WHN\0CK_MMH\0CK_MHN\0CUSTOM_KEYCODE_END\0"
};

const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
  /*
  レイアウトの枠だけ作成する。
  実際のキーマップは、keymap.jsnで定義する。
  */
  [_JPB] = LAYOUT( /* QWERTY */
    _______, KC_ESC,   KC_1,    KC_2,    KC_3,    KC_4,    KC_5,    KC_6,    KC_7,    KC_8,    KC_9,    KC_0,    KC_MINS, KC_EQL,  KC_BSLS, KC_BSPC,
    _______, KC_PGUP,  KC_TAB,  KC_Q,    KC_W,    KC_E,    KC_R,    KC_T,    KC_Y,    KC_U,    KC_I,    KC_O,    KC_P,    KC_LBRC, KC_RBRC, KC_DEL,
    _______, KC_PGDN,  KC_LCTL, KC_A,    KC_S,    KC_D,    KC_F,    KC_G,    KC_H,    KC_J,    KC_K,    KC_L,    KC_SCLN, KC_QUOT, KC_GRV,  KC_ENT,
    _______, KC_HOME,  KC_LSFT, KC_Z,    KC_X,    KC_C,    KC_V,    KC_B,    KC_N,    KC_M,    KC_COMM, KC_DOT,  KC_SLSH, KC_RSFT, KC_RSFT, KC_RCTL,
    _______, KC_END,   KC_LCTL, KC_LGUI, KC_LALT, KC_LALT, CK_WMH,  KC_SPC,  KC_SPC,  CK_WHN,  KC_RALT, KC_APP,  KC_LEFT, KC_DOWN, KC_UP,   KC_RIGHT
    ),
  
  [_MJB] = LAYOUT( /* QWERTY */
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______
    ),
  
  [__CS] = LAYOUT( /* FUNCTION */
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______
    ),
  
  [__FN] = LAYOUT( /* FUNCTION */
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______
    ),
  
  [_FN2] = LAYOUT( /* FUNCTION */
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______
    ),
  
  [_FN3] = LAYOUT( /* FUNCTION */
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______,
    _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,  _______,  _______
    ),
};

void clear(uint8_t kind) {
  for (int i = 0; i < KEYS_MAX; i++ ) {
    keys[i][kind] = false;
  }
}

bool isPressed(uint8_t kind, uint16_t keycode) {
  return keys[keycode][kind];
}

void pressKey(uint8_t kind, uint16_t keycode) {
  keys[keycode][kind] = true;
}

void releaseKey(uint8_t kind, uint16_t keycode) {
  keys[keycode][kind] = false;
}

// 最終的に必要なキーのみを設定。足りないものを押して、不要なものをリリースする。
void fixKey(uint8_t kind) {
  // 離すときは文字キーから離す
  for (int i = 0; i < KEYS_MAX; i++) {
    if (keys[i][kind] != keys[i][LOGI] && !keys[i][kind]) {
      unregister_code(i);
      keys[i][LOGI] = keys[i][kind];
    }
  }
  
  // 押すときは修飾キーから押す
  for (int i = KEYS_MAX - 1; i >= 0 ; i--) {
    if (keys[i][kind] != keys[i][LOGI] && keys[i][kind]) {
      register_code(i);
      keys[i][LOGI] = keys[i][kind];
    }
  }
}

bool processKey(uint32_t keycode, bool pressed) {
  clear(TRAN);
  bool trans = false;
  
  if (mode == MODE_MAC) {
    if (layer == __CS || layer == __FN || layer == _FN2 || layer == _FN3) {
      if (isPressed(PHYS, KC_UP)) {
        pressKey(TRAN, KC_UP);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      } 
      
      if (isPressed(PHYS, KC_LEFT)) {
        pressKey(TRAN, KC_LEFT);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      }
      
      if (isPressed(PHYS, KC_DOWN)) {
        pressKey(TRAN, KC_DOWN);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      } 
      
      if (isPressed(PHYS, KC_RIGHT)) {
        pressKey(TRAN, KC_RIGHT);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      } 
      
      if (isPressed(PHYS, KC_HOME)) {
        pressKey(TRAN, KC_LGUI);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_UP);
        } else {
          pressKey(TRAN, KC_LEFT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      }
      
      if (isPressed(PHYS, KC_END)) {
        pressKey(TRAN, KC_LGUI);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_DOWN);
        } else {
          pressKey(TRAN, KC_RIGHT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      } 
      
      if (isPressed(PHYS, KC_BSPC)) {
        pressKey(TRAN, KC_BSPC);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      } 
      
      if (isPressed(PHYS, KC_DEL)) {
        pressKey(TRAN, KC_DEL);
        if (isPressed(PHYS, KC_LGUI)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LSFT)) {
          pressKey(TRAN, KC_LSFT);
        }
        if (isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_RSFT);
        }
        if (isPressed(PHYS, KC_LALT)) {
          pressKey(TRAN, KC_LALT);
        }
        if (isPressed(PHYS, KC_LCTL)) {
          pressKey(TRAN, KC_LCTL);
        }
        trans = true;
      }
      
      if (isPressed(PHYS, KC_P)) {
        pressKey(TRAN, KC_PGUP);
        if (isPressed(PHYS, KC_LSFT) || isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_LCTL);
        } else {
          pressKey(TRAN, KC_LALT);
        }
        trans = true;
      }
      
      if (isPressed(PHYS, KC_LBRC)) {
        pressKey(TRAN, KC_PGDN);
        if (isPressed(PHYS, KC_LSFT) || isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_LCTL);
        } else {
          pressKey(TRAN, KC_LALT);
        }
        trans = true;
      }
    }
  } else {
    if (layer == __CS || layer == __FN || layer == _FN2 || layer == _FN3) {
      if (isPressed(PHYS, KC_P)) {
        pressKey(TRAN, KC_PGUP);
        if (isPressed(PHYS, KC_LSFT) || isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_LCTL);
        } else {
          pressKey(TRAN, KC_LALT);
        }
        trans = true;
      }
      
      if (isPressed(PHYS, KC_LBRC)) {
        pressKey(TRAN, KC_PGDN);
        if (isPressed(PHYS, KC_LSFT) || isPressed(PHYS, KC_RSFT)) {
          pressKey(TRAN, KC_LCTL);
        } else {
          pressKey(TRAN, KC_LALT);
        }
        trans = true;
      }
    }
  }
  
  if (trans) {
    fixKey(TRAN);
  } else {
    fixKey(PHYS);
  }
  
  return false;
}

void processHenkanMuhenkan(uint16_t keycode, bool pressed) {
  if (pressed) {
    switch (keycode) {
    case CK_WMH:
    case CK_MMH:
      layer_on(__CS);
      break;
    case CK_WHN:
    case CK_MHN:
      layer_on(__FN);
      break;
    }
    last_press_keycode = keycode;
  } else {
    switch (keycode) {
    case CK_WMH:
    case CK_MMH:
      layer_off(__CS);
      break;
    case CK_WHN:
    case CK_MHN:
      layer_off(__FN);
      break;
    }
    
    if (last_press_keycode == keycode) {
      switch (keycode) {
      case CK_WMH:
        register_code(KC_MHEN);
        unregister_code(KC_MHEN);
        break;
      case CK_MMH:
        register_code(KC_LANG2);
        unregister_code(KC_LANG2);
        break;
      case CK_WHN:
        register_code(KC_HENK);
        unregister_code(KC_HENK);
        break;
      case CK_MHN:
        register_code(KC_LANG1);
        unregister_code(KC_LANG1);
        break;
      }
    }
  }
}

bool process_record_user(uint16_t keycode, keyrecord_t *record) {
  bool pressed = record->event.pressed;
  
  processHenkanMuhenkan(keycode, pressed);
  
  if (KEYS_MAX <= keycode) {
    return true;
  }
  
  if (pressed) {
    keys[keycode][PHYS] = true;
  } else {
    keys[keycode][PHYS] = false;
  }
  
  return processKey(keycode, pressed);
}

void matrix_init_user(void) {
  // rgblight_enable();
  // rgblight_mode(1);
  // rgblight_setrgb(0, 0, 0);
}

void matrix_scan_user(void) {
}

void led_set_user(uint8_t usb_led) {}

uint32_t layer_state_set_user(uint32_t state) {
  layer = biton32(state);
  
  // if (layer == _MNT) {
  //   rgblight_setrgb(30, 0, 0);
  // } else {
  //   rgblight_setrgb(0, 0, 0);
  // }
  
  if (_MJB <= layer && layer <= _MJB) {
    mode = MODE_MAC;
  } else if (layer == __CS || layer == __FN || layer == _FN2 || layer == _FN3) {
    // do nothing
  } else {
    mode = MODE_NORMAL;
  }
  return state;
}