From 3dc029968b2c3fa06acdafd373237ddb31a1a4f4 Mon Sep 17 00:00:00 2001
From: Donkie <daniel.cf.hultgren@gmail.com>
Date: Mon, 10 Jul 2023 01:45:21 +0200
Subject: [PATCH] Added a version bumping script

---
 pyproject.toml   |  3 +++
 spoolman/bump.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 spoolman/bump.py

diff --git a/pyproject.toml b/pyproject.toml
index a09a746..4a07199 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,6 +38,9 @@ dev = [
 [tool.pdm.scripts.docs]
 call = "spoolman.docs:generate_docs"
 
+[tool.pdm.scripts.bump]
+call = "spoolman.bump:bump"
+
 [tool.ruff]
 select = [
     "ALL",
diff --git a/spoolman/bump.py b/spoolman/bump.py
new file mode 100644
index 0000000..7af580a
--- /dev/null
+++ b/spoolman/bump.py
@@ -0,0 +1,52 @@
+"""A python script that bumps the version number of a project."""
+
+# ruff: noqa: PLR2004, T201, S603, S607
+
+import re
+import subprocess
+import sys
+from pathlib import Path
+
+
+def bump() -> None:
+    """Bump the version number of the project."""
+    project_root = Path(__file__).parent.parent
+
+    if len(sys.argv) < 2:
+        print("Please specify a bump type, e.g. major, minor, micro.")
+        sys.exit(1)
+
+    if subprocess.run(["git", "diff", "--quiet", "pyproject.toml"], cwd=project_root).returncode != 0:
+        print("The pyproject.toml file is dirty, please commit your changes before bumping the version number.")
+        sys.exit(1)
+
+    if subprocess.run(["git", "diff", "--cached", "--quiet"], cwd=project_root).returncode != 0:
+        print("There are staged changes, please commit them before bumping the version number.")
+        sys.exit(1)
+
+    if subprocess.run(["pip", "show", "pdm-bump"], cwd=project_root, capture_output=True).returncode != 0:
+        print("Please install pdm-bump using pip.")
+        sys.exit(1)
+
+    # Bump the version number, read the pdm bump output to determine the new version number
+    bump_type = sys.argv[1]
+    bump_output = subprocess.run(["pdm", "bump", bump_type], cwd=project_root, capture_output=True, check=True)
+    # Example output: "Performing increment of version: 0.7.0 -> 0.8.0\nSome more text"
+    # Parse using regex
+    new_version_match = re.search(r"-> ([A-Za-z0-9\.\-]+)", bump_output.stdout.decode())
+    if new_version_match is None:
+        print("Failed to parse pdm bump output, did it fail?")
+        sys.exit(1)
+    new_version = new_version_match.group(1)
+
+    # Stage the pyproject.toml file
+    subprocess.run(["git", "add", "pyproject.toml"], cwd=project_root, check=True)
+
+    # Commit the changes
+    subprocess.run(["git", "commit", "-m", f"Bump version to {new_version}"], cwd=project_root, check=True)
+
+    # Tag the commit, prefix with "v"
+    subprocess.run(["git", "tag", f"v{new_version}"], cwd=project_root, check=True)
+
+    # Notify user that the process is complete
+    print(f"Bumped version to {new_version}.")