chore: bump instrument-hooks

Author: not-matthias

src/pytest_codspeed/instruments/hooks/__init__.py

@@ -10,6 +10,11 @@
 if TYPE_CHECKING:
     from .dist_instrument_hooks import InstrumentHooksPointer, LibType
 
+MARKER_TYPE_SAMPLE_START = 0
+MARKER_TYPE_SAMPLE_END = 1
+MARKER_TYPE_BENCHMARK_START = 2
+MARKER_TYPE_BENCHMARK_END = 3
+
 
 class InstrumentHooks:
     """Zig library wrapper class providing benchmark measurement functionality."""
@@ -80,3 +85,24 @@ def set_integration(self, name: str, version: str) -> None:
     def is_instrumented(self) -> bool:
         """Check if instrumentation is active."""
         return self.lib.instrument_hooks_is_instrumented(self.instance)
+
+    def add_marker(self, pid: int, marker_type: int, timestamp: int) -> int:
+        """Add a marker with the specified type and timestamp.
+
+        Args:
+            pid: Process ID
+            marker_type: Type of marker (use MARKER_TYPE_* constants)
+            timestamp: Timestamp value
+
+        Returns:
+            0 on success, non-zero on error
+        """
+        return self.lib.instrument_hooks_add_marker(self.instance, pid, marker_type, timestamp)
+
+    def current_timestamp(self) -> int:
+        """Get the current timestamp.
+
+        Returns:
+            Current timestamp value
+        """
+        return self.lib.instrument_hooks_current_timestamp()

src/pytest_codspeed/instruments/hooks/build.py

@@ -6,10 +6,35 @@
 
 includes_dir = Path(__file__).parent.joinpath("instrument-hooks/includes")
 header_text = (includes_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)
+
+
+# Manually copied from `instrument-hooks/includes/core.h` to avoid parsing issues
+ffibuilder.cdef("""
+typedef uint64_t *InstrumentHooks;
+
+InstrumentHooks *instrument_hooks_init(void);
+void instrument_hooks_deinit(InstrumentHooks *);
+
+bool instrument_hooks_is_instrumented(InstrumentHooks *);
+int8_t instrument_hooks_start_benchmark(InstrumentHooks *);
+int8_t instrument_hooks_stop_benchmark(InstrumentHooks *);
+int8_t instrument_hooks_set_executed_benchmark(InstrumentHooks *, int32_t pid,
+                                               const char *uri);
+int8_t instrument_hooks_set_integration(InstrumentHooks *, const char *name,
+                                        const char *version);
+
+#define MARKER_TYPE_SAMPLE_START 0
+#define MARKER_TYPE_SAMPLE_END 1
+#define MARKER_TYPE_BENCHMARK_START 2
+#define MARKER_TYPE_BENCHMARK_END 3
+
+int8_t instrument_hooks_add_marker(InstrumentHooks *, uint32_t pid,
+                                   uint8_t marker_type, uint64_t timestamp);
+uint64_t instrument_hooks_current_timestamp(void);
+
+int8_t callgrind_start_instrumentation();
+int8_t callgrind_stop_instrumentation();
+""")
 
 ffibuilder.set_source(
     "pytest_codspeed.instruments.hooks.dist_instrument_hooks",

src/pytest_codspeed/instruments/hooks/dist_instrument_hooks.pyi

@@ -20,6 +20,12 @@ class lib:
         hooks: InstrumentHooksPointer, name: bytes, version: bytes
     ) -> int: ...
     @staticmethod
+    def instrument_hooks_add_marker(
+        hooks: InstrumentHooksPointer, pid: int, marker_type: int, timestamp: int
+    ) -> int: ...
+    @staticmethod
+    def instrument_hooks_current_timestamp() -> int: ...
+    @staticmethod
     def callgrind_start_instrumentation() -> int: ...
     @staticmethod
     def callgrind_stop_instrumentation() -> int: ...