/* Copyright 2022 Christopher Swenson
 *
 * 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

enum planck_layers {
    _ENIGMA,
    _QWERTY,
    _FN,
    _LOWER,
    _RAISE
};

enum planck_normal_keycodes {
    QWERTY = SAFE_RANGE,
    ENIGMA,
    EN_A,
    EN_B,
    EN_C,
    EN_D,
    EN_E,
    EN_F,
    EN_G,
    EN_H,
    EN_I,
    EN_J,
    EN_K,
    EN_L,
    EN_M,
    EN_N,
    EN_O,
    EN_P,
    EN_Q,
    EN_R,
    EN_S,
    EN_T,
    EN_U,
    EN_V,
    EN_W,
    EN_X,
    EN_Y,
    EN_Z,
    EN_RES,
    EN_TEST,
    EN_DIAG,
    EN_BSPC,
    EN_SREF,
    EN_SROT,
    EN_SPOS,
    EN_SRIN,
    EN_SPLU
};

const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
    /* Enigma
    * ,-----------------------------------------------------------------------------------.
    * | Esc  |   Q  |   W  |   E  |   R  |   T  |   Y  |   U  |   I  |   O  |   P  | Bksp |
    * |------+------+------+------+------+-------------+------+------+------+------+------|
    * | Tab  |   A  |   S  |   D  |   F  |   G  |   H  |   J  |   K  |   L  |   ;  |  "   |
    * |------+------+------+------+------+------|------+------+------+------+------+------|
    * | Shift|   Z  |   X  |   C  |   V  |   B  |   N  |   M  |   ,  |   .  |   /  |Enter |
    * |------+------+------+------+------+------+------+------+------+------+------+------|
    * | Ctrl |  Fn  | Alt  | GUI  |Lower |    Space    |Raise | Left | Down |  Up  |Right |
    * `-----------------------------------------------------------------------------------'
    */
    [_ENIGMA] = LAYOUT_planck_mit(
        QK_GESC, EN_Q,    EN_W,    EN_E,    EN_R,       EN_T,    EN_Y, EN_U,       EN_I,    EN_O,    EN_P,    KC_BSPC,
        KC_TAB,  EN_A,    EN_S,    EN_D,    EN_F,       EN_G,    EN_H, EN_J,       EN_K,    EN_L,    KC_SCLN, KC_QUOT,
        KC_LSFT, EN_Z,    EN_X,    EN_C,    EN_V,       EN_B,    EN_N, EN_M,       KC_COMM, KC_DOT,  KC_SLSH, KC_ENT,
        KC_LCTL, MO(_FN), KC_LGUI, KC_LALT, MO(_LOWER),    KC_SPC,     MO(_RAISE), KC_LEFT, KC_DOWN, KC_UP,   KC_RGHT
    ),
    /* Qwerty
    * ,-----------------------------------------------------------------------------------.
    * | Esc  |   Q  |   W  |   E  |   R  |   T  |   Y  |   U  |   I  |   O  |   P  | Bksp |
    * |------+------+------+------+------+-------------+------+------+------+------+------|
    * | Tab  |   A  |   S  |   D  |   F  |   G  |   H  |   J  |   K  |   L  |   ;  |  "   |
    * |------+------+------+------+------+------|------+------+------+------+------+------|
    * | Shift|   Z  |   X  |   C  |   V  |   B  |   N  |   M  |   ,  |   .  |   /  |Enter |
    * |------+------+------+------+------+------+------+------+------+------+------+------|
    * | Ctrl |  Fn  | Alt  | GUI  |Lower |    Space    |Raise | Left | Down |  Up  |Right |
    * `-----------------------------------------------------------------------------------'
    */
    [_QWERTY] = LAYOUT_planck_mit(
        QK_GESC, KC_Q,    KC_W,    KC_E,    KC_R,      KC_T,    KC_Y,    KC_U,       KC_I,    KC_O,    KC_P,    KC_BSPC,
        KC_TAB,  KC_A,    KC_S,    KC_D,    KC_F,      KC_G,    KC_H,    KC_J,       KC_K,    KC_L,    KC_SCLN, KC_QUOT,
        KC_LSFT, KC_Z,    KC_X,    KC_C,    KC_V,      KC_B,    KC_N,    KC_M,       KC_COMM, KC_DOT,  KC_SLSH, KC_ENT,
        KC_LCTL, MO(_FN), KC_LGUI, KC_LALT, MO(_LOWER),    KC_SPC,       MO(_RAISE), KC_LEFT, KC_DOWN, KC_UP,   KC_RGHT
    ),
    /* Function
    * ,-----------------------------------------------------------------------------------.
    * | Boot | Reset| Diag |      |      |      |      |      |      |      | Test |Revers|
    * |------+------+------+------+------+-------------+------+------+------+------+------|
    * |      |      |      |      |      |      |      |      |      |      |      |      |
    * |------+------+------+------+------+------|------+------+------+------+------+------|
    * |      |Reflec|Rotors|Posn.s|Rings |Plugs |      |      |Enigma|Qwerty|      |      |
    * |------+------+------+------+------+------+------+------+------+------+------+------|
    * |      |      |      |      |      |             |      |      |      |      |      |
    * `-----------------------------------------------------------------------------------'
    */
    [_FN] = LAYOUT_planck_mit(
        QK_BOOT, EN_RES,  EN_DIAG, _______, _______, _______, _______, _______, _______, _______, EN_TEST, EN_BSPC,
        _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______,
        _______, EN_SREF, EN_SROT, EN_SPOS, EN_SRIN, EN_SPLU, _______, _______, QWERTY,  ENIGMA,  _______, _______,
        _______, _______, _______, _______, _______,      _______,     _______, _______, _______, _______, _______
    ),
    /* Lower
    * ,-----------------------------------------------------------------------------------.
    * | ~    |   !  |   @  |   #  |   $  |   %  |   ^  |   &  |   *  |   (  |   )  | Bksp |
    * |------+------+------+------+------+-------------+------+------+------+------+------|
    * |      |      |      |      |      |      |      |   _  |   +  |   {  |   }  |  |   |
    * |------+------+------+------+------+------|------+------+------+------+------+------|
    * | Shift|      |      |      |      |      |      |      |      |      |      |      |
    * |------+------+------+------+------+------+------+------+------+------+------+------|
    * | Ctrl |  Fn  | Alt  | GUI  |Lower |    Space    |Raise | Left | Down |  Up  |Right |
    * `-----------------------------------------------------------------------------------'
    */
    [_LOWER] = LAYOUT_planck_mit(
        KC_TILD, KC_EXLM, KC_AT,   KC_HASH, KC_DLR,  KC_PERC, KC_CIRC, KC_AMPR,  KC_ASTR, KC_LPRN, KC_RPRN, KC_BSPC, 
        _______, _______, _______, _______, _______, _______, _______, KC_UNDS,  KC_PLUS, KC_LCBR, KC_RCBR, KC_PIPE, 
        KC_LSFT, _______, _______, _______, _______, _______, _______,  _______, _______, _______, _______, _______, 
        KC_LCTL, _______, KC_LGUI, KC_LALT, _______,       KC_SPC,      _______, KC_LEFT, KC_DOWN, KC_UP,   KC_RGHT
    ),
    /* RAISE
    * ,-----------------------------------------------------------------------------------.
    * | `    |   1  |   2  |   3  |   4  |   5  |   6  |   7  |   8  |   9  |   0  | Bksp |
    * |------+------+------+------+------+-------------+------+------+------+------+------|
    * |      |      |      |      |      |      |      |   -  |   =  |   [  |   ]  |  \   |
    * |------+------+------+------+------+------|------+------+------+------+------+------|
    * | Shift|      |      |      |      |      |      |      |      |      |      |      |
    * |------+------+------+------+------+------+------+------+------+------+------+------|
    * | Ctrl |  Fn  | Alt  | GUI  |Lower |    Space    |Raise | Left | Down |  Up  |Right |
    * `-----------------------------------------------------------------------------------'
    */
    [_RAISE] = LAYOUT_planck_mit(
        KC_GRV,  KC_1,    KC_2,    KC_3,    KC_4,    KC_5,    KC_6,    KC_7,    KC_8,    KC_9,    KC_0,    KC_BSPC, 
        _______, _______, _______, _______, _______, _______, _______, KC_MINS, KC_EQL,  KC_LBRC, KC_RBRC, KC_BSLS, 
        KC_LSFT, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, 
        KC_LCTL, _______, KC_LGUI, KC_LALT, _______,      KC_SPC,      _______, KC_LEFT, KC_DOWN, KC_UP,   KC_RGHT
    ),
};

char rotor_definitions[5][26] = {
    "EKMFLGDQVZNTOWYHXUSPAIBRCJ",
    "AJDKSIRUXBLHWTMCQGZNPYFVOE",
    "BDFHJLCPRTXVZNYEIWGAKMUSQO",
    "ESOVPZJAYQUIRHXLNFTGKDCMWB",
    "VZBRGITYUPSDNHLXAWMJQOFECK" 
};

char reflector_definitions[3][26] = {
    "EJMZALYXVBWFCRQUONTSPIKHGD",
    "YRUHQSLDPXNGOKMIEBFZCWVJAT",
    "FVPJIAOYEDRZXWGCTKUQSBNMHL"
};

char notch[5] = "QEVJZ";

typedef struct Settings {
    char rotor_order[3];
    char rotor_rings[3];
    char rotor_positions[3];
    char plugs[25];
    int plug_count;
    char reflector;
} Settings;

typedef struct KeyboardState {
    bool is_setting_reflector;
    bool is_setting_rotors;
    bool is_setting_rotor_positions;
    bool is_setting_rotor_rings;
    bool is_setting_plugs;
    char setting_progress[26];
    int setting_index;
    Settings current_settings;
    Settings default_settings;
} KeyboardState;

int bound(int letter) {
    return ((letter % 26) + 26) % 26;
}

char to_char(int letter) {
    return 'A' + letter;
}

int to_int(char letter) {
    return letter - 'A';
}

char encipher(char letter, Settings *settings) {
    int rotor_2_step = settings->rotor_positions[2] == notch[settings->rotor_order[2] - 1];
    int rotor_1_step = settings->rotor_positions[1] == notch[settings->rotor_order[1] - 1];
      
    // Advance the third rotor
    settings->rotor_positions[2]++;
    if (settings->rotor_positions[2] > 'Z') {
      settings->rotor_positions[2] -= 26;
    }

    // Maybe advance the second rotor, including double steps
    if (rotor_2_step || rotor_1_step) {
        settings->rotor_positions[1]++;
        if (settings->rotor_positions[1] > 'Z') {
            settings->rotor_positions[1] -= 26;
        }
    }

    // Maybe advance the first rotor
    if (rotor_1_step) {
        settings->rotor_positions[0]++;
        if (settings->rotor_positions[0] > 'Z') {
            settings->rotor_positions[0] -= 26;
        }
    }

    // Swap letters on plugboard
    for (int i = 0; i < settings->plug_count * 2; i += 2) {
        if (letter == settings->plugs[i]) {
            letter = settings->plugs[i + 1];
        } else if (letter == settings->plugs[i + 1]) {
            letter = settings->plugs[i];
        }  
    }

    // Rotors (right to left)
    for (int rotor_index = 2; rotor_index >= 0; rotor_index--) {
        char *rotor_definition = rotor_definitions[settings->rotor_order[rotor_index] - 1];
        int position = to_int(settings->rotor_positions[rotor_index]);
        int ring = to_int(settings->rotor_rings[rotor_index]);
        int char_index = to_int(letter);
        letter = to_char(bound(
            to_int(rotor_definition[bound(char_index + position - ring)]) + ring - position
        ));
    }

    // Swap via reflector
    letter = reflector_definitions[to_int(settings->reflector)][to_int(letter)];

    // Rotors in reverse (left to right)
    for (int rotor_index = 0; rotor_index < 3; rotor_index++) {
        char *rotor_definition = rotor_definitions[settings->rotor_order[rotor_index] - 1];
        int position = to_int(settings->rotor_positions[rotor_index]);
        int ring = to_int(settings->rotor_rings[rotor_index]);
        int search_index;
        for (search_index = 0; search_index < 26; search_index++) {
            if (
                rotor_definition[search_index] 
                == to_char(bound(to_int(letter) - ring + position))
            ) {
                break;
            }
        }
        letter = to_char(bound(search_index - position + ring));
    }
    
    // Plugboard again
    for (int i = 0; i < settings->plug_count * 2; i += 2) {
        if (letter == settings->plugs[i]) {
            letter = settings->plugs[i + 1];
        } else if (letter == settings->plugs[i + 1]) {
            letter = settings->plugs[i];
        }
    }

    return letter;
}

void init_enigma_default(Settings *settings) {
    settings->rotor_order[0] = 1;
    settings->rotor_rings[0] = 'A';
    settings->rotor_positions[0] = 'A';

    settings->rotor_order[1] = 2;
    settings->rotor_rings[1] = 'A';
    settings->rotor_positions[1] = 'A';

    settings->rotor_order[2] =  3;
    settings->rotor_rings[2] = 'A';
    settings->rotor_positions[2] = 'A';

    strcpy(settings->plugs, "");
    settings->plug_count = 0;

    settings->reflector = 'B';
}

void copy_settings(Settings *from, Settings *to) {
    to->rotor_order[0] = from->rotor_order[0];
    to->rotor_rings[0] = from->rotor_rings[0];
    to->rotor_positions[0] = from->rotor_positions[0];

    to->rotor_order[1] = from->rotor_order[1];
    to->rotor_rings[1] = from->rotor_rings[1];
    to->rotor_positions[1] = from->rotor_positions[1];

    to->rotor_order[2] = from->rotor_order[2];
    to->rotor_rings[2] = from->rotor_rings[2];
    to->rotor_positions[2] = from->rotor_positions[2];

    strncpy(to->plugs, from->plugs, from->plug_count * 2);
    to->plug_count = from->plug_count;

    to->reflector = from->reflector;
}

char *rotor_name(int rotor_number) {
    if (rotor_number == 1) {
        return "I";
    } else if (rotor_number == 2) {
        return "II";
    } else if (rotor_number == 3) {
        return "III";
    } else if (rotor_number == 4) {
        return "IV";
    } else if (rotor_number == 5) {
        return "V";
    }
    return "?";
}

void rotors_reverse(Settings *settings) {
    int rotor_2_step = settings->rotor_positions[2] 
        == to_char(bound(to_int(notch[settings->rotor_order[2] - 1]) + 1));
    int rotor_1_step = settings->rotor_positions[1] 
        == to_char(bound(to_int(notch[settings->rotor_order[1] - 1]) + 1));
      
    // Reverse third rotor
    settings->rotor_positions[2]--;
    if (settings->rotor_positions[2] < 'A') {
        settings->rotor_positions[2] += 26;
    }

    // Maybe reverse second rotor (including double steps)
    if (rotor_2_step || rotor_1_step) {
        settings->rotor_positions[1]--;
        if (settings->rotor_positions[1] < 'A') {
            settings->rotor_positions[1] += 26;
        }
    }

    // Maybe recerse first rotor
    if (rotor_1_step) {
        settings->rotor_positions[0]--;
        if (settings->rotor_positions[0] < 'A') {
            settings->rotor_positions[0] += 26;
        }
    }
}

void reset_settings(KeyboardState *state) {
  copy_settings(&state->default_settings, &state->current_settings);
}

void set_layer(uint8_t default_layer) {
    default_layer_set((layer_state_t)1 << default_layer);
}

void set_backlight(uint8_t mode, uint8_t hue, uint8_t sat, uint8_t val) {
    #ifdef RGBLIGHT_ENABLE
        rgblight_enable_noeeprom(); 
        rgblight_mode_noeeprom(mode);
        rgblight_sethsv_noeeprom(hue, sat, val);
    #endif
}

void clear_working_settings(KeyboardState *state) {
    state->is_setting_reflector = false;
    state->is_setting_rotors = false;
    state->is_setting_rotor_positions = false;
    state->is_setting_rotor_rings = false;
    state->is_setting_plugs = false;
    state->setting_index = 0;
    set_layer(_ENIGMA);
    set_backlight(RGBLIGHT_MODE_STATIC_LIGHT, HSV_RED);
}

void send_settings_string(Settings *settings) {
    send_char(settings->reflector);
    send_string(". ");
    send_string(rotor_name(settings->rotor_order[0]));
    send_char(settings->rotor_rings[0]);
    send_string("-");
    send_string(rotor_name(settings->rotor_order[1]));
    send_char(settings->rotor_rings[1]);
    send_string("-");
    send_string(rotor_name(settings->rotor_order[2]));
    send_char(settings->rotor_rings[2]);
    send_string(" (");
    send_char(settings->rotor_positions[0]);
    send_string(", ");
    send_char(settings->rotor_positions[1]);
    send_string(", ");
    send_char(settings->rotor_positions[2]);
    send_string(") ");
    for (int i = 0; i < settings->plug_count; i++) {
        send_char(settings->plugs[i * 2]);
        send_char(settings->plugs[i * 2 + 1]);
        send_string(" ");
    }
    send_string("\n");
}

void perform_test(Settings *settings) {
    for (int i = 0; i < 1000; i++) {
        send_char(encipher('A', settings) - ('A' - 'a'));
    }
}

void commit_plug_settings(KeyboardState *state) {
    state->default_settings.plug_count = state->setting_index / 2; 
    for (int i = 0; i < state->default_settings.plug_count; i++) {
        state->default_settings.plugs[i * 2] = state->setting_progress[i * 2];
        state->default_settings.plugs[i * 2 + 1] = state->setting_progress[i * 2 + 1];
    }
    reset_settings(state);
    clear_working_settings(state);
}

void handle_set_reflector(char letter, KeyboardState *state) {
    if (letter >= 'A' && letter <= 'C') {
        state->default_settings.reflector = letter;
        reset_settings(state);
        clear_working_settings(state);
    }
}

bool handle_set_rotor(char letter, KeyboardState *state) {
    bool is_valid = letter >= 'A' && letter <= 'E' && state->setting_index < 3;
    if (is_valid) {
        state->setting_progress[state->setting_index] = to_int(letter) + 1;
        state->setting_index += 1;
        if (state->setting_index == 3) {
            state->default_settings.rotor_order[0] = state->setting_progress[0];
            state->default_settings.rotor_order[1] = state->setting_progress[1];
            state->default_settings.rotor_order[2] = state->setting_progress[2];
            reset_settings(state);
            clear_working_settings(state);
        }
    }
    return is_valid;
}

bool handle_set_rotor_position(char letter, KeyboardState *state) {
    bool is_valid = state->setting_index < 3; // Guaranteed to be A-Z already
    if (is_valid) {
        state->setting_progress[state->setting_index] = letter;
        state->setting_index += 1;
        if (state->setting_index == 3) {
            state->default_settings.rotor_positions[0] = state->setting_progress[0];
            state->default_settings.rotor_positions[1] = state->setting_progress[1];
            state->default_settings.rotor_positions[2] = state->setting_progress[2]; 
            reset_settings(state);
            clear_working_settings(state);
        }
    }
    return is_valid;
}

bool handle_set_rotor_ring(char letter, KeyboardState *state) {
    bool is_valid = state->setting_index < 3;
    if (is_valid) {
        state->setting_progress[state->setting_index] = letter;
        state->setting_index += 1;
        if (state->setting_index == 3) {
            state->default_settings.rotor_rings[0] = state->setting_progress[0];
            state->default_settings.rotor_rings[1] = state->setting_progress[1];
            state->default_settings.rotor_rings[2] = state->setting_progress[2];
            reset_settings(state);
            clear_working_settings(state);
        }
    }
    return is_valid;
}

bool handle_set_plug(char letter, KeyboardState *state) {
    bool is_valid = state->setting_index < 26;
    if (is_valid) {
        state->setting_progress[state->setting_index] = letter;
        state->setting_index += 1;
    }
    return is_valid;
}

void handle_enigma_keypress(char letter, bool any_mods, KeyboardState *state) {
    bool settings_valid = true;
    if (letter) {
        if (any_mods) {
            tap_code(KC_A + to_int(letter));
        } else if (state->is_setting_reflector) {
            handle_set_reflector(letter, state);
        } else if (state->is_setting_rotors) {
            settings_valid = handle_set_rotor(letter, state);
        } else if (state->is_setting_rotor_positions) {
            settings_valid = handle_set_rotor_position(letter, state);
        } else if (state->is_setting_rotor_rings) {
            settings_valid = handle_set_rotor_ring(letter, state);
        } else if (state->is_setting_plugs) {
            settings_valid = handle_set_plug(letter, state);
        } else {
            char c2 = encipher(letter, &state->current_settings);
            send_char(c2 - ('A' - 'a'));
        }
    }
    if (!settings_valid) {
        clear_working_settings(state);
    }
}

KeyboardState STATE;

void keyboard_pre_init_user(void) {
    init_enigma_default(&STATE.default_settings);
    init_enigma_default(&STATE.current_settings);
    clear_working_settings(&STATE);
}

bool process_record_user(uint16_t keycode, keyrecord_t *record) {
    uint8_t letter_index;
    bool letter_found = false;
    if (record->event.pressed) {
        switch (keycode) {
          case QWERTY:
              set_layer(_QWERTY);
              set_backlight(RGBLIGHT_MODE_RAINBOW_SWIRL + 4, HSV_PURPLE);
              break;
          case ENIGMA:
              set_layer(_ENIGMA);
              set_backlight(RGBLIGHT_MODE_STATIC_LIGHT, HSV_RED);
              break;
          case EN_SREF:
              reset_settings(&STATE);
              STATE.is_setting_reflector = true;
              set_layer(_ENIGMA);
              set_backlight(RGBLIGHT_MODE_SNAKE, HSV_RED);
              break;
          case EN_SROT:
              reset_settings(&STATE);
              STATE.is_setting_rotors = true;
              STATE.setting_index = 0;
              set_layer(_ENIGMA);
              set_backlight(RGBLIGHT_MODE_SNAKE, 10, 255, 255);
              break;
          case EN_SPOS:
              reset_settings(&STATE);
              STATE.is_setting_rotor_positions = true;
              STATE.setting_index = 0;
              set_layer(_ENIGMA);
              set_backlight(RGBLIGHT_MODE_SNAKE, HSV_ORANGE);
              break;
          case EN_SRIN:
              reset_settings(&STATE);
              STATE.is_setting_rotor_rings = true;
              STATE.setting_index = 0;
              set_layer(_ENIGMA);
              set_backlight(RGBLIGHT_MODE_SNAKE, HSV_GREEN);
              break;
          case EN_SPLU:
              reset_settings(&STATE);
              STATE.is_setting_plugs = true;
              STATE.setting_index = 0;
              set_layer(_ENIGMA);
              set_backlight(RGBLIGHT_MODE_SNAKE, HSV_BLUE);
              break;
          case QK_GESC:
              if (
                  STATE.is_setting_reflector
                  || STATE.is_setting_rotors
                  || STATE.is_setting_rotor_positions
                  || STATE.is_setting_rotor_rings
                  || STATE.is_setting_plugs
              ) {
                  clear_working_settings(&STATE);
                  return false;
              }
              break;
          case KC_ENT:
              if (STATE.is_setting_plugs) { 
                  commit_plug_settings(&STATE);
                  return false;
              }
              break;
          case EN_A ... EN_Z:
              letter_index = keycode - EN_A;
              letter_found = true;
              break;
          case EN_RES:
              reset_settings(&STATE);
              break;
          case EN_TEST:
              perform_test(&STATE.current_settings);
              break;
          case EN_DIAG:
              send_settings_string(&STATE.current_settings);
              break;
          case EN_BSPC:
              rotors_reverse(&STATE.current_settings);
              tap_code(KC_BSPC);
              break;
      }
    }
    char letter = letter_found ? 'A' + letter_index : 0;
    uint8_t mods = get_mods();
    bool any_mods = (mods & MOD_MASK_CTRL) || (mods & MOD_MASK_ALT) || (mods & MOD_MASK_GUI);
    handle_enigma_keypress(letter, any_mods, &STATE);
    return true;
}

void keyboard_post_init_user(void) {
    set_layer(_QWERTY);
    set_backlight(RGBLIGHT_MODE_RAINBOW_SWIRL + 4, HSV_PURPLE);
}