Fix rust build with stubs (#1001)

* Fix rust build with stubs

Signed-off-by: Alexandre Milesi <[email protected]>

* Add stub build test

Signed-off-by: Alexandre Milesi <[email protected]>

---------

Signed-off-by: Alexandre Milesi <[email protected]>
Co-authored-by: Raul Akhmetshin <[email protected]>

Author: milesial

.gitlab/test_rust.sh

@@ -46,4 +46,7 @@ export NIXL_PREFIX=${INSTALL_DIR}
 
 cargo test -- --test-threads=1
 
+# test stubs build
+cargo build --features stub-api
+
 cargo package

src/bindings/rust/build.rs

@@ -201,6 +201,21 @@ fn build_stubs(cc_builder: &mut cc::Build) {
     // Tell cargo to invalidate the built crate whenever the stubs change
     println!("cargo:rerun-if-changed=stubs.cpp");
     println!("cargo:rerun-if-changed=wrapper.h");
+
+    // Get the output path for bindings
+    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
+
+    // Generate bindings with minimal configuration
+    bindgen::Builder::default()
+        .header("wrapper.h")
+        .clang_arg("-std=c++17")
+        .clang_arg("-x")
+        .clang_arg("c++")
+        .parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
+        .generate()
+        .expect("Unable to generate bindings")
+        .write_to_file(out_path.join("bindings.rs"))
+        .expect("Couldn't write bindings!");
 }
 
 fn run_build(use_stub_api: bool) {