Add zstd_errors.h to bindgen generator when using experimental feature (

#224) * Add zstd_errors.h to bindgen generator when using experimental feature * Refactor to use ifdef instead of separate header
gyscos · Jul 21, 2023 · 042f231 · 042f231
1 parent 5872d29
commit 042f231
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/zstd-safe/zstd-sys/build.rs b/zstd-safe/zstd-sys/build.rs
@@ -7,6 +7,8 @@ fn generate_bindings(defs: Vec<&str>, headerpaths: Vec<PathBuf>) {
     let bindings = bindgen::Builder::default().header("zstd.h");
     #[cfg(feature = "zdict_builder")]
     let bindings = bindings.header("zdict.h");
+    #[cfg(feature = "experimental")]
+    let bindings = bindings.clang_arg("-DZSTD_RUST_BINDINGS_EXPERIMENTAL");
     let bindings = bindings
         .blocklist_type("max_align_t")
         .size_t_is_usize(true)

diff --git a/zstd-safe/zstd-sys/zstd.h b/zstd-safe/zstd-sys/zstd.h
@@ -2,11 +2,16 @@
 
 /* Just use installed headers */
 #include <zstd.h>
-// Don't use experimental features like zstdmt
+#ifdef ZSTD_RUST_BINDINGS_EXPERIMENTAL
+#include <zstd_errors.h>
+#endif
 
 #else // #ifdef PKG_CONFIG
 
 #include "zstd/lib/zstd.h"
+#ifdef ZSTD_RUST_BINDINGS_EXPERIMENTAL
+#include "zstd/lib/zstd_errors.h"
+#endif
 
 #endif // #ifdef PKG_CONFIG