From a36fb83452726feb9740d05efb5261c5c262f6d5 Mon Sep 17 00:00:00 2001
From: matejcik <ja@matejcik.cz>
Date: Wed, 6 Sep 2023 11:02:16 +0200
Subject: [PATCH] fix(tools): more general support for version-bumping that
 also includes boardloader

---
 tools/bump-version.py | 23 ++++++++---------------
 1 file changed, 8 insertions(+), 15 deletions(-)

diff --git a/tools/bump-version.py b/tools/bump-version.py
index 7a78e1cc7c..f045dc76aa 100755
--- a/tools/bump-version.py
+++ b/tools/bump-version.py
@@ -64,7 +64,7 @@ def cli(project, version):
     major, minor, patch = m.group(1, 2, 3)
 
     parts = project.parts
-    if parts[-3:] == ("core", "embed", "bootloader"):
+    if (project / "version.h").is_file():
         bump_header(
             project / "version.h",
             VERSION_MAJOR=major,
@@ -78,24 +78,17 @@ def cli(project, version):
             VERSION_MINOR=minor,
             VERSION_PATCH=patch,
         )
+    elif parts[-1] == "legacy":
+        bump_header(
+            project / "firmware" / "version.h",
+            VERSION_MAJOR=major,
+            VERSION_MINOR=minor,
+            VERSION_PATCH=patch,
+        )
     elif parts[-1] == "python":
         bump_python(
             project / "src" / "trezorlib" / "__init__.py", f"{major}.{minor}.{patch}"
         )
-    elif parts[-2:] == ("legacy", "firmware"):
-        bump_header(
-            project / "version.h",
-            VERSION_MAJOR=major,
-            VERSION_MINOR=minor,
-            VERSION_PATCH=patch,
-        )
-    elif parts[-2:] == ("legacy", "bootloader"):
-        bump_header(
-            project / "version.h",
-            VERSION_MAJOR=major,
-            VERSION_MINOR=minor,
-            VERSION_PATCH=patch,
-        )
     else:
         raise click.ClickException(f"Unknown project {project}.")