Merge pull request #3087 from rust-lang-nursery/run
Use the compilers exit code computation instead of rolling our own
This commit is contained in:
Oliver Schneider
GitHub
commit
628934424e
1 changed files with 110 additions and 112 deletions
222
src/driver.rs
222
src/driver.rs
|
|
@ -13,125 +13,123 @@ fn show_version() {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn main() {
|
pub fn main() {
|
||||||
use std::env;
|
exit(rustc_driver::run(move || {
|
||||||
|
use std::env;
|
||||||
|
|
||||||
if std::env::args().any(|a| a == "--version" || a == "-V") {
|
if std::env::args().any(|a| a == "--version" || a == "-V") {
|
||||||
show_version();
|
show_version();
|
||||||
return;
|
exit(0);
|
||||||
}
|
|
||||||
|
|
||||||
let sys_root = option_env!("SYSROOT")
|
|
||||||
.map(String::from)
|
|
||||||
.or_else(|| std::env::var("SYSROOT").ok())
|
|
||||||
.or_else(|| {
|
|
||||||
let home = option_env!("RUSTUP_HOME").or(option_env!("MULTIRUST_HOME"));
|
|
||||||
let toolchain = option_env!("RUSTUP_TOOLCHAIN").or(option_env!("MULTIRUST_TOOLCHAIN"));
|
|
||||||
home.and_then(|home| toolchain.map(|toolchain| format!("{}/toolchains/{}", home, toolchain)))
|
|
||||||
})
|
|
||||||
.or_else(|| {
|
|
||||||
Command::new("rustc")
|
|
||||||
.arg("--print")
|
|
||||||
.arg("sysroot")
|
|
||||||
.output()
|
|
||||||
.ok()
|
|
||||||
.and_then(|out| String::from_utf8(out.stdout).ok())
|
|
||||||
.map(|s| s.trim().to_owned())
|
|
||||||
})
|
|
||||||
.expect("need to specify SYSROOT env var during clippy compilation, or use rustup or multirust");
|
|
||||||
|
|
||||||
// Setting RUSTC_WRAPPER causes Cargo to pass 'rustc' as the first argument.
|
|
||||||
// We're invoking the compiler programmatically, so we ignore this/
|
|
||||||
let mut orig_args: Vec<String> = env::args().collect();
|
|
||||||
if orig_args.len() <= 1 {
|
|
||||||
std::process::exit(1);
|
|
||||||
}
|
|
||||||
if orig_args[1] == "rustc" {
|
|
||||||
// we still want to be able to invoke it normally though
|
|
||||||
orig_args.remove(1);
|
|
||||||
}
|
|
||||||
// this conditional check for the --sysroot flag is there so users can call
|
|
||||||
// `clippy_driver` directly
|
|
||||||
// without having to pass --sysroot or anything
|
|
||||||
let mut args: Vec<String> = if orig_args.iter().any(|s| s == "--sysroot") {
|
|
||||||
orig_args.clone()
|
|
||||||
} else {
|
|
||||||
orig_args
|
|
||||||
.clone()
|
|
||||||
.into_iter()
|
|
||||||
.chain(Some("--sysroot".to_owned()))
|
|
||||||
.chain(Some(sys_root))
|
|
||||||
.collect()
|
|
||||||
};
|
|
||||||
|
|
||||||
// this check ensures that dependencies are built but not linted and the final
|
|
||||||
// crate is
|
|
||||||
// linted but not built
|
|
||||||
let clippy_enabled = env::var("CLIPPY_TESTS").ok().map_or(false, |val| val == "true")
|
|
||||||
|| orig_args.iter().any(|s| s == "--emit=dep-info,metadata");
|
|
||||||
|
|
||||||
if clippy_enabled {
|
|
||||||
args.extend_from_slice(&["--cfg".to_owned(), r#"feature="cargo-clippy""#.to_owned()]);
|
|
||||||
if let Ok(extra_args) = env::var("CLIPPY_ARGS") {
|
|
||||||
args.extend(
|
|
||||||
extra_args
|
|
||||||
.split("__CLIPPY_HACKERY__")
|
|
||||||
.filter(|s| !s.is_empty())
|
|
||||||
.map(str::to_owned),
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
let mut controller = CompileController::basic();
|
let sys_root = option_env!("SYSROOT")
|
||||||
if clippy_enabled {
|
.map(String::from)
|
||||||
controller.after_parse.callback = Box::new(move |state| {
|
.or_else(|| std::env::var("SYSROOT").ok())
|
||||||
let mut registry = rustc_plugin::registry::Registry::new(
|
.or_else(|| {
|
||||||
state.session,
|
let home = option_env!("RUSTUP_HOME").or(option_env!("MULTIRUST_HOME"));
|
||||||
state
|
let toolchain = option_env!("RUSTUP_TOOLCHAIN").or(option_env!("MULTIRUST_TOOLCHAIN"));
|
||||||
.krate
|
home.and_then(|home| toolchain.map(|toolchain| format!("{}/toolchains/{}", home, toolchain)))
|
||||||
.as_ref()
|
})
|
||||||
.expect(
|
.or_else(|| {
|
||||||
"at this compilation stage \
|
Command::new("rustc")
|
||||||
the crate must be parsed",
|
.arg("--print")
|
||||||
)
|
.arg("sysroot")
|
||||||
.span,
|
.output()
|
||||||
);
|
.ok()
|
||||||
registry.args_hidden = Some(Vec::new());
|
.and_then(|out| String::from_utf8(out.stdout).ok())
|
||||||
|
.map(|s| s.trim().to_owned())
|
||||||
|
})
|
||||||
|
.expect("need to specify SYSROOT env var during clippy compilation, or use rustup or multirust");
|
||||||
|
|
||||||
let conf = clippy_lints::read_conf(®istry);
|
// Setting RUSTC_WRAPPER causes Cargo to pass 'rustc' as the first argument.
|
||||||
clippy_lints::register_plugins(&mut registry, &conf);
|
// We're invoking the compiler programmatically, so we ignore this/
|
||||||
|
let mut orig_args: Vec<String> = env::args().collect();
|
||||||
|
if orig_args.len() <= 1 {
|
||||||
|
std::process::exit(1);
|
||||||
|
}
|
||||||
|
if orig_args[1] == "rustc" {
|
||||||
|
// we still want to be able to invoke it normally though
|
||||||
|
orig_args.remove(1);
|
||||||
|
}
|
||||||
|
// this conditional check for the --sysroot flag is there so users can call
|
||||||
|
// `clippy_driver` directly
|
||||||
|
// without having to pass --sysroot or anything
|
||||||
|
let mut args: Vec<String> = if orig_args.iter().any(|s| s == "--sysroot") {
|
||||||
|
orig_args.clone()
|
||||||
|
} else {
|
||||||
|
orig_args
|
||||||
|
.clone()
|
||||||
|
.into_iter()
|
||||||
|
.chain(Some("--sysroot".to_owned()))
|
||||||
|
.chain(Some(sys_root))
|
||||||
|
.collect()
|
||||||
|
};
|
||||||
|
|
||||||
let rustc_plugin::registry::Registry {
|
// this check ensures that dependencies are built but not linted and the final
|
||||||
early_lint_passes,
|
// crate is
|
||||||
late_lint_passes,
|
// linted but not built
|
||||||
lint_groups,
|
let clippy_enabled = env::var("CLIPPY_TESTS").ok().map_or(false, |val| val == "true")
|
||||||
llvm_passes,
|
|| orig_args.iter().any(|s| s == "--emit=dep-info,metadata");
|
||||||
attributes,
|
|
||||||
..
|
if clippy_enabled {
|
||||||
} = registry;
|
args.extend_from_slice(&["--cfg".to_owned(), r#"feature="cargo-clippy""#.to_owned()]);
|
||||||
let sess = &state.session;
|
if let Ok(extra_args) = env::var("CLIPPY_ARGS") {
|
||||||
let mut ls = sess.lint_store.borrow_mut();
|
args.extend(
|
||||||
for pass in early_lint_passes {
|
extra_args
|
||||||
ls.register_early_pass(Some(sess), true, pass);
|
.split("__CLIPPY_HACKERY__")
|
||||||
}
|
.filter(|s| !s.is_empty())
|
||||||
for pass in late_lint_passes {
|
.map(str::to_owned),
|
||||||
ls.register_late_pass(Some(sess), true, pass);
|
);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
for (name, to) in lint_groups {
|
let mut controller = CompileController::basic();
|
||||||
ls.register_group(Some(sess), true, name, to);
|
if clippy_enabled {
|
||||||
}
|
controller.after_parse.callback = Box::new(move |state| {
|
||||||
clippy_lints::register_pre_expansion_lints(sess, &mut ls, &conf);
|
let mut registry = rustc_plugin::registry::Registry::new(
|
||||||
|
state.session,
|
||||||
|
state
|
||||||
|
.krate
|
||||||
|
.as_ref()
|
||||||
|
.expect(
|
||||||
|
"at this compilation stage \
|
||||||
|
the crate must be parsed",
|
||||||
|
)
|
||||||
|
.span,
|
||||||
|
);
|
||||||
|
registry.args_hidden = Some(Vec::new());
|
||||||
|
|
||||||
sess.plugin_llvm_passes.borrow_mut().extend(llvm_passes);
|
let conf = clippy_lints::read_conf(®istry);
|
||||||
sess.plugin_attributes.borrow_mut().extend(attributes);
|
clippy_lints::register_plugins(&mut registry, &conf);
|
||||||
});
|
|
||||||
}
|
|
||||||
controller.compilation_done.stop = Compilation::Stop;
|
|
||||||
|
|
||||||
if rustc_driver::run_compiler(&args, Box::new(controller), None, None)
|
let rustc_plugin::registry::Registry {
|
||||||
.0
|
early_lint_passes,
|
||||||
.is_err()
|
late_lint_passes,
|
||||||
{
|
lint_groups,
|
||||||
exit(101);
|
llvm_passes,
|
||||||
}
|
attributes,
|
||||||
|
..
|
||||||
|
} = registry;
|
||||||
|
let sess = &state.session;
|
||||||
|
let mut ls = sess.lint_store.borrow_mut();
|
||||||
|
for pass in early_lint_passes {
|
||||||
|
ls.register_early_pass(Some(sess), true, pass);
|
||||||
|
}
|
||||||
|
for pass in late_lint_passes {
|
||||||
|
ls.register_late_pass(Some(sess), true, pass);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (name, to) in lint_groups {
|
||||||
|
ls.register_group(Some(sess), true, name, to);
|
||||||
|
}
|
||||||
|
clippy_lints::register_pre_expansion_lints(sess, &mut ls, &conf);
|
||||||
|
|
||||||
|
sess.plugin_llvm_passes.borrow_mut().extend(llvm_passes);
|
||||||
|
sess.plugin_attributes.borrow_mut().extend(attributes);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
controller.compilation_done.stop = Compilation::Stop;
|
||||||
|
|
||||||
|
let args = args;
|
||||||
|
rustc_driver::run_compiler(&args, Box::new(controller), None, None)
|
||||||
|
}) as i32)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue