chore: bump instrument-hooks (#87)

* fix(docs): update to v4

* chore: bump instrument-hooks

Author: not-matthias

README.md

@@ -111,8 +111,9 @@ jobs:
         run: pip install -r requirements.txt
 
       - name: Run benchmarks
-        uses: CodSpeedHQ/action@v3
+        uses: CodSpeedHQ/action@v4
         with:
+          mode: instrumentation   # or `walltime`
           token: ${{ secrets.CODSPEED_TOKEN }}
           run: pytest tests/ --codspeed
 ```

src/pytest_codspeed/instruments/hooks/__init__.py

@@ -63,7 +63,7 @@ def set_executed_benchmark(self, uri: str, pid: int | None = None) -> None:
         if pid is None:
             pid = os.getpid()
 
-        ret = self.lib.instrument_hooks_executed_benchmark(
+        ret = self.lib.instrument_hooks_set_executed_benchmark(
             self.instance, pid, uri.encode("ascii")
         )
         if ret != 0:

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 *);
+uint8_t instrument_hooks_start_benchmark(InstrumentHooks *);
+uint8_t instrument_hooks_stop_benchmark(InstrumentHooks *);
+uint8_t instrument_hooks_set_executed_benchmark(InstrumentHooks *, int32_t pid,
+                                               const char *uri);
+uint8_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
+
+uint8_t instrument_hooks_add_marker(InstrumentHooks *, uint32_t pid,
+                                   uint8_t marker_type, uint64_t timestamp);
+uint64_t instrument_hooks_current_timestamp(void);
+
+void callgrind_start_instrumentation();
+void callgrind_stop_instrumentation();
+""")
 
 ffibuilder.set_source(
     "pytest_codspeed.instruments.hooks.dist_instrument_hooks",

src/pytest_codspeed/instruments/hooks/dist_instrument_hooks.pyi

@@ -12,14 +12,20 @@ class lib:
     @staticmethod
     def instrument_hooks_stop_benchmark(hooks: InstrumentHooksPointer) -> int: ...
     @staticmethod
-    def instrument_hooks_executed_benchmark(
+    def instrument_hooks_set_executed_benchmark(
         hooks: InstrumentHooksPointer, pid: int, uri: bytes
     ) -> int: ...
     @staticmethod
     def instrument_hooks_set_integration(
         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: ...