code.py

import time
import usb_hid

from pmk import PMK
from pmk.platform.keybow2040 import Keybow2040 as Hardware

from adafruit_hid.keyboard import Keyboard
from adafruit_hid.keyboard_layout_us import KeyboardLayoutUS
from adafruit_hid.keycode import Keycode

DEBUG = True

# Set up Keybow
keybow = PMK(Hardware())
keys = keybow.keys

# Set up the keyboard and layout
keyboard = Keyboard(usb_hid.devices)
layout = KeyboardLayoutUS(keyboard)

# Either control (win) or cmd (mac)
command_mod = "command"
# command_mod = "control"

# A map of keycodes that will be mapped sequentially to each of the keys, 0-15
keymap = [
    ["f9"],
    ["delete"],
    [command_mod, "y"],
    [command_mod, "z"],  # 0x0-3
    ["left", "f11"],
    ["grave"],
    ["left"],
    ["tab"],  # 0x4-7
    ["shift", "f12"],
    ["insert"],
    ["down"],
    ["up"],  # 0x8-B
    [],
    ["backspace"],
    ["right"],
    ["space"],  # 0xC-F
]

fire_time_interval = 0.1
last_fired = {}

# https://docs.circuitpython.org/projects/hid/en/latest/api.html#adafruit-hid-keycode-keycode
keycodes = {
    "backspace": Keycode.BACKSPACE,
    "control": Keycode.LEFT_CONTROL,
    "command": Keycode.COMMAND,
    "delete": Keycode.DELETE,
    "down": Keycode.DOWN_ARROW,
    "f11": Keycode.F11,
    "f12": Keycode.F12,
    "f9": Keycode.F9,
    "grave": Keycode.GRAVE_ACCENT,
    "insert": Keycode.INSERT,
    "left": Keycode.LEFT_ARROW,
    "right": Keycode.RIGHT_ARROW,
    "shift": Keycode.LEFT_SHIFT,
    "space": Keycode.SPACEBAR,
    "tab": Keycode.TAB,
    "up": Keycode.UP_ARROW,
    "y": Keycode.Y,
    "z": Keycode.Z,
}

colors = {
    "red": (255, 0, 0),
    "off": (0, 0, 0),
    "yellow": (255, 255 // 2, 0),
    "green": (0, 255, 0),
    "blue": (0, 0, 255),
}

rgb_map = [
    "red",
    "off",
    "yellow",
    "yellow",  # 0-3
    "off",
    "green",
    "blue",
    "off",  # 4-7
    "off",
    "green",
    "blue",
    "blue",  # 8-B
    "red",
    "off",
    "blue",
    "off",  # C-F
]

def log(key, event):
    if DEBUG == False:
        return

    now = time.monotonic()
    mapping = keymap[key.number]
    codes = [keycodes[c] for c in mapping]
    print(f"{event} {key.number} {mapping} {codes} {now}")

def send_keycode(key):
    mapping = keymap[key.number]
    codes = [keycodes[c] for c in mapping]
    keyboard.send(*codes)
    log(key, 'press')

for key in keys:
    color = rgb_map[key.number]
    key.set_led(*colors[color])

    @keybow.on_press(key)
    def press_handler(key):
        send_keycode(key)

    @keybow.on_release(key)
    def release_handler(key):
        mapping = keymap[key.number]
        try:
            del last_fired[key.number]
        except KeyError:
            pass

        log(key, 'release')

    @keybow.on_hold(key)
    def hold_handler(key):
        mapping = keymap[key.number]
        now = time.monotonic()
        last_fired[key.number] = now
        log(key, 'hold')

while True:
    keybow.update()
    if(not last_fired):
        continue

    now = time.monotonic()
    for key in keys:
        try:
            last_fire = last_fired[key.number]
        except KeyError:
            continue
        else:
            next_fire = last_fire + fire_time_interval
            if now > next_fire:
                last_fired[key.number] = now
                send_keycode(key)