mirror of
https://github.com/rust-lang/rust.git
synced 2024-12-05 05:04:24 +00:00
158 lines
4.5 KiB
Rust
158 lines
4.5 KiB
Rust
use std::env;
|
|
use std::path::PathBuf;
|
|
use std::process;
|
|
|
|
use self::utils::is_ci;
|
|
|
|
mod abi_cafe;
|
|
mod build_backend;
|
|
mod build_sysroot;
|
|
mod config;
|
|
mod prepare;
|
|
mod rustc_info;
|
|
mod tests;
|
|
mod utils;
|
|
|
|
fn usage() {
|
|
eprintln!("Usage:");
|
|
eprintln!(" ./y.rs prepare");
|
|
eprintln!(
|
|
" ./y.rs build [--debug] [--sysroot none|clif|llvm] [--target-dir DIR] [--no-unstable-features]"
|
|
);
|
|
eprintln!(
|
|
" ./y.rs test [--debug] [--sysroot none|clif|llvm] [--target-dir DIR] [--no-unstable-features]"
|
|
);
|
|
}
|
|
|
|
macro_rules! arg_error {
|
|
($($err:tt)*) => {{
|
|
eprintln!($($err)*);
|
|
usage();
|
|
std::process::exit(1);
|
|
}};
|
|
}
|
|
|
|
#[derive(PartialEq, Debug)]
|
|
enum Command {
|
|
Build,
|
|
Test,
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug)]
|
|
pub(crate) enum SysrootKind {
|
|
None,
|
|
Clif,
|
|
Llvm,
|
|
}
|
|
|
|
pub fn main() {
|
|
env::set_var("CG_CLIF_DISPLAY_CG_TIME", "1");
|
|
env::set_var("CG_CLIF_DISABLE_INCR_CACHE", "1");
|
|
// The target dir is expected in the default location. Guard against the user changing it.
|
|
env::set_var("CARGO_TARGET_DIR", "target");
|
|
|
|
if is_ci() {
|
|
// Disabling incr comp reduces cache size and incr comp doesn't save as much on CI anyway
|
|
env::set_var("CARGO_BUILD_INCREMENTAL", "false");
|
|
}
|
|
|
|
let mut args = env::args().skip(1);
|
|
let command = match args.next().as_deref() {
|
|
Some("prepare") => {
|
|
if args.next().is_some() {
|
|
arg_error!("./y.rs prepare doesn't expect arguments");
|
|
}
|
|
prepare::prepare();
|
|
process::exit(0);
|
|
}
|
|
Some("build") => Command::Build,
|
|
Some("test") => Command::Test,
|
|
Some(flag) if flag.starts_with('-') => arg_error!("Expected command found flag {}", flag),
|
|
Some(command) => arg_error!("Unknown command {}", command),
|
|
None => {
|
|
usage();
|
|
process::exit(0);
|
|
}
|
|
};
|
|
|
|
let mut target_dir = PathBuf::from("build");
|
|
let mut channel = "release";
|
|
let mut sysroot_kind = SysrootKind::Clif;
|
|
let mut use_unstable_features = true;
|
|
while let Some(arg) = args.next().as_deref() {
|
|
match arg {
|
|
"--target-dir" => {
|
|
target_dir = PathBuf::from(args.next().unwrap_or_else(|| {
|
|
arg_error!("--target-dir requires argument");
|
|
}))
|
|
}
|
|
"--debug" => channel = "debug",
|
|
"--sysroot" => {
|
|
sysroot_kind = match args.next().as_deref() {
|
|
Some("none") => SysrootKind::None,
|
|
Some("clif") => SysrootKind::Clif,
|
|
Some("llvm") => SysrootKind::Llvm,
|
|
Some(arg) => arg_error!("Unknown sysroot kind {}", arg),
|
|
None => arg_error!("--sysroot requires argument"),
|
|
}
|
|
}
|
|
"--no-unstable-features" => use_unstable_features = false,
|
|
flag if flag.starts_with("-") => arg_error!("Unknown flag {}", flag),
|
|
arg => arg_error!("Unexpected argument {}", arg),
|
|
}
|
|
}
|
|
target_dir = std::env::current_dir().unwrap().join(target_dir);
|
|
|
|
let host_triple = if let Ok(host_triple) = std::env::var("HOST_TRIPLE") {
|
|
host_triple
|
|
} else if let Some(host_triple) = config::get_value("host") {
|
|
host_triple
|
|
} else {
|
|
rustc_info::get_host_triple()
|
|
};
|
|
let target_triple = if let Ok(target_triple) = std::env::var("TARGET_TRIPLE") {
|
|
if target_triple != "" {
|
|
target_triple
|
|
} else {
|
|
host_triple.clone() // Empty target triple can happen on GHA
|
|
}
|
|
} else if let Some(target_triple) = config::get_value("target") {
|
|
target_triple
|
|
} else {
|
|
host_triple.clone()
|
|
};
|
|
|
|
let cg_clif_dylib = build_backend::build_backend(channel, &host_triple, use_unstable_features);
|
|
match command {
|
|
Command::Test => {
|
|
tests::run_tests(
|
|
channel,
|
|
sysroot_kind,
|
|
&target_dir,
|
|
&cg_clif_dylib,
|
|
&host_triple,
|
|
&target_triple,
|
|
);
|
|
|
|
abi_cafe::run(
|
|
channel,
|
|
sysroot_kind,
|
|
&target_dir,
|
|
&cg_clif_dylib,
|
|
&host_triple,
|
|
&target_triple,
|
|
);
|
|
}
|
|
Command::Build => {
|
|
build_sysroot::build_sysroot(
|
|
channel,
|
|
sysroot_kind,
|
|
&target_dir,
|
|
&cg_clif_dylib,
|
|
&host_triple,
|
|
&target_triple,
|
|
);
|
|
}
|
|
}
|
|
}
|