✨ feat(version): update cargo toml version in package section only

Author: Jannchie

tests/unit/test_version.py

@@ -24,6 +24,7 @@
     get_version_from_version_file,
     get_version_from_version_txt,
     show_file_diff,
+    update_cargo_toml_version,
 )
 
 
@@ -738,3 +739,86 @@ def test_show_file_diff_user_cancels(self, mock_confirm):
         with pytest.raises(SystemExit):
             show_file_diff(old_content, new_content, "test.txt")
         mock_confirm.assert_called_once_with("Do you want to continue?", default=True)
+
+
+class TestCargoTomlVersionUpdate:
+    """Test cases for Cargo.toml version updating."""
+
+    def test_update_cargo_toml_version_package_section_only(self, tmp_path):
+        """Test that update_cargo_toml_version only updates version in [package] section."""
+        cargo_toml = tmp_path / "Cargo.toml"
+        cargo_toml_content = """[package]
+name = "test-package"
+version = "1.0.0"
+authors = ["Test Author"]
+
+[dependencies]
+serde = { version = "1.0.0", features = ["derive"] }
+
+[dev-dependencies]
+tokio = { version = "1.0.0", features = ["full"] }
+
+[workspace]
+members = ["subcrate"]
+
+# Some other version reference that should NOT be changed
+# version = "should-not-change"
+"""
+        cargo_toml.write_text(cargo_toml_content)
+
+        # Update version
+        update_cargo_toml_version(str(cargo_toml), "2.0.0", 0, show_diff=False)
+
+        # Read updated content
+        updated_content = cargo_toml.read_text()
+
+        # Verify only the package version was updated
+        assert 'version = "2.0.0"' in updated_content
+        assert 'serde = { version = "1.0.0"' in updated_content  # Dependency version unchanged
+        assert 'tokio = { version = "1.0.0"' in updated_content  # Dev dependency version unchanged
+        assert '# version = "should-not-change"' in updated_content  # Comment unchanged
+
+        # Count occurrences to ensure only one version was changed
+        assert updated_content.count('version = "2.0.0"') == 1
+        assert updated_content.count('version = "1.0.0"') == 2  # The two dependency versions remain
+
+    def test_update_cargo_toml_version_complex_package_section(self, tmp_path):
+        """Test updating version in a more complex [package] section."""
+        cargo_toml = tmp_path / "Cargo.toml"
+        cargo_toml_content = """[package]
+name = "complex-package"
+version = "0.1.0"
+edition = "2021"
+authors = ["Author One", "Author Two"]
+license = "MIT"
+description = "A test package"
+repository = "https://github.com/test/test"
+
+[lib]
+name = "complex_package"
+path = "src/lib.rs"
+
+[dependencies]
+log = { version = "0.4.0" }
+
+[workspace]
+members = ["other-crate"]
+"""
+        cargo_toml.write_text(cargo_toml_content)
+
+        update_cargo_toml_version(str(cargo_toml), "0.2.0", 0, show_diff=False)
+
+        updated_content = cargo_toml.read_text()
+
+        # Verify correct update
+        assert 'version = "0.2.0"' in updated_content
+        assert 'log = { version = "0.4.0" }' in updated_content  # Dependency unchanged
+        assert updated_content.count('version = "0.2.0"') == 1
+        assert updated_content.count('version = "0.4.0"') == 1
+
+    def test_update_cargo_toml_version_file_not_exists(self, tmp_path):
+        """Test that function handles non-existent file gracefully."""
+        non_existent_file = str(tmp_path / "nonexistent.toml")
+        
+        # Should not raise an error
+        update_cargo_toml_version(non_existent_file, "1.0.0", 0, show_diff=False)

tgit/version.py

@@ -555,7 +555,7 @@ def update_version_in_file(verbose: int, next_version_str: str, file: str, file_
     elif file == "setup.py":
         update_file(str(file_path), r"version=['\"].*?['\"]", f"version='{next_version_str}'", verbose, show_diff=show_diff)
     elif file == "Cargo.toml":
-        update_file(str(file_path), r'version\s*=\s*".*?"', f'version = "{next_version_str}"', verbose, show_diff=show_diff)
+        update_cargo_toml_version(str(file_path), next_version_str, verbose, show_diff=show_diff)
     elif file in ("VERSION", "VERSION.txt"):
         update_file(str(file_path), None, next_version_str, verbose, show_diff=show_diff)
 
@@ -564,7 +564,7 @@ def update_file_in_root(next_version_str: str, verbose: int, root_path: Path, *,
     update_file(str(root_path / "package.json"), r'"version":\s*".*?"', f'"version": "{next_version_str}"', verbose, show_diff=show_diff)
     update_file(str(root_path / "pyproject.toml"), r'version\s*=\s*".*?"', f'version = "{next_version_str}"', verbose, show_diff=show_diff)
     update_file(str(root_path / "setup.py"), r"version=['\"].*?['\"]", f"version='{next_version_str}'", verbose, show_diff=show_diff)
-    update_file(str(root_path / "Cargo.toml"), r'(?m)^version\s*=\s*".*?"', f'version = "{next_version_str}"', verbose, show_diff=show_diff)
+    update_cargo_toml_version(str(root_path / "Cargo.toml"), next_version_str, verbose, show_diff=show_diff)
     update_file(
         str(root_path / "build.gradle.kts"),
         r'version\s*=\s*".*?"',
@@ -591,6 +591,37 @@ def update_file(filename: str, search_pattern: str | None, replace_text: str, ve
         f.write(new_content)
 
 
+def update_cargo_toml_version(filename: str, next_version_str: str, verbose: int, *, show_diff: bool = True) -> None:
+    """Update version in Cargo.toml, specifically in the [package] section only."""
+    file_path = Path(filename)
+    if not file_path.exists():
+        return
+    if verbose > 0:
+        console.print(f"Updating {file_path}")
+
+    with file_path.open() as f:
+        content = f.read()
+
+    # Use regex to match version in [package] section only
+    # This pattern matches:
+    # 1. [package] section header
+    # 2. Any content until version = "..."
+    # 3. Captures the version line to replace
+    pattern = r'(\[package\].*?)(version\s*=\s*"[^"]*")'
+
+    def replace_version(match: re.Match[str]) -> str:
+        package_section = match.group(1)
+        return f'{package_section}version = "{next_version_str}"'
+
+    new_content = re.sub(pattern, replace_version, content, flags=re.DOTALL)
+
+    if show_diff:
+        show_file_diff(content, new_content, str(file_path))
+
+    with file_path.open("w") as f:
+        f.write(new_content)
+
+
 def show_file_diff(old_content: str, new_content: str, filename: str) -> None:
     old_lines = old_content.splitlines()
     new_lines = new_content.splitlines()