Rebuild on Linker change

1 files changed, 20 insertions, 0 deletions

diff --git a/build.rs b/build.rs
new file mode 100644
index 0000000..721d6d6
--- /dev/null
+++ b/build.rs
@@ -0,0 +1,20 @@
+use std::{env,fs,process};
+
+fn main() {
+	let ld_script_path = match env::var("LD_SCRIPT_PATH") {
+		Ok(var) => var,
+		_ => process::exit(0),
+	};
+
+	let files = fs::read_dir(ld_script_path).unwrap();
+	files
+		.filter_map(Result::ok)
+		.filter(|d| {
+			if let Some(e) = d.path().extension() {
+				e == "ld"
+			} else {
+				false
+			}
+		})
+		.for_each(|f| println!("cargo:rerun-if-changed={}", f.path().display()));
+}