Add FilterList.apply_to_buffer, apply_to_file, apply_to_blob

Author: jorio

pygit2/decl/filter.h

@@ -26,6 +26,23 @@ int git_filter_list_contains(
 	git_filter_list *filters,
 	const char *name);
 
+int git_filter_list_apply_to_buffer(
+	git_buf *out,
+	git_filter_list *filters,
+	const char* in,
+	size_t in_len);
+
+int git_filter_list_apply_to_file(
+	git_buf *out,
+	git_filter_list *filters,
+	git_repository *repo,
+	const char *path);
+
+int git_filter_list_apply_to_blob(
+	git_buf *out,
+	git_filter_list *filters,
+	git_blob *blob);
+
 size_t git_filter_list_length(
 	const git_filter_list *fl);
 

pygit2/filter.py

@@ -26,7 +26,7 @@
 import weakref
 from collections.abc import Callable
 
-from ._pygit2 import FilterSource, Repository
+from ._pygit2 import Blob, FilterSource, Repository
 from .errors import check_error
 from .ffi import C, ffi
 from .utils import to_bytes
@@ -146,5 +146,49 @@ def __contains__(self, name: str) -> bool:
     def __len__(self) -> int:
         return C.git_filter_list_length(self._pointer)
 
+    def apply_to_buffer(self, data: bytes) -> bytes:
+        """
+        Apply a filter list to a data buffer.
+        Return the filtered contents.
+        """
+        buf = ffi.new('git_buf *')
+        err = C.git_filter_list_apply_to_buffer(buf, self._pointer, data, len(data))
+        check_error(err)
+        try:
+            return ffi.string(buf.ptr)
+        finally:
+            C.git_buf_dispose(buf)
+
+    def apply_to_file(self, repo: Repository, path: str) -> bytes:
+        """
+        Apply a filter list to the contents of a file on disk.
+        Return the filtered contents.
+        """
+        buf = ffi.new('git_buf *')
+        c_path = to_bytes(path)
+        err = C.git_filter_list_apply_to_file(buf, self._pointer, repo._repo, c_path)
+        check_error(err)
+        try:
+            return ffi.string(buf.ptr)
+        finally:
+            C.git_buf_dispose(buf)
+
+    def apply_to_blob(self, blob: Blob) -> bytes:
+        """
+        Apply a filter list to a data buffer.
+        Return the filtered contents.
+        """
+        buf = ffi.new('git_buf *')
+
+        c_blob = ffi.new('git_blob **')
+        ffi.buffer(c_blob)[:] = blob._pointer[:]
+
+        err = C.git_filter_list_apply_to_blob(buf, self._pointer, c_blob[0])
+        check_error(err)
+        try:
+            return ffi.string(buf.ptr)
+        finally:
+            C.git_buf_dispose(buf)
+
     def __del__(self):
         C.git_filter_list_free(self._pointer)

test/test_filter.py

@@ -6,7 +6,7 @@
 
 import pygit2
 from pygit2 import Blob, Filter, FilterSource, Repository
-from pygit2.enums import BlobFilter
+from pygit2.enums import BlobFilter, FilterMode
 from pygit2.errors import Passthrough
 
 
@@ -155,6 +155,20 @@ def test_filterlist_crlf(testrepo: Repository) -> None:
         1234 in fl
 
 
+def test_filterlist_crlf_clean(testrepo: Repository) -> None:
+    testrepo.config['core.autocrlf'] = True
+    fl = testrepo.load_filter_list('whatever.txt', mode=FilterMode.CLEAN)
+    filtered = fl.apply_to_buffer(b'hello\r\nworld\r\n')
+    assert filtered == b'hello\nworld\n'
+
+
+def test_filterlist_crlf_smudge(testrepo: Repository) -> None:
+    testrepo.config['core.autocrlf'] = True
+    fl = testrepo.load_filter_list('whatever.txt', mode=FilterMode.SMUDGE)
+    filtered = fl.apply_to_buffer(b'hello\nworld\n')
+    assert filtered == b'hello\r\nworld\r\n'
+
+
 def test_filterlist_rot13(testrepo: Repository, rot13_filter: Filter) -> None:
     fl = testrepo.load_filter_list('hello.txt')
     assert fl is not None
@@ -177,3 +191,29 @@ def test_filterlist_rot13_dangerous_unregister(testrepo: Repository) -> None:
     # to unregister the filter.
     del fl
     pygit2.filter_unregister('rot13')
+
+
+def test_filterlist_rot13_apply_to_buffer(
+    testrepo: Repository, rot13_filter: Filter
+) -> None:
+    fl = testrepo.load_filter_list('whatever.txt')
+    filtered = fl.apply_to_buffer(b'bye world\n')
+    assert filtered == b'olr jbeyq\n'
+
+
+def test_filterlist_rot13_apply_to_file(
+    testrepo: Repository, rot13_filter: Filter
+) -> None:
+    fl = testrepo.load_filter_list('bye.txt')
+    filtered = fl.apply_to_file(testrepo, 'bye.txt')
+    assert filtered == b'olr jbeyq\n'
+
+
+def test_filterlist_rot13_apply_to_blob(
+    testrepo: Repository, rot13_filter: Filter
+) -> None:
+    fl = testrepo.load_filter_list('whatever.txt')
+    blob_oid = testrepo.create_blob(b'bye world\n')
+    blob = testrepo[blob_oid]
+    filtered = fl.apply_to_blob(blob)
+    assert filtered == b'olr jbeyq\n'