Allow getting board serial number

Author: xarantolus

.gitignore

@@ -1 +1,2 @@
 *.yml
+**/__pycache__

Dockerfile

@@ -39,8 +39,8 @@ RUN cd /actions-runner && \
 # Fix for Fedora not finding libstlink.so.1
 
 RUN pip install PyYAML
-COPY register-runner.py /actions-runner
-RUN chmod +x /actions-runner/register-runner.py
+COPY register_runner.py /actions-runner
+RUN chmod +x /actions-runner/register_runner.py
 
 ENV PATH="/actions-runner:${PATH}"
 
@@ -49,4 +49,4 @@ ENV LD_LIBRARY_PATH=/usr/local/lib:/usr/local/lib64
 
 WORKDIR /actions-runner
 
-ENTRYPOINT [ "/actions-runner/register-runner.py" ]
+ENTRYPOINT [ "/actions-runner/register_runner.py" ]

board_serial

@@ -0,0 +1,26 @@
+#!/bin/python3
+import os
+import sys
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+import register_runner
+
+if __name__ == "__main__":
+	chips = register_runner.get_chips()
+	if len(sys.argv) != 3:
+		print("Usage: %s <device>" % sys.argv[0])
+		print("Available devices:")
+		for chip in chips:
+			print("  %s" % chip)
+		sys.exit(1)
+
+	devices = list(filter(lambda x: x["dev_type"] == sys.argv[1], chips))
+
+	if len(devices) == 0:
+		print("Device not found")
+		sys.exit(1)
+
+	device = devices[0]
+
+	print(device['serial'])
+

register-runner.py register_runner.pyregister-runner.py renamed to register_runner.py

@@ -1,4 +1,4 @@
-#!/bin/python
+#!/bin/python3
 import os
 import re
 import subprocess
@@ -115,7 +115,14 @@ class Config(TypedDict):
     serial_map: Dict[str, str]
     runner_name: str
 
-def parse_config_file(config_file: str) -> dict:
+def parse_config_file(config_file: str = None) -> dict:
+    if config_file is None:
+        env_value = os.getenv("CHIPS_FILE_PATH")
+        if env_value:
+            config_file = env_value
+        else:
+            config_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "chips.yml")
+
     with open(config_file) as f:
         return yaml.safe_load(f)
 
@@ -138,10 +145,8 @@ def parse_config_file(config_file: str) -> dict:
 
     return config
 
-def main():
-    config_file_path = os.getenv("CHIPS_FILE_PATH") or "chips.yml"
-
-    config = parse_config_file(config_file_path)
+def get_chips():
+    config = parse_config_file()
     probe_info = run_command("st-info --probe")
     if not probe_info:
         print("No output from st-info --probe")
@@ -156,6 +161,11 @@ def main():
         if probe["serial"] in config["serial_map"]:
             chips[i]["dev_type"] = config["serial_map"][probe["serial"]]
 
+    return chips
+
+def main():
+    chips = get_chips()
+
     print(yaml.dump(chips))
 
     labels = set()