43 lines
1.2 KiB
Rust
43 lines
1.2 KiB
Rust
#![feature(rustc_private)]
|
|
|
|
extern crate rustc_driver;
|
|
extern crate rustc_interface;
|
|
|
|
use std::env;
|
|
use std::ffi::OsString;
|
|
use std::path::Path;
|
|
|
|
fn main() {
|
|
let mut callbacks = Callbacks;
|
|
let args = rustc_args(env::args_os().skip(1).collect());
|
|
rustc_driver::run_compiler(&args, &mut callbacks);
|
|
}
|
|
|
|
struct Callbacks;
|
|
|
|
impl rustc_driver::Callbacks for Callbacks {
|
|
fn config(&mut self, config: &mut rustc_interface::Config) {
|
|
let previous = config.register_lints.take();
|
|
config.register_lints = Some(Box::new(move |sess, lint_store| {
|
|
if let Some(previous) = &previous {
|
|
previous(sess, lint_store);
|
|
}
|
|
argument_comment_lint::register_lints(sess, lint_store);
|
|
}));
|
|
}
|
|
}
|
|
|
|
fn rustc_args(args: Vec<OsString>) -> Vec<String> {
|
|
let mut rustc_args: Vec<String> = args
|
|
.into_iter()
|
|
.map(|arg| arg.to_string_lossy().into_owned())
|
|
.collect();
|
|
if rustc_args.first().is_none_or(|arg| !is_rustc(arg)) {
|
|
rustc_args.insert(0, "rustc".to_string());
|
|
}
|
|
rustc_args
|
|
}
|
|
|
|
fn is_rustc(arg: &str) -> bool {
|
|
Path::new(arg).file_stem().and_then(|stem| stem.to_str()) == Some("rustc")
|
|
}
|