Support file substitutions

src/rooster/_cli.py

@@ -6,7 +6,7 @@
     Changelog,
     VersionSection,
 )
-from rooster._config import Config
+from rooster._config import Config, SubstitutionEntry
 from rooster._git import (
     GitLookupError,
     get_commit_for_tag,
@@ -23,6 +23,7 @@
     Version,
     bump_version,
     get_latest_version,
+    process_substitutions,
     update_version_file,
     versions_from_git_tags,
 )
@@ -286,10 +287,15 @@ def release(
     typer.echo("Updated changelog")
 
     if update_version_files:
+        old_version = last_version or Version("0.0.0")
         for version_file in config.version_files:
-            update_version_file(
-                version_file, last_version or Version("0.0.0"), new_version
-            )
+            if isinstance(version_file, SubstitutionEntry):
+                for match in directory.glob(version_file.target):
+                    process_substitutions(
+                        match, old_version, new_version, version_file.replace
+                    )
+            else:
+                update_version_file(version_file, old_version, new_version)
             typer.echo(f"Updated version in {version_file}")
 
 

src/rooster/_config.py

@@ -47,7 +47,9 @@ class Config(pydantic.BaseModel):
     trim_title_prefixes: frozenset[str] = frozenset()
 
     # Paths to files to replace versions at
-    version_files: list[Path | VersionFile] = [Path("pyproject.toml")]
+    version_files: list[Path | VersionFile | SubstitutionEntry] = [
+        Path("pyproject.toml")
+    ]
 
     # The default version bump to use
     default_bump_type: BumpType = BumpType.patch
@@ -108,6 +110,14 @@ def __str__(self):
         return str(self.path)
 
 
+class SubstitutionEntry(pydantic.BaseModel):
+    target: str
+    replace: str | None = None
+
+    def __str__(self):
+        return str(self.target)
+
+
 class SubmoduleLabels(pydantic.BaseModel):
     submodule: str
     labels: frozenset[str] = frozenset()

src/rooster/_versions.py

@@ -170,6 +170,26 @@ def update_version_file(
         )
 
 
+def process_substitutions(
+    match: Path,
+    old_version: Version,
+    new_version: Version,
+    replacement: str | None = None,
+) -> None:
+    """Process substitutions in a matched file.
+
+    When the replacement is None it will replace instances
+    of the last version with the new version.
+    """
+    new_version_str = str(new_version)
+    contents = match.read_text()
+    if replacement is None:
+        contents = contents.replace(str(old_version), new_version_str)
+    else:
+        contents = contents.replace(replacement, new_version_str)
+    match.write_text(contents)
+
+
 def update_text_version(path: Path, old_version: str, new_version: str) -> None:
     """
     Update the version in a basic text file.