feat: add instrument-hook library

Author: not-matthias

.gitmodules

@@ -1,3 +1,6 @@
 [submodule "tests/benchmarks/TheAlgorithms"]
 	path = tests/benchmarks/TheAlgorithms
 	url = git@github.com:TheAlgorithms/Python.git
+[submodule "src/pytest_codspeed/instruments/hooks"]
+	path = src/pytest_codspeed/instruments/hooks
+	url = https://github.com/CodSpeedHQ/instrument-hooks

setup.py

@@ -6,7 +6,7 @@
 from setuptools import setup
 
 build_path = (
-    Path(__file__).parent / "src/pytest_codspeed/instruments/valgrind/_wrapper/build.py"
+    Path(__file__).parent / "src/pytest_codspeed/instruments/hooks/build.py"
 )
 
 spec = importlib.util.spec_from_file_location("build", build_path)

src/pytest_codspeed/instruments/hooks/__init__.py

@@ -0,0 +1,39 @@
+import os
+from typing import Optional
+
+from pytest_codspeed.core import lib
+
+class InstrumentHooks:
+    """Core library wrapper class providing benchmark measurement functionality."""
+
+    @staticmethod
+    def start_benchmark() -> None:
+        """Start a new benchmark measurement."""
+        lib.start_benchmark()
+
+    @staticmethod
+    def stop_benchmark() -> None:
+        """Stop the current benchmark measurement."""
+        lib.stop_benchmark()
+
+    @staticmethod
+    def set_current_benchmark(uri: str, pid: Optional[int] = None) -> None:
+        """Set the current benchmark URI and process ID.
+
+        Args:
+            uri: The benchmark URI string identifier
+            pid: Optional process ID (defaults to current process)
+        """
+        if pid is None:
+            pid = os.getpid()
+        lib.current_benchmark(pid, uri.encode('ascii'))
+
+    @staticmethod
+    def set_integration(name: str, version: str) -> None:
+        """Set the integration name and version."""
+        lib.set_integration(name.encode('ascii'), version.encode('ascii'))
+
+    @staticmethod
+    def is_instrumented() -> bool:
+        """Check if instrumentation is active."""
+        return lib.is_instrumented()

src/pytest_codspeed/instruments/hooks/build.py

@@ -0,0 +1,27 @@
+from pathlib import Path
+from cffi import FFI  # type: ignore
+
+ffibuilder = FFI()
+
+dist_dir = Path(__file__).parent.joinpath("instrument-hooks/dist")
+header_text = (dist_dir / "core.h").read_text()
+filtered_header = "\n".join(
+    line for line in header_text.splitlines()
+    if not line.strip().startswith("#")
+)
+ffibuilder.cdef(filtered_header)
+
+ffibuilder.set_source(
+    "pytest_codspeed.core",
+    "",
+    libraries=["m", "quadmath"],
+    sources=[
+        "src/pytest_codspeed/instruments/core/instrument-hooks/dist/core.c",
+        "src/pytest_codspeed/instruments/core/instrument-hooks/dist/wrapper.c",
+    ],
+    include_dirs=[str(dist_dir)],
+    extra_compile_args=["-lm", "-lcs50"],
+)
+
+if __name__ == "__main__":
+    ffibuilder.compile(verbose=True)

src/pytest_codspeed/instruments/hooks/core/instrument-hooks

@@ -0,0 +1 @@
+Subproject commit 3363c83d15532a981c4a4c9985871ceebb546ed4