import board

### OLED
# import adafruit_displayio_ssd1306
# import displayio
# import terminalio
# from adafruit_display_text import label
from kb import KMKKeyboard

from kmk.hid import HIDModes
from kmk.keys import KC
from kmk.modules.layers import Layers
from kmk.modules.split import Split, SplitType

keyboard = KMKKeyboard()


keyboard.tap_time = 320
keyboard.debug_enabled = True

split = Split(split_type=SplitType.UART)
layers = Layers()

keyboard.modules = [split, layers]


keyboard.keymap = [
    [
        # Alpha
        KC.ESC,   KC.Q,    KC.W,    KC.E,    KC.R,    KC.T,            KC.Y,    KC.U,    KC.I,     KC.O,   KC.P,    KC.LBKT, \
        KC.CAPS,  KC.A,    KC.S,    KC.D,    KC.F,    KC.G,            KC.H,    KC.J,    KC.K,     KC.L,   KC.SEMI, KC.SQT, \
        KC.LSFT,  KC.Z,    KC.X,    KC.C,    KC.V,    KC.B,            KC.N,    KC.M,    KC.COMMA, KC.DOT, KC.FSLH, KC.MO(2), \
                                    KC.LGUI, KC.LCTRL, KC.SPC,       KC.LCTRL(KC.LALT),  KC.MO(1), KC.LALT,
    ],
    [
        # Control
        KC.NO,   KC.PG_UP, KC.BSPC, KC.UP,     KC.DEL,   KC.PG_DN,      KC.NO,  KC.N7,  KC.N8,  KC.N9,  KC.NO,  KC.NO, \
        KC.NO,   KC.HOME,  KC.LEFT, KC.DOWN,   KC.RIGHT, KC.END,        KC.NO,  KC.N4,  KC.N5,  KC.N6,  KC.N0,  KC.NO, \
        KC.LSFT, KC.ESC,   KC.TAB,  KC.INSERT, KC.ENTER, KC.K_UNDO,     KC.N0,  KC.N1,  KC.N2,  KC.N3,  KC.NO,  KC.NO, \

        KC.ESC,   KC.QUOT, KC.COMM, KC.DOT,  KC.P,    KC.Y,            KC.F,    KC.G,    KC.C,    KC.R,    KC.L,    KC.BSPC, \
        KC.LCTL,  KC.A,    KC.O,    KC.E,    KC.U,    KC.I,            KC.D,    KC.H,    KC.T,    KC.N,    KC.S,    KC.ENT, \
        KC.LSFT,  KC.SCLN, KC.Q,    KC.J,    KC.K,    KC.X,            KC.B,    KC.M,    KC.W,    KC.V,    KC.Z,    KC.SLSH, \
                                   KC.LGUI, KC.LCTRL, KC.SPC,        KC.LCTRL(KC.LALT),  KC.MO(1), KC.LALT,
    ],
    [
        # Function
        KC.C_MUTE, KC.C_VOL_DN, KC.C_VOL_UP, KC.C_PREV, KC.C_NEXT, KC.C_PP,           KC.NO, KC.NO, KC.NO, KC.NO, KC.NO, KC.NO, \
        KC.F1,     KC.F2,       KC.F3,       KC.F4,     KC.F5,     KC.F6,             KC.NO, KC.NO, KC.NO, KC.NO, KC.NO, KC.NO, \
        KC.F7,     KC.F8,       KC.F9,       KC.F10,    KC.F11,    KC.F12,            KC.NO, KC.NO, KC.NO, KC.NO, KC.NO, KC.PAUSE_BREAK, \
                                                   KC.LGUI, KC.LCTRL, KC.SPC,       KC.LCTRL(KC.LALT),  KC.MO(1), KC.LALT,
    ]
]

if __name__ == '__main__':
    keyboard.go(hid_type=HIDModes.USB)