diff options
Diffstat (limited to 'lib/python/qmk/cli/generate')
| -rwxr-xr-x | lib/python/qmk/cli/generate/api.py | 15 | ||||
| -rw-r--r-- | lib/python/qmk/cli/generate/keycodes.py | 55 | ||||
| -rw-r--r-- | lib/python/qmk/cli/generate/keycodes_tests.py | 39 | 
3 files changed, 108 insertions, 1 deletions
| diff --git a/lib/python/qmk/cli/generate/api.py b/lib/python/qmk/cli/generate/api.py index 8650a36b84..d662e14e00 100755 --- a/lib/python/qmk/cli/generate/api.py +++ b/lib/python/qmk/cli/generate/api.py @@ -10,8 +10,9 @@ from qmk.datetime import current_datetime  from qmk.info import info_json  from qmk.json_encoders import InfoJSONEncoder  from qmk.json_schema import json_load +from qmk.keymap import list_keymaps  from qmk.keyboard import find_readme, list_keyboards -from qmk.keycodes import load_spec, list_versions +from qmk.keycodes import load_spec, list_versions, list_languages  DATA_PATH = Path('data')  TEMPLATE_PATH = DATA_PATH / 'templates/api/' @@ -44,6 +45,13 @@ def _resolve_keycode_specs(output_folder):          output_file = output_folder / f'constants/keycodes_{version}.json'          output_file.write_text(json.dumps(overall, indent=4), encoding='utf-8') +    for lang in list_languages(): +        for version in list_versions(lang): +            overall = load_spec(version, lang) + +            output_file = output_folder / f'constants/keycodes_{lang}_{version}.json' +            output_file.write_text(json.dumps(overall, indent=4), encoding='utf-8') +      # Purge files consumed by 'load_spec'      shutil.rmtree(output_folder / 'constants/keycodes/') @@ -104,6 +112,11 @@ def generate_api(cli):      # Generate and write keyboard specific JSON files      for keyboard_name in keyboard_list:          kb_all[keyboard_name] = info_json(keyboard_name) + +        # Populate the list of JSON keymaps +        for keymap in list_keymaps(keyboard_name, c=False, fullpath=True): +            kb_all[keyboard_name]['keymaps'][keymap.name] = {'url': f'https://raw.githubusercontent.com/qmk/qmk_firmware/master/{keymap}/keymap.json'} +          keyboard_dir = v1_dir / 'keyboards' / keyboard_name          keyboard_info = keyboard_dir / 'info.json'          keyboard_readme = keyboard_dir / 'readme.md' diff --git a/lib/python/qmk/cli/generate/keycodes.py b/lib/python/qmk/cli/generate/keycodes.py index 29b7db3c80..2ed84cd589 100644 --- a/lib/python/qmk/cli/generate/keycodes.py +++ b/lib/python/qmk/cli/generate/keycodes.py @@ -8,6 +8,24 @@ from qmk.path import normpath  from qmk.keycodes import load_spec +def _render_key(key): +    width = 7 +    if 'S(' in key: +        width += len('S()') +    if 'A(' in key: +        width += len('A()') +    if 'RCTL(' in key: +        width += len('RCTL()') +    if 'ALGR(' in key: +        width += len('ALGR()') +    return key.ljust(width) + + +def _render_label(label): +    label = label.replace("\\", "(backslash)") +    return label + +  def _generate_ranges(lines, keycodes):      lines.append('')      lines.append('enum qk_keycode_ranges {') @@ -67,6 +85,23 @@ def _generate_helpers(lines, keycodes):          lines.append(f'#define IS_{ group.upper() }_KEYCODE(code) ((code) >= {lo} && (code) <= {hi})') +def _generate_aliases(lines, keycodes): +    lines.append('') +    lines.append('// Aliases') +    for key, value in keycodes["aliases"].items(): +        define = _render_key(value.get("key")) +        val = _render_key(key) +        if 'label' in value: +            lines.append(f'#define {define} {val} // {_render_label(value.get("label"))}') +        else: +            lines.append(f'#define {define} {val}') + +    lines.append('') +    for key, value in keycodes["aliases"].items(): +        for alias in value.get("aliases", []): +            lines.append(f'#define {alias} {value.get("key")}') + +  @cli.argument('-v', '--version', arg_only=True, required=True, help='Version of keycodes to generate.')  @cli.argument('-o', '--output', arg_only=True, type=normpath, help='File to write to')  @cli.argument('-q', '--quiet', arg_only=True, action='store_true', help="Quiet mode, only output error messages") @@ -86,3 +121,23 @@ def generate_keycodes(cli):      # Show the results      dump_lines(cli.args.output, keycodes_h_lines, cli.args.quiet) + + +@cli.argument('-v', '--version', arg_only=True, required=True, help='Version of keycodes to generate.') +@cli.argument('-l', '--lang', arg_only=True, required=True, help='Language of keycodes to generate.') +@cli.argument('-o', '--output', arg_only=True, type=normpath, help='File to write to') +@cli.argument('-q', '--quiet', arg_only=True, action='store_true', help="Quiet mode, only output error messages") +@cli.subcommand('Used by the make system to generate keymap_{lang}.h from keycodes_{lang}_{version}.json', hidden=True) +def generate_keycode_extras(cli): +    """Generates the header file. +    """ + +    # Build the header file. +    keycodes_h_lines = [GPL2_HEADER_C_LIKE, GENERATED_HEADER_C_LIKE, '#pragma once', '#include "keymap.h"', '// clang-format off'] + +    keycodes = load_spec(cli.args.version, cli.args.lang) + +    _generate_aliases(keycodes_h_lines, keycodes) + +    # Show the results +    dump_lines(cli.args.output, keycodes_h_lines, cli.args.quiet) diff --git a/lib/python/qmk/cli/generate/keycodes_tests.py b/lib/python/qmk/cli/generate/keycodes_tests.py new file mode 100644 index 0000000000..453b4693a7 --- /dev/null +++ b/lib/python/qmk/cli/generate/keycodes_tests.py @@ -0,0 +1,39 @@ +"""Used by the make system to generate a keycode lookup table from keycodes_{version}.json +""" +from milc import cli + +from qmk.constants import GPL2_HEADER_C_LIKE, GENERATED_HEADER_C_LIKE +from qmk.commands import dump_lines +from qmk.path import normpath +from qmk.keycodes import load_spec + + +def _generate_defines(lines, keycodes): +    lines.append('') +    lines.append('std::map<uint16_t, std::string> KEYCODE_ID_TABLE = {') +    for key, value in keycodes["keycodes"].items(): +        lines.append(f'    {{{value.get("key")}, "{value.get("key")}"}},') +    lines.append('};') + + +@cli.argument('-v', '--version', arg_only=True, required=True, help='Version of keycodes to generate.') +@cli.argument('-o', '--output', arg_only=True, type=normpath, help='File to write to') +@cli.argument('-q', '--quiet', arg_only=True, action='store_true', help="Quiet mode, only output error messages") +@cli.subcommand('Used by the make system to generate a keycode lookup table from keycodes_{version}.json', hidden=True) +def generate_keycodes_tests(cli): +    """Generates a keycode to identifier lookup table for unit test output. +    """ + +    # Build the keycodes.h file. +    keycodes_h_lines = [GPL2_HEADER_C_LIKE, GENERATED_HEADER_C_LIKE, '// clang-format off'] +    keycodes_h_lines.append('extern "C" {\n#include <keycode.h>\n}') +    keycodes_h_lines.append('#include <map>') +    keycodes_h_lines.append('#include <string>') +    keycodes_h_lines.append('#include <cstdint>') + +    keycodes = load_spec(cli.args.version) + +    _generate_defines(keycodes_h_lines, keycodes) + +    # Show the results +    dump_lines(cli.args.output, keycodes_h_lines, cli.args.quiet) | 
