feat: compile the callgrind wrapper at build time

Author: art049

.gitignore

@@ -122,6 +122,7 @@ celerybeat.pid
 # Environments
 .env
 .venv
+.venv.*
 env/
 venv/
 ENV/

pyproject.toml

@@ -1,7 +1,3 @@
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
 [project.urls]
 Homepage = "https://codspeed.io/"
 Documentation = "https://docs.codspeed.io/"
@@ -35,11 +31,7 @@ classifiers = [
 ]
 dependencies = [
     "cffi >= 1.17.1",
-    # cffi doesn't automatically install setuptools with python 3.12+
-    # cf https://github.com/python-cffi/cffi/releases/tag/v1.16.0
-    "setuptools; python_full_version >= '3.12.0'",
     "pytest>=3.8",
-    "filelock >= 3.12.2",
     "rich>=13.8.1",
     "importlib-metadata>=8.5.0; python_version < '3.10'",
 ]
@@ -53,9 +45,17 @@ compat = [
     # "pytest-speed>=0.3.5",
 ]
 test = ["pytest ~= 7.0", "pytest-cov ~= 4.0.0"]
+
 [project.entry-points]
 pytest11 = { codspeed = "pytest_codspeed.plugin" }
 
+[build-system]
+requires = ["setuptools >= 61", "cffi >= 1.17.1"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.dynamic]
+version = { attr = "pytest_codspeed.__version__" }
+
 [tool.hatch.envs.default]
 python = "3.11"
 features = ["lint", "test", "compat"]
@@ -67,12 +67,6 @@ features = ["test"]
 python = ["3.9", "3.10", "3.11", "3.12", "3.13"]
 features = ["compat", "test"]
 
-[tool.hatch.version]
-path = "src/pytest_codspeed/__init__.py"
-
-[tool.hatch.build.targets.sdist]
-include = ["/src"]
-
 [tool.mypy]
 python_version = "3.12"
 

setup.py

@@ -0,0 +1,31 @@
+import os
+import platform
+
+from setuptools import setup
+
+system = platform.system()
+current_arch = platform.machine()
+
+IS_EXTENSION_REQUIRED = (
+    system == "Linux"
+    and current_arch
+    in [
+        "x86_64",
+        "arm64",
+    ]
+    or os.environ.get("PYTEST_CODSPEED_FORCE_EXTENSION") is not None
+)
+
+setup(
+    package_data={
+        "pytest_codspeed": [
+            "instruments/valgrind/_wrapper/*.h",
+            "instruments/valgrind/_wrapper/*.c",
+        ]
+    },
+    cffi_modules=[
+        "src/pytest_codspeed/instruments/valgrind/_wrapper/build.py:ffibuilder"
+    ]
+    if IS_EXTENSION_REQUIRED
+    else [],
+)

src/pytest_codspeed/instruments/valgrind/_wrapper/.gitignore

@@ -1,38 +1,13 @@
 from __future__ import annotations
 
-import os
 from typing import TYPE_CHECKING
 
-from cffi import FFI  # type: ignore
-from filelock import FileLock
-
 if TYPE_CHECKING:
     from .wrapper import lib as LibType
 
-_wrapper_dir = os.path.dirname(os.path.abspath(__file__))
-
-
-def _get_ffi():
-    ffi = FFI()
-    with open(f"{_wrapper_dir}/wrapper.h") as f:
-        ffi.cdef(f.read())
-    ffi.set_source(
-        "dist_callgrind_wrapper",
-        '#include "wrapper.h"',
-        sources=["wrapper.c"],
-    )
-    return ffi
-
 
 def get_lib() -> LibType:
     try:
-        ffi = _get_ffi()
-        build_lock = FileLock(f"{_wrapper_dir}/build.lock")
-        with build_lock:
-            ffi.compile(
-                target="dist_callgrind_wrapper.*",
-                tmpdir=_wrapper_dir,
-            )
         from .dist_callgrind_wrapper import lib  # type: ignore
 
         return lib

src/pytest_codspeed/instruments/valgrind/_wrapper/__init__.py

@@ -0,0 +1,19 @@
+from pathlib import Path
+
+from cffi import FFI  # type: ignore
+
+wrapper_dir = Path(__file__).parent
+
+ffibuilder = FFI()
+
+ffibuilder.cdef((wrapper_dir / "wrapper.h").read_text())
+
+ffibuilder.set_source(
+    "pytest_codspeed.instruments.valgrind._wrapper.dist_callgrind_wrapper",
+    '#include "wrapper.h"',
+    sources=["src/pytest_codspeed/instruments/valgrind/_wrapper/wrapper.c"],
+    include_dirs=[str(wrapper_dir)],
+)
+
+if __name__ == "__main__":
+    ffibuilder.compile(verbose=True)