feat(build): Use RUSTFMT to find rustfmt binary (#566)

hyperium · Mar 11, 2021 · ea56e2e · ea56e2e
1 parent 61555ff
commit ea56e2e
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/tonic-build/src/lib.rs b/tonic-build/src/lib.rs
@@ -165,13 +165,14 @@ pub fn fmt(out_dir: &str) {
         if !file.ends_with(".rs") {
             continue;
         }
-        let result = Command::new("rustfmt")
-            .arg("--emit")
-            .arg("files")
-            .arg("--edition")
-            .arg("2018")
-            .arg(format!("{}/{}", out_dir, file))
-            .output();
+        let result =
+            Command::new(std::env::var("RUSTFMT").unwrap_or_else(|_| "rustfmt".to_owned()))
+                .arg("--emit")
+                .arg("files")
+                .arg("--edition")
+                .arg("2018")
+                .arg(format!("{}/{}", out_dir, file))
+                .output();
 
         match result {
             Err(e) => {