Support a simple macro of a sequence of keycodes (basis for SEND_STRING)

kmk/common/event_defs.py

@@ -9,6 +9,9 @@ KEY_DOWN_EVENT = const(2)
 INIT_FIRMWARE_EVENT = const(3)
 NEW_MATRIX_EVENT = const(4)
 HID_REPORT_EVENT = const(5)
+KEYCODE_UP_EVENT = const(6)
+KEYCODE_DOWN_EVENT = const(7)
+MACRO_COMPLETE_EVENT = const(8)
 
 logger = logging.getLogger(__name__)
 
@@ -39,6 +42,28 @@ def key_down_event(row, col):
     }
 
 
+def keycode_up_event(keycode):
+    '''
+    Press a key by Keycode object, bypassing the keymap. Used mostly for
+    macros.
+    '''
+    return {
+        'type': KEYCODE_UP_EVENT,
+        'keycode': keycode,
+    }
+
+
+def keycode_down_event(keycode):
+    '''
+    Release a key by Keycode object, bypassing the keymap. Used mostly for
+    macros.
+    '''
+    return {
+        'type': KEYCODE_DOWN_EVENT,
+        'keycode': keycode,
+    }
+
+
 def new_matrix_event(matrix):
     return {
         'type': NEW_MATRIX_EVENT,
@@ -52,6 +77,13 @@ def hid_report_event():
     }
 
 
+def macro_complete_event(macro):
+    return {
+        'type': MACRO_COMPLETE_EVENT,
+        'macro': macro,
+    }
+
+
 def matrix_changed(new_matrix):
     def _key_pressed(dispatch, get_state):
         state = get_state()
@@ -94,4 +126,12 @@ def matrix_changed(new_matrix):
                 except ImportError:
                     logger.warning('Tried to reset to bootloader, but not supported on this chip?')
 
+        while get_state().macros_pending:
+            macro = get_state().macros_pending[0]
+
+            for event in macro():
+                dispatch(event)
+
+            dispatch(macro_complete_event(macro))
+
     return _key_pressed

kmk/common/internal_state.py

@@ -4,9 +4,11 @@ import sys
 from kmk.common.consts import DiodeOrientation
 from kmk.common.event_defs import (HID_REPORT_EVENT, INIT_FIRMWARE_EVENT,
                                    KEY_DOWN_EVENT, KEY_UP_EVENT,
-                                   NEW_MATRIX_EVENT)
+                                   KEYCODE_DOWN_EVENT, KEYCODE_UP_EVENT,
+                                   MACRO_COMPLETE_EVENT, NEW_MATRIX_EVENT)
 from kmk.common.internal_keycodes import process_internal_key_event
 from kmk.common.keycodes import FIRST_KMK_INTERNAL_KEYCODE, Keycodes
+from kmk.macros import KMKMacro
 
 
 class ReduxStore:
@@ -52,6 +54,7 @@ class ReduxStore:
 class InternalState:
     modifiers_pressed = frozenset()
     keys_pressed = frozenset()
+    macros_pending = []
     keymap = []
     row_pins = []
     col_pins = []
@@ -133,6 +136,20 @@ def kmk_reducer(state=None, action=None, logger=None):
             matrix=action['matrix'],
         )
 
+    if action['type'] == KEYCODE_UP_EVENT:
+        return state.update(
+            keys_pressed=frozenset(
+                key for key in state.keys_pressed if key != action['keycode']
+            ),
+        )
+
+    if action['type'] == KEYCODE_DOWN_EVENT:
+        return state.update(
+            keys_pressed=(
+                state.keys_pressed | {action['keycode']}
+            ),
+        )
+
     if action['type'] == KEY_UP_EVENT:
         row = action['row']
         col = action['col']
@@ -144,6 +161,14 @@ def kmk_reducer(state=None, action=None, logger=None):
         if not changed_key:
             return state
 
+        if isinstance(changed_key, KMKMacro):
+            if changed_key.keyup:
+                return state.update(
+                    macros_pending=state.macros_pending + [changed_key.keyup],
+                )
+
+            return state
+
         newstate = state.update(
             keys_pressed=frozenset(
                 key for key in state.keys_pressed if key != changed_key
@@ -166,6 +191,14 @@ def kmk_reducer(state=None, action=None, logger=None):
         if not changed_key:
             return state
 
+        if isinstance(changed_key, KMKMacro):
+            if changed_key.keydown:
+                return state.update(
+                    macros_pending=state.macros_pending + [changed_key.keydown],
+                )
+
+            return state
+
         newstate = state.update(
             keys_pressed=(
                 state.keys_pressed | {changed_key}
@@ -196,6 +229,14 @@ def kmk_reducer(state=None, action=None, logger=None):
     if action['type'] == HID_REPORT_EVENT:
         return state
 
+    if action['type'] == MACRO_COMPLETE_EVENT:
+        return state.update(
+            macros_pending=[
+                m for m in state.macros_pending
+                if m != action['macro']
+            ],
+        )
+
     # On unhandled events, log and do not mutate state
     logger.warning('Unhandled event! Returning state unmodified.')
     return state