mirror of
https://github.com/Keychron/qmk_firmware.git
synced 2024-12-01 13:16:50 +06:00
8ea955c72f
* Improve argument handling of c2json * Add automagic
74 lines
3.2 KiB
Python
74 lines
3.2 KiB
Python
"""Generate a keymap.json from a keymap.c file.
|
|
"""
|
|
import re
|
|
import json
|
|
|
|
from argcomplete.completers import FilesCompleter
|
|
from milc import cli
|
|
|
|
import qmk.path
|
|
from qmk.json_encoders import InfoJSONEncoder
|
|
from qmk.decorators import automagic_keyboard, automagic_keymap
|
|
from qmk.keyboard import keyboard_completer, keyboard_folder
|
|
from qmk.keymap import locate_keymap, find_keymap_from_dir, generate_json, c2json as c2json_impl
|
|
from qmk.errors import CppError
|
|
from qmk.commands import dump_lines
|
|
|
|
|
|
@cli.argument('--no-cpp', arg_only=True, action='store_false', help='Do not use \'cpp\' on keymap.c')
|
|
@cli.argument('-o', '--output', arg_only=True, type=qmk.path.normpath, help='File to write to')
|
|
@cli.argument('-q', '--quiet', arg_only=True, action='store_true', help="Quiet mode, only output error messages")
|
|
@cli.argument('-kb', '--keyboard', type=keyboard_folder, completer=keyboard_completer, help='The keyboard\'s name')
|
|
@cli.argument('-km', '--keymap', help='The keymap\'s name')
|
|
@cli.argument('filename', nargs='?', type=qmk.path.FileType('r'), arg_only=True, completer=FilesCompleter('.c'), help='keymap.c file')
|
|
@cli.subcommand('Creates a keymap.json from a keymap.c file.')
|
|
@automagic_keyboard
|
|
@automagic_keymap
|
|
def c2json(cli):
|
|
"""Generate a keymap.json from a keymap.c file.
|
|
|
|
This command uses the `qmk.keymap` module to generate a keymap.json from a keymap.c file. The generated keymap is written to stdout, or to a file if -o is provided.
|
|
"""
|
|
filename = cli.args.filename
|
|
keyboard = cli.config.c2json.keyboard
|
|
keymap = cli.config.c2json.keymap
|
|
|
|
if filename:
|
|
if not keyboard and not keymap:
|
|
# fallback to inferring keyboard/keymap from path
|
|
(keymap, found_type) = find_keymap_from_dir(filename)
|
|
if found_type == 'keymap_directory':
|
|
keyboard = re.search(fr"keyboards/(.+)/keymaps/{keymap}/.*", filename.as_posix()).group(1)
|
|
|
|
elif keyboard and keymap:
|
|
if not filename:
|
|
# fallback to inferring keyboard/keymap from path
|
|
filename = locate_keymap(keyboard, keymap)
|
|
|
|
if not all((filename, keyboard, keymap)):
|
|
cli.log.error('You must supply keyboard and keymap, a path to a keymap.c within qmk_firmware, or absolute filename and keyboard and keymap')
|
|
cli.print_help()
|
|
return False
|
|
|
|
try:
|
|
keymap_json = c2json_impl(keyboard, keymap, filename, use_cpp=cli.args.no_cpp)
|
|
except CppError as e:
|
|
if cli.config.general.verbose:
|
|
cli.log.debug('The C pre-processor ran into a fatal error: %s', e)
|
|
cli.log.error('Something went wrong. Try to use --no-cpp.\nUse the CLI in verbose mode to find out more.')
|
|
return False
|
|
|
|
# Generate the keymap.json
|
|
try:
|
|
keymap_json = generate_json(keymap_json['keymap'], keymap_json['keyboard'], keymap_json['layout'], keymap_json['layers'])
|
|
except KeyError:
|
|
cli.log.error('Something went wrong. Try to use --no-cpp.')
|
|
return False
|
|
|
|
if cli.args.output:
|
|
keymap_lines = [json.dumps(keymap_json, cls=InfoJSONEncoder, sort_keys=True)]
|
|
else:
|
|
keymap_lines = [json.dumps(keymap_json)]
|
|
|
|
dump_lines(cli.args.output, keymap_lines, cli.args.quiet)
|