scripts/cargo.rs

   1 use std::env;
   2 use std::os::unix::process::CommandExt;
   3 use std::path::PathBuf;
   4 use std::process::{Command, Stdio};
   5
   6 fn main() {
   7     if env::var("RUSTC_WRAPPER").map_or(false, |wrapper| wrapper.contains("sccache")) {
   8         eprintln!(
   9             "\x1b[1;93m=== Warning: Unsetting RUSTC_WRAPPER to prevent interference with sccache ===\x1b[0m"
  10         );
  11         env::remove_var("RUSTC_WRAPPER");
  12     }
  13
  14     let sysroot = PathBuf::from(env::current_exe().unwrap().parent().unwrap());
  15
  16     env::set_var("RUSTC", sysroot.join("bin/cg_clif".to_string() + env::consts::EXE_SUFFIX));
  17
  18     let mut rustdoc_flags = env::var("RUSTDOCFLAGS").unwrap_or(String::new());
  19     rustdoc_flags.push_str(" -Cpanic=abort -Zpanic-abort-tests -Zcodegen-backend=");
  20     rustdoc_flags.push_str(
  21         sysroot
  22             .join(if cfg!(windows) { "bin" } else { "lib" })
  23             .join(
  24                 env::consts::DLL_PREFIX.to_string()
  25                     + "rustc_codegen_cranelift"
  26                     + env::consts::DLL_SUFFIX,
  27             )
  28             .to_str()
  29             .unwrap(),
  30     );
  31     rustdoc_flags.push_str(" --sysroot ");
  32     rustdoc_flags.push_str(sysroot.to_str().unwrap());
  33     env::set_var("RUSTDOCFLAGS", rustdoc_flags);
  34
  35     let default_sysroot = Command::new("rustc")
  36         .stderr(Stdio::inherit())
  37         .args(&["--print", "sysroot"])
  38         .output()
  39         .unwrap()
  40         .stdout;
  41     let default_sysroot = std::str::from_utf8(&default_sysroot).unwrap().trim();
  42
  43     let extra_ld_lib_path =
  44         default_sysroot.to_string() + ":" + sysroot.join("lib").to_str().unwrap();
  45     if cfg!(target_os = "macos") {
  46         env::set_var(
  47             "DYLD_LIBRARY_PATH",
  48             env::var("DYLD_LIBRARY_PATH").unwrap_or(String::new()) + ":" + &extra_ld_lib_path,
  49         );
  50     } else if cfg!(unix) {
  51         env::set_var(
  52             "LD_LIBRARY_PATH",
  53             env::var("LD_LIBRARY_PATH").unwrap_or(String::new()) + ":" + &extra_ld_lib_path,
  54         );
  55     }
  56
  57     // Ensure that the right toolchain is used
  58     env::set_var("RUSTUP_TOOLCHAIN", env!("RUSTUP_TOOLCHAIN"));
  59
  60     let args: Vec<_> = match env::args().nth(1).as_deref() {
  61         Some("jit") => {
  62             env::set_var(
  63                 "RUSTFLAGS",
  64                 env::var("RUSTFLAGS").unwrap_or(String::new()) + " -Cprefer-dynamic",
  65             );
  66             std::array::IntoIter::new(["rustc".to_string()])
  67                 .chain(env::args().skip(2))
  68                 .chain(["--".to_string(), "-Cllvm-args=mode=jit".to_string()])
  69                 .collect()
  70         }
  71         Some("lazy-jit") => {
  72             env::set_var(
  73                 "RUSTFLAGS",
  74                 env::var("RUSTFLAGS").unwrap_or(String::new()) + " -Cprefer-dynamic",
  75             );
  76             std::array::IntoIter::new(["rustc".to_string()])
  77                 .chain(env::args().skip(2))
  78                 .chain(["--".to_string(), "-Cllvm-args=mode=jit-lazy".to_string()])
  79                 .collect()
  80         }
  81         _ => env::args().skip(1).collect(),
  82     };
  83
  84     Command::new("cargo").args(args).exec();
  85 }