mirror of
				https://github.com/mfulz/qmk_firmware.git
				synced 2025-11-04 07:12:33 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			130 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			130 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
"""Command to look over a keyboard/keymap and check for common mistakes.
 | 
						|
"""
 | 
						|
from pathlib import Path
 | 
						|
 | 
						|
from milc import cli
 | 
						|
 | 
						|
from qmk.decorators import automagic_keyboard, automagic_keymap
 | 
						|
from qmk.info import info_json
 | 
						|
from qmk.keyboard import keyboard_completer, list_keyboards
 | 
						|
from qmk.keymap import locate_keymap
 | 
						|
from qmk.path import is_keyboard, keyboard
 | 
						|
 | 
						|
 | 
						|
def keymap_check(kb, km):
 | 
						|
    """Perform the keymap level checks.
 | 
						|
    """
 | 
						|
    ok = True
 | 
						|
    keymap_path = locate_keymap(kb, km)
 | 
						|
 | 
						|
    if not keymap_path:
 | 
						|
        ok = False
 | 
						|
        cli.log.error("%s: Can't find %s keymap.", kb, km)
 | 
						|
 | 
						|
    return ok
 | 
						|
 | 
						|
 | 
						|
def rules_mk_assignment_only(keyboard_path):
 | 
						|
    """Check the keyboard-level rules.mk to ensure it only has assignments.
 | 
						|
    """
 | 
						|
    current_path = Path()
 | 
						|
    errors = []
 | 
						|
 | 
						|
    for path_part in keyboard_path.parts:
 | 
						|
        current_path = current_path / path_part
 | 
						|
        rules_mk = current_path / 'rules.mk'
 | 
						|
 | 
						|
        if rules_mk.exists():
 | 
						|
            continuation = None
 | 
						|
 | 
						|
            for i, line in enumerate(rules_mk.open()):
 | 
						|
                line = line.strip()
 | 
						|
 | 
						|
                if '#' in line:
 | 
						|
                    line = line[:line.index('#')]
 | 
						|
 | 
						|
                if continuation:
 | 
						|
                    line = continuation + line
 | 
						|
                    continuation = None
 | 
						|
 | 
						|
                if line:
 | 
						|
                    if line[-1] == '\\':
 | 
						|
                        continuation = line[:-1]
 | 
						|
                        continue
 | 
						|
 | 
						|
                    if line and '=' not in line:
 | 
						|
                        errors.append(f'Non-assignment code on line +{i} {rules_mk}: {line}')
 | 
						|
 | 
						|
    return errors
 | 
						|
 | 
						|
 | 
						|
@cli.argument('--strict', action='store_true', help='Treat warnings as errors')
 | 
						|
@cli.argument('-kb', '--keyboard', completer=keyboard_completer, help='Comma separated list of keyboards to check')
 | 
						|
@cli.argument('-km', '--keymap', help='The keymap to check')
 | 
						|
@cli.argument('--all-kb', action='store_true', arg_only=True, help='Check all keyboards')
 | 
						|
@cli.subcommand('Check keyboard and keymap for common mistakes.')
 | 
						|
@automagic_keyboard
 | 
						|
@automagic_keymap
 | 
						|
def lint(cli):
 | 
						|
    """Check keyboard and keymap for common mistakes.
 | 
						|
    """
 | 
						|
    failed = []
 | 
						|
 | 
						|
    # Determine our keyboard list
 | 
						|
    if cli.args.all_kb:
 | 
						|
        if cli.args.keyboard:
 | 
						|
            cli.log.warning('Both --all-kb and --keyboard passed, --all-kb takes presidence.')
 | 
						|
 | 
						|
        keyboard_list = list_keyboards()
 | 
						|
    elif not cli.config.lint.keyboard:
 | 
						|
        cli.log.error('Missing required arguments: --keyboard or --all-kb')
 | 
						|
        cli.print_help()
 | 
						|
        return False
 | 
						|
    else:
 | 
						|
        keyboard_list = cli.config.lint.keyboard.split(',')
 | 
						|
 | 
						|
    # Lint each keyboard
 | 
						|
    for kb in keyboard_list:
 | 
						|
        if not is_keyboard(kb):
 | 
						|
            cli.log.error('No such keyboard: %s', kb)
 | 
						|
            continue
 | 
						|
 | 
						|
        # Gather data about the keyboard.
 | 
						|
        ok = True
 | 
						|
        keyboard_path = keyboard(kb)
 | 
						|
        keyboard_info = info_json(kb)
 | 
						|
 | 
						|
        # Check for errors in the info.json
 | 
						|
        if keyboard_info['parse_errors']:
 | 
						|
            ok = False
 | 
						|
            cli.log.error('%s: Errors found when generating info.json.', kb)
 | 
						|
 | 
						|
        if cli.config.lint.strict and keyboard_info['parse_warnings']:
 | 
						|
            ok = False
 | 
						|
            cli.log.error('%s: Warnings found when generating info.json (Strict mode enabled.)', kb)
 | 
						|
 | 
						|
        # Check the rules.mk file(s)
 | 
						|
        rules_mk_assignment_errors = rules_mk_assignment_only(keyboard_path)
 | 
						|
        if rules_mk_assignment_errors:
 | 
						|
            ok = False
 | 
						|
            cli.log.error('%s: Non-assignment code found in rules.mk. Move it to post_rules.mk instead.', kb)
 | 
						|
            for assignment_error in rules_mk_assignment_errors:
 | 
						|
                cli.log.error(assignment_error)
 | 
						|
 | 
						|
        # Keymap specific checks
 | 
						|
        if cli.config.lint.keymap:
 | 
						|
            if not keymap_check(kb, cli.config.lint.keymap):
 | 
						|
                ok = False
 | 
						|
 | 
						|
        # Report status
 | 
						|
        if not ok:
 | 
						|
            failed.append(kb)
 | 
						|
 | 
						|
    # Check and report the overall status
 | 
						|
    if failed:
 | 
						|
        cli.log.error('Lint check failed for: %s', ', '.join(failed))
 | 
						|
        return False
 | 
						|
 | 
						|
    cli.log.info('Lint check passed!')
 | 
						|
    return True
 |