1use std::any::Any;
2use std::default::Default;
3use std::iter;
4use std::path::Component::Prefix;
5use std::path::PathBuf;
6use std::rc::Rc;
7use std::sync::Arc;
8
9use rustc_ast::attr::MarkedAttrs;
10use rustc_ast::tokenstream::TokenStream;
11use rustc_ast::visit::{AssocCtxt, Visitor};
12use rustc_ast::{self as ast, AttrVec, Attribute, HasAttrs, Item, NodeId, PatKind, Safety};
13use rustc_data_structures::fx::{FxHashMap, FxIndexMap};
14use rustc_data_structures::sync;
15use rustc_errors::{BufferedEarlyLint, DiagCtxtHandle, ErrorGuaranteed, PResult};
16use rustc_feature::Features;
17use rustc_hir as hir;
18use rustc_hir::attrs::{CfgEntry, CollapseMacroDebuginfo, Deprecation};
19use rustc_hir::def::MacroKinds;
20use rustc_hir::limit::Limit;
21use rustc_hir::{Stability, find_attr};
22use rustc_lint_defs::RegisteredTools;
23use rustc_parse::MACRO_ARGUMENTS;
24use rustc_parse::parser::Parser;
25use rustc_session::Session;
26use rustc_session::parse::ParseSess;
27use rustc_span::def_id::{CrateNum, DefId, LocalDefId};
28use rustc_span::edition::Edition;
29use rustc_span::hygiene::{AstPass, ExpnData, ExpnKind, LocalExpnId, MacroKind};
30use rustc_span::source_map::SourceMap;
31use rustc_span::{DUMMY_SP, Ident, Span, Symbol, kw};
32use smallvec::{SmallVec, smallvec};
33use thin_vec::ThinVec;
34
35use crate::diagnostics;
36use crate::expand::{self, AstFragment, Invocation};
37use crate::mbe::macro_rules::ParserAnyMacro;
38use crate::module::DirOwnership;
39use crate::stats::MacroStat;
40
41#[derive(#[automatically_derived]
impl ::core::fmt::Debug for Annotatable {
#[inline]
fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
match self {
Annotatable::Item(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f, "Item",
&__self_0),
Annotatable::AssocItem(__self_0, __self_1) =>
::core::fmt::Formatter::debug_tuple_field2_finish(f,
"AssocItem", __self_0, &__self_1),
Annotatable::ForeignItem(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"ForeignItem", &__self_0),
Annotatable::Stmt(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f, "Stmt",
&__self_0),
Annotatable::Expr(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f, "Expr",
&__self_0),
Annotatable::Arm(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f, "Arm",
&__self_0),
Annotatable::ExprField(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"ExprField", &__self_0),
Annotatable::PatField(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"PatField", &__self_0),
Annotatable::GenericParam(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"GenericParam", &__self_0),
Annotatable::Param(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f, "Param",
&__self_0),
Annotatable::FieldDef(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"FieldDef", &__self_0),
Annotatable::Variant(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"Variant", &__self_0),
Annotatable::WherePredicate(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f,
"WherePredicate", &__self_0),
Annotatable::Crate(__self_0) =>
::core::fmt::Formatter::debug_tuple_field1_finish(f, "Crate",
&__self_0),
}
}
}Debug, #[automatically_derived]
impl ::core::clone::Clone for Annotatable {
#[inline]
fn clone(&self) -> Annotatable {
match self {
Annotatable::Item(__self_0) =>
Annotatable::Item(::core::clone::Clone::clone(__self_0)),
Annotatable::AssocItem(__self_0, __self_1) =>
Annotatable::AssocItem(::core::clone::Clone::clone(__self_0),
::core::clone::Clone::clone(__self_1)),
Annotatable::ForeignItem(__self_0) =>
Annotatable::ForeignItem(::core::clone::Clone::clone(__self_0)),
Annotatable::Stmt(__self_0) =>
Annotatable::Stmt(::core::clone::Clone::clone(__self_0)),
Annotatable::Expr(__self_0) =>
Annotatable::Expr(::core::clone::Clone::clone(__self_0)),
Annotatable::Arm(__self_0) =>
Annotatable::Arm(::core::clone::Clone::clone(__self_0)),
Annotatable::ExprField(__self_0) =>
Annotatable::ExprField(::core::clone::Clone::clone(__self_0)),
Annotatable::PatField(__self_0) =>
Annotatable::PatField(::core::clone::Clone::clone(__self_0)),
Annotatable::GenericParam(__self_0) =>
Annotatable::GenericParam(::core::clone::Clone::clone(__self_0)),
Annotatable::Param(__self_0) =>
Annotatable::Param(::core::clone::Clone::clone(__self_0)),
Annotatable::FieldDef(__self_0) =>
Annotatable::FieldDef(::core::clone::Clone::clone(__self_0)),
Annotatable::Variant(__self_0) =>
Annotatable::Variant(::core::clone::Clone::clone(__self_0)),
Annotatable::WherePredicate(__self_0) =>
Annotatable::WherePredicate(::core::clone::Clone::clone(__self_0)),
Annotatable::Crate(__self_0) =>
Annotatable::Crate(::core::clone::Clone::clone(__self_0)),
}
}
}Clone)]
45pub enum Annotatable {
46 Item(Box<ast::Item>),
47 AssocItem(Box<ast::AssocItem>, AssocCtxt),
48 ForeignItem(Box<ast::ForeignItem>),
49 Stmt(Box<ast::Stmt>),
50 Expr(Box<ast::Expr>),
51 Arm(ast::Arm),
52 ExprField(ast::ExprField),
53 PatField(ast::PatField),
54 GenericParam(ast::GenericParam),
55 Param(ast::Param),
56 FieldDef(ast::FieldDef),
57 Variant(ast::Variant),
58 WherePredicate(ast::WherePredicate),
59 Crate(ast::Crate),
60}
61
62impl Annotatable {
63 pub fn span(&self) -> Span {
64 match self {
65 Annotatable::Item(item) => item.span,
66 Annotatable::AssocItem(assoc_item, _) => assoc_item.span,
67 Annotatable::ForeignItem(foreign_item) => foreign_item.span,
68 Annotatable::Stmt(stmt) => stmt.span,
69 Annotatable::Expr(expr) => expr.span,
70 Annotatable::Arm(arm) => arm.span,
71 Annotatable::ExprField(field) => field.span,
72 Annotatable::PatField(fp) => fp.pat.span,
73 Annotatable::GenericParam(gp) => gp.ident.span,
74 Annotatable::Param(p) => p.span,
75 Annotatable::FieldDef(sf) => sf.span,
76 Annotatable::Variant(v) => v.span,
77 Annotatable::WherePredicate(wp) => wp.span,
78 Annotatable::Crate(c) => c.spans.inner_span,
79 }
80 }
81
82 pub fn visit_attrs(&mut self, f: impl FnOnce(&mut AttrVec)) {
83 match self {
84 Annotatable::Item(item) => item.visit_attrs(f),
85 Annotatable::AssocItem(assoc_item, _) => assoc_item.visit_attrs(f),
86 Annotatable::ForeignItem(foreign_item) => foreign_item.visit_attrs(f),
87 Annotatable::Stmt(stmt) => stmt.visit_attrs(f),
88 Annotatable::Expr(expr) => expr.visit_attrs(f),
89 Annotatable::Arm(arm) => arm.visit_attrs(f),
90 Annotatable::ExprField(field) => field.visit_attrs(f),
91 Annotatable::PatField(fp) => fp.visit_attrs(f),
92 Annotatable::GenericParam(gp) => gp.visit_attrs(f),
93 Annotatable::Param(p) => p.visit_attrs(f),
94 Annotatable::FieldDef(sf) => sf.visit_attrs(f),
95 Annotatable::Variant(v) => v.visit_attrs(f),
96 Annotatable::WherePredicate(wp) => wp.visit_attrs(f),
97 Annotatable::Crate(c) => c.visit_attrs(f),
98 }
99 }
100
101 pub fn visit_with<'a, V: Visitor<'a>>(&'a self, visitor: &mut V) -> V::Result {
102 match self {
103 Annotatable::Item(item) => visitor.visit_item(item),
104 Annotatable::AssocItem(item, ctxt) => visitor.visit_assoc_item(item, *ctxt),
105 Annotatable::ForeignItem(foreign_item) => visitor.visit_foreign_item(foreign_item),
106 Annotatable::Stmt(stmt) => visitor.visit_stmt(stmt),
107 Annotatable::Expr(expr) => visitor.visit_expr(expr),
108 Annotatable::Arm(arm) => visitor.visit_arm(arm),
109 Annotatable::ExprField(field) => visitor.visit_expr_field(field),
110 Annotatable::PatField(fp) => visitor.visit_pat_field(fp),
111 Annotatable::GenericParam(gp) => visitor.visit_generic_param(gp),
112 Annotatable::Param(p) => visitor.visit_param(p),
113 Annotatable::FieldDef(sf) => visitor.visit_field_def(sf),
114 Annotatable::Variant(v) => visitor.visit_variant(v),
115 Annotatable::WherePredicate(wp) => visitor.visit_where_predicate(wp),
116 Annotatable::Crate(c) => visitor.visit_crate(c),
117 }
118 }
119
120 pub fn to_tokens(&self) -> TokenStream {
121 match self {
122 Annotatable::Item(node) => TokenStream::from_ast(node),
123 Annotatable::AssocItem(node, _) => TokenStream::from_ast(node),
124 Annotatable::ForeignItem(node) => TokenStream::from_ast(node),
125 Annotatable::Stmt(node) => {
126 if !!#[allow(non_exhaustive_omitted_patterns)] match node.kind {
ast::StmtKind::Empty => true,
_ => false,
} {
::core::panicking::panic("assertion failed: !matches!(node.kind, ast::StmtKind::Empty)")
};assert!(!matches!(node.kind, ast::StmtKind::Empty));
127 TokenStream::from_ast(node)
128 }
129 Annotatable::Expr(node) => TokenStream::from_ast(node),
130 Annotatable::Arm(..)
131 | Annotatable::ExprField(..)
132 | Annotatable::PatField(..)
133 | Annotatable::GenericParam(..)
134 | Annotatable::Param(..)
135 | Annotatable::FieldDef(..)
136 | Annotatable::Variant(..)
137 | Annotatable::WherePredicate(..)
138 | Annotatable::Crate(..) => { ::core::panicking::panic_fmt(format_args!("unexpected annotatable")); }panic!("unexpected annotatable"),
139 }
140 }
141
142 pub fn expect_item(self) -> Box<ast::Item> {
143 match self {
144 Annotatable::Item(i) => i,
145 _ => { ::core::panicking::panic_fmt(format_args!("expected Item")); }panic!("expected Item"),
146 }
147 }
148
149 pub fn expect_trait_item(self) -> Box<ast::AssocItem> {
150 match self {
151 Annotatable::AssocItem(i, AssocCtxt::Trait) => i,
152 _ => { ::core::panicking::panic_fmt(format_args!("expected trait item")); }panic!("expected trait item"),
153 }
154 }
155
156 pub fn expect_impl_item(self) -> Box<ast::AssocItem> {
157 match self {
158 Annotatable::AssocItem(i, AssocCtxt::Impl { .. }) => i,
159 _ => { ::core::panicking::panic_fmt(format_args!("expected impl item")); }panic!("expected impl item"),
160 }
161 }
162
163 pub fn expect_foreign_item(self) -> Box<ast::ForeignItem> {
164 match self {
165 Annotatable::ForeignItem(i) => i,
166 _ => { ::core::panicking::panic_fmt(format_args!("expected foreign item")); }panic!("expected foreign item"),
167 }
168 }
169
170 pub fn expect_stmt(self) -> ast::Stmt {
171 match self {
172 Annotatable::Stmt(stmt) => *stmt,
173 _ => { ::core::panicking::panic_fmt(format_args!("expected statement")); }panic!("expected statement"),
174 }
175 }
176
177 pub fn expect_expr(self) -> Box<ast::Expr> {
178 match self {
179 Annotatable::Expr(expr) => expr,
180 _ => { ::core::panicking::panic_fmt(format_args!("expected expression")); }panic!("expected expression"),
181 }
182 }
183
184 pub fn expect_arm(self) -> ast::Arm {
185 match self {
186 Annotatable::Arm(arm) => arm,
187 _ => { ::core::panicking::panic_fmt(format_args!("expected match arm")); }panic!("expected match arm"),
188 }
189 }
190
191 pub fn expect_expr_field(self) -> ast::ExprField {
192 match self {
193 Annotatable::ExprField(field) => field,
194 _ => { ::core::panicking::panic_fmt(format_args!("expected field")); }panic!("expected field"),
195 }
196 }
197
198 pub fn expect_pat_field(self) -> ast::PatField {
199 match self {
200 Annotatable::PatField(fp) => fp,
201 _ => { ::core::panicking::panic_fmt(format_args!("expected field pattern")); }panic!("expected field pattern"),
202 }
203 }
204
205 pub fn expect_generic_param(self) -> ast::GenericParam {
206 match self {
207 Annotatable::GenericParam(gp) => gp,
208 _ => { ::core::panicking::panic_fmt(format_args!("expected generic parameter")); }panic!("expected generic parameter"),
209 }
210 }
211
212 pub fn expect_param(self) -> ast::Param {
213 match self {
214 Annotatable::Param(param) => param,
215 _ => { ::core::panicking::panic_fmt(format_args!("expected parameter")); }panic!("expected parameter"),
216 }
217 }
218
219 pub fn expect_field_def(self) -> ast::FieldDef {
220 match self {
221 Annotatable::FieldDef(sf) => sf,
222 _ => { ::core::panicking::panic_fmt(format_args!("expected struct field")); }panic!("expected struct field"),
223 }
224 }
225
226 pub fn expect_variant(self) -> ast::Variant {
227 match self {
228 Annotatable::Variant(v) => v,
229 _ => { ::core::panicking::panic_fmt(format_args!("expected variant")); }panic!("expected variant"),
230 }
231 }
232
233 pub fn expect_where_predicate(self) -> ast::WherePredicate {
234 match self {
235 Annotatable::WherePredicate(wp) => wp,
236 _ => { ::core::panicking::panic_fmt(format_args!("expected where predicate")); }panic!("expected where predicate"),
237 }
238 }
239
240 pub fn expect_crate(self) -> ast::Crate {
241 match self {
242 Annotatable::Crate(krate) => krate,
243 _ => { ::core::panicking::panic_fmt(format_args!("expected krate")); }panic!("expected krate"),
244 }
245 }
246}
247
248pub enum ExpandResult<T, U> {
251 Ready(T),
253 Retry(U),
255}
256
257impl<T, U> ExpandResult<T, U> {
258 pub fn map<E, F: FnOnce(T) -> E>(self, f: F) -> ExpandResult<E, U> {
259 match self {
260 ExpandResult::Ready(t) => ExpandResult::Ready(f(t)),
261 ExpandResult::Retry(u) => ExpandResult::Retry(u),
262 }
263 }
264}
265
266impl<'cx> MacroExpanderResult<'cx> {
267 pub fn from_tts(
271 cx: &'cx mut ExtCtxt<'_>,
272 tts: TokenStream,
273 site_span: Span,
274 arm_span: Span,
275 macro_ident: Ident,
276 ) -> Self {
277 let is_local = true;
279
280 let parser =
281 ParserAnyMacro::from_tts(cx, tts, site_span, arm_span, is_local, macro_ident, &[], &[]);
282 ExpandResult::Ready(Box::new(parser))
283 }
284}
285
286pub trait MultiItemModifier {
287 fn expand(
289 &self,
290 ecx: &mut ExtCtxt<'_>,
291 span: Span,
292 meta_item: &ast::MetaItem,
293 item: Annotatable,
294 is_derive_const: bool,
295 ) -> ExpandResult<Vec<Annotatable>, Annotatable>;
296}
297
298impl<F> MultiItemModifier for F
299where
300 F: Fn(&mut ExtCtxt<'_>, Span, &ast::MetaItem, Annotatable) -> Vec<Annotatable>,
301{
302 fn expand(
303 &self,
304 ecx: &mut ExtCtxt<'_>,
305 span: Span,
306 meta_item: &ast::MetaItem,
307 item: Annotatable,
308 _is_derive_const: bool,
309 ) -> ExpandResult<Vec<Annotatable>, Annotatable> {
310 ExpandResult::Ready(self(ecx, span, meta_item, item))
311 }
312}
313
314pub trait BangProcMacro {
315 fn expand<'cx>(
316 &self,
317 ecx: &'cx mut ExtCtxt<'_>,
318 span: Span,
319 ts: TokenStream,
320 ) -> Result<TokenStream, ErrorGuaranteed>;
321}
322
323impl<F> BangProcMacro for F
324where
325 F: Fn(&mut ExtCtxt<'_>, Span, TokenStream) -> Result<TokenStream, ErrorGuaranteed>,
326{
327 fn expand<'cx>(
328 &self,
329 ecx: &'cx mut ExtCtxt<'_>,
330 span: Span,
331 ts: TokenStream,
332 ) -> Result<TokenStream, ErrorGuaranteed> {
333 self(ecx, span, ts)
335 }
336}
337
338pub trait AttrProcMacro {
339 fn expand<'cx>(
340 &self,
341 ecx: &'cx mut ExtCtxt<'_>,
342 span: Span,
343 annotation: TokenStream,
344 annotated: TokenStream,
345 ) -> Result<TokenStream, ErrorGuaranteed>;
346
347 fn expand_with_safety<'cx>(
349 &self,
350 ecx: &'cx mut ExtCtxt<'_>,
351 safety: Safety,
352 span: Span,
353 annotation: TokenStream,
354 annotated: TokenStream,
355 ) -> Result<TokenStream, ErrorGuaranteed> {
356 if let Safety::Unsafe(span) = safety {
357 ecx.dcx().span_err(span, "unnecessary `unsafe` on safe attribute");
358 }
359 self.expand(ecx, span, annotation, annotated)
360 }
361}
362
363impl<F> AttrProcMacro for F
364where
365 F: Fn(TokenStream, TokenStream) -> TokenStream,
366{
367 fn expand<'cx>(
368 &self,
369 _ecx: &'cx mut ExtCtxt<'_>,
370 _span: Span,
371 annotation: TokenStream,
372 annotated: TokenStream,
373 ) -> Result<TokenStream, ErrorGuaranteed> {
374 Ok(self(annotation, annotated))
376 }
377}
378
379pub trait TTMacroExpander: Any {
381 fn expand<'cx, 'a: 'cx>(
382 &'a self,
383 ecx: &'cx mut ExtCtxt<'_>,
384 span: Span,
385 input: TokenStream,
386 ) -> MacroExpanderResult<'cx>;
387}
388
389pub type MacroExpanderResult<'cx> = ExpandResult<Box<dyn MacResult + 'cx>, ()>;
390
391pub type MacroExpanderFn =
392 for<'cx> fn(&'cx mut ExtCtxt<'_>, Span, TokenStream) -> MacroExpanderResult<'cx>;
393
394impl<F: 'static> TTMacroExpander for F
395where
396 F: for<'cx> Fn(&'cx mut ExtCtxt<'_>, Span, TokenStream) -> MacroExpanderResult<'cx>,
397{
398 fn expand<'cx, 'a: 'cx>(
399 &'a self,
400 ecx: &'cx mut ExtCtxt<'_>,
401 span: Span,
402 input: TokenStream,
403 ) -> MacroExpanderResult<'cx> {
404 self(ecx, span, input)
405 }
406}
407
408pub trait GlobDelegationExpander {
409 fn expand(&self, ecx: &mut ExtCtxt<'_>) -> ExpandResult<Vec<(Ident, Option<Ident>)>, ()>;
410}
411
412fn make_stmts_default(expr: Option<Box<ast::Expr>>) -> Option<SmallVec<[ast::Stmt; 1]>> {
413 expr.map(|e| {
414 {
let count = 0usize + 1usize;
let mut vec = ::smallvec::SmallVec::new();
if count <= vec.inline_size() {
vec.push(ast::Stmt {
id: ast::DUMMY_NODE_ID,
span: e.span,
kind: ast::StmtKind::Expr(e),
});
vec
} else {
::smallvec::SmallVec::from_vec(::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[ast::Stmt {
id: ast::DUMMY_NODE_ID,
span: e.span,
kind: ast::StmtKind::Expr(e),
}])))
}
}smallvec![ast::Stmt { id: ast::DUMMY_NODE_ID, span: e.span, kind: ast::StmtKind::Expr(e) }]
415 })
416}
417
418pub trait MacResult {
421 fn make_expr(self: Box<Self>) -> Option<Box<ast::Expr>> {
423 None
424 }
425
426 fn make_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::Item>; 1]>> {
428 None
429 }
430
431 fn make_impl_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
433 None
434 }
435
436 fn make_trait_impl_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
438 None
439 }
440
441 fn make_trait_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
443 None
444 }
445
446 fn make_foreign_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::ForeignItem>; 1]>> {
448 None
449 }
450
451 fn make_pat(self: Box<Self>) -> Option<Box<ast::Pat>> {
453 None
454 }
455
456 fn make_stmts(self: Box<Self>) -> Option<SmallVec<[ast::Stmt; 1]>> {
461 make_stmts_default(self.make_expr())
462 }
463
464 fn make_ty(self: Box<Self>) -> Option<Box<ast::Ty>> {
465 None
466 }
467
468 fn make_arms(self: Box<Self>) -> Option<SmallVec<[ast::Arm; 1]>> {
469 None
470 }
471
472 fn make_expr_fields(self: Box<Self>) -> Option<SmallVec<[ast::ExprField; 1]>> {
473 None
474 }
475
476 fn make_pat_fields(self: Box<Self>) -> Option<SmallVec<[ast::PatField; 1]>> {
477 None
478 }
479
480 fn make_generic_params(self: Box<Self>) -> Option<SmallVec<[ast::GenericParam; 1]>> {
481 None
482 }
483
484 fn make_params(self: Box<Self>) -> Option<SmallVec<[ast::Param; 1]>> {
485 None
486 }
487
488 fn make_field_defs(self: Box<Self>) -> Option<SmallVec<[ast::FieldDef; 1]>> {
489 None
490 }
491
492 fn make_variants(self: Box<Self>) -> Option<SmallVec<[ast::Variant; 1]>> {
493 None
494 }
495
496 fn make_where_predicates(self: Box<Self>) -> Option<SmallVec<[ast::WherePredicate; 1]>> {
497 None
498 }
499
500 fn make_crate(self: Box<Self>) -> Option<ast::Crate> {
501 ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
503 }
504}
505
506#[derive(#[automatically_derived]
impl ::core::default::Default for MacEager {
#[inline]
fn default() -> MacEager {
MacEager {
expr: ::core::default::Default::default(),
items: ::core::default::Default::default(),
ty: ::core::default::Default::default(),
}
}
}Default)]
509pub struct MacEager {
510 pub expr: Option<Box<ast::Expr>>,
511 pub items: Option<SmallVec<[Box<ast::Item>; 1]>>,
512 pub ty: Option<Box<ast::Ty>>,
513}
514
515impl MacEager {
516 pub fn expr(v: Box<ast::Expr>) -> Box<dyn MacResult> {
517 Box::new(MacEager { expr: Some(v), ..Default::default() })
518 }
519
520 pub fn items(v: SmallVec<[Box<ast::Item>; 1]>) -> Box<dyn MacResult> {
521 Box::new(MacEager { items: Some(v), ..Default::default() })
522 }
523
524 pub fn ty(v: Box<ast::Ty>) -> Box<dyn MacResult> {
525 Box::new(MacEager { ty: Some(v), ..Default::default() })
526 }
527}
528
529impl MacResult for MacEager {
530 fn make_expr(self: Box<Self>) -> Option<Box<ast::Expr>> {
531 self.expr
532 }
533
534 fn make_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::Item>; 1]>> {
535 self.items
536 }
537
538 fn make_pat(self: Box<Self>) -> Option<Box<ast::Pat>> {
539 if let Some(e) = self.expr {
540 if #[allow(non_exhaustive_omitted_patterns)] match e.kind {
ast::ExprKind::Lit(_) | ast::ExprKind::IncludedBytes(_) => true,
_ => false,
}matches!(e.kind, ast::ExprKind::Lit(_) | ast::ExprKind::IncludedBytes(_)) {
541 return Some(Box::new(ast::Pat {
542 id: ast::DUMMY_NODE_ID,
543 span: e.span,
544 kind: PatKind::Expr(e),
545 }));
546 }
547 }
548 None
549 }
550
551 fn make_ty(self: Box<Self>) -> Option<Box<ast::Ty>> {
552 self.ty
553 }
554}
555
556#[derive(#[automatically_derived]
impl ::core::marker::Copy for DummyResult { }Copy, #[automatically_derived]
impl ::core::clone::Clone for DummyResult {
#[inline]
fn clone(&self) -> DummyResult {
let _: ::core::clone::AssertParamIsClone<Option<ErrorGuaranteed>>;
let _: ::core::clone::AssertParamIsClone<Span>;
*self
}
}Clone)]
559pub struct DummyResult {
560 guar: Option<ErrorGuaranteed>,
561 span: Span,
562}
563
564impl DummyResult {
565 pub fn any(span: Span, guar: ErrorGuaranteed) -> Box<dyn MacResult + 'static> {
570 Box::new(DummyResult { guar: Some(guar), span })
571 }
572
573 pub fn any_valid(span: Span) -> Box<dyn MacResult + 'static> {
575 Box::new(DummyResult { guar: None, span })
576 }
577
578 pub fn raw_expr(sp: Span, guar: Option<ErrorGuaranteed>) -> Box<ast::Expr> {
580 Box::new(ast::Expr {
581 id: ast::DUMMY_NODE_ID,
582 kind: if let Some(guar) = guar {
583 ast::ExprKind::Err(guar)
584 } else {
585 ast::ExprKind::Tup(ThinVec::new())
586 },
587 span: sp,
588 attrs: ast::AttrVec::new(),
589 tokens: None,
590 })
591 }
592}
593
594impl MacResult for DummyResult {
595 fn make_expr(self: Box<DummyResult>) -> Option<Box<ast::Expr>> {
596 Some(DummyResult::raw_expr(self.span, self.guar))
597 }
598
599 fn make_pat(self: Box<DummyResult>) -> Option<Box<ast::Pat>> {
600 Some(Box::new(ast::Pat { id: ast::DUMMY_NODE_ID, kind: PatKind::Wild, span: self.span }))
601 }
602
603 fn make_items(self: Box<DummyResult>) -> Option<SmallVec<[Box<ast::Item>; 1]>> {
604 Some(SmallVec::new())
605 }
606
607 fn make_impl_items(self: Box<DummyResult>) -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
608 Some(SmallVec::new())
609 }
610
611 fn make_trait_impl_items(self: Box<DummyResult>) -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
612 Some(SmallVec::new())
613 }
614
615 fn make_trait_items(self: Box<DummyResult>) -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
616 Some(SmallVec::new())
617 }
618
619 fn make_foreign_items(self: Box<Self>) -> Option<SmallVec<[Box<ast::ForeignItem>; 1]>> {
620 Some(SmallVec::new())
621 }
622
623 fn make_stmts(self: Box<DummyResult>) -> Option<SmallVec<[ast::Stmt; 1]>> {
624 Some({
let count = 0usize + 1usize;
let mut vec = ::smallvec::SmallVec::new();
if count <= vec.inline_size() {
vec.push(ast::Stmt {
id: ast::DUMMY_NODE_ID,
kind: ast::StmtKind::Expr(DummyResult::raw_expr(self.span,
self.guar)),
span: self.span,
});
vec
} else {
::smallvec::SmallVec::from_vec(::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
[ast::Stmt {
id: ast::DUMMY_NODE_ID,
kind: ast::StmtKind::Expr(DummyResult::raw_expr(self.span,
self.guar)),
span: self.span,
}])))
}
}smallvec![ast::Stmt {
625 id: ast::DUMMY_NODE_ID,
626 kind: ast::StmtKind::Expr(DummyResult::raw_expr(self.span, self.guar)),
627 span: self.span,
628 }])
629 }
630
631 fn make_ty(self: Box<DummyResult>) -> Option<Box<ast::Ty>> {
632 Some(Box::new(ast::Ty {
636 id: ast::DUMMY_NODE_ID,
637 kind: ast::TyKind::Tup(ThinVec::new()),
638 span: self.span,
639 }))
640 }
641
642 fn make_arms(self: Box<DummyResult>) -> Option<SmallVec<[ast::Arm; 1]>> {
643 Some(SmallVec::new())
644 }
645
646 fn make_expr_fields(self: Box<DummyResult>) -> Option<SmallVec<[ast::ExprField; 1]>> {
647 Some(SmallVec::new())
648 }
649
650 fn make_pat_fields(self: Box<DummyResult>) -> Option<SmallVec<[ast::PatField; 1]>> {
651 Some(SmallVec::new())
652 }
653
654 fn make_generic_params(self: Box<DummyResult>) -> Option<SmallVec<[ast::GenericParam; 1]>> {
655 Some(SmallVec::new())
656 }
657
658 fn make_params(self: Box<DummyResult>) -> Option<SmallVec<[ast::Param; 1]>> {
659 Some(SmallVec::new())
660 }
661
662 fn make_field_defs(self: Box<DummyResult>) -> Option<SmallVec<[ast::FieldDef; 1]>> {
663 Some(SmallVec::new())
664 }
665
666 fn make_variants(self: Box<DummyResult>) -> Option<SmallVec<[ast::Variant; 1]>> {
667 Some(SmallVec::new())
668 }
669
670 fn make_crate(self: Box<DummyResult>) -> Option<ast::Crate> {
671 Some(ast::Crate {
672 attrs: Default::default(),
673 items: Default::default(),
674 spans: Default::default(),
675 id: ast::DUMMY_NODE_ID,
676 is_placeholder: Default::default(),
677 })
678 }
679}
680
681#[derive(#[automatically_derived]
impl ::core::clone::Clone for SyntaxExtensionKind {
#[inline]
fn clone(&self) -> SyntaxExtensionKind {
match self {
SyntaxExtensionKind::MacroRules(__self_0) =>
SyntaxExtensionKind::MacroRules(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::Bang(__self_0) =>
SyntaxExtensionKind::Bang(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::LegacyBang(__self_0) =>
SyntaxExtensionKind::LegacyBang(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::Attr(__self_0) =>
SyntaxExtensionKind::Attr(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::LegacyAttr(__self_0) =>
SyntaxExtensionKind::LegacyAttr(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::NonMacroAttr =>
SyntaxExtensionKind::NonMacroAttr,
SyntaxExtensionKind::Derive(__self_0) =>
SyntaxExtensionKind::Derive(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::LegacyDerive(__self_0) =>
SyntaxExtensionKind::LegacyDerive(::core::clone::Clone::clone(__self_0)),
SyntaxExtensionKind::GlobDelegation(__self_0) =>
SyntaxExtensionKind::GlobDelegation(::core::clone::Clone::clone(__self_0)),
}
}
}Clone)]
683pub enum SyntaxExtensionKind {
684 MacroRules(Arc<crate::MacroRulesMacroExpander>),
686
687 Bang(
689 Arc<dyn BangProcMacro + sync::DynSync + sync::DynSend>,
691 ),
692
693 LegacyBang(
695 Arc<dyn TTMacroExpander + sync::DynSync + sync::DynSend>,
697 ),
698
699 Attr(
701 Arc<dyn AttrProcMacro + sync::DynSync + sync::DynSend>,
705 ),
706
707 LegacyAttr(
709 Arc<dyn MultiItemModifier + sync::DynSync + sync::DynSend>,
713 ),
714
715 NonMacroAttr,
720
721 Derive(
723 Arc<dyn MultiItemModifier + sync::DynSync + sync::DynSend>,
731 ),
732
733 LegacyDerive(
735 Arc<dyn MultiItemModifier + sync::DynSync + sync::DynSend>,
738 ),
739
740 GlobDelegation(Arc<dyn GlobDelegationExpander + sync::DynSync + sync::DynSend>),
744}
745
746impl SyntaxExtensionKind {
747 pub fn as_legacy_bang(&self) -> Option<&(dyn TTMacroExpander + sync::DynSync + sync::DynSend)> {
751 match self {
752 SyntaxExtensionKind::LegacyBang(exp) => Some(exp.as_ref()),
753 SyntaxExtensionKind::MacroRules(exp) if exp.kinds().contains(MacroKinds::BANG) => {
754 Some(exp.as_ref())
755 }
756 _ => None,
757 }
758 }
759
760 pub fn as_attr(&self) -> Option<&(dyn AttrProcMacro + sync::DynSync + sync::DynSend)> {
764 match self {
765 SyntaxExtensionKind::Attr(exp) => Some(exp.as_ref()),
766 SyntaxExtensionKind::MacroRules(exp) if exp.kinds().contains(MacroKinds::ATTR) => {
767 Some(exp.as_ref())
768 }
769 _ => None,
770 }
771 }
772}
773
774pub struct SyntaxExtension {
776 pub kind: SyntaxExtensionKind,
778 pub span: Span,
780 pub allow_internal_unstable: Option<Arc<[Symbol]>>,
782 pub stability: Option<Stability>,
784 pub deprecation: Option<Deprecation>,
786 pub helper_attrs: Vec<Symbol>,
788 pub edition: Edition,
790 pub builtin_name: Option<Symbol>,
793 pub allow_internal_unsafe: bool,
795 pub local_inner_macros: bool,
797 pub collapse_debuginfo: bool,
800 pub hide_backtrace: bool,
803}
804
805impl SyntaxExtension {
806 pub fn macro_kinds(&self) -> MacroKinds {
808 match self.kind {
809 SyntaxExtensionKind::Bang(..)
810 | SyntaxExtensionKind::LegacyBang(..)
811 | SyntaxExtensionKind::GlobDelegation(..) => MacroKinds::BANG,
812 SyntaxExtensionKind::Attr(..)
813 | SyntaxExtensionKind::LegacyAttr(..)
814 | SyntaxExtensionKind::NonMacroAttr => MacroKinds::ATTR,
815 SyntaxExtensionKind::Derive(..) | SyntaxExtensionKind::LegacyDerive(..) => {
816 MacroKinds::DERIVE
817 }
818 SyntaxExtensionKind::MacroRules(ref m) => m.kinds(),
819 }
820 }
821
822 pub fn default(kind: SyntaxExtensionKind, edition: Edition) -> SyntaxExtension {
824 SyntaxExtension {
825 span: DUMMY_SP,
826 allow_internal_unstable: None,
827 stability: None,
828 deprecation: None,
829 helper_attrs: Vec::new(),
830 edition,
831 builtin_name: None,
832 kind,
833 allow_internal_unsafe: false,
834 local_inner_macros: false,
835 collapse_debuginfo: false,
836 hide_backtrace: false,
837 }
838 }
839
840 fn get_collapse_debuginfo(sess: &Session, attrs: &[hir::Attribute], ext: bool) -> bool {
847 let flag = sess.opts.cg.collapse_macro_debuginfo;
848 let attr = if let Some(info) = {
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(CollapseDebugInfo(info)) => {
break 'done Some(info);
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(attrs, CollapseDebugInfo(info) => info) {
849 *info
850 } else if {
{
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(RustcBuiltinMacro { .. }) => {
break 'done Some(());
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}.is_some()
}find_attr!(attrs, RustcBuiltinMacro { .. }) {
851 CollapseMacroDebuginfo::Yes
852 } else {
853 CollapseMacroDebuginfo::Unspecified
854 };
855
856 #[rustfmt::skip]
857 let collapse_table = [
858 [false, false, false, false],
859 [false, ext, ext, true],
860 [false, ext, ext, true],
861 [true, true, true, true],
862 ];
863 collapse_table[flag as usize][attr as usize]
864 }
865
866 fn get_hide_backtrace(attrs: &[hir::Attribute]) -> bool {
867 {
{
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(RustcDiagnosticItem(..)) => {
break 'done Some(());
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}.is_some()
}find_attr!(attrs, RustcDiagnosticItem(..))
870 }
871
872 pub fn new(
875 sess: &Session,
876 kind: SyntaxExtensionKind,
877 span: Span,
878 helper_attrs: Vec<Symbol>,
879 edition: Edition,
880 name: Symbol,
881 attrs: &[hir::Attribute],
882 is_local: bool,
883 ) -> SyntaxExtension {
884 let allow_internal_unstable = {
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(AllowInternalUnstable(i, _)) => {
break 'done Some(i);
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(attrs, AllowInternalUnstable(i, _) => i)
885 .map(|i| i.as_slice())
886 .unwrap_or_default();
887 let allow_internal_unsafe = {
{
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(AllowInternalUnsafe(_)) => {
break 'done Some(());
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}.is_some()
}find_attr!(attrs, AllowInternalUnsafe(_));
888
889 let local_inner_macros =
890 *{
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(MacroExport {
local_inner_macros: l, .. }) => {
break 'done Some(l);
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(attrs, MacroExport {local_inner_macros: l, ..} => l).unwrap_or(&false);
891 let collapse_debuginfo = Self::get_collapse_debuginfo(sess, attrs, !is_local);
892 {
use ::tracing::__macro_support::Callsite as _;
static __CALLSITE: ::tracing::callsite::DefaultCallsite =
{
static META: ::tracing::Metadata<'static> =
{
::tracing_core::metadata::Metadata::new("event compiler/rustc_expand/src/base.rs:892",
"rustc_expand::base", ::tracing::Level::DEBUG,
::tracing_core::__macro_support::Option::Some("compiler/rustc_expand/src/base.rs"),
::tracing_core::__macro_support::Option::Some(892u32),
::tracing_core::__macro_support::Option::Some("rustc_expand::base"),
::tracing_core::field::FieldSet::new(&["name",
"local_inner_macros", "collapse_debuginfo",
"allow_internal_unsafe"],
::tracing_core::callsite::Identifier(&__CALLSITE)),
::tracing::metadata::Kind::EVENT)
};
::tracing::callsite::DefaultCallsite::new(&META)
};
let enabled =
::tracing::Level::DEBUG <= ::tracing::level_filters::STATIC_MAX_LEVEL
&&
::tracing::Level::DEBUG <=
::tracing::level_filters::LevelFilter::current() &&
{
let interest = __CALLSITE.interest();
!interest.is_never() &&
::tracing::__macro_support::__is_enabled(__CALLSITE.metadata(),
interest)
};
if enabled {
(|value_set: ::tracing::field::ValueSet|
{
let meta = __CALLSITE.metadata();
::tracing::Event::dispatch(meta, &value_set);
;
})({
#[allow(unused_imports)]
use ::tracing::field::{debug, display, Value};
let mut iter = __CALLSITE.metadata().fields().iter();
__CALLSITE.metadata().fields().value_set(&[(&::tracing::__macro_support::Iterator::next(&mut iter).expect("FieldSet corrupted (this is a bug)"),
::tracing::__macro_support::Option::Some(&debug(&name) as
&dyn Value)),
(&::tracing::__macro_support::Iterator::next(&mut iter).expect("FieldSet corrupted (this is a bug)"),
::tracing::__macro_support::Option::Some(&debug(&local_inner_macros)
as &dyn Value)),
(&::tracing::__macro_support::Iterator::next(&mut iter).expect("FieldSet corrupted (this is a bug)"),
::tracing::__macro_support::Option::Some(&debug(&collapse_debuginfo)
as &dyn Value)),
(&::tracing::__macro_support::Iterator::next(&mut iter).expect("FieldSet corrupted (this is a bug)"),
::tracing::__macro_support::Option::Some(&debug(&allow_internal_unsafe)
as &dyn Value))])
});
} else { ; }
};tracing::debug!(?name, ?local_inner_macros, ?collapse_debuginfo, ?allow_internal_unsafe);
893
894 let (builtin_name, helper_attrs) = match {
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(RustcBuiltinMacro {
builtin_name, helper_attrs }) => {
break 'done Some((builtin_name, helper_attrs));
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(attrs, RustcBuiltinMacro { builtin_name, helper_attrs } => (builtin_name, helper_attrs))
895 {
896 Some((Some(name), helper_attrs)) => {
899 (Some(*name), helper_attrs.iter().copied().collect())
900 }
901 Some((None, _)) => (Some(name), Vec::new()),
902
903 None => (None, helper_attrs),
905 };
906 let hide_backtrace = builtin_name.is_some() || Self::get_hide_backtrace(attrs);
907
908 let stability = {
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(Stability { stability, .. }) => {
break 'done Some(*stability);
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(attrs, Stability { stability, .. } => *stability);
909
910 if let Some(sp) = {
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(RustcBodyStability { span, .. })
=> {
break 'done Some(*span);
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(attrs, RustcBodyStability{ span, .. } => *span) {
911 sess.dcx().emit_err(diagnostics::MacroBodyStability {
912 span: sp,
913 head_span: sess.source_map().guess_head_span(span),
914 });
915 }
916
917 SyntaxExtension {
918 kind,
919 span,
920 allow_internal_unstable: (!allow_internal_unstable.is_empty())
921 .then(|| allow_internal_unstable.iter().map(|i| i.0).collect::<Vec<_>>().into()),
923 stability,
924 deprecation: {
'done:
{
for i in attrs {
#[allow(unused_imports)]
use rustc_hir::attrs::AttributeKind::*;
let i: &rustc_hir::Attribute = i;
match i {
rustc_hir::Attribute::Parsed(Deprecated { deprecation, .. })
=> {
break 'done Some(*deprecation);
}
rustc_hir::Attribute::Unparsed(..) =>
{}
#[deny(unreachable_patterns)]
_ => {}
}
}
None
}
}find_attr!(
925 attrs,
926 Deprecated { deprecation, .. } => *deprecation
927 ),
928 helper_attrs,
929 edition,
930 builtin_name,
931 allow_internal_unsafe,
932 local_inner_macros,
933 collapse_debuginfo,
934 hide_backtrace,
935 }
936 }
937
938 pub fn dummy_bang(edition: Edition) -> SyntaxExtension {
940 fn expand(
941 ecx: &mut ExtCtxt<'_>,
942 span: Span,
943 _ts: TokenStream,
944 ) -> Result<TokenStream, ErrorGuaranteed> {
945 Err(ecx.dcx().span_delayed_bug(span, "expanded a dummy bang macro"))
946 }
947 SyntaxExtension::default(SyntaxExtensionKind::Bang(Arc::new(expand)), edition)
948 }
949
950 pub fn dummy_derive(edition: Edition) -> SyntaxExtension {
952 fn expander(
953 _: &mut ExtCtxt<'_>,
954 _: Span,
955 _: &ast::MetaItem,
956 _: Annotatable,
957 ) -> Vec<Annotatable> {
958 Vec::new()
959 }
960 SyntaxExtension::default(SyntaxExtensionKind::Derive(Arc::new(expander)), edition)
961 }
962
963 pub fn non_macro_attr(edition: Edition) -> SyntaxExtension {
964 SyntaxExtension::default(SyntaxExtensionKind::NonMacroAttr, edition)
965 }
966
967 pub fn glob_delegation(
968 trait_def_id: DefId,
969 impl_def_id: LocalDefId,
970 star_span: Span,
971 edition: Edition,
972 ) -> SyntaxExtension {
973 struct GlobDelegationExpanderImpl {
974 trait_def_id: DefId,
975 impl_def_id: LocalDefId,
976 star_span: Span,
977 }
978 impl GlobDelegationExpander for GlobDelegationExpanderImpl {
979 fn expand(
980 &self,
981 ecx: &mut ExtCtxt<'_>,
982 ) -> ExpandResult<Vec<(Ident, Option<Ident>)>, ()> {
983 match ecx.resolver.glob_delegation_suffixes(
984 self.trait_def_id,
985 self.impl_def_id,
986 self.star_span,
987 ) {
988 Ok(suffixes) => ExpandResult::Ready(suffixes),
989 Err(Indeterminate) if ecx.force_mode => ExpandResult::Ready(Vec::new()),
990 Err(Indeterminate) => ExpandResult::Retry(()),
991 }
992 }
993 }
994
995 let expander = GlobDelegationExpanderImpl { trait_def_id, impl_def_id, star_span };
996 SyntaxExtension::default(SyntaxExtensionKind::GlobDelegation(Arc::new(expander)), edition)
997 }
998
999 pub fn expn_data(
1000 &self,
1001 parent: LocalExpnId,
1002 call_site: Span,
1003 descr: Symbol,
1004 kind: MacroKind,
1005 macro_def_id: Option<DefId>,
1006 parent_module: Option<DefId>,
1007 ) -> ExpnData {
1008 ExpnData::new(
1009 ExpnKind::Macro(kind, descr),
1010 parent.to_expn_id(),
1011 call_site,
1012 self.span,
1013 self.allow_internal_unstable.clone(),
1014 self.edition,
1015 macro_def_id,
1016 parent_module,
1017 self.allow_internal_unsafe,
1018 self.local_inner_macros,
1019 self.collapse_debuginfo,
1020 self.hide_backtrace,
1021 )
1022 }
1023}
1024
1025pub struct Indeterminate;
1027
1028pub struct DeriveResolution {
1029 pub path: ast::Path,
1030 pub item: Annotatable,
1031 pub exts: Option<Arc<SyntaxExtension>>,
1035 pub is_const: bool,
1036}
1037
1038pub trait ResolverExpand {
1039 fn next_node_id(&mut self) -> NodeId;
1040 fn invocation_parent(&self, id: LocalExpnId) -> LocalDefId;
1041
1042 fn resolve_dollar_crates(&self);
1043 fn visit_ast_fragment_with_placeholders(
1044 &mut self,
1045 expn_id: LocalExpnId,
1046 fragment: &AstFragment,
1047 );
1048 fn register_builtin_macro(&mut self, name: Symbol, ext: SyntaxExtensionKind);
1049
1050 fn expansion_for_ast_pass(
1051 &mut self,
1052 call_site: Span,
1053 pass: AstPass,
1054 features: &[Symbol],
1055 parent_module_id: Option<NodeId>,
1056 ) -> LocalExpnId;
1057
1058 fn resolve_imports(&mut self);
1059
1060 fn resolve_macro_invocation(
1061 &mut self,
1062 invoc: &Invocation,
1063 eager_expansion_root: LocalExpnId,
1064 force: bool,
1065 ) -> Result<Arc<SyntaxExtension>, Indeterminate>;
1066
1067 fn record_macro_rule_usage(&mut self, mac_id: NodeId, rule_index: usize);
1068
1069 fn check_unused_macros(&mut self);
1070
1071 fn has_derive_copy(&self, expn_id: LocalExpnId) -> bool;
1074 fn has_derive_ord(&self, expn_id: LocalExpnId) -> bool;
1076 fn resolve_derives(
1078 &mut self,
1079 expn_id: LocalExpnId,
1080 force: bool,
1081 derive_paths: &dyn Fn() -> Vec<DeriveResolution>,
1082 ) -> Result<(), Indeterminate>;
1083 fn take_derive_resolutions(&mut self, expn_id: LocalExpnId) -> Option<Vec<DeriveResolution>>;
1086 fn cfg_accessible(
1088 &mut self,
1089 expn_id: LocalExpnId,
1090 path: &ast::Path,
1091 ) -> Result<bool, Indeterminate>;
1092 fn macro_accessible(
1093 &mut self,
1094 expn_id: LocalExpnId,
1095 path: &ast::Path,
1096 ) -> Result<bool, Indeterminate>;
1097
1098 fn get_proc_macro_quoted_span(&self, krate: CrateNum, id: usize) -> Span;
1101
1102 fn declare_proc_macro(&mut self, id: NodeId);
1109
1110 fn append_stripped_cfg_item(
1111 &mut self,
1112 parent_node: NodeId,
1113 ident: Ident,
1114 cfg: CfgEntry,
1115 cfg_span: Span,
1116 );
1117
1118 fn registered_tools(&self) -> &RegisteredTools;
1120
1121 fn register_glob_delegation(&mut self, invoc_id: LocalExpnId);
1123
1124 fn glob_delegation_suffixes(
1126 &self,
1127 trait_def_id: DefId,
1128 impl_def_id: LocalDefId,
1129 star_span: Span,
1130 ) -> Result<Vec<(Ident, Option<Ident>)>, Indeterminate>;
1131
1132 fn insert_impl_trait_name(&mut self, id: NodeId, name: Symbol);
1135
1136 fn mark_scope_with_compile_error(&mut self, parent_node: NodeId);
1139}
1140
1141pub trait LintStoreExpand {
1142 fn pre_expansion_lint(
1143 &self,
1144 sess: &Session,
1145 features: &Features,
1146 registered_tools: &RegisteredTools,
1147 node_id: NodeId,
1148 attrs: &[Attribute],
1149 items: &[Box<Item>],
1150 name: Symbol,
1151 );
1152}
1153
1154type LintStoreExpandDyn<'a> = Option<&'a (dyn LintStoreExpand + 'a)>;
1155
1156#[derive(#[automatically_derived]
impl ::core::fmt::Debug for ModuleData {
#[inline]
fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
::core::fmt::Formatter::debug_struct_field3_finish(f, "ModuleData",
"mod_path", &self.mod_path, "file_path_stack",
&self.file_path_stack, "dir_path", &&self.dir_path)
}
}Debug, #[automatically_derived]
impl ::core::default::Default for ModuleData {
#[inline]
fn default() -> ModuleData {
ModuleData {
mod_path: ::core::default::Default::default(),
file_path_stack: ::core::default::Default::default(),
dir_path: ::core::default::Default::default(),
}
}
}Default)]
1157pub struct ModuleData {
1158 pub mod_path: Vec<Ident>,
1160 pub file_path_stack: Vec<PathBuf>,
1163 pub dir_path: PathBuf,
1166}
1167
1168impl ModuleData {
1169 pub fn with_dir_path(&self, dir_path: PathBuf) -> ModuleData {
1170 ModuleData {
1171 mod_path: self.mod_path.clone(),
1172 file_path_stack: self.file_path_stack.clone(),
1173 dir_path,
1174 }
1175 }
1176}
1177
1178#[derive(#[automatically_derived]
impl ::core::clone::Clone for ExpansionData {
#[inline]
fn clone(&self) -> ExpansionData {
ExpansionData {
id: ::core::clone::Clone::clone(&self.id),
depth: ::core::clone::Clone::clone(&self.depth),
module: ::core::clone::Clone::clone(&self.module),
dir_ownership: ::core::clone::Clone::clone(&self.dir_ownership),
lint_node_id: ::core::clone::Clone::clone(&self.lint_node_id),
is_trailing_mac: ::core::clone::Clone::clone(&self.is_trailing_mac),
}
}
}Clone)]
1179pub struct ExpansionData {
1180 pub id: LocalExpnId,
1181 pub depth: usize,
1182 pub module: Rc<ModuleData>,
1183 pub dir_ownership: DirOwnership,
1184 pub lint_node_id: NodeId,
1186 pub is_trailing_mac: bool,
1187}
1188
1189pub struct ExtCtxt<'a> {
1193 pub sess: &'a Session,
1194 pub ecfg: expand::ExpansionConfig<'a>,
1195 pub num_standard_library_imports: usize,
1196 pub reduced_recursion_limit: Option<(Limit, ErrorGuaranteed)>,
1197 pub root_path: PathBuf,
1198 pub resolver: &'a mut dyn ResolverExpand,
1199 pub current_expansion: ExpansionData,
1200 pub force_mode: bool,
1203 pub expansions: FxIndexMap<Span, Vec<String>>,
1204 pub(super) lint_store: LintStoreExpandDyn<'a>,
1206 pub buffered_early_lint: Vec<BufferedEarlyLint>,
1208 pub(super) expanded_inert_attrs: MarkedAttrs,
1212 pub macro_stats: FxHashMap<(Symbol, MacroKind), MacroStat>,
1214 pub nb_macro_errors: usize,
1215}
1216
1217impl<'a> ExtCtxt<'a> {
1218 pub fn new(
1219 sess: &'a Session,
1220 ecfg: expand::ExpansionConfig<'a>,
1221 resolver: &'a mut dyn ResolverExpand,
1222 lint_store: LintStoreExpandDyn<'a>,
1223 ) -> ExtCtxt<'a> {
1224 ExtCtxt {
1225 sess,
1226 ecfg,
1227 num_standard_library_imports: 0,
1228 reduced_recursion_limit: None,
1229 resolver,
1230 lint_store,
1231 root_path: PathBuf::new(),
1232 current_expansion: ExpansionData {
1233 id: LocalExpnId::ROOT,
1234 depth: 0,
1235 module: Default::default(),
1236 dir_ownership: DirOwnership::Owned { relative: None },
1237 lint_node_id: ast::CRATE_NODE_ID,
1238 is_trailing_mac: false,
1239 },
1240 force_mode: false,
1241 expansions: FxIndexMap::default(),
1242 expanded_inert_attrs: MarkedAttrs::new(),
1243 buffered_early_lint: ::alloc::vec::Vec::new()vec![],
1244 macro_stats: Default::default(),
1245 nb_macro_errors: 0,
1246 }
1247 }
1248
1249 pub fn dcx(&self) -> DiagCtxtHandle<'a> {
1250 self.sess.dcx()
1251 }
1252
1253 pub fn expander<'b>(&'b mut self) -> expand::MacroExpander<'b, 'a> {
1255 expand::MacroExpander::new(self, false)
1256 }
1257
1258 pub fn monotonic_expander<'b>(&'b mut self) -> expand::MacroExpander<'b, 'a> {
1261 expand::MacroExpander::new(self, true)
1262 }
1263 pub fn new_parser_from_tts(&self, stream: TokenStream) -> Parser<'a> {
1264 Parser::new(&self.sess.psess, stream, MACRO_ARGUMENTS)
1265 }
1266 pub fn source_map(&self) -> &'a SourceMap {
1267 self.sess.psess.source_map()
1268 }
1269 pub fn psess(&self) -> &'a ParseSess {
1270 &self.sess.psess
1271 }
1272 pub fn call_site(&self) -> Span {
1273 self.current_expansion.id.expn_data().call_site
1274 }
1275
1276 pub(crate) fn expansion_descr(&self) -> String {
1278 let expn_data = self.current_expansion.id.expn_data();
1279 expn_data.kind.descr()
1280 }
1281
1282 pub fn with_def_site_ctxt(&self, span: Span) -> Span {
1285 span.with_def_site_ctxt(self.current_expansion.id.to_expn_id())
1286 }
1287
1288 pub fn with_call_site_ctxt(&self, span: Span) -> Span {
1291 span.with_call_site_ctxt(self.current_expansion.id.to_expn_id())
1292 }
1293
1294 pub fn with_mixed_site_ctxt(&self, span: Span) -> Span {
1297 span.with_mixed_site_ctxt(self.current_expansion.id.to_expn_id())
1298 }
1299
1300 pub fn expansion_cause(&self) -> Option<Span> {
1304 self.current_expansion.id.expansion_cause()
1305 }
1306
1307 pub fn macro_error_and_trace_macros_diag(&mut self) {
1309 self.nb_macro_errors += 1;
1310 self.trace_macros_diag();
1311 }
1312
1313 pub fn trace_macros_diag(&mut self) {
1314 for (span, notes) in self.expansions.iter() {
1315 let mut db = self.dcx().create_note(diagnostics::TraceMacro { span: *span });
1316 for note in notes {
1317 db.note(note.clone());
1318 }
1319 db.emit();
1320 }
1321 self.expansions.clear();
1323 }
1324 pub fn trace_macros(&self) -> bool {
1325 self.ecfg.trace_mac
1326 }
1327 pub fn set_trace_macros(&mut self, x: bool) {
1328 self.ecfg.trace_mac = x
1329 }
1330 pub fn std_path(&self, components: &[Symbol]) -> Vec<Ident> {
1331 let def_site = self.with_def_site_ctxt(DUMMY_SP);
1332 iter::once(Ident::new(kw::DollarCrate, def_site))
1333 .chain(components.iter().map(|&s| Ident::new(s, def_site)))
1334 .collect()
1335 }
1336 pub fn def_site_path(&self, components: &[Symbol]) -> Vec<Ident> {
1337 let def_site = self.with_def_site_ctxt(DUMMY_SP);
1338 components.iter().map(|&s| Ident::new(s, def_site)).collect()
1339 }
1340
1341 pub fn check_unused_macros(&mut self) {
1342 self.resolver.check_unused_macros();
1343 }
1344}
1345
1346pub fn resolve_path(sess: &Session, path: impl Into<PathBuf>, span: Span) -> PResult<'_, PathBuf> {
1350 let path = path.into();
1351
1352 if !path.is_absolute() {
1355 let callsite = span.source_callsite();
1356 let source_map = sess.source_map();
1357 let Some(mut base_path) = source_map.span_to_filename(callsite).into_local_path() else {
1358 return Err(sess.dcx().create_err(diagnostics::ResolveRelativePath {
1359 span,
1360 path: source_map
1361 .filename_for_diagnostics(&source_map.span_to_filename(callsite))
1362 .to_string(),
1363 }));
1364 };
1365 base_path.pop();
1366 base_path.push(path);
1367 Ok(base_path)
1368 } else {
1369 match path.components().next() {
1372 Some(Prefix(prefix)) if prefix.kind().is_verbatim() => Ok(path.components().collect()),
1373 _ => Ok(path),
1374 }
1375 }
1376}