From 47b2611bb3a02d0081c27438f0cfe17550989865 Mon Sep 17 00:00:00 2001
From: SageMik <sagemik@163.com>
Date: Sun, 3 Aug 2025 03:58:09 +0800
Subject: [PATCH] Support toolchain and custom args for CMake

---
 build/cmake_probe.rs | 6 ++++++
 build/library.rs     | 4 ++++
 2 files changed, 10 insertions(+)

diff --git a/build/cmake_probe.rs b/build/cmake_probe.rs
index 9c2e75564..fd433f85a 100644
--- a/build/cmake_probe.rs
+++ b/build/cmake_probe.rs
@@ -110,6 +110,7 @@ pub struct CmakeProbe<'r> {
 	package_name: &'r str,
 	toolchain: Option<&'r Path>,
 	is_release: bool,
+	args: Option<&'r str>,
 }
 
 impl<'r> CmakeProbe<'r> {
@@ -120,6 +121,7 @@ impl<'r> CmakeProbe<'r> {
 		package_name: &'r str,
 		toolchain: Option<&'r Path>,
 		is_release: bool,
+		args: Option<&'r str>,
 	) -> Self {
 		Self {
 			cmake_bin: cmake_bin.unwrap_or_else(|| "cmake".into()),
@@ -128,6 +130,7 @@ impl<'r> CmakeProbe<'r> {
 			package_name,
 			toolchain,
 			is_release,
+			args,
 		}
 	}
 
@@ -162,6 +165,9 @@ impl<'r> CmakeProbe<'r> {
 		} else {
 			out.arg("-DCMAKE_BUILD_TYPE=Debug");
 		}
+		if let Some(args) = &self.args {
+			out.args(args.split(' '));
+		}
 		out
 	}
 
diff --git a/build/library.rs b/build/library.rs
index e83726813..91c8a7e9a 100644
--- a/build/library.rs
+++ b/build/library.rs
@@ -341,6 +341,8 @@ impl Library {
 		cmake_bin: Option<&Path>,
 		ninja_bin: Option<&Path>,
 	) -> Result<Self> {
+		let toolchain_var = env::var_os("OPENCV_CMAKE_TOOLCHAIN_FILE").map(PathBuf::from);
+		let toolchain = toolchain.or_else(|| toolchain_var.as_ref().map(PathBuf::as_path));
 		eprintln!(
 			"=== Probing OpenCV library using cmake{}",
 			toolchain.map_or_else(|| "".to_string(), |tc| format!(" with toolchain: {}", tc.display()))
@@ -348,6 +350,7 @@ impl Library {
 
 		let src_dir = MANIFEST_DIR.join("cmake");
 		let package_name = PackageName::cmake();
+		let args = env::var_os("OPENCV_CMAKE_ARGS");
 		let cmake = CmakeProbe::new(
 			env::var_os("OPENCV_CMAKE_BIN")
 				.map(PathBuf::from)
@@ -357,6 +360,7 @@ impl Library {
 			package_name.as_ref(),
 			toolchain,
 			env::var_os("PROFILE").is_some_and(|p| p == "release"),
+			args.as_ref().and_then(|p| p.to_str()),
 		);
 		let mut probe_result = cmake
 			.probe_ninja(ninja_bin)