Encoder Support in CLI qmk compile

keyboards/handwired/onekey/keymaps/default_json/keymap.json

@@ -0,0 +1,12 @@
+{
+  "keyboard": "handwired/onekey/pytest",
+  "keymap": "default_json",
+  "layout": "LAYOUT",
+  "layers": [
+    [
+      "KC_ENTER"
+    ]
+  ],
+  "author": "",
+  "notes": ""
+}

keyboards/handwired/onekey/keymaps/encodertest/keymap.json

@@ -0,0 +1,18 @@
+{
+  "keyboard": "handwired/onekey/pytest",
+  "keymap": "encodertest",
+  "layout": "LAYOUT",
+  "layers": [
+    [
+      "KC_ENTER"
+    ]
+  ],
+  "author": "",
+  "notes": "",
+  "encoders": [
+    {
+      "clockwise": "KC_UP",
+      "counter": "KC_DOWN"
+    }
+  ]
+}

lib/python/qmk/cli/flash.py

@@ -61,7 +61,7 @@ def flash(cli):
         # Handle compiling a configurator JSON
         user_keymap = parse_configurator_json(cli.args.filename)
         keymap_path = qmk.path.keymap(user_keymap['keyboard'])
-        command = compile_configurator_json(user_keymap, cli.args.bootloader)
+        command = compile_configurator_json(user_keymap, bootloader=cli.args.bootloader)
 
         cli.log.info('Wrote keymap to {fg_cyan}%s/%s/keymap.c', keymap_path, user_keymap['keymap'])
 

lib/python/qmk/commands.py

@@ -53,7 +53,7 @@ def compile_configurator_json(user_keymap, bootloader=None):
         A command to run to compile and flash the C file.
     """
     # Write the keymap C file
-    qmk.keymap.write(user_keymap['keyboard'], user_keymap['keymap'], user_keymap['layout'], user_keymap['layers'])
+    qmk.keymap.write(user_keymap['keyboard'], user_keymap['keymap'], user_keymap['layout'], user_keymap['layers'], user_keymap.get('encoders'))
 
     # Return a command that can be run to make the keymap and flash if given
     if bootloader is None:

lib/python/qmk/keymap.py

@@ -19,6 +19,19 @@
 };
 """
 
+ENCODER_SUPPORT = """void encoder_update_user(uint8_t index, bool clockwise) {
+__ENCODERS_GO_HERE__
+};
+"""
+
+ENCODER_IF = """\t_ELIF_ (index == _INDEX_) {
+\t\tif (clockwise) {
+\t\t\ttap_code16(_CLOCKWISE_);
+\t\t} else {
+\t\t\ttap_code16(_COUNTER_);
+\t\t}
+\t}"""
+
 
 def template(keyboard):
     """Returns the `keymap.c` template for a keyboard.
@@ -47,7 +60,7 @@ def _strip_any(keycode):
     return keycode
 
 
-def generate(keyboard, layout, layers):
+def generate(keyboard, layout, layers, encoders=None):
     """Returns a keymap.c for the specified keyboard, layout, and layers.
 
     Args:
@@ -59,6 +72,9 @@ def generate(keyboard, layout, layers):
 
         layers
             An array of arrays describing the keymap. Each item in the inner array should be a string that is a valid QMK keycode.
+
+        encoders
+            An array of encoders on the keyboard, if any. Each item includes the index and the counter- and clockwise key codes.
     """
     layer_txt = []
 
@@ -73,10 +89,39 @@ def generate(keyboard, layout, layers):
     keymap = '\n'.join(layer_txt)
     keymap_c = template(keyboard)
 
-    return keymap_c.replace('__KEYMAP_GOES_HERE__', keymap)
+    keymap_c = keymap_c.replace('__KEYMAP_GOES_HERE__', keymap)
+
+    if encoders:
+        keymap_c += ENCODER_SUPPORT
+        encoders_txt = []
+        for encoder_num, encoder_set in enumerate(encoders):
+            curr_encoder = ENCODER_IF.replace('_ELIF_', 'if' if encoder_num == 0 else 'else if')
+            curr_encoder = curr_encoder.replace('_INDEX_', str(encoder_num))
+
+            clockwise = parse_basic_code(encoder_set.get('clockwise'))
+            counter = parse_basic_code(encoder_set.get('counter'))
+
+            curr_encoder = curr_encoder.replace('_CLOCKWISE_', clockwise)
+            curr_encoder = curr_encoder.replace('_COUNTER_', counter)
+
+            encoders_txt.append(curr_encoder)
 
+        keymap_c = keymap_c.replace('ENCODERS_GO_HERE__', '\n'.join(encoders_txt))
 
-def write(keyboard, keymap, layout, layers):
+    return keymap_c
+
+
+def parse_basic_code(keycode):
+    """Performs a simplistic check for a 'basic' keycode. Otherwise it returns 'KC_NO'
+
+    Args:
+        keycode
+            The keycode the user is trying to apply
+    """
+    return keycode if keycode[:2].lower() == 'kc' else 'KC_NO'
+
+
+def write(keyboard, keymap, layout, layers, encoders=None):
     """Generate the `keymap.c` and write it to disk.
 
     Returns the filename written to.
@@ -93,8 +138,11 @@ def write(keyboard, keymap, layout, layers):
 
         layers
             An array of arrays describing the keymap. Each item in the inner array should be a string that is a valid QMK keycode.
+
+        encoders
+            An array of encoders on the keyboard. Each item includes the index and the counter- and clockwise key codes.
     """
-    keymap_c = generate(keyboard, layout, layers)
+    keymap_c = generate(keyboard, layout, layers, encoders)
     keymap_file = qmk.path.keymap(keyboard) / keymap / 'keymap.c'
 
     keymap_file.parent.mkdir(parents=True, exist_ok=True)

lib/python/qmk/tests/test_cli_commands.py

@@ -12,7 +12,16 @@ def test_cformat():
     assert result.returncode == 0
 
 
+def test_compile_json():
+    assert check_subcommand('compile', 'keyboards/handwired/onekey/keymaps/default_json/keymap.json').returncode == 0
+
+
+def test_compile_json_encoder():
+    assert check_subcommand('compile', 'keyboards/handwired/onekey/keymaps/encodertest/keymap.json').returncode == 0
+
+
 def test_compile():
+    # keyboard compile
     assert check_subcommand('compile', '-kb', 'handwired/onekey/pytest', '-km', 'default').returncode == 0