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

UebelAndre · web-flow · commit ea56e2e2b89d · 2021-03-11T10:47:07.000-05:00
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) => {
