1 files changed, 72 insertions, 0 deletions
diff --git a/check_layout.py b/check_layout.py
new file mode 100644
index 0000000..c38705e
--- /dev/null
+++ b/check_layout.py
@@ -0,0 +1,72 @@
+import xml.etree.ElementTree as ET
+import sys
+
+warning_count = 0
+
+def warn(msg):
+    global warning_count
+    print(msg)
+    warning_count += 1
+
+def key_list_str(keys):
+    return ", ".join(sorted(list(keys)))
+
+def missing_some_of(keys, symbols, class_name=None):
+    if class_name is None:
+        class_name = "of [" + ", ".join(symbols) + "]"
+    missing = set(symbols).difference(keys)
+    if len(missing) > 0 and len(missing) != len(symbols):
+        warn("Layout includes some %s but not all, missing: %s" % (
+            class_name, key_list_str(missing)))
+
+def missing_required(keys, symbols, msg):
+    missing = set(symbols).difference(keys)
+    if len(missing) > 0:
+        warn("%s, missing: %s" % (msg, key_list_str(missing)))
+
+def unexpected_keys(keys, symbols, msg):
+    unexpected = set(symbols).intersection(keys)
+    if len(unexpected) > 0:
+        warn("%s, unexpected: %s" % (msg, key_list_str(unexpected)))
+
+def parse_layout(fname):
+    keys = set()
+    root = ET.parse(fname).getroot()
+    if root.tag != "keyboard":
+        return None
+    for row in root:
+        for key in row:
+            for attr in key.keys():
+                keys.add(key.get(attr).removeprefix("\\"))
+    return root, keys
+
+def check_layout(layout):
+    root, keys = layout
+    missing_some_of(keys, "~!@#$%^&*(){}`[]=\\-_;:/.,?<>'\"+|", "ASCII punctuation")
+    missing_some_of(keys, "0123456789", "digits")
+    missing_some_of(keys, ["f11_placeholder", "f12_placeholder"])
+    missing_some_of(keys, ["esc", "tab"])
+    missing_required(keys, ["backspace", "delete"], "Layout doesn't define some important keys")
+
+    bottom_row_keys = [
+            "ctrl", "fn", "switch_numeric", "change_method", "switch_emoji",
+            "config", "switch_second", "enter", "action", "left", "up", "right",
+            "down", "space"
+            ]
+
+    if root.get("bottom_row") == "false":
+        missing_required(keys, bottom_row_keys,
+                         "Layout redefines the bottom row but some important keys are missing")
+    else:
+        unexpected_keys(keys, bottom_row_keys,
+                        "Layout contains keys present in the bottom row")
+
+for fname in sys.argv[1:]:
+    layout = parse_layout(fname)
+    if layout == None:
+        print("Not a layout file: %s" % fname)
+    else:
+        print("# %s" % fname)
+        warning_count = 0
+        check_layout(layout)
+        print("%d warnings" % warning_count)