summaryrefslogtreecommitdiff
path: root/lib/python
diff options
context:
space:
mode:
Diffstat (limited to 'lib/python')
-rw-r--r--lib/python/qmk/cli/lint.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/lib/python/qmk/cli/lint.py b/lib/python/qmk/cli/lint.py
index 6b077c250f..c67809a697 100644
--- a/lib/python/qmk/cli/lint.py
+++ b/lib/python/qmk/cli/lint.py
@@ -10,6 +10,7 @@ from qmk.keyboard import keyboard_completer, list_keyboards
from qmk.keymap import locate_keymap, list_keymaps
from qmk.path import is_keyboard, keyboard
from qmk.git import git_get_ignored_files
+from qmk.c_parse import c_source_files
def _list_defaultish_keymaps(kb):
@@ -25,6 +26,29 @@ def _list_defaultish_keymaps(kb):
return keymaps
+def _get_code_files(kb, km=None):
+ """Return potential keyboard/keymap code files
+ """
+ search_path = locate_keymap(kb, km).parent if km else keyboard(kb)
+
+ code_files = []
+ for file in c_source_files([search_path]):
+ # Ignore keymaps when only globing keyboard files
+ if not km and 'keymaps' in file.parts:
+ continue
+ code_files.append(file)
+
+ return code_files
+
+
+def _has_license(file):
+ """Check file has a license header
+ """
+ # Crude assumption that first line of license header is a comment
+ fline = open(file).readline().rstrip()
+ return fline.startswith(("/*", "//"))
+
+
def _handle_json_errors(kb, info):
"""Convert any json errors into lint errors
"""
@@ -92,6 +116,11 @@ def keymap_check(kb, km):
cli.log.error(f'{kb}/{km}: The file "{file}" should not exist!')
ok = False
+ for file in _get_code_files(kb, km):
+ if not _has_license(file):
+ cli.log.error(f'{kb}/{km}: The file "{file}" does not have a license header!')
+ ok = False
+
return ok
@@ -119,6 +148,11 @@ def keyboard_check(kb):
cli.log.error(f'{kb}: The file "{file}" should not exist!')
ok = False
+ for file in _get_code_files(kb):
+ if not _has_license(file):
+ cli.log.error(f'{kb}: The file "{file}" does not have a license header!')
+ ok = False
+
return ok