From a6d0d7b5982b319c24a9adbd109ed8d20581bbab Mon Sep 17 00:00:00 2001
From: Maxim Prokhorov <prokhorov.max@outlook.com>
Date: Wed, 31 Jul 2024 00:25:07 +0300
Subject: [PATCH 1/2] CI - restyle script summary and annotations in PRs

similar to .ino builder, prepare a short 'diff' summary of all the requried changes
https://docs.github.com/en/actions/using-workflows/workflow-commands-for-github-actions#adding-a-job-summary

provide inline annotations, so it is apparent clang-format job is the
cause of the PR actions check failure
https://docs.github.com/en/actions/using-workflows/workflow-commands-for-github-actions#setting-a-notice-message
---
 tests/restyle.py | 270 +++++++++++++++++++++++++++++++++++++++++++++++
 tests/restyle.sh |  56 ++--------
 2 files changed, 279 insertions(+), 47 deletions(-)
 create mode 100755 tests/restyle.py

diff --git a/tests/restyle.py b/tests/restyle.py
new file mode 100755
index 0000000000..909f089a3c
--- /dev/null
+++ b/tests/restyle.py
@@ -0,0 +1,270 @@
+#!/usr/bin/env python
+
+import argparse
+import os
+import sys
+import pathlib
+import subprocess
+import contextlib
+
+from dataclasses import dataclass
+
+
+GIT_ROOT = pathlib.Path(
+    subprocess.check_output(
+        ["git", "rev-parse", "--show-toplevel"], universal_newlines=True
+    ).strip()
+)
+
+
+def clang_format(clang_format, config, files):
+    if not files:
+        raise ValueError("Files list cannot be empty")
+
+    cmd = [clang_format, "--verbose", f"--style=file:{config.as_posix()}", "-i"]
+    cmd.extend(files)
+
+    subprocess.run(cmd, check=True)
+
+
+def ls_files(patterns):
+    """Git-only search, but rather poor at matching complex patterns (at least w/ <=py3.12)"""
+    proc = subprocess.run(
+        ["git", "--no-pager", "ls-files"],
+        capture_output=True,
+        check=True,
+        universal_newlines=True,
+    )
+
+    out = []
+    for line in proc.stdout.split("\n"):
+        path = pathlib.Path(line.strip())
+        if any(path.match(pattern) for pattern in patterns):
+            out.append(path)
+
+    return out
+
+
+def find_files(patterns):
+    """Filesystem search, matches both git and non-git files"""
+    return [
+        file
+        for pattern in patterns
+        for file in [found for found in GIT_ROOT.rglob(pattern)]
+    ]
+
+
+def find_core_files():
+    """Returns a subset of Core files that should be formatted"""
+    return [
+        file
+        for file in find_files(
+            (
+                "cores/esp8266/Lwip*",
+                "libraries/ESP8266mDNS/**/*",
+                "libraries/Wire/**/*",
+                "libraries/lwIP*/**/*",
+                "cores/esp8266/debug*",
+                "cores/esp8266/core_esp8266_si2c*",
+                "cores/esp8266/StreamString*",
+                "cores/esp8266/StreamSend*",
+                "libraries/Netdump/**/*",
+                "tests/**/*",
+            )
+        )
+        if file.is_file()
+        and file.suffix in (".c", ".cpp", ".h", ".hpp")
+        and not GIT_ROOT / "tests/host/bin" in file.parents
+        and not GIT_ROOT / "tests/host/common/catch.hpp" == file
+    ]
+
+
+def find_arduino_files():
+    """Returns every .ino file available in the repository, excluding submodule ones"""
+    return [
+        ino
+        for library in find_files(("libraries/*",))
+        if library.is_dir() and not (library / ".git").exists()
+        for ino in library.rglob("**/*.ino")
+    ]
+
+
+FILES_PRESETS = {
+    "core": find_core_files,
+    "arduino": find_arduino_files,
+}
+
+
+@dataclass
+class Changed:
+    file: str
+    hunk: list[str]
+    lines: list[int]
+
+
+class Context:
+    def __init__(self):
+        self.append_hunk = False
+        self.deleted = False
+        self.file = ""
+        self.hunk = []
+        self.markers = []
+
+    def reset(self):
+        self.__init__()
+
+    def reset_with_line(self, line):
+        self.reset()
+        self.hunk.append(line)
+
+    def pop(self, out, line):
+        if self.file and self.hunk and self.markers:
+            out.append(
+                Changed(file=self.file, hunk="\n".join(self.hunk), lines=self.markers)
+            )
+
+        self.reset_with_line(line)
+
+
+def changed_files():
+    """
+    Naive git-diff output parser. Generates list[Changed] for every file changed after clang-format.
+    """
+    proc = subprocess.run(
+        ["git", "--no-pager", "diff"],
+        capture_output=True,
+        check=True,
+        universal_newlines=True,
+    )
+
+    ctx = Context()
+    out = []
+
+    # TODO: pygit2?
+    # ref. https://github.com/cpp-linter/cpp-linter/blob/main/cpp_linter/git/__init__.py ::parse_diff
+    # ref. https://github.com/libgit2/pygit2/blob/master/src/diff.c ::parse_diff
+    for line in proc.stdout.split("\n"):
+        # '--- a/path/to/changed/file' most likely
+        # '--- a/dev/null' aka created file. should be ignored, same as removed ones
+        if line.startswith("---"):
+            ctx.pop(out, line)
+            ctx.deleted = "/dev/null" in file
+
+        # '+++ b/path/to/changed/file' most likely
+        # '+++ /dev/null' aka removed file
+        elif not ctx.deleted and line.startswith("+++"):
+            ctx.hunk.append(line)
+
+            _, file = line.split(" ")
+            ctx.deleted = "/dev/null" in file
+            if not ctx.deleted:
+                ctx.file = file[2:]
+
+        # @@ from-file-line-numbers to-file-line-numbers @@
+        elif not ctx.deleted and line.startswith("@@"):
+            ctx.hunk.append(line)
+
+            _, _, numbers, _ = line.split(" ", 3)
+            if "," in numbers:
+                numbers, _ = numbers.split(",")  # drop count
+
+            numbers = numbers.replace("+", "")
+            numbers = numbers.replace("-", "")
+
+            ctx.markers.append(int(numbers))
+            ctx.append_hunk = True
+
+        # capture diff for the summary
+        elif ctx.append_hunk and line.startswith(("+", "-", " ")):
+            ctx.hunk.append(line)
+
+    ctx.pop(out, line)
+
+    return out
+
+
+def errors_changed(changed: Changed):
+    all_lines = ", ".join(str(x) for x in changed.lines)
+    for line in changed.lines:
+        print(
+            f"::error file={changed.file},title=Run tests/restyle.sh and re-commit {changed.file},line={line}::File {changed.file} failed clang-format style check. (lines {all_lines})"
+        )
+
+
+SUMMARY_PATH = pathlib.Path(os.environ.get("GITHUB_STEP_SUMMARY", os.devnull))
+SUMMARY_OUTPUT = SUMMARY_PATH.open("a")
+
+
+def summary_diff(changed: Changed):
+    with contextlib.redirect_stdout(SUMMARY_OUTPUT):
+        print(f"# {changed.file} (suggested change)")
+        print("```diff")
+        print(changed.hunk)
+        print("```")
+
+
+def stdout_diff():
+    subprocess.run(["git", "--no-pager", "diff"])
+
+
+def assert_unchanged():
+    subprocess.run(
+        ["git", "diff", "--exit-code"], check=True, stdout=subprocess.DEVNULL
+    )
+
+
+def run_format(args):
+    targets = []
+
+    for include in args.include:
+        targets.append(
+            (GIT_ROOT / f"tests/clang-format-{include}.yaml", FILES_PRESETS[include]())
+        )
+
+    if not targets:
+        targets.append((args.config, args.files))
+
+    for target in targets:
+        clang_format(args.clang_format, *target)
+
+
+def run_assert(args):
+    for changed in changed_files():
+        if args.with_errors:
+            errors_changed(changed)
+        if args.with_summary:
+            summary_diff(changed)
+
+    if args.with_diff:
+        stdout_diff()
+
+    assert_unchanged()
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+
+    cmd = parser.add_subparsers(required=True)
+    format_ = cmd.add_parser("format")
+    format_.set_defaults(func=run_format)
+    format_.add_argument("--clang-format", default="clang-format")
+
+    fmt = format_.add_subparsers(required=True)
+
+    preset = fmt.add_parser("preset")
+    preset.add_argument(
+        "--include", action="append", required=True, choices=tuple(FILES_PRESETS.keys())
+    )
+
+    files = fmt.add_parser("files")
+    files.add_argument("--config", type=pathlib.Path, required=True)
+    files.add_argument("files", type=pathlib.Path, nargs="+")
+
+    assert_ = cmd.add_parser("assert")
+    assert_.set_defaults(func=run_assert)
+    assert_.add_argument("--with-diff", action="store_true")
+    assert_.add_argument("--with-errors", action="store_true")
+    assert_.add_argument("--with-summary", action="store_true")
+
+    args = parser.parse_args()
+    args.func(args)
diff --git a/tests/restyle.sh b/tests/restyle.sh
index ce8e906a76..6e10cab6a2 100755
--- a/tests/restyle.sh
+++ b/tests/restyle.sh
@@ -1,5 +1,5 @@
 #!/bin/sh
-# requires clang-format, git, python3 with pyyaml
+# requires python3, git, and runnable clang-format (specified below)
 
 set -e -x
 
@@ -11,51 +11,13 @@ test -d ${root}/libraries
 # default to v15, latest stable version from ubuntu-latest Github Actions image
 CLANG_FORMAT=${CLANG_FORMAT:-clang-format-15}
 
-#########################################
-# 'all' variable should be "cores/esp8266 libraries"
-
-all=${1:-"
-cores/esp8266/Lwip*
-libraries/ESP8266mDNS
-libraries/Wire
-libraries/lwIP*
-cores/esp8266/debug*
-cores/esp8266/core_esp8266_si2c.cpp
-cores/esp8266/StreamString.*
-cores/esp8266/StreamSend.*
-libraries/Netdump
-tests
-"}
-
-#########################################
-# restyling core & libraries
-
 cd $root
-
-style=${root}/tests/clang-format-core.yaml
-for target in $all; do
-    if [ -d "$target" ]; then
-        find $target \
-            '(' -name "*.cpp" -o -name "*.c" -o -name "*.h" ')' \
-            -exec $CLANG_FORMAT --verbose --style="file:$style" -i {} \;
-    else
-        $CLANG_FORMAT --verbose --style="file:$style" -i $target
-    fi
-done
-
-#########################################
-# restyling arduino examples
-
-# TODO should not be matched, these are formatted externally
-# exclude=$(git submodule --quiet foreach git rev-parse --show-toplevel | grep libraries)
-
-if [ -z $1 ] ; then
-    style=${root}/tests/clang-format-arduino.yaml
-    find libraries \
-        -path libraries/ESP8266SdFat -prune -o \
-        -path libraries/Ethernet -prune -o \
-        -path libraries/SoftwareSerial -prune -o \
-        -name '*.ino' -exec $CLANG_FORMAT --verbose --style="file:$style" -i {} \;
+python $root/tests/restyle.py format --clang-format=$CLANG_FORMAT preset --include core --include arduino
+
+if [ $CI = "true" ] ; then
+    echo foo
+    python $root/tests/restyle.py assert --with-summary --with-errors
+else
+    echo bar
+    python $root/tests/restyle.py assert --with-diff
 fi
-
-#########################################

From 1af821ba7ebb3bbe8ea2a29c707dcad8a2376b40 Mon Sep 17 00:00:00 2001
From: Maxim Prokhorov <prokhorov.max@outlook.com>
Date: Thu, 1 Aug 2024 02:13:46 +0300
Subject: [PATCH 2/2] fixup! CI - restyle script summary and annotations in PRs

---
 .github/workflows/style-check.yml  |   1 +
 tests/host/common/ArduinoCatch.hpp |   8 +-
 tests/restyle.py                   |  46 +++++---
 tests/test_restyle.py              | 182 +++++++++++++++++++++++++++++
 4 files changed, 221 insertions(+), 16 deletions(-)
 create mode 100644 tests/test_restyle.py

diff --git a/.github/workflows/style-check.yml b/.github/workflows/style-check.yml
index 50a99f2e53..263278e14e 100644
--- a/.github/workflows/style-check.yml
+++ b/.github/workflows/style-check.yml
@@ -26,6 +26,7 @@ jobs:
     - name: Style check
       run: |
           sudo apt update
+          python ./tests/test_restyle.py --quiet
           bash ./tests/ci/style_check.sh
 
   # Validate orthography
diff --git a/tests/host/common/ArduinoCatch.hpp b/tests/host/common/ArduinoCatch.hpp
index 8cb81afb7c..d30d30e4ec 100644
--- a/tests/host/common/ArduinoCatch.hpp
+++ b/tests/host/common/ArduinoCatch.hpp
@@ -24,13 +24,15 @@
 
 std::ostream& operator<<(std::ostream&, const String&);
 
-namespace Catch {
+namespace Catch
+{
 
 std::string toString(const String&);
 
 template<>
-struct StringMaker<String> {
+struct StringMaker<String>
+{
     static std::string convert(const String&);
 };
 
-} // namespace Catch
+}  // namespace Catch
diff --git a/tests/restyle.py b/tests/restyle.py
index 909f089a3c..78a8816000 100755
--- a/tests/restyle.py
+++ b/tests/restyle.py
@@ -45,6 +45,17 @@ def ls_files(patterns):
     return out
 
 
+def diff_lines():
+    proc = subprocess.run(
+        ["git", "--no-pager", "diff", "--ignore-submodules"],
+        capture_output=True,
+        check=True,
+        universal_newlines=True,
+    )
+
+    return proc.stdout.split("\n")
+
+
 def find_files(patterns):
     """Filesystem search, matches both git and non-git files"""
     return [
@@ -98,7 +109,7 @@ def find_arduino_files():
 @dataclass
 class Changed:
     file: str
-    hunk: list[str]
+    hunk: str
     lines: list[int]
 
 
@@ -126,16 +137,17 @@ def pop(self, out, line):
         self.reset_with_line(line)
 
 
-def changed_files():
+def changed_files_for_diff(lines: list[str] | str) -> list[Changed]:
     """
-    Naive git-diff output parser. Generates list[Changed] for every file changed after clang-format.
+    Naive git-diff output parser. Generates list of objects for every file changed after clang-format.
     """
-    proc = subprocess.run(
-        ["git", "--no-pager", "diff"],
-        capture_output=True,
-        check=True,
-        universal_newlines=True,
-    )
+    match lines:
+        case str():
+            lines = lines.split("\n")
+        case list():
+            pass
+        case _:
+            raise ValueError("Unknown 'lines' type, can be either list[str] or str")
 
     ctx = Context()
     out = []
@@ -143,11 +155,13 @@ def changed_files():
     # TODO: pygit2?
     # ref. https://github.com/cpp-linter/cpp-linter/blob/main/cpp_linter/git/__init__.py ::parse_diff
     # ref. https://github.com/libgit2/pygit2/blob/master/src/diff.c ::parse_diff
-    for line in proc.stdout.split("\n"):
+    for line in lines:
         # '--- a/path/to/changed/file' most likely
-        # '--- a/dev/null' aka created file. should be ignored, same as removed ones
+        # '--- /dev/null' aka created file. should be ignored, same as removed ones
         if line.startswith("---"):
             ctx.pop(out, line)
+
+            _, file = line.split(" ")
             ctx.deleted = "/dev/null" in file
 
         # '+++ b/path/to/changed/file' most likely
@@ -183,6 +197,10 @@ def changed_files():
     return out
 
 
+def changed_files() -> list[Changed]:
+    return changed_files_for_diff(diff_lines())
+
+
 def errors_changed(changed: Changed):
     all_lines = ", ".join(str(x) for x in changed.lines)
     for line in changed.lines:
@@ -204,12 +222,14 @@ def summary_diff(changed: Changed):
 
 
 def stdout_diff():
-    subprocess.run(["git", "--no-pager", "diff"])
+    subprocess.run(["git", "--no-pager", "diff", "--ignore-submodules"])
 
 
 def assert_unchanged():
     subprocess.run(
-        ["git", "diff", "--exit-code"], check=True, stdout=subprocess.DEVNULL
+        ["git", "diff", "--ignore-submodules", "--exit-code"],
+        check=True,
+        stdout=subprocess.DEVNULL,
     )
 
 
diff --git a/tests/test_restyle.py b/tests/test_restyle.py
new file mode 100644
index 0000000000..7264f1c09e
--- /dev/null
+++ b/tests/test_restyle.py
@@ -0,0 +1,182 @@
+import unittest
+
+from restyle import changed_files_for_diff
+
+# small git-diff samples from https://queirozf.com/entries/git-diff-reference-and-examples
+
+
+class BaseTest(unittest.TestCase):
+    def testNewLine(self):
+        diff = """
+diff --git a/file.txt b/file.txt
+index 257cc56..3bd1f0e 100644
+--- a/file.txt
++++ b/file.txt
+@@ -1 +1,2 @@
+ foo
++bar
+"""
+        changed = changed_files_for_diff(diff)
+        self.assertEqual(1, len(changed))
+        self.assertEqual("file.txt", changed[0].file)
+        self.assertEqual(1, len(changed[0].lines))
+        self.assertEqual(1, changed[0].lines[0])
+
+        expected = """
+--- a/file.txt
++++ b/file.txt
+@@ -1 +1,2 @@
+ foo
++bar
+""".strip()
+        self.assertEqual(expected, changed[0].hunk.strip())
+
+    def testNewLines(self):
+        diff = """
+diff --git a/file.txt b/file.txt
+index 257cc56..3bd1f0e 100644
+--- a/file2.txt
++++ b/file2.txt
+@@ -1 +1,2 @@
+ foo
++bar
+ baz
+@@ -1 +10,2 @@
+ 222
+-222
+ 333
+@@ -1 +100,3 @@
+ aaa
++bbb
++ccc
+ ddd
+"""
+        changed = changed_files_for_diff(diff)
+        self.assertEqual(1, len(changed))
+        self.assertEqual("file2.txt", changed[0].file)
+
+        lines = changed[0].lines
+        self.assertEqual(3, len(lines))
+
+        first, second, third = lines
+        self.assertEqual(1, first)
+        self.assertEqual(10, second)
+        self.assertEqual(100, third)
+
+        expected = """
+--- a/file2.txt
++++ b/file2.txt
+@@ -1 +1,2 @@
+ foo
++bar
+ baz
+@@ -1 +10,2 @@
+ 222
+-222
+ 333
+@@ -1 +100,3 @@
+ aaa
++bbb
++ccc
+ ddd
+""".strip()
+        self.assertEqual(expected, changed[0].hunk.strip())
+
+    def testRemovedLineAndDeletedFile(self):
+        diff = """
+diff --git a/file.txt b/file.txt
+index 3bd1f0e..257cc56 100644
+--- a/file.txt
++++ b/file.txt
+@@ -1,2 +1 @@
+ foo
+-bar
+diff --git a/file2.txt b/file2.txt
+deleted file mode 100644
+index 85553e8..0000000
+--- a/file2.txt
++++ /dev/null
+@@ -1,2 +0,0 @@
+-aaaaaa
+-bbbbbb
+"""
+        changed = changed_files_for_diff(diff)
+        self.assertEqual(1, len(changed))
+        self.assertEqual("file.txt", changed[0].file)
+        self.assertEqual(1, len(changed[0].lines))
+        self.assertEqual(1, changed[0].lines[0])
+
+        expected = """
+--- a/file.txt
++++ b/file.txt
+@@ -1,2 +1 @@
+ foo
+-bar
+""".strip()
+        self.assertEqual(expected, changed[0].hunk.strip())
+
+    def testNewLineAndDeletedFile(self):
+        diff = """
+diff --git a/file.txt b/file.txt
+index 3bd1f0e..86e041d 100644
+--- a/file.txt
++++ b/file.txt
+@@ -1,2 +1,3 @@
+ foo
+ bar
++baz
+diff --git a/file2.txt b/file2.txt
+deleted file mode 100644
+index 85553e8..0000000
+--- a/file2.txt
++++ /dev/null
+@@ -1,2 +0,0 @@
+-aaaaaa
+-bbbbbb
+"""
+        changed = changed_files_for_diff(diff)
+        self.assertEqual(1, len(changed))
+        self.assertEqual("file.txt", changed[0].file)
+        self.assertEqual(1, len(changed[0].lines))
+        self.assertEqual(1, changed[0].lines[0])
+
+        expected = """
+--- a/file.txt
++++ b/file.txt
+@@ -1,2 +1,3 @@
+ foo
+ bar
++baz
+""".strip()
+        self.assertEqual(expected, changed[0].hunk.strip())
+
+    def testDeletedFile(self):
+        diff = """
+diff --git a/file2.txt b/file2.txt
+deleted file mode 100644
+index 85553e8..0000000
+--- a/file2.txt
++++ /dev/null
+@@ -1,2 +0,0 @@
+-aaaaaa
+-bbbbbb
+"""
+        changed = changed_files_for_diff(diff)
+        self.assertEqual(0, len(changed))
+
+    def testNewFile(self):
+        diff = """
+diff --git a/file3.txt b/file3.txt
+new file mode 100644
+index 0000000..a309e46
+--- /dev/null
++++ b/file3.txt
+@@ -0,0 +1 @@
++this is file3
+"""
+        changed = changed_files_for_diff(diff)
+        self.assertEqual(0, len(changed))
+
+
+if __name__ == '__main__':
+    unittest.main()