rust-project: add support for alternative Buck2 commands

This is useful for users of wrappers such as Buckle, as well as developers hacking on Buck2 that
need to use a custom build.

integrations/rust-project/src/buck.rs

@@ -50,9 +50,10 @@ pub(crate) fn to_json_project(
     aliases: FxHashMap<Target, AliasedTargetInfo>,
     relative_paths: bool,
     check_cycles: bool,
+    buck2_command: Option<String>,
 ) -> Result<JsonProject, anyhow::Error> {
     let mode = select_mode(None);
-    let buck = Buck::new(mode);
+    let buck = Buck::new(buck2_command, mode);
     let project_root = buck.resolve_project_root()?;
 
     let ExpandedAndResolved {
@@ -436,12 +437,16 @@ fn merge_unit_test_targets(
 
 #[derive(Debug, Default)]
 pub(crate) struct Buck {
+    command: String,
     mode: Option<String>,
 }
 
 impl Buck {
-    pub(crate) fn new(mode: Option<String>) -> Self {
-        Buck { mode }
+    pub(crate) fn new(command: Option<String>, mode: Option<String>) -> Self {
+        Buck {
+            command: command.unwrap_or_else(|| "buck2".into()),
+            mode,
+        }
     }
 
     /// Invoke `buck2` with the given subcommands.
@@ -471,7 +476,7 @@ impl Buck {
         I: IntoIterator<Item = S>,
         S: AsRef<OsStr>,
     {
-        let mut cmd = Command::new("buck2");
+        let mut cmd = Command::new(&self.command);
 
         // rust-analyzer invokes the check-on-save command with `RUST_BACKTRACE=short`
         // set. Unfortunately, buck2 doesn't handle that well and becomes extremely

integrations/rust-project/src/cli/check.rs

@@ -23,11 +23,16 @@ pub(crate) struct Check {
 }
 
 impl Check {
-    pub(crate) fn new(mode: Option<String>, use_clippy: bool, saved_file: PathBuf) -> Self {
+    pub(crate) fn new(
+        buck2_command: Option<String>,
+        mode: Option<String>,
+        use_clippy: bool,
+        saved_file: PathBuf,
+    ) -> Self {
         let saved_file = canonicalize(&saved_file).unwrap_or(saved_file);
 
         let mode = select_mode(mode.as_deref());
-        let buck = buck::Buck::new(mode);
+        let buck = buck::Buck::new(buck2_command, mode);
         Self {
             buck,
             use_clippy,

integrations/rust-project/src/cli/develop.rs

@@ -38,6 +38,7 @@ pub(crate) struct Develop {
     pub(crate) buck: buck::Buck,
     pub(crate) check_cycles: bool,
     pub(crate) invoked_by_ra: bool,
+    pub(crate) buck2_command: Option<String>,
 }
 
 pub(crate) struct OutputCfg {
@@ -64,6 +65,7 @@ impl Develop {
             relative_paths,
             mode,
             check_cycles,
+            buck2_command,
         } = command
         {
             let out = if stdout {
@@ -81,14 +83,15 @@ impl Develop {
             };
 
             let mode = select_mode(mode.as_deref());
-            let buck = buck::Buck::new(mode);
+            let buck = buck::Buck::new(buck2_command.clone(), mode);
 
             let develop = Develop {
                 sysroot,
                 relative_paths,
                 buck,
                 check_cycles,
                 invoked_by_ra: false,
+                buck2_command,
             };
             let out = OutputCfg { out, pretty };
 
@@ -102,7 +105,12 @@ impl Develop {
             return (develop, input, out);
         }
 
-        if let crate::Command::DevelopJson { sysroot_mode, args } = command {
+        if let crate::Command::DevelopJson {
+            sysroot_mode,
+            args,
+            buck2_command,
+        } = command
+        {
             let out = Output::Stdout;
             let mode = select_mode(None);
 
@@ -119,14 +127,15 @@ impl Develop {
                 }
             };
 
-            let buck = buck::Buck::new(mode);
+            let buck = buck::Buck::new(buck2_command.clone(), mode);
 
             let develop = Develop {
                 sysroot,
                 relative_paths: false,
                 buck,
                 check_cycles: false,
                 invoked_by_ra: true,
+                buck2_command,
             };
             let out = OutputCfg { out, pretty: false };
 
@@ -226,6 +235,7 @@ impl Develop {
             relative_paths,
             buck,
             check_cycles,
+            buck2_command,
             ..
         } = self;
 
@@ -265,6 +275,7 @@ impl Develop {
             aliased_libraries,
             *relative_paths,
             *check_cycles,
+            buck2_command.clone(),
         )?;
 
         Ok(rust_project)

integrations/rust-project/src/main.rs

@@ -107,6 +107,10 @@ enum Command {
         /// Optional argument specifying build mode.
         #[clap(short = 'm', long)]
         mode: Option<String>,
+
+        /// Command used to run Buck2. Defaults to `buck2`.
+        #[clap(long)]
+        buck2_command: Option<String>,
     },
     /// `DevelopJson` is a more limited, stripped down [`Command::Develop`].
     ///
@@ -123,6 +127,10 @@ enum Command {
         sysroot_mode: SysrootMode,
 
         args: JsonArguments,
+
+        /// Command used to run Buck2. Defaults to `buck2`.
+        #[clap(long)]
+        buck2_command: Option<String>,
     },
     /// Build the saved file's owning target. This is meant to be used by IDEs to provide diagnostics on save.
     Check {
@@ -133,6 +141,9 @@ enum Command {
         use_clippy: bool,
         /// The file saved by the user. `rust-project` will infer the owning target(s) of the saved file and build them.
         saved_file: PathBuf,
+        /// Command used to run Buck2. Defaults to `buck2`.
+        #[clap(long)]
+        buck2_command: Option<String>,
     },
 }
 
@@ -266,11 +277,12 @@ fn main() -> Result<(), anyhow::Error> {
             mode,
             use_clippy,
             saved_file,
+            buck2_command,
         } => {
             let subscriber = tracing_subscriber::registry().with(fmt.with_filter(filter));
             tracing::subscriber::set_global_default(subscriber)?;
 
-            cli::Check::new(mode, use_clippy, saved_file).run()
+            cli::Check::new(buck2_command, mode, use_clippy, saved_file).run()
         }
     }
 }