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