mirror of https://github.com/rust-lang/rust
41 lines
1.2 KiB
Rust
41 lines
1.2 KiB
Rust
//! Attributes injected into the crate root from command line using `-Z crate-attr`.
|
|
|
|
use crate::errors;
|
|
use rustc_ast::attr::mk_attr;
|
|
use rustc_ast::token;
|
|
use rustc_ast::{self as ast, AttrItem, AttrStyle};
|
|
use rustc_session::parse::ParseSess;
|
|
use rustc_span::FileName;
|
|
|
|
pub fn inject(krate: &mut ast::Crate, psess: &ParseSess, attrs: &[String]) {
|
|
for raw_attr in attrs {
|
|
let mut parser = rustc_parse::new_parser_from_source_str(
|
|
psess,
|
|
FileName::cli_crate_attr_source_code(raw_attr),
|
|
raw_attr.clone(),
|
|
);
|
|
|
|
let start_span = parser.token.span;
|
|
let AttrItem { path, args, tokens: _ } = match parser.parse_attr_item(false) {
|
|
Ok(ai) => ai,
|
|
Err(err) => {
|
|
err.emit();
|
|
continue;
|
|
}
|
|
};
|
|
let end_span = parser.token.span;
|
|
if parser.token != token::Eof {
|
|
psess.dcx.emit_err(errors::InvalidCrateAttr { span: start_span.to(end_span) });
|
|
continue;
|
|
}
|
|
|
|
krate.attrs.push(mk_attr(
|
|
&psess.attr_id_generator,
|
|
AttrStyle::Inner,
|
|
path,
|
|
args,
|
|
start_span.to(end_span),
|
|
));
|
|
}
|
|
}
|