Add FilterList, Repository.load_filter_list

Author: jorio

pygit2/_run.py

@@ -77,6 +77,7 @@
     'net.h',
     'refspec.h',
     'repository.h',
+    'filter.h',
     'commit.h',
     'revert.h',
     'stash.h',

pygit2/decl/filter.h

@@ -0,0 +1,26 @@
+typedef enum {
+	GIT_FILTER_TO_WORKTREE = ...,
+	GIT_FILTER_TO_ODB = ...,
+} git_filter_mode_t;
+
+typedef enum {
+	GIT_FILTER_DEFAULT = ...,
+	GIT_FILTER_ALLOW_UNSAFE = ...,
+	GIT_FILTER_NO_SYSTEM_ATTRIBUTES = ...,
+	GIT_FILTER_ATTRIBUTES_FROM_HEAD = ...,
+	GIT_FILTER_ATTRIBUTES_FROM_COMMIT = ...,
+} git_filter_flag_t;
+
+int git_filter_list_load(
+	git_filter_list **filters,
+	git_repository *repo,
+	git_blob *blob,
+	const char *path,
+	git_filter_mode_t mode,
+	uint32_t flags);
+
+int git_filter_list_contains(
+	git_filter_list *filters,
+	const char *name);
+
+void git_filter_list_free(git_filter_list *filters);

pygit2/decl/types.h

@@ -1,6 +1,8 @@
+typedef struct git_blob git_blob;
 typedef struct git_commit git_commit;
 typedef struct git_annotated_commit git_annotated_commit;
 typedef struct git_config git_config;
+typedef struct git_filter_list git_filter_list;
 typedef struct git_index git_index;
 typedef struct git_index_conflict_iterator git_index_conflict_iterator;
 typedef struct git_object git_object;

pygit2/filter.py

@@ -25,7 +25,10 @@
 
 from collections.abc import Callable
 
-from ._pygit2 import FilterSource
+from ._pygit2 import FilterSource, Repository
+from .errors import check_error
+from .ffi import C, ffi
+from .utils import to_bytes
 
 
 class Filter:
@@ -107,3 +110,23 @@ def close(self, write_next: Callable[[bytes], None]) -> None:
                 Any remaining filtered output data must be written to
                 `write_next` before returning.
         """
+
+
+class FilterList:
+    @classmethod
+    def _from_c(cls, ptr):
+        if ptr == ffi.NULL:
+            return None
+        fl = cls.__new__(cls)
+        fl._pointer = ptr
+        return fl
+
+    def __contains__(self, name: str) -> bool:
+        if not isinstance(name, str):
+            raise TypeError('argument must be str')
+        c_name = to_bytes(name)
+        result = C.git_filter_list_contains(self._pointer, c_name)
+        return bool(result)
+
+    def __del__(self):
+        C.git_filter_list_free(self._pointer)

pygit2/repository.py

@@ -64,6 +64,7 @@
     DescribeStrategy,
     DiffOption,
     FileMode,
+    FilterMode,
     MergeFavor,
     MergeFileFlag,
     MergeFlag,
@@ -73,6 +74,7 @@
 )
 from .errors import check_error
 from .ffi import C, ffi
+from .filter import FilterList
 from .index import Index, IndexEntry, MergeFileResult
 from .packbuilder import PackBuilder
 from .references import References
@@ -235,6 +237,30 @@ def hashfile(
         oid = Oid(raw=bytes(ffi.buffer(c_oid.id)[:]))
         return oid
 
+    def load_filter_list(
+        self, path: str, mode: FilterMode = FilterMode.TO_ODB
+    ) -> FilterList | None:
+        """
+        Load the filter list for a given path.
+
+        Parameters:
+
+        path
+            Relative path of the file to be filtered
+
+        mode
+            Filtering direction: ODB to worktree (SMUDGE), or worktree to ODB
+            (CLEAN).
+        """
+        c_filters = ffi.new('git_filter_list **')
+        c_path = to_bytes(path)
+        c_mode = int(mode)
+
+        err = C.git_filter_list_load(c_filters, self._repo, ffi.NULL, c_path, c_mode, 0)
+        check_error(err)
+        fl = FilterList._from_c(c_filters[0])
+        return fl
+
     def __iter__(self) -> Iterator[Oid]:
         return iter(self.odb)
 

test/test_filter.py

@@ -136,3 +136,19 @@ def test_filter_cleanup(dirtyrepo: Repository, rot13_filter: Filter) -> None:
     # Indirectly test that pygit2_filter_cleanup has the GIL
     # before calling pygit2_filter_payload_free.
     dirtyrepo.diff()
+
+
+def test_filterlist_none(testrepo: Repository) -> None:
+    fl = testrepo.load_filter_list('hello.txt')
+    assert fl is None
+
+
+def test_filterlist_crlf(testrepo: Repository) -> None:
+    testrepo.config['core.autocrlf'] = True
+
+    fl = testrepo.load_filter_list('hello.txt')
+    assert fl is not None
+    assert 'crlf' in fl
+
+    with pytest.raises(TypeError):
+        1234 in fl