feat: check buildability and fallback when build doesn't work

Author: art049

.github/workflows/ci.yml

@@ -54,9 +54,12 @@ jobs:
         run: sudo apt-get install valgrind -y
       - name: Install dependencies with pytest${{ matrix.pytest-version }}
         run: |
-          uv sync --all-extras --dev --locked
+          if [ "${{ matrix.config }}" == "valgrind" ]; then
+            export PYTEST_CODSPEED_FORCE_EXTENSION=1
+          fi
+          uv sync --all-extras --dev --locked --verbose
           uv pip install "pytest${{ matrix.pytest-version }}"
-          uv pip uninstall -y pytest-benchmark
+          uv pip uninstall pytest-benchmark
       - if: matrix.config == 'pytest-benchmark-4'
         name: Install pytest-benchmark 4.0.0
         run: uv pip install pytest-benchmark~=4.0.0

setup.py

@@ -1,19 +1,41 @@
+import importlib.util
 import os
 import platform
+from pathlib import Path
 
 from setuptools import setup
 
+build_path = (
+    Path(__file__).parent / "src/pytest_codspeed/instruments/valgrind/_wrapper/build.py"
+)
+
+spec = importlib.util.spec_from_file_location("build", build_path)
+assert spec is not None, "The spec should be initialized"
+build = importlib.util.module_from_spec(spec)
+assert spec.loader is not None, "The loader should be initialized"
+spec.loader.exec_module(build)
+
 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
+IS_EXTENSION_BUILDABLE = system == "Linux" and current_arch in [
+    "x86_64",
+    "arm64",
+]
+
+IS_EXTENSION_REQUIRED = os.environ.get("PYTEST_CODSPEED_FORCE_EXTENSION") is not None
+if IS_EXTENSION_REQUIRED and not IS_EXTENSION_BUILDABLE:
+    raise ValueError(
+        "The extension is required but the current platform is not supported"
+    )
+
+
+ffi_extension = build.ffibuilder.distutils_extension()
+ffi_extension.optional = not IS_EXTENSION_REQUIRED
+
+print(
+    "CodSpeed native extension is "
+    + ("required" if IS_EXTENSION_REQUIRED else "optional")
 )
 
 setup(
@@ -23,9 +45,5 @@
             "instruments/valgrind/_wrapper/*.c",
         ]
     },
-    cffi_modules=[
-        "src/pytest_codspeed/instruments/valgrind/_wrapper/build.py:ffibuilder"
-    ]
-    if IS_EXTENSION_REQUIRED
-    else [],
+    ext_modules=[ffi_extension] if IS_EXTENSION_BUILDABLE else [],
 )