#include QMK_KEYBOARD_H
#include <stdio.h>

enum layer { LAYER_0, LAYER_1, LAYER_2, LAYER_3, LAYER_4 };

const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {

    [LAYER_0] = 
        LAYOUT(KC_ESC, KC_1, KC_2, KC_3, KC_4, KC_5, KC_6, KC_7, KC_8,
            KC_9, KC_0, KC_PSCR, KC_TAB, KC_Q, KC_W, KC_E, KC_F,
            KC_P, KC_J, KC_L, KC_U, KC_Y, KC_SCLN, KC_DEL, KC_CAPS,
            KC_LALT, KC_A, KC_R, KC_S, KC_T, KC_G, KC_M, KC_N, KC_E,
            KC_I, KC_O, TG(2), KC_RBRC, KC_LCTL, KC_Z, KC_X, KC_C,
            KC_V, KC_B, KC_LCBR, KC_NO, KC_NO, KC_RCBR, KC_N, KC_M,
            KC_COMM, KC_DOT, KC_SLSH, TG(1), KC_LGUI, MO(4), KC_LSFT,
            KC_ENT, KC_HOME, KC_END, KC_SPC, KC_BSPC, MO(3), KC_NO),

    [LAYER_1] =
        LAYOUT(KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_I, KC_TAB, KC_Q,
            KC_W, KC_E, KC_R, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_T, KC_M, KC_LSFT, KC_A, KC_S, KC_D, KC_F, KC_TRNS,
            KC_TRNS, KC_TRNS, KC_UP, KC_TRNS, KC_TRNS, KC_TRNS, KC_B, KC_U,
            KC_Z, KC_X, KC_C, KC_V, KC_G, KC_J, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_LEFT, KC_DOWN, KC_RGHT, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_LCTL, KC_SPC, KC_Y, KC_TRNS, KC_ENT, KC_TRNS, KC_NO, KC_TRNS),
    [LAYER_2] =
        LAYOUT(KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_F1,
            KC_F2, KC_F3, KC_F4, KC_F5, KC_F6, KC_TRNS, KC_Q, KC_W, KC_E,
            KC_R, KC_T, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_TRNS, KC_A, KC_S, KC_D, KC_F, KC_G, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_Z, KC_X,
            KC_C, KC_V, KC_B, KC_P, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_LSFT, KC_SPC, KC_Y, KC_TRNS, KC_ENT, KC_TRNS, KC_NO, KC_TRNS),
    [LAYER_3] = 
        LAYOUT(KC_F1, KC_F2, KC_F3, KC_F4, KC_F5, KC_F6, KC_F7, KC_F8, KC_F9, KC_F10,
            KC_F11, KC_F12, KC_TRNS, KC_TILD, KC_NO, KC_NO, KC_QUOT, KC_EQL, KC_NO,
            KC_NO, RALT(KC_Y), KC_NO, KC_NO, KC_NO, KC_GRV, KC_TRNS, RALT(KC_Q),
            KC_BSLS, RALT(KC_S), KC_EXLM, KC_LPRN, KC_RPRN, KC_QUES, KC_NO, KC_UP,
            RALT(KC_P), KC_NO, KC_NO, KC_TRNS, RALT(KC_5), KC_PSLS, KC_PAST,
            KC_PMNS, KC_PPLS, KC_LBRC, KC_TRNS, KC_TRNS, KC_RBRC, KC_NO, KC_UNDS,
            KC_LEFT, KC_DOWN, KC_RGHT, KC_NO, KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS,
            KC_TRNS, KC_TRNS, KC_TRNS, KC_TRNS, MO(3), KC_TRNS),
    [LAYER_4] = 
        LAYOUT(KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO,
            KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO,
            KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO,
            KC_NO, KC_NO, KC_MPRV, KC_MNXT, KC_MSTP, KC_MPLY, KC_NO, KC_NO, KC_NO,
            KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO,
            KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO, KC_NO,
            KC_NO, KC_NO, KC_NO, KC_NO, KC_NO),

};

const char *read_keylog(void);
const char *read_keylogs(void);
void        set_keylog(uint16_t keycode, keyrecord_t *record);

static void render_status(void) {
    // // WPM
    // oled_write_P(PSTR("      "), false);
    // sprintf(wpm_str, "%03d", get_current_wpm());
    // oled_write(wpm_str, false);
    // oled_write_P(PSTR("   WPM"), false);

    switch (get_highest_layer(layer_state)) {
        case 0:
            oled_write_ln("colemak", false);
            break;
        case 1:
            oled_write_ln("epic gaming mode", false);
            break;
        case 2:
            oled_write_ln("epic gaming mode 2", false);
            break;
        case 3:
            oled_write_ln("symbols", false);
            break;
        case 4:
            oled_write_ln("actions", false);
            break;
        default:
            oled_write_ln("colemak", false);
            break;
    }

    oled_write_ln(read_keylog(), false);
    oled_write_ln(read_keylogs(), false);
}

bool process_record_user(uint16_t keycode, keyrecord_t *record) {
    if (record->event.pressed) {
        set_keylog(keycode, record);
        // set_timelog();
    }
    return true;
}

//
// OLED display rendering
//
bool oled_task_user(void) {
    if (is_keyboard_master()) {
        // Left side
        render_status();
    } else {
        // Right side
        // render_anim();
    }

    return 0;
}