feat: add instrument hooks bindings

GuillaumeLagrange · GuillaumeLagrange · commit 440ac304d4f8 · 2025-08-24T20:56:21.000+02:00
diff --git a/.eslintrc.json b/.eslintrc.json
@@ -10,7 +10,8 @@
     "**/dist/**",
     "**/node_modules/**",
     "**/rollup.config.ts",
-    "**/jest.config.js"
+    "**/jest.config.js",
+    "packages/core/src/native_core/instruments/hooks/**"
   ],
   "settings": {
     "import/parsers": {
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -13,6 +13,7 @@ jobs:
       - uses: "actions/checkout@v4"
         with:
           fetch-depth: 0
+          submodules: true
       - name: Install valgrind
         run: |
           sudo apt-get update
@@ -32,6 +33,9 @@ jobs:
       examples: ${{ steps.list-examples.outputs.examples }}
     steps:
       - uses: "actions/checkout@v4"
+        with:
+          fetch-depth: 0
+          submodules: true
       # list the directories in ./examples and output them to a github action workflow variables as a JSON array
       - run: |
           examples=$(find ./examples -maxdepth 1 -mindepth 1 -type d -printf '%f\n' | jq -R -s -c 'split("\n") | map(select(length > 0))')
@@ -51,6 +55,7 @@ jobs:
       - uses: "actions/checkout@v4"
         with:
           fetch-depth: 0
+          submodules: true
       - name: Install valgrind
         run: |
           sudo apt-get update
diff --git a/.github/workflows/codspeed.yml b/.github/workflows/codspeed.yml
@@ -14,6 +14,7 @@ jobs:
       - uses: "actions/checkout@v4"
         with:
           fetch-depth: 0
+          submodules: true
       - name: Install valgrind
         run: |
           sudo apt-get update
@@ -43,6 +44,7 @@ jobs:
       - uses: "actions/checkout@v4"
         with:
           fetch-depth: 0
+          submodules: true
       - name: Install valgrind
         run: |
           sudo apt-get update
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -17,6 +17,7 @@ jobs:
       - uses: actions/checkout@v4
         with:
           fetch-depth: 0
+          submodules: true
       - uses: pnpm/action-setup@v2
       - uses: actions/setup-node@v3
         with:
diff --git a/.gitmodules b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "packages/core/src/native_core/instruments/hooks"]
+	path = packages/core/src/native_core/instruments/hooks
+	url = git@github.com:CodSpeedHQ/instrument-hooks.git
diff --git a/.prettierignore b/.prettierignore
@@ -2,3 +2,4 @@ node_modules
 .rollup.cache
 dist
 generated
+packages/core/src/native_core/instruments/hooks
diff --git a/packages/core/binding.gyp b/packages/core/binding.gyp
@@ -8,14 +8,31 @@
       "cflags_cc!": [
         "-fno-exceptions"
       ],
+      "cflags": [
+        "-Wno-maybe-uninitialized",
+        "-Wno-unused-variable",
+        "-Wno-unused-parameter",
+        "-Wno-unused-but-set-variable",
+        "-Wno-type-limits"
+      ],
+      "cflags_cc": [
+        "-Wno-maybe-uninitialized",
+        "-Wno-unused-variable",
+        "-Wno-unused-parameter",
+        "-Wno-unused-but-set-variable",
+        "-Wno-type-limits"
+      ],
       "sources": [
         "src/native_core/measurement/measurement.cc",
         "src/native_core/linux_perf/linux_perf.cc",
         "src/native_core/linux_perf/linux_perf_listener.cc",
+        "src/native_core/instruments/hooks_wrapper.cc",
+        "src/native_core/instruments/hooks/dist/core.c",
         "src/native_core/native_core.cc"
       ],
       "include_dirs": [
-        "<!@(node -p \"require('node-addon-api').include\")"
+        "<!@(node -p \"require('node-addon-api').include\")",
+        "src/native_core/instruments/hooks/includes"
       ],
       "defines": [
         "NAPI_DISABLE_CPP_EXCEPTIONS"
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
@@ -53,3 +53,4 @@ export { optimizeFunction, optimizeFunctionSync } from "./optimization";
 export * from "./utils";
 export * from "./walltime";
 export const Measurement = native_core.Measurement;
+export const InstrumentHooks = native_core.InstrumentHooks;
diff --git a/packages/core/src/native_core/index.ts b/packages/core/src/native_core/index.ts
@@ -1,8 +1,10 @@
 import path from "path";
+import { InstrumentHooks } from "./instruments/hooks";
 import { LinuxPerf } from "./linux_perf/linux_perf";
 import { Measurement } from "./measurement/measurement";
 interface NativeCore {
   Measurement: Measurement;
+  InstrumentHooks: InstrumentHooks;
   LinuxPerf: typeof LinuxPerf;
 }
 
@@ -37,7 +39,23 @@ try {
         return false;
       }
     },
-
+    InstrumentHooks: {
+      isInstrumented: () => {
+        return false;
+      },
+      startBenchmark: () => {
+        return 0;
+      },
+      stopBenchmark: () => {
+        return 0;
+      },
+      setExecutedBenchmark: (_pid: number, _uri: string) => {
+        return 0;
+      },
+      setIntegration: (_name: string, _version: string) => {
+        return 0;
+      },
+    },
     isBound: false,
   };
 }
diff --git a/packages/core/src/native_core/instruments/hooks b/packages/core/src/native_core/instruments/hooks
@@ -0,0 +1 @@
+Subproject commit 568ce111be7842d1ce5cd7cfdbfe23b1a3085d0b
diff --git a/packages/core/src/native_core/instruments/hooks.ts b/packages/core/src/native_core/instruments/hooks.ts
@@ -0,0 +1,34 @@
+export interface InstrumentHooks {
+  /**
+   * Check if instrumentation is enabled
+   */
+  isInstrumented(): boolean;
+
+  /**
+   * Start benchmark measurement
+   * @returns 0 on success, non-zero on error
+   */
+  startBenchmark(): number;
+
+  /**
+   * Stop benchmark measurement
+   * @returns 0 on success, non-zero on error
+   */
+  stopBenchmark(): number;
+
+  /**
+   * Set the executed benchmark metadata
+   * @param pid Process ID
+   * @param uri Benchmark URI/identifier
+   * @returns 0 on success, non-zero on error
+   */
+  setExecutedBenchmark(pid: number, uri: string): number;
+
+  /**
+   * Set integration metadata
+   * @param name Integration name
+   * @param version Integration version
+   * @returns 0 on success, non-zero on error
+   */
+  setIntegration(name: string, version: string): number;
+}
diff --git a/packages/core/src/native_core/instruments/hooks_wrapper.cc b/packages/core/src/native_core/instruments/hooks_wrapper.cc
@@ -0,0 +1,137 @@
+#include "hooks_wrapper.h"
+#include "hooks/includes/core.h"
+#include <memory>
+
+namespace codspeed_native {
+namespace instruments {
+namespace hooks_wrapper {
+
+// Global instance to maintain state across calls
+static std::unique_ptr<InstrumentHooks, decltype(&instrument_hooks_deinit)>
+    g_hooks{nullptr, &instrument_hooks_deinit};
+
+// Initialize instrument hooks if not already done
+static InstrumentHooks *ensureInitialized() {
+  if (!g_hooks) {
+    InstrumentHooks *hooks = instrument_hooks_init();
+    if (hooks) {
+      g_hooks.reset(hooks);
+    }
+  }
+  return g_hooks.get();
+}
+
+Napi::Boolean IsInstrumented(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+  InstrumentHooks *hooks = ensureInitialized();
+
+  if (!hooks) {
+    return Napi::Boolean::New(env, false);
+  }
+
+  bool instrumented = instrument_hooks_is_instrumented(hooks);
+  return Napi::Boolean::New(env, instrumented);
+}
+
+Napi::Number StartBenchmark(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+  InstrumentHooks *hooks = ensureInitialized();
+
+  if (!hooks) {
+    return Napi::Number::New(env, 1); // Return error code
+  }
+
+  uint8_t result = instrument_hooks_start_benchmark(hooks);
+  return Napi::Number::New(env, result);
+}
+
+Napi::Number StopBenchmark(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+  InstrumentHooks *hooks = ensureInitialized();
+
+  if (!hooks) {
+    return Napi::Number::New(env, 1); // Return error code
+  }
+
+  uint8_t result = instrument_hooks_stop_benchmark(hooks);
+  return Napi::Number::New(env, result);
+}
+
+Napi::Number SetExecutedBenchmark(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+
+  if (info.Length() != 2) {
+    Napi::TypeError::New(env, "Expected 2 arguments: pid and uri")
+        .ThrowAsJavaScriptException();
+    return Napi::Number::New(env, 1);
+  }
+
+  if (!info[0].IsNumber() || !info[1].IsString()) {
+    Napi::TypeError::New(env, "Expected number (pid) and string (uri)")
+        .ThrowAsJavaScriptException();
+    return Napi::Number::New(env, 1);
+  }
+
+  InstrumentHooks *hooks = ensureInitialized();
+  if (!hooks) {
+    return Napi::Number::New(env, 1);
+  }
+
+  uint32_t pid = info[0].As<Napi::Number>().Uint32Value();
+  std::string uri = info[1].As<Napi::String>().Utf8Value();
+
+  uint8_t result =
+      instrument_hooks_set_executed_benchmark(hooks, pid, uri.c_str());
+  return Napi::Number::New(env, result);
+}
+
+Napi::Number SetIntegration(const Napi::CallbackInfo &info) {
+  Napi::Env env = info.Env();
+
+  if (info.Length() != 2) {
+    Napi::TypeError::New(env, "Expected 2 arguments: name and version")
+        .ThrowAsJavaScriptException();
+    return Napi::Number::New(env, 1);
+  }
+
+  if (!info[0].IsString() || !info[1].IsString()) {
+    Napi::TypeError::New(env, "Expected string (name) and string (version)")
+        .ThrowAsJavaScriptException();
+    return Napi::Number::New(env, 1);
+  }
+
+  InstrumentHooks *hooks = ensureInitialized();
+  if (!hooks) {
+    return Napi::Number::New(env, 1);
+  }
+
+  std::string name = info[0].As<Napi::String>().Utf8Value();
+  std::string version = info[1].As<Napi::String>().Utf8Value();
+
+  uint8_t result =
+      instrument_hooks_set_integration(hooks, name.c_str(), version.c_str());
+  return Napi::Number::New(env, result);
+}
+
+Napi::Object Initialize(Napi::Env env, Napi::Object exports) {
+  Napi::Object instrumentHooksObj = Napi::Object::New(env);
+
+  instrumentHooksObj.Set(Napi::String::New(env, "isInstrumented"),
+                         Napi::Function::New(env, IsInstrumented));
+  instrumentHooksObj.Set(Napi::String::New(env, "startBenchmark"),
+                         Napi::Function::New(env, StartBenchmark));
+  instrumentHooksObj.Set(Napi::String::New(env, "stopBenchmark"),
+                         Napi::Function::New(env, StopBenchmark));
+  instrumentHooksObj.Set(Napi::String::New(env, "setExecutedBenchmark"),
+                         Napi::Function::New(env, SetExecutedBenchmark));
+  instrumentHooksObj.Set(Napi::String::New(env, "setIntegration"),
+                         Napi::Function::New(env, SetIntegration));
+
+  exports.Set(Napi::String::New(env, "InstrumentHooks"), instrumentHooksObj);
+
+  return exports;
+}
+
+} // namespace hooks_wrapper
+} // namespace instruments
+} // namespace codspeed_native
diff --git a/packages/core/src/native_core/instruments/hooks_wrapper.h b/packages/core/src/native_core/instruments/hooks_wrapper.h
@@ -0,0 +1,21 @@
+#ifndef INSTRUMENTS_HOOKS_WRAPPER_H
+#define INSTRUMENTS_HOOKS_WRAPPER_H
+
+#include <napi.h>
+
+namespace codspeed_native {
+namespace instruments {
+namespace hooks_wrapper {
+
+Napi::Boolean IsInstrumented(const Napi::CallbackInfo &info);
+Napi::Number StartBenchmark(const Napi::CallbackInfo &info);
+Napi::Number StopBenchmark(const Napi::CallbackInfo &info);
+Napi::Number SetExecutedBenchmark(const Napi::CallbackInfo &info);
+Napi::Number SetIntegration(const Napi::CallbackInfo &info);
+Napi::Object Initialize(Napi::Env env, Napi::Object exports);
+
+} // namespace hooks_wrapper
+} // namespace instruments
+} // namespace codspeed_native
+
+#endif // INSTRUMENTS_HOOKS_WRAPPER_H
diff --git a/packages/core/src/native_core/native_core.cc b/packages/core/src/native_core/native_core.cc
@@ -1,12 +1,14 @@
 #include "linux_perf/linux_perf.h"
 #include "measurement/measurement.h"
+#include "instruments/hooks_wrapper.h"
 #include <napi.h>
 
 namespace codspeed_native {
 
 Napi::Object Initialize(Napi::Env env, Napi::Object exports) {
   codspeed_native::LinuxPerf::Initialize(env, exports);
   codspeed_native::Measurement::Initialize(env, exports);
+  codspeed_native::instruments::hooks_wrapper::Initialize(env, exports);
 
   return exports;
 }

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+[submodule "packages/core/src/native_core/instruments/hooks"]`
	`2`	`+ path = packages/core/src/native_core/instruments/hooks`
	`3`	`+ url = [email protected]:CodSpeedHQ/instrument-hooks.git`