From 31c486470591409767d797dba6ffcfe456ca1f13 Mon Sep 17 00:00:00 2001
From: zvecr <git@zvecr.com>
Date: Wed, 16 Mar 2022 19:45:37 +0000
Subject: [PATCH] Crude CLI device discovery

---
 lib/python/qmk/cli/__init__.py     |  1 +
 lib/python/qmk/cli/xap/__init__.py | 79 ++++++++++++++++++++++++++++++
 2 files changed, 80 insertions(+)

diff --git a/lib/python/qmk/cli/__init__.py b/lib/python/qmk/cli/__init__.py
index 3197d9d7f22..f0e86dd1240 100644
--- a/lib/python/qmk/cli/__init__.py
+++ b/lib/python/qmk/cli/__init__.py
@@ -70,6 +70,7 @@ subcommands = [
     'qmk.cli.new.keymap',
     'qmk.cli.pyformat',
     'qmk.cli.pytest',
+    'qmk.cli.xap',
     'qmk.cli.xap.generate_docs',
     'qmk.cli.xap.generate_json',
     'qmk.cli.xap.generate_qmk',
diff --git a/lib/python/qmk/cli/xap/__init__.py b/lib/python/qmk/cli/xap/__init__.py
index e69de29bb2d..963617a6eee 100755
--- a/lib/python/qmk/cli/xap/__init__.py
+++ b/lib/python/qmk/cli/xap/__init__.py
@@ -0,0 +1,79 @@
+"""Interactions with compatible XAP devices
+"""
+import json
+import random
+
+from milc import cli
+
+
+def _is_xap_usage(x):
+    return x['usage_page'] == 0xFF51 and x['usage'] == 0x0058
+
+
+def _is_filtered_device(x):
+    name = "%04x:%04x" % (x['vendor_id'], x['product_id'])
+    return name.lower().startswith(cli.args.device.lower())
+
+
+def _search():
+    devices = filter(_is_xap_usage, hid.enumerate())
+    if cli.args.device:
+        devices = filter(_is_filtered_device, devices)
+
+    return list(devices)
+
+
+def _list_devices():
+    """Dump out available devices
+    """
+    cli.log.info('Available devices:')
+    devices = _search()
+    for dev in devices:
+        device = hid.Device(path=dev['path'])
+
+        data = _query_device(device)
+        cli.log.info("  %04x:%04x %s %s [API:%s]", dev['vendor_id'], dev['product_id'], dev['manufacturer_string'], dev['product_string'], data['ver'])
+
+        if cli.config.general.verbose:
+            # TODO: better formatting like "lsusb -v"
+            cli.log.info("    " + json.dumps(data))
+
+
+def _query_device(device):
+    # gen token
+    tok = random.getrandbits(16)
+    temp = tok.to_bytes(2, byteorder='big')
+
+    # send with padding
+    padding = b"\x00" * 59
+    device.write(temp + b'\x02\x00\x00' + padding)
+
+    # get resp
+    array_alpha = device.read(8, 100)
+    # hex_string = " ".join("%02x" % b for b in array_alpha)
+
+    # validate tok sent == resp
+    ver = "UNKNOWN"
+    if str(temp) == str(array_alpha[:2]):
+        # to BCD string
+        a = (array_alpha[7] << 24) + (array_alpha[6] << 16) + (array_alpha[5] << 8) + (array_alpha[4])
+        ver = f'{a>>24}.{a>>16 & 0xFF}.{a & 0xFFFF}'
+
+    return {'ver': ver}
+
+
+@cli.argument('-d', '--device', help='device to select - uses format <pid>:<vid>.')
+@cli.argument('-i', '--index', default=0, help='device index to select.')
+@cli.argument('-l', '--list', arg_only=True, action='store_true', help='List available devices.')
+@cli.subcommand('Acquire debugging information from usb XAP devices.', hidden=False if cli.config.user.developer else True)
+def xap(cli):
+    """Acquire debugging information from XAP devices
+    """
+    # Lazy load to avoid issues
+    global hid
+    import hid
+
+    if cli.args.list:
+        return _list_devices()
+
+    cli.log.warn("TODO: Device specific stuff")