Allow for the specification of the rustfmt path

tonic-build/src/lib.rs

@@ -158,14 +158,22 @@ pub trait Method {
 #[cfg(feature = "rustfmt")]
 #[cfg_attr(docsrs, doc(cfg(feature = "rustfmt")))]
 pub fn fmt(out_dir: &str) {
+    // Since fmt is public, preserve the API of fmt by dispatching to a helper function that is
+    // called within the crate.
+    format_with("rustfmt", &out_dir);
+}
+
+#[cfg(feature = "rustfmt")]
+#[cfg_attr(docsrs, doc(cfg(feature = "rustfmt")))]
+fn format_with(rustfmt_path: impl AsRef<std::ffi::OsStr>, out_dir: &str) {
     let dir = std::fs::read_dir(out_dir).unwrap();
 
     for entry in dir {
         let file = entry.unwrap().file_name().into_string().unwrap();
         if !file.ends_with(".rs") {
             continue;
         }
-        let result = Command::new("rustfmt")
+        let result = Command::new(rustfmt_path.as_ref())
             .arg("--emit")
             .arg("files")
             .arg("--edition")

tonic-build/src/prost.rs

@@ -21,6 +21,8 @@ pub fn configure() -> Builder {
         compile_well_known_types: false,
         #[cfg(feature = "rustfmt")]
         format: true,
+        #[cfg(feature = "rustfmt")]
+        rustfmt_path: None,
         emit_package: true,
     }
 }
@@ -210,6 +212,8 @@ pub struct Builder {
     out_dir: Option<PathBuf>,
     #[cfg(feature = "rustfmt")]
     format: bool,
+    #[cfg(feature = "rustfmt")]
+    rustfmt_path: Option<PathBuf>,
 }
 
 impl Builder {
@@ -239,6 +243,13 @@ impl Builder {
         self
     }
 
+    /// The path that should be used to find rustfmt, if format is enabled.
+    #[cfg(feature = "rustfmt")]
+    pub fn format_with(mut self, path: impl AsRef<Path>) -> Self {
+        self.rustfmt_path = Some(path.as_ref().to_path_buf());
+        self
+    }
+
     /// Set the output directory to generate code to.
     ///
     /// Defaults to the `OUT_DIR` environment variable.
@@ -331,6 +342,11 @@ impl Builder {
 
         #[cfg(feature = "rustfmt")]
         let format = self.format;
+        #[cfg(feature = "rustfmt")]
+        let rustfmt_path = self
+            .rustfmt_path
+            .clone()
+            .unwrap_or_else(|| "rustfmt".into());
 
         config.out_dir(out_dir.clone());
         if let Some(path) = self.file_descriptor_set_path.as_ref() {
@@ -355,7 +371,10 @@ impl Builder {
         #[cfg(feature = "rustfmt")]
         {
             if format {
-                super::fmt(out_dir.to_str().expect("Expected utf8 out_dir"));
+                super::format_with(
+                    rustfmt_path,
+                    out_dir.to_str().expect("Expected utf8 out_dir"),
+                );
             }
         }