src/libsyntax_ext/cmdline_attrs.rs

   1 //! Attributes injected into the crate root from command line using `-Z crate-attr`.
   2
   3 use syntax::ast::{self, AttrItem, AttrStyle};
   4 use syntax::attr::mk_attr;
   5 use syntax::parse;
   6 use syntax::token;
   7 use syntax::sess::ParseSess;
   8 use syntax_expand::panictry;
   9 use syntax_pos::FileName;
  10
  11 pub fn inject(mut krate: ast::Crate, parse_sess: &ParseSess, attrs: &[String]) -> ast::Crate {
  12     for raw_attr in attrs {
  13         let mut parser = parse::new_parser_from_source_str(
  14             parse_sess,
  15             FileName::cli_crate_attr_source_code(&raw_attr),
  16             raw_attr.clone(),
  17         );
  18
  19         let start_span = parser.token.span;
  20         let AttrItem { path, tokens } = panictry!(parser.parse_attr_item());
  21         let end_span = parser.token.span;
  22         if parser.token != token::Eof {
  23             parse_sess.span_diagnostic
  24                 .span_err(start_span.to(end_span), "invalid crate attribute");
  25             continue;
  26         }
  27
  28         krate.attrs.push(mk_attr(AttrStyle::Inner, path, tokens, start_span.to(end_span)));
  29     }
  30
  31     krate
  32 }