compiler/rustc_builtin_macros/src/lib.rs

   1 //! This crate contains implementations of built-in macros and other code generating facilities
   2 //! injecting code into the crate before it is lowered to HIR.
   3
   4 #![allow(rustc::potential_query_instability)]
   5 #![doc(html_root_url = "https://doc.rust-lang.org/nightly/nightly-rustc/")]
   6 #![feature(array_windows)]
   7 #![feature(box_patterns)]
   8 #![feature(decl_macro)]
   9 #![feature(if_let_guard)]
  10 #![feature(is_some_and)]
  11 #![feature(is_sorted)]
  12 #![feature(let_chains)]
  13 #![feature(proc_macro_internals)]
  14 #![feature(proc_macro_quote)]
  15 #![recursion_limit = "256"]
  16
  17 extern crate proc_macro;
  18
  19 #[macro_use]
  20 extern crate tracing;
  21
  22 use crate::deriving::*;
  23
  24 use rustc_expand::base::{MacroExpanderFn, ResolverExpand, SyntaxExtensionKind};
  25 use rustc_expand::proc_macro::BangProcMacro;
  26 use rustc_span::symbol::sym;
  27
  28 mod alloc_error_handler;
  29 mod assert;
  30 mod cfg;
  31 mod cfg_accessible;
  32 mod cfg_eval;
  33 mod compile_error;
  34 mod concat;
  35 mod concat_bytes;
  36 mod concat_idents;
  37 mod derive;
  38 mod deriving;
  39 mod edition_panic;
  40 mod env;
  41 mod format;
  42 mod format_foreign;
  43 mod global_allocator;
  44 mod log_syntax;
  45 mod source_util;
  46 mod test;
  47 mod trace_macros;
  48 mod util;
  49
  50 pub mod asm;
  51 pub mod cmdline_attrs;
  52 pub mod proc_macro_harness;
  53 pub mod standard_library_imports;
  54 pub mod test_harness;
  55
  56 pub fn register_builtin_macros(resolver: &mut dyn ResolverExpand) {
  57     let mut register = |name, kind| resolver.register_builtin_macro(name, kind);
  58     macro register_bang($($name:ident: $f:expr,)*) {
  59         $(register(sym::$name, SyntaxExtensionKind::LegacyBang(Box::new($f as MacroExpanderFn)));)*
  60     }
  61     macro register_attr($($name:ident: $f:expr,)*) {
  62         $(register(sym::$name, SyntaxExtensionKind::LegacyAttr(Box::new($f)));)*
  63     }
  64     macro register_derive($($name:ident: $f:expr,)*) {
  65         $(register(sym::$name, SyntaxExtensionKind::LegacyDerive(Box::new(BuiltinDerive($f))));)*
  66     }
  67
  68     register_bang! {
  69         asm: asm::expand_asm,
  70         assert: assert::expand_assert,
  71         cfg: cfg::expand_cfg,
  72         column: source_util::expand_column,
  73         compile_error: compile_error::expand_compile_error,
  74         concat_bytes: concat_bytes::expand_concat_bytes,
  75         concat_idents: concat_idents::expand_concat_idents,
  76         concat: concat::expand_concat,
  77         env: env::expand_env,
  78         file: source_util::expand_file,
  79         format_args_nl: format::expand_format_args_nl,
  80         format_args: format::expand_format_args,
  81         const_format_args: format::expand_format_args,
  82         global_asm: asm::expand_global_asm,
  83         include_bytes: source_util::expand_include_bytes,
  84         include_str: source_util::expand_include_str,
  85         include: source_util::expand_include,
  86         line: source_util::expand_line,
  87         log_syntax: log_syntax::expand_log_syntax,
  88         module_path: source_util::expand_mod,
  89         option_env: env::expand_option_env,
  90         core_panic: edition_panic::expand_panic,
  91         std_panic: edition_panic::expand_panic,
  92         unreachable: edition_panic::expand_unreachable,
  93         stringify: source_util::expand_stringify,
  94         trace_macros: trace_macros::expand_trace_macros,
  95     }
  96
  97     register_attr! {
  98         alloc_error_handler: alloc_error_handler::expand,
  99         bench: test::expand_bench,
 100         cfg_accessible: cfg_accessible::Expander,
 101         cfg_eval: cfg_eval::expand,
 102         derive: derive::Expander(false),
 103         derive_const: derive::Expander(true),
 104         global_allocator: global_allocator::expand,
 105         test: test::expand_test,
 106         test_case: test::expand_test_case,
 107     }
 108
 109     register_derive! {
 110         Clone: clone::expand_deriving_clone,
 111         Copy: bounds::expand_deriving_copy,
 112         Debug: debug::expand_deriving_debug,
 113         Default: default::expand_deriving_default,
 114         Eq: eq::expand_deriving_eq,
 115         Hash: hash::expand_deriving_hash,
 116         Ord: ord::expand_deriving_ord,
 117         PartialEq: partial_eq::expand_deriving_partial_eq,
 118         PartialOrd: partial_ord::expand_deriving_partial_ord,
 119         RustcDecodable: decodable::expand_deriving_rustc_decodable,
 120         RustcEncodable: encodable::expand_deriving_rustc_encodable,
 121     }
 122
 123     let client = proc_macro::bridge::client::Client::expand1(proc_macro::quote);
 124     register(sym::quote, SyntaxExtensionKind::Bang(Box::new(BangProcMacro { client })));
 125 }