rustc_builtin_macros/
cmdline_attrs.rs

Help
1//! Attributes injected into the crate root from command line using `-Z crate-attr`.
2
3use rustc_ast::attr::mk_attr;
4use rustc_ast::{self as ast, AttrItem, AttrStyle, token};
5use rustc_parse::parser::ForceCollect;
6use rustc_parse::{new_parser_from_source_str, unwrap_or_emit_fatal};
7use rustc_session::parse::ParseSess;
8use rustc_span::FileName;
9
10use crate::errors;
11
12pub fn inject(krate: &mut ast::Crate, psess: &ParseSess, attrs: &[String]) {
13    for raw_attr in attrs {
14        let mut parser = unwrap_or_emit_fatal(new_parser_from_source_str(
15            psess,
16            FileName::cli_crate_attr_source_code(raw_attr),
17            raw_attr.clone(),
18        ));
19
20        let start_span = parser.token.span;
21        let AttrItem { unsafety, path, args, tokens: _ } =
22            match parser.parse_attr_item(ForceCollect::No) {
23                Ok(ai) => ai,
24                Err(err) => {
25                    err.emit();
26                    continue;
27                }
28            };
29        let end_span = parser.token.span;
30        if parser.token != token::Eof {
31            psess.dcx().emit_err(errors::InvalidCrateAttr { span: start_span.to(end_span) });
32            continue;
33        }
34
35        krate.attrs.push(mk_attr(
36            &psess.attr_id_generator,
37            AttrStyle::Inner,
38            unsafety,
39            path,
40            args,
41            start_span.to(end_span),
42        ));
43    }
44}
rustc_builtin_macros/cmdline_attrs.rs

rustc_builtin_macros/
cmdline_attrs.rs
