compiler/rustc_builtin_macros/src/cmdline_attrs.rs

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