import os

import board
from kmk.quickpin.pro_micro.kb2040 import pinout as pins

from kmk.kmk_keyboard import KMKKeyboard
from kmk.scanners.keypad import KeysScanner

from kmk.modules.layers import Layers
from kmk.modules.combos import Combos, Chord
from kmk.modules.mouse_keys import MouseKeys

from kmk.extensions.media_keys import MediaKeys

from ardux.chord import ArduxChord
from kmk.keys import KC

class ArduxKeyboard(KMKKeyboard):
    coord_mapping = [
        0,  1,  2,  3,
        4,  5,  6,  7,
    ]

    keymap = [
        # std - left - base
        [KC.S, KC.T, KC.R, KC.A,
         KC.O, KC.I, KC.Y, KC.E],
        # std - left - parenthesis
        [KC.RIGHT_CURLY_BRACE, KC.LEFT_PAREN, KC.RIGHT_PAREN, KC.TRANSPARENT,
         KC.LEFT_CURLY_BRACE, KC.LBRACKET, KC.RBRACKET, KC.NO],
        # std - left - number
        [KC.TRANSPARENT, KC.N3, KC.N2, KC.N1,
         KC.NO, KC.N6, KC.N5, KC.N4],
        # std - left - navigation
        [KC.PGUP, KC.END, KC.UP, KC.HOME,
         KC.PGDOWN, KC.LEFT, KC.DOWN, KC.RIGHT],
        # std - left - symbols
        [KC.GRAVE, KC.SCOLON, KC.BSLASH, KC.EXCLAIM,
         KC.EQUAL, KC.MINUS, KC.QUESTION, KC.TRANSPARENT],
        # std - left - custom
        [KC.NO, KC.AUDIO_VOL_UP, KC.INSERT, KC.AUDIO_MUTE,
         KC.TRANSPARENT, KC.AUDIO_VOL_DOWN, KC.PSCREEN, KC.RSHIFT],
        # std - left - mouse
        [KC.MW_DOWN, KC.MB_RMB, KC.MS_UP, KC.MB_LMB,
         KC.MW_DOWN, KC.MS_LEFT, KC.MS_DOWN, KC.MS_RIGHT]
    ]

    # Init / constructor / setup
    def __init__(self):
        # Enable debugging if appropriate
        if os.getenv('ARDUX_KMK_DEBUGGING'):
            self.debug_enabled = True

        # setup modules/extensions arrays
        self.modules = []
        self.extensions = []
        
        # Direct wire & matrix setup
        self.matrix = KeysScanner([pins[16], pins[17], pins[18], pins[19], pins[12], pins[13], pins[14], pins[15]])

        # Layers
        self.modules.append(Layers())

        # Media Keys
        self.extensions.append(MediaKeys())

        # Mouse Keys
        self.modules.append(MouseKeys())

        # Combos
        self.combo_module = Combos()
        self.modules.append(self.combo_module)
        self.setup_combos()

    # Define combos for ardux
    def setup_combos(self):
        self.combo_module.combos = []
        
        combo_enter = ArduxChord((KC.A, KC.E), KC.ENTER, ardux_keyboard=[self], layers=[0])
        self.combo_module.combos.append(combo_enter)
        
        combo_space = ArduxChord((KC.O, KC.I, KC.Y, KC.E), KC.SPACE, ardux_keyboard=[self], layers=[1])
        self.combo_module.combos.append(combo_space)