fix(precommit): handle filenames with spaces in Commit.from_merge

Author: gg-jonathangriffe

ggshield/core/scan/commit_utils.py

@@ -369,7 +369,7 @@ def get_file_sha_in_ref(
     """
     output = git(["ls-tree", "-z", ref] + files, cwd=cwd)
     for line in output.split("\0")[:-1]:
-        _, _, sha, path = line.split()
+        _, _, sha, path = line.split(maxsplit=3)
         yield (path, sha)
 
 
@@ -381,7 +381,7 @@ def get_file_sha_stage(
     """
     output = git(["ls-files", "--stage", "-z"] + files, cwd=cwd)
     for line in output.split("\0")[:-1]:
-        _, sha, _, path = line.split()
+        _, sha, _, path = line.split(maxsplit=3)
         yield (path, sha)
 
 

tests/unit/core/scan/test_commit.py

@@ -504,3 +504,41 @@ def test_from_merge(tmp_path):
         (Path("conflict.md"), Filemode.MODIFY),
         (Path("new.md"), Filemode.NEW),
     ]
+
+
+def test_from_merge_filename_with_spaces(tmp_path):
+    """
+    GIVEN two commits on different branches with a conflict
+    involving a filename with spaces
+    WHEN Commit.from_merge() is called
+    THEN it returns successfully
+    AND get_files returns the correct filename
+    """
+    repo = Repository.create(tmp_path, initial_branch="master")
+
+    Path(tmp_path / "inital.md").write_text("Initial")
+    repo.add(".")
+    repo.create_commit("Initial commit on master")
+
+    repo.create_branch("feature_branch")
+    repo.checkout("master")
+    conflict_file = tmp_path / "file with spaces.md"
+    conflict_file.write_text("Hello")
+    repo.add(".")
+    repo.create_commit("Commit on master")
+
+    repo.checkout("feature_branch")
+    conflict_file.write_text("World")
+    repo.add(".")
+    repo.create_commit("Commit on feature_branch")
+
+    # Create merge commit with conflict
+    with pytest.raises(subprocess.CalledProcessError):
+        repo.git("merge", "master")
+
+    conflict_file.write_text("Hello World !")
+    repo.add(".")
+    commit = Commit.from_merge(cwd=tmp_path)
+    files = list(commit.get_files())
+    assert len(files) == 1
+    assert files[0].path == Path("file with spaces.md")