src/libsyntax_ext/lib.rs

   1 //! Syntax extensions in the Rust compiler.
   2
   3 #![doc(html_logo_url = "https://www.rust-lang.org/logos/rust-logo-128x128-blk-v2.png",
   4        html_favicon_url = "https://doc.rust-lang.org/favicon.ico",
   5        html_root_url = "https://doc.rust-lang.org/nightly/")]
   6
   7 #![deny(rust_2018_idioms)]
   8
   9 #![feature(in_band_lifetimes)]
  10 #![feature(proc_macro_diagnostic)]
  11 #![feature(proc_macro_internals)]
  12 #![feature(proc_macro_span)]
  13 #![feature(decl_macro)]
  14 #![feature(str_escape)]
  15 #![feature(rustc_diagnostic_macros)]
  16
  17 #![recursion_limit="256"]
  18
  19 extern crate proc_macro;
  20
  21 use rustc_errors as errors;
  22
  23 mod diagnostics;
  24
  25 mod asm;
  26 mod assert;
  27 mod cfg;
  28 mod compile_error;
  29 mod concat;
  30 mod concat_idents;
  31 mod env;
  32 mod format;
  33 mod format_foreign;
  34 mod global_asm;
  35 mod log_syntax;
  36 mod proc_macro_server;
  37 mod test;
  38 mod test_case;
  39 mod trace_macros;
  40
  41 pub mod deriving;
  42 pub mod proc_macro_decls;
  43 pub mod proc_macro_impl;
  44
  45 use rustc_data_structures::sync::Lrc;
  46 use syntax::ast;
  47 use syntax::ext::base::{MacroExpanderFn, NormalTT, NamedSyntaxExtension, MultiModifier};
  48 use syntax::ext::hygiene;
  49 use syntax::symbol::Symbol;
  50
  51 pub fn register_builtins(resolver: &mut dyn syntax::ext::base::Resolver,
  52                          user_exts: Vec<NamedSyntaxExtension>) {
  53     deriving::register_builtin_derives(resolver);
  54
  55     let mut register = |name, ext| {
  56         resolver.add_builtin(ast::Ident::with_empty_ctxt(name), Lrc::new(ext));
  57     };
  58
  59     macro_rules! register {
  60         ($( $name:ident: $f:expr, )*) => { $(
  61             register(Symbol::intern(stringify!($name)),
  62                      NormalTT {
  63                         expander: Box::new($f as MacroExpanderFn),
  64                         def_info: None,
  65                         allow_internal_unstable: false,
  66                         allow_internal_unsafe: false,
  67                         local_inner_macros: false,
  68                         unstable_feature: None,
  69                         edition: hygiene::default_edition(),
  70                     });
  71         )* }
  72     }
  73
  74     use syntax::ext::source_util::*;
  75     register! {
  76         line: expand_line,
  77         __rust_unstable_column: expand_column_gated,
  78         column: expand_column,
  79         file: expand_file,
  80         stringify: expand_stringify,
  81         include: expand_include,
  82         include_str: expand_include_str,
  83         include_bytes: expand_include_bytes,
  84         module_path: expand_mod,
  85
  86         asm: asm::expand_asm,
  87         global_asm: global_asm::expand_global_asm,
  88         cfg: cfg::expand_cfg,
  89         concat: concat::expand_syntax_ext,
  90         concat_idents: concat_idents::expand_syntax_ext,
  91         env: env::expand_env,
  92         option_env: env::expand_option_env,
  93         log_syntax: log_syntax::expand_syntax_ext,
  94         trace_macros: trace_macros::expand_trace_macros,
  95         compile_error: compile_error::expand_compile_error,
  96         assert: assert::expand_assert,
  97     }
  98
  99     register(Symbol::intern("test_case"), MultiModifier(Box::new(test_case::expand)));
 100     register(Symbol::intern("test"), MultiModifier(Box::new(test::expand_test)));
 101     register(Symbol::intern("bench"), MultiModifier(Box::new(test::expand_bench)));
 102
 103     // format_args uses `unstable` things internally.
 104     register(Symbol::intern("format_args"),
 105              NormalTT {
 106                 expander: Box::new(format::expand_format_args),
 107                 def_info: None,
 108                 allow_internal_unstable: true,
 109                 allow_internal_unsafe: false,
 110                 local_inner_macros: false,
 111                 unstable_feature: None,
 112                 edition: hygiene::default_edition(),
 113             });
 114     register(Symbol::intern("format_args_nl"),
 115              NormalTT {
 116                  expander: Box::new(format::expand_format_args_nl),
 117                  def_info: None,
 118                  allow_internal_unstable: true,
 119                  allow_internal_unsafe: false,
 120                  local_inner_macros: false,
 121                  unstable_feature: None,
 122                  edition: hygiene::default_edition(),
 123              });
 124
 125     for (name, ext) in user_exts {
 126         register(name, ext);
 127     }
 128 }