python · JelleZijlstra · Mar 12, 2024 · Feb 7, 2024 · Feb 8, 2024 · Feb 8, 2024
diff --git a/pyrightconfig.stricter.json b/pyrightconfig.stricter.json
@@ -62,6 +62,7 @@
         "stubs/psycopg2",
         "stubs/pyasn1",
         "stubs/pyflakes",
+        "stubs/pygit2",
         "stubs/Pygments",
         "stubs/PyMySQL",
         "stubs/python-dateutil",

diff --git a/stubs/pygit2/@tests/stubtest_allowlist.txt b/stubs/pygit2/@tests/stubtest_allowlist.txt
@@ -0,0 +1,17 @@
+# stubtest wrongly checks the Python implementation with the cffi type's
+# stubs, likely due to name shadowing that happens at import-time
+pygit2.Repository.__init__
+pygit2.Repository._from_c
+# @cached_property that fail to get recognized as such by stubtest
+pygit2.config.ConfigEntry.level
+pygit2.config.ConfigEntry.raw_name
+pygit2.config.ConfigEntry.raw_value
+# Actual signatures enforced by C-level checks not visible to stubtest
+pygit2.Mailmap.__init__
+pygit2.OdbBackend.__init__
+pygit2.RefLogEntry.__init__
+pygit2.RefdbBackend.__init__
+pygit2._pygit2.Mailmap.__init__
+pygit2._pygit2.OdbBackend.__init__
+pygit2._pygit2.RefLogEntry.__init__
+pygit2._pygit2.RefdbBackend.__init__
diff --git a/stubs/pygit2/METADATA.toml b/stubs/pygit2/METADATA.toml
@@ -0,0 +1,6 @@
+version = "1.14.*"
+upstream_repository = "https://github.com/libgit2/pygit2"
+requires = ["types-cffi"]
+
+[tool.stubtest]
+platforms = ["darwin", "linux", "win32"]
diff --git a/stubs/pygit2/pygit2/__init__.pyi b/stubs/pygit2/pygit2/__init__.pyi
@@ -0,0 +1,54 @@
+from _typeshed import StrOrBytesPath, SupportsAllComparisons
+from collections.abc import Callable
+
+from . import enums
+from ._build import __version__ as __version__
+from ._pygit2 import *
+from .blame import Blame as Blame, BlameHunk as BlameHunk
+from .blob import BlobIO as BlobIO
+from .callbacks import (
+    CheckoutCallbacks as CheckoutCallbacks,
+    Payload as Payload,
+    RemoteCallbacks,
+    StashApplyCallbacks as StashApplyCallbacks,
+    get_credentials as get_credentials,
+)
+from .config import Config as Config
+from .credentials import *
+from .errors import Passthrough as Passthrough
+from .filter import Filter as Filter
+from .index import Index as Index, IndexEntry as IndexEntry
+from .legacyenums import *
+from .packbuilder import PackBuilder as PackBuilder
+from .remotes import Remote as Remote
+from .repository import Repository  # type: ignore[assignment]
+from .settings import Settings
+from .submodules import Submodule as Submodule
+
+features: enums.Feature
+LIBGIT2_VER: tuple[int, int, int]
+
+def init_repository(
+    path: StrOrBytesPath | None,
+    bare: bool = False,
+    flags: enums.RepositoryInitFlag = ...,
+    mode: int | enums.RepositoryInitMode = ...,
+    workdir_path: str | None = None,
+    description: str | None = None,
+    template_path: str | None = None,
+    initial_head: str | None = None,
+    origin_url: str | None = None,
+) -> Repository: ...
+def clone_repository(
+    url: str,
+    path: str,
+    bare: bool = False,
+    repository: Callable[[str, bool], Repository] | None = None,
+    remote: Callable[[Repository, str, str], Remote] | None = None,
+    checkout_branch: str | None = None,
+    callbacks: RemoteCallbacks | None = None,
+    depth: int = 0,
+) -> Repository: ...
+
+tree_entry_key: Callable[[Object], SupportsAllComparisons]  # functools.cmp_to_key(tree_entry_cmp)
+settings: Settings
diff --git a/stubs/pygit2/pygit2/_build.pyi b/stubs/pygit2/pygit2/_build.pyi
@@ -0,0 +1,5 @@
+from pathlib import Path
+
+__version__: str
+
+def get_libgit2_paths() -> tuple[Path, dict[str, list[str]]]: ...
diff --git a/stubs/pygit2/pygit2/_libgit2.pyi b/stubs/pygit2/pygit2/_libgit2.pyi
@@ -0,0 +1,4 @@
+from _cffi_backend import FFI, Lib
+
+ffi: FFI
+lib: Lib