feat(toolchain): consider external rust-analyzer when calling a proxy

Author: rami3l

src/test/mock_bin_src.rs

@@ -102,6 +102,10 @@ fn main() {
                 panic!("CARGO environment variable not set");
             }
         }
+        Some("--echo-current-exe") => {
+            let mut out = io::stderr();
+            writeln!(out, "{}", std::env::current_exe().unwrap().display()).unwrap();
+        }
         arg => panic!("bad mock proxy commandline: {:?}", arg),
     }
 }

src/toolchain.rs

@@ -12,6 +12,7 @@ use std::{
 
 use anyhow::{Context, anyhow, bail};
 use fs_at::OpenOptions;
+use same_file::is_same_file;
 use tracing::info;
 use url::Url;
 use wait_timeout::ChildExt;
@@ -331,6 +332,8 @@ impl<'a> Toolchain<'a> {
         // perhaps a trait that create command layers on?
         if let Some(cmd) = self.maybe_do_cargo_fallback(binary)? {
             return Ok(cmd);
+        } else if let Some(cmd) = self.maybe_do_rust_analyzer_fallback(binary)? {
+            return Ok(cmd);
         }
 
         self.create_command(binary)
@@ -371,6 +374,37 @@ impl<'a> Toolchain<'a> {
         Ok(None)
     }
 
+    fn maybe_do_rust_analyzer_fallback(&self, binary: &str) -> anyhow::Result<Option<Command>> {
+        if binary != "rust-analyzer" && binary != "rust-analyzer.exe"
+            || self.binary_file("rust-analyzer").exists()
+        {
+            return Ok(None);
+        }
+
+        let proc = self.cfg.process;
+        let Some(path) = proc.var_os("PATH") else {
+            return Ok(None);
+        };
+
+        let me = env::current_exe()?;
+
+        // Try to find the first `rust-analyzer` under the `$PATH` that is both
+        // an existing file and not the same file as `me`, i.e. not a rustup proxy.
+        Ok(env::split_paths(&path).find_map(|mut p| {
+            p.push(binary);
+            let is_external_ra = p.is_file()
+                // We report `true` on `is_same_file()` error to prevent an invalid `p`
+                // from becoming the candidate.
+                && !is_same_file(&me, &p).unwrap_or(true);
+            if !is_external_ra {
+                return None;
+            }
+            let mut ra = Command::new(p);
+            self.set_env(&mut ra);
+            Some(ra)
+        }))
+    }
+
     #[cfg_attr(feature="otel", tracing::instrument(err, fields(binary, recursion = self.cfg.process.var("RUST_RECURSION_COUNT").ok())))]
     fn create_command<T: AsRef<OsStr> + Debug>(&self, binary: T) -> Result<Command, anyhow::Error> {
         // Create the path to this binary within the current toolchain sysroot

tests/suite/cli_misc.rs

@@ -1277,3 +1277,73 @@ async fn rustup_updates_cargo_env_if_proxy() {
         )
         .await;
 }
+
+#[tokio::test]
+async fn rust_analyzer_proxy_falls_back_external() {
+    let mut cx = CliTestContext::new(Scenario::SimpleV2).await;
+    cx.config
+        .expect_ok(&[
+            "rustup",
+            "toolchain",
+            "install",
+            "stable",
+            "--profile=minimal",
+            "--component=rls",
+        ])
+        .await;
+    cx.config.expect_ok(&["rustup", "default", "stable"]).await;
+
+    // We pretend to have a `rust-analyzer` installation by reusing the `rls`
+    // proxy and mock binary.
+    let rls = format!("rls{EXE_SUFFIX}");
+    let ra = format!("rust-analyzer{EXE_SUFFIX}");
+    let exedir = &cx.config.exedir;
+    let bindir = &cx
+        .config
+        .rustupdir
+        .join("toolchains")
+        .join(for_host!("stable-{0}"))
+        .join("bin");
+    for dir in [exedir, bindir] {
+        fs::rename(dir.join(&rls), dir.join(&ra)).unwrap();
+    }
+
+    // Base case: rustup-hosted RA installed, external RA unavailable,
+    // use the former.
+    let real_path = cx
+        .config
+        .run("rust-analyzer", &["--echo-current-exe"], &[])
+        .await;
+    assert!(real_path.ok);
+    let real_path_str = real_path.stderr.lines().next().unwrap();
+    let real_path = Path::new(real_path_str);
+
+    assert!(real_path.is_file());
+
+    let tempdir = tempfile::Builder::new().prefix("rustup").tempdir().unwrap();
+    let extern_dir = tempdir.path();
+    let extern_path = &extern_dir.join("rust-analyzer");
+    fs::copy(real_path, extern_path).unwrap();
+
+    // First case: rustup-hosted and external RA both installed,
+    // prioritize the former.
+    cx.config
+        .expect_ok_ex_env(
+            &["rust-analyzer", "--echo-current-exe"],
+            &[("PATH", &extern_dir.to_string_lossy())],
+            "",
+            &format!("{real_path_str}\n"),
+        )
+        .await;
+
+    // Second case: rustup-hosted RA unavailable, fallback on the external RA.
+    fs::remove_file(bindir.join(&ra)).unwrap();
+    cx.config
+        .expect_ok_ex_env(
+            &["rust-analyzer", "--echo-current-exe"],
+            &[("PATH", &extern_dir.to_string_lossy())],
+            "",
+            &format!("{}\n", extern_path.display()),
+        )
+        .await;
+}