From 44aa80d678a53cd21eb9e16dec87db9e2cbc877b Mon Sep 17 00:00:00 2001
From: Douglas Creager <dcreager@dcreager.net>
Date: Mon, 20 Nov 2023 09:39:19 -0500
Subject: [PATCH] feat: include build script links in Rust bindings

This patch updates the Rust binding's build script to output [build
metadata][links].  This makes it easier for downstream crates to
determine the include path, in case they need to compile their own C
code that requires the tree-sitter headers.

[links]: https://doc.rust-lang.org/cargo/reference/build-scripts.html#the-links-manifest-key
---
 lib/Cargo.toml            |  1 +
 lib/binding_rust/build.rs | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/lib/Cargo.toml b/lib/Cargo.toml
index 30e5b405..1b97e423 100644
--- a/lib/Cargo.toml
+++ b/lib/Cargo.toml
@@ -12,6 +12,7 @@ repository = "https://github.com/tree-sitter/tree-sitter"
 rust-version.workspace = true
 
 build = "binding_rust/build.rs"
+links = "tree-sitter"
 
 include = [
 	"/binding_rust/*",
diff --git a/lib/binding_rust/build.rs b/lib/binding_rust/build.rs
index f357f3fa..d028dc35 100644
--- a/lib/binding_rust/build.rs
+++ b/lib/binding_rust/build.rs
@@ -32,8 +32,11 @@ fn main() {
         config.define("TREE_SITTER_FEATURE_WASM", "");
     }
 
-    let src_path = Path::new("src");
-    for entry in fs::read_dir(src_path).unwrap() {
+    let manifest_path = Path::new(env!("CARGO_MANIFEST_DIR"));
+    let include_path = manifest_path.join("include");
+    let src_path = manifest_path.join("src");
+    let wasm_path = src_path.join("wasm");
+    for entry in fs::read_dir(&src_path).unwrap() {
         let entry = entry.unwrap();
         let path = src_path.join(entry.file_name());
         println!("cargo:rerun-if-changed={}", path.to_str().unwrap());
@@ -44,11 +47,13 @@ fn main() {
         .flag_if_supported("-fvisibility=hidden")
         .flag_if_supported("-Wshadow")
         .flag_if_supported("-Wno-unused-parameter")
-        .include(src_path)
-        .include(src_path.join("wasm"))
-        .include("include")
+        .include(&src_path)
+        .include(&wasm_path)
+        .include(&include_path)
         .file(src_path.join("lib.c"))
         .compile("tree-sitter");
+
+    println!("cargo:include={}", include_path.display());
 }
 
 #[cfg(feature = "bindgen")]