Add PR to bump version after finalizing release

Author: rvermeulen

.github/workflows/finalize-release.yml

@@ -56,4 +56,29 @@ jobs:
           version=${BASE_REF#rc/}
           echo "Finalizing release v$version"
 
-          gh release edit "v$version" --draft=false --tag=v$version
+          gh release edit "v$version" --draft=false --tag=v$version
+
+      - name: Determine if release was a hotfix release
+        run: |
+          version=${BASE_REF#rc/}
+          echo "HOTFIX_RELEASE=$(python scripts/release/is-hotfix.py $version)" >> "$GITHUB_ENV"
+
+      - name: Bump main version
+        if: env.HOTFIX_RELEASE == 'false'
+        run: |
+          version=${BASE_REF#rc/}
+          next_version="$version-dev"
+          echo "Bumping main version to $next_version"
+
+          git switch main
+          git pull --ff-only origin main
+
+          git switch -c release-automation/bump-version
+
+          python scripts/bump_version.sh "$next_version"
+
+          git add -u .
+          git commit -m "Bump version to $next_version"
+          git push --set-upstream origin release-automation/bump-version
+
+          gh pr create --repo $GITHUB_REPOSITORY --base main --head release-automation/bump-version --body "Bump the version of main to the dev label of the just released version $next_version" --title "Bump version to $next_version"

scripts/release/is-hotfix-release.py

@@ -0,0 +1,56 @@
+from semantic_version import Version # type: ignore
+from subprocess import run
+from typing import List, Literal, TYPE_CHECKING
+from sys import stderr
+
+if TYPE_CHECKING:
+    from argparse import Namespace
+
+def get_merge_base_of_ref() -> str:
+    cp = run(["git", "merge-base", "HEAD", "origin/main"], capture_output=True, text=True)
+    if cp.returncode != 0:
+        raise RuntimeError("Failed to get merge base")
+    return cp.stdout.strip()
+
+def get_release_branches_containing(commit: str) -> List[Version]:
+    cp = run(["git", "branch", "--list", "rc/*", "--contains", commit], capture_output=True, text=True)
+    if cp.returncode != 0:
+        raise RuntimeError("Failed to get branches containing commit")
+    release_versions: List[Version] = []
+    for version in [b.strip() for b in cp.stdout.splitlines()]:
+        try:
+            if version.startswith("rc/"):
+                version = version[3:]
+                release_versions.append(Version(version))
+        except ValueError:
+            print(f"Warning: Skipping invalid version string: {version}", file=stderr)
+
+    return release_versions
+
+def main(args: 'Namespace') -> Literal[0,1]:
+    try:
+        merge_base = get_merge_base_of_ref()
+        release_versions = get_release_branches_containing(merge_base)
+        if len(release_versions) == 0:
+            print(f"Info: No release branches found containing merge base {merge_base}", file=stderr)
+            print("false")
+            return 0
+
+        for version in release_versions:
+            if version.next_patch() == Version(args.version):
+                print("true")
+                return 0
+            
+        print("false")
+        return 0
+    except RuntimeError as e:
+        print(f"Error: {e}", file=stderr)
+        return 1
+    
+if __name__ == '__main__':
+    from sys import stderr, exit
+    import argparse
+
+    parser = argparse.ArgumentParser(description="Check if a version is a hotfix release")
+    parser.add_argument("version", help="The version string to compare against the base branches")
+    exit(main(parser.parse_args()))