Use arduino-cli built uf2 file if available

build_platform.py

@@ -3,9 +3,9 @@ import glob
 import time
 import os
 import shutil
-from distutils.dir_util import copy_tree
 import subprocess
 import collections
+from contextlib import contextmanager
 
 # optional wall option cause build failed if has warnings
 BUILD_WALL = False
@@ -248,6 +248,13 @@ if not IS_LEARNING_SYS:
 
 ################################ UF2 Utils.
 
+def glob01(pattern):
+    result = glob.glob(pattern)
+    print("glob01", pattern, result)
+    if len(result) > 1:
+        raise RuntimeError(f"Required pattern {pattern} to match at most 1 file, got {result}")
+    return result[0] if result else None
+
 def glob1(pattern):
     result = glob.glob(pattern)
     if len(result) != 1:
@@ -276,11 +283,26 @@ def generate_uf2(example_path):
     """
     if not download_uf2_utils():
         return None
+
+    os.system(f"find {example_path} -print")
+
+    cli_build_uf2_path = "build/*.*." + fqbn.split(':')[2] + "/*.uf2"
+    uf2_input_file = glob01(os.path.join(example_path, cli_build_uf2_path))
+
+    # Some platforms, like rp2040, directly generate a uf2 file, so no need to do it ourselves
+    if uf2_input_file is not None:
+        output_file = os.path.splitext(uf2_input_file)[0] + ".uf2"
+        shutil.copy(uf2_input_file, output_file)
+        ColorPrint.print_pass(CHECK)
+        ColorPrint.print_info("Used uf2 generated by arduino-cli")
+        return
+
+    # For other uf2-supporting platforms, we can generate it from a hex file
     cli_build_path = "build/*.*." + fqbn.split(':')[2] + "/*.hex"
-    input_file = glob1(os.path.join(example_path, cli_build_path))
-    output_file = os.path.splitext(input_file)[0] + ".uf2"
+    hex_input_file = glob1(os.path.join(example_path, cli_build_hex_path))
+    output_file = os.path.splitext(nex_input_file)[0] + ".uf2"
     family_id = ALL_PLATFORMS[platform][1]
-    cmd = ['python3', 'uf2conv.py', input_file, '-c', '-f', family_id, '-o', output_file]
+    cmd = ['python3', 'uf2conv.py', hex_input_file, '-c', '-f', family_id, '-o', output_file]
     proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     r = proc.wait(timeout=60)
     out = proc.stdout.read()
@@ -311,6 +333,20 @@ for arg in sys.argv[1:]:
         print("Unknown platform: ", arg)
         exit(-1)
 
+@contextmanager
+def group_output(title):
+    sys.stdout.flush()
+    sys.stderr.flush()
+    print(f"::group::{title}")
+    try:
+        yield
+    finally:
+        sys.stdout.flush()
+        sys.stderr.flush()
+        print(f"::endgroup::")
+        sys.stdout.flush()
+
+
 def test_examples_in_folder(folderpath):
     global success
     for example in sorted(os.listdir(folderpath)):
@@ -366,6 +402,7 @@ def test_examples_in_folder(folderpath):
             ColorPrint.print_pass(CHECK)
             if err:
                 # also print out warning message
+                with group_output(f"{example} {fqbn} build output"):
                     ColorPrint.print_fail(err.decode("utf-8"))
             if os.path.exists(gen_file_name):
                 if ALL_PLATFORMS[platform][1] == None:
@@ -383,6 +420,7 @@ def test_examples_in_folder(folderpath):
                         os.system("ls -lR "+BUILD_DIR+"/build")
         else:
             ColorPrint.print_fail(CROSS)
+            with group_output(f"{example} {fqbn} built output"):
                 ColorPrint.print_fail(out.decode("utf-8"))
                 ColorPrint.print_fail(err.decode("utf-8"))
             success = 1