Skip to main content

rustc_expand/
expand.rs

1use std::path::PathBuf;
2use std::rc::Rc;
3use std::sync::Arc;
4use std::{iter, mem, slice};
5
6use rustc_ast::mut_visit::*;
7use rustc_ast::tokenstream::TokenStream;
8use rustc_ast::visit::{self, AssocCtxt, Visitor, VisitorResult, try_visit, walk_list};
9use rustc_ast::{
10    self as ast, AssocItemKind, AstNodeWrapper, AttrArgs, AttrItemKind, AttrStyle, AttrVec,
11    DUMMY_NODE_ID, EarlyParsedAttribute, ExprKind, ForeignItemKind, HasAttrs, HasNodeId, Inline,
12    ItemKind, MacStmtStyle, MetaItemInner, MetaItemKind, ModKind, NodeId, PatKind, StmtKind,
13    TyKind, token,
14};
15use rustc_ast_pretty::pprust;
16use rustc_attr_parsing::parser::AllowExprMetavar;
17use rustc_attr_parsing::{
18    AttributeParser, CFG_TEMPLATE, Early, EvalConfigResult, ShouldEmit, eval_config_entry,
19    parse_cfg, validate_attr,
20};
21use rustc_data_structures::flat_map_in_place::FlatMapInPlace;
22use rustc_data_structures::stack::ensure_sufficient_stack;
23use rustc_errors::{PResult, msg};
24use rustc_feature::Features;
25use rustc_hir::Target;
26use rustc_hir::def::MacroKinds;
27use rustc_hir::limit::Limit;
28use rustc_parse::parser::{
29    AllowConstBlockItems, AttemptLocalParseRecovery, CommaRecoveryMode, ForceCollect, Parser,
30    RecoverColon, RecoverComma, Recovery, token_descr,
31};
32use rustc_session::Session;
33use rustc_session::lint::builtin::{UNUSED_ATTRIBUTES, UNUSED_DOC_COMMENTS};
34use rustc_session::parse::feature_err;
35use rustc_span::hygiene::SyntaxContext;
36use rustc_span::{ErrorGuaranteed, FileName, Ident, LocalExpnId, Span, Symbol, sym};
37use smallvec::SmallVec;
38
39use crate::base::*;
40use crate::config::{StripUnconfigured, attr_into_trace};
41use crate::errors::{
42    EmptyDelegationMac, GlobDelegationOutsideImpls, GlobDelegationTraitlessQpath, IncompleteParse,
43    RecursionLimitReached, RemoveExprNotSupported, RemoveNodeNotSupported, UnsupportedKeyValue,
44    WrongFragmentKind,
45};
46use crate::mbe::diagnostics::annotate_err_with_kind;
47use crate::module::{
48    DirOwnership, ParsedExternalMod, mod_dir_path, mod_file_path_from_attr, parse_external_mod,
49};
50use crate::placeholders::{PlaceholderExpander, placeholder};
51use crate::stats::*;
52
53macro_rules! ast_fragments {
54    (
55        $($Kind:ident($AstTy:ty) {
56            $kind_name:expr;
57            $(one
58                fn $mut_visit_ast:ident;
59                fn $visit_ast:ident;
60                fn $ast_to_string:path;
61            )?
62            $(many
63                fn $flat_map_ast_elt:ident;
64                fn $visit_ast_elt:ident($($args:tt)*);
65                fn $ast_to_string_elt:path;
66            )?
67            fn $make_ast:ident;
68        })*
69    ) => {
70        /// A fragment of AST that can be produced by a single macro expansion.
71        /// Can also serve as an input and intermediate result for macro expansion operations.
72        pub enum AstFragment {
73            OptExpr(Option<Box<ast::Expr>>),
74            MethodReceiverExpr(Box<ast::Expr>),
75            $($Kind($AstTy),)*
76        }
77
78        /// "Discriminant" of an AST fragment.
79        #[derive(Copy, Clone, Debug, PartialEq, Eq)]
80        pub enum AstFragmentKind {
81            OptExpr,
82            MethodReceiverExpr,
83            $($Kind,)*
84        }
85
86        impl AstFragmentKind {
87            pub fn name(self) -> &'static str {
88                match self {
89                    AstFragmentKind::OptExpr => "expression",
90                    AstFragmentKind::MethodReceiverExpr => "expression",
91                    $(AstFragmentKind::$Kind => $kind_name,)*
92                }
93            }
94
95            fn make_from(self, result: Box<dyn MacResult + '_>) -> Option<AstFragment> {
96                match self {
97                    AstFragmentKind::OptExpr =>
98                        result.make_expr().map(Some).map(AstFragment::OptExpr),
99                    AstFragmentKind::MethodReceiverExpr =>
100                        result.make_expr().map(AstFragment::MethodReceiverExpr),
101                    $(AstFragmentKind::$Kind => result.$make_ast().map(AstFragment::$Kind),)*
102                }
103            }
104        }
105
106        impl AstFragment {
107            fn add_placeholders(&mut self, placeholders: &[NodeId]) {
108                if placeholders.is_empty() {
109                    return;
110                }
111                match self {
112                    $($(AstFragment::$Kind(ast) => ast.extend(placeholders.iter().flat_map(|id| {
113                        ${ignore($flat_map_ast_elt)}
114                        placeholder(AstFragmentKind::$Kind, *id, None).$make_ast()
115                    })),)?)*
116                    _ => panic!("unexpected AST fragment kind")
117                }
118            }
119
120            pub(crate) fn make_opt_expr(self) -> Option<Box<ast::Expr>> {
121                match self {
122                    AstFragment::OptExpr(expr) => expr,
123                    _ => panic!("AstFragment::make_* called on the wrong kind of fragment"),
124                }
125            }
126
127            pub(crate) fn make_method_receiver_expr(self) -> Box<ast::Expr> {
128                match self {
129                    AstFragment::MethodReceiverExpr(expr) => expr,
130                    _ => panic!("AstFragment::make_* called on the wrong kind of fragment"),
131                }
132            }
133
134            $(pub fn $make_ast(self) -> $AstTy {
135                match self {
136                    AstFragment::$Kind(ast) => ast,
137                    _ => panic!("AstFragment::make_* called on the wrong kind of fragment"),
138                }
139            })*
140
141            fn make_ast<T: InvocationCollectorNode>(self) -> T::OutputTy {
142                T::fragment_to_output(self)
143            }
144
145            pub(crate) fn mut_visit_with(&mut self, vis: &mut impl MutVisitor) {
146                match self {
147                    AstFragment::OptExpr(opt_expr) => {
148                        if let Some(expr) = opt_expr.take() {
149                            *opt_expr = vis.filter_map_expr(expr)
150                        }
151                    }
152                    AstFragment::MethodReceiverExpr(expr) => vis.visit_method_receiver_expr(expr),
153                    $($(AstFragment::$Kind(ast) => vis.$mut_visit_ast(ast),)?)*
154                    $($(AstFragment::$Kind(ast) =>
155                        ast.flat_map_in_place(|ast| vis.$flat_map_ast_elt(ast, $($args)*)),)?)*
156                }
157            }
158
159            pub fn visit_with<'a, V: Visitor<'a>>(&'a self, visitor: &mut V) -> V::Result {
160                match self {
161                    AstFragment::OptExpr(Some(expr)) => try_visit!(visitor.visit_expr(expr)),
162                    AstFragment::OptExpr(None) => {}
163                    AstFragment::MethodReceiverExpr(expr) => try_visit!(visitor.visit_method_receiver_expr(expr)),
164                    $($(AstFragment::$Kind(ast) => try_visit!(visitor.$visit_ast(ast)),)?)*
165                    $($(AstFragment::$Kind(ast) => walk_list!(visitor, $visit_ast_elt, &ast[..], $($args)*),)?)*
166                }
167                V::Result::output()
168            }
169
170            pub(crate) fn to_string(&self) -> String {
171                match self {
172                    AstFragment::OptExpr(Some(expr)) => pprust::expr_to_string(expr),
173                    AstFragment::OptExpr(None) => unreachable!(),
174                    AstFragment::MethodReceiverExpr(expr) => pprust::expr_to_string(expr),
175                    $($(AstFragment::$Kind(ast) => $ast_to_string(ast),)?)*
176                    $($(
177                        AstFragment::$Kind(ast) => {
178                            // The closure unwraps a `P` if present, or does nothing otherwise.
179                            elems_to_string(&*ast, |ast| $ast_to_string_elt(&*ast))
180                        }
181                    )?)*
182                }
183            }
184        }
185
186        impl<'a, 'b> MacResult for crate::mbe::macro_rules::ParserAnyMacro<'a, 'b> {
187            $(fn $make_ast(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
188                           -> Option<$AstTy> {
189                Some(self.make(AstFragmentKind::$Kind).$make_ast())
190            })*
191        }
192    }
193}
194
195/// A fragment of AST that can be produced by a single macro expansion.
/// Can also serve as an input and intermediate result for macro expansion operations.
pub enum AstFragment {
    OptExpr(Option<Box<ast::Expr>>),
    MethodReceiverExpr(Box<ast::Expr>),
    Expr(Box<ast::Expr>),
    Pat(Box<ast::Pat>),
    Ty(Box<ast::Ty>),
    Stmts(SmallVec<[ast::Stmt; 1]>),
    Items(SmallVec<[Box<ast::Item>; 1]>),
    TraitItems(SmallVec<[Box<ast::AssocItem>; 1]>),
    ImplItems(SmallVec<[Box<ast::AssocItem>; 1]>),
    TraitImplItems(SmallVec<[Box<ast::AssocItem>; 1]>),
    ForeignItems(SmallVec<[Box<ast::ForeignItem>; 1]>),
    Arms(SmallVec<[ast::Arm; 1]>),
    ExprFields(SmallVec<[ast::ExprField; 1]>),
    PatFields(SmallVec<[ast::PatField; 1]>),
    GenericParams(SmallVec<[ast::GenericParam; 1]>),
    Params(SmallVec<[ast::Param; 1]>),
    FieldDefs(SmallVec<[ast::FieldDef; 1]>),
    Variants(SmallVec<[ast::Variant; 1]>),
    WherePredicates(SmallVec<[ast::WherePredicate; 1]>),
    Crate(ast::Crate),
}
/// "Discriminant" of an AST fragment.
pub enum AstFragmentKind {
    OptExpr,
    MethodReceiverExpr,
    Expr,
    Pat,
    Ty,
    Stmts,
    Items,
    TraitItems,
    ImplItems,
    TraitImplItems,
    ForeignItems,
    Arms,
    ExprFields,
    PatFields,
    GenericParams,
    Params,
    FieldDefs,
    Variants,
    WherePredicates,
    Crate,
}
#[automatically_derived]
impl ::core::marker::Copy for AstFragmentKind { }
#[automatically_derived]
#[doc(hidden)]
unsafe impl ::core::clone::TrivialClone for AstFragmentKind { }
#[automatically_derived]
impl ::core::clone::Clone for AstFragmentKind {
    #[inline]
    fn clone(&self) -> AstFragmentKind { *self }
}
#[automatically_derived]
impl ::core::fmt::Debug for AstFragmentKind {
    #[inline]
    fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {
        ::core::fmt::Formatter::write_str(f,
            match self {
                AstFragmentKind::OptExpr => "OptExpr",
                AstFragmentKind::MethodReceiverExpr => "MethodReceiverExpr",
                AstFragmentKind::Expr => "Expr",
                AstFragmentKind::Pat => "Pat",
                AstFragmentKind::Ty => "Ty",
                AstFragmentKind::Stmts => "Stmts",
                AstFragmentKind::Items => "Items",
                AstFragmentKind::TraitItems => "TraitItems",
                AstFragmentKind::ImplItems => "ImplItems",
                AstFragmentKind::TraitImplItems => "TraitImplItems",
                AstFragmentKind::ForeignItems => "ForeignItems",
                AstFragmentKind::Arms => "Arms",
                AstFragmentKind::ExprFields => "ExprFields",
                AstFragmentKind::PatFields => "PatFields",
                AstFragmentKind::GenericParams => "GenericParams",
                AstFragmentKind::Params => "Params",
                AstFragmentKind::FieldDefs => "FieldDefs",
                AstFragmentKind::Variants => "Variants",
                AstFragmentKind::WherePredicates => "WherePredicates",
                AstFragmentKind::Crate => "Crate",
            })
    }
}
#[automatically_derived]
impl ::core::marker::StructuralPartialEq for AstFragmentKind { }
#[automatically_derived]
impl ::core::cmp::PartialEq for AstFragmentKind {
    #[inline]
    fn eq(&self, other: &AstFragmentKind) -> bool {
        let __self_discr = ::core::intrinsics::discriminant_value(self);
        let __arg1_discr = ::core::intrinsics::discriminant_value(other);
        __self_discr == __arg1_discr
    }
}
#[automatically_derived]
impl ::core::cmp::Eq for AstFragmentKind {
    #[inline]
    #[doc(hidden)]
    #[coverage(off)]
    fn assert_fields_are_eq(&self) {}
}
impl AstFragmentKind {
    pub fn name(self) -> &'static str {
        match self {
            AstFragmentKind::OptExpr => "expression",
            AstFragmentKind::MethodReceiverExpr => "expression",
            AstFragmentKind::Expr => "expression",
            AstFragmentKind::Pat => "pattern",
            AstFragmentKind::Ty => "type",
            AstFragmentKind::Stmts => "statement",
            AstFragmentKind::Items => "item",
            AstFragmentKind::TraitItems => "trait item",
            AstFragmentKind::ImplItems => "impl item",
            AstFragmentKind::TraitImplItems => "impl item",
            AstFragmentKind::ForeignItems => "foreign item",
            AstFragmentKind::Arms => "match arm",
            AstFragmentKind::ExprFields => "field expression",
            AstFragmentKind::PatFields => "field pattern",
            AstFragmentKind::GenericParams => "generic parameter",
            AstFragmentKind::Params => "function parameter",
            AstFragmentKind::FieldDefs => "field",
            AstFragmentKind::Variants => "variant",
            AstFragmentKind::WherePredicates => "where predicate",
            AstFragmentKind::Crate => "crate",
        }
    }
    fn make_from(self, result: Box<dyn MacResult + '_>)
        -> Option<AstFragment> {
        match self {
            AstFragmentKind::OptExpr =>
                result.make_expr().map(Some).map(AstFragment::OptExpr),
            AstFragmentKind::MethodReceiverExpr =>
                result.make_expr().map(AstFragment::MethodReceiverExpr),
            AstFragmentKind::Expr =>
                result.make_expr().map(AstFragment::Expr),
            AstFragmentKind::Pat => result.make_pat().map(AstFragment::Pat),
            AstFragmentKind::Ty => result.make_ty().map(AstFragment::Ty),
            AstFragmentKind::Stmts =>
                result.make_stmts().map(AstFragment::Stmts),
            AstFragmentKind::Items =>
                result.make_items().map(AstFragment::Items),
            AstFragmentKind::TraitItems =>
                result.make_trait_items().map(AstFragment::TraitItems),
            AstFragmentKind::ImplItems =>
                result.make_impl_items().map(AstFragment::ImplItems),
            AstFragmentKind::TraitImplItems =>
                result.make_trait_impl_items().map(AstFragment::TraitImplItems),
            AstFragmentKind::ForeignItems =>
                result.make_foreign_items().map(AstFragment::ForeignItems),
            AstFragmentKind::Arms =>
                result.make_arms().map(AstFragment::Arms),
            AstFragmentKind::ExprFields =>
                result.make_expr_fields().map(AstFragment::ExprFields),
            AstFragmentKind::PatFields =>
                result.make_pat_fields().map(AstFragment::PatFields),
            AstFragmentKind::GenericParams =>
                result.make_generic_params().map(AstFragment::GenericParams),
            AstFragmentKind::Params =>
                result.make_params().map(AstFragment::Params),
            AstFragmentKind::FieldDefs =>
                result.make_field_defs().map(AstFragment::FieldDefs),
            AstFragmentKind::Variants =>
                result.make_variants().map(AstFragment::Variants),
            AstFragmentKind::WherePredicates =>
                result.make_where_predicates().map(AstFragment::WherePredicates),
            AstFragmentKind::Crate =>
                result.make_crate().map(AstFragment::Crate),
        }
    }
}
impl AstFragment {
    fn add_placeholders(&mut self, placeholders: &[NodeId]) {
        if placeholders.is_empty() { return; }
        match self {
            AstFragment::Stmts(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::Stmts, *id, None).make_stmts()
                            })),
            AstFragment::Items(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::Items, *id, None).make_items()
                            })),
            AstFragment::TraitItems(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::TraitItems, *id,
                                        None).make_trait_items()
                            })),
            AstFragment::ImplItems(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::ImplItems, *id,
                                        None).make_impl_items()
                            })),
            AstFragment::TraitImplItems(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::TraitImplItems, *id,
                                        None).make_trait_impl_items()
                            })),
            AstFragment::ForeignItems(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::ForeignItems, *id,
                                        None).make_foreign_items()
                            })),
            AstFragment::Arms(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::Arms, *id, None).make_arms()
                            })),
            AstFragment::ExprFields(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::ExprFields, *id,
                                        None).make_expr_fields()
                            })),
            AstFragment::PatFields(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::PatFields, *id,
                                        None).make_pat_fields()
                            })),
            AstFragment::GenericParams(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::GenericParams, *id,
                                        None).make_generic_params()
                            })),
            AstFragment::Params(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::Params, *id,
                                        None).make_params()
                            })),
            AstFragment::FieldDefs(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::FieldDefs, *id,
                                        None).make_field_defs()
                            })),
            AstFragment::Variants(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::Variants, *id,
                                        None).make_variants()
                            })),
            AstFragment::WherePredicates(ast) =>
                ast.extend(placeholders.iter().flat_map(|id|
                            {
                                placeholder(AstFragmentKind::WherePredicates, *id,
                                        None).make_where_predicates()
                            })),
            _ => {
                ::core::panicking::panic_fmt(format_args!("unexpected AST fragment kind"));
            }
        }
    }
    pub(crate) fn make_opt_expr(self) -> Option<Box<ast::Expr>> {
        match self {
            AstFragment::OptExpr(expr) => expr,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub(crate) fn make_method_receiver_expr(self) -> Box<ast::Expr> {
        match self {
            AstFragment::MethodReceiverExpr(expr) => expr,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_expr(self) -> Box<ast::Expr> {
        match self {
            AstFragment::Expr(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_pat(self) -> Box<ast::Pat> {
        match self {
            AstFragment::Pat(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_ty(self) -> Box<ast::Ty> {
        match self {
            AstFragment::Ty(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_stmts(self) -> SmallVec<[ast::Stmt; 1]> {
        match self {
            AstFragment::Stmts(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_items(self) -> SmallVec<[Box<ast::Item>; 1]> {
        match self {
            AstFragment::Items(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_trait_items(self) -> SmallVec<[Box<ast::AssocItem>; 1]> {
        match self {
            AstFragment::TraitItems(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_impl_items(self) -> SmallVec<[Box<ast::AssocItem>; 1]> {
        match self {
            AstFragment::ImplItems(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_trait_impl_items(self) -> SmallVec<[Box<ast::AssocItem>; 1]> {
        match self {
            AstFragment::TraitImplItems(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_foreign_items(self) -> SmallVec<[Box<ast::ForeignItem>; 1]> {
        match self {
            AstFragment::ForeignItems(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_arms(self) -> SmallVec<[ast::Arm; 1]> {
        match self {
            AstFragment::Arms(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_expr_fields(self) -> SmallVec<[ast::ExprField; 1]> {
        match self {
            AstFragment::ExprFields(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_pat_fields(self) -> SmallVec<[ast::PatField; 1]> {
        match self {
            AstFragment::PatFields(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_generic_params(self) -> SmallVec<[ast::GenericParam; 1]> {
        match self {
            AstFragment::GenericParams(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_params(self) -> SmallVec<[ast::Param; 1]> {
        match self {
            AstFragment::Params(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_field_defs(self) -> SmallVec<[ast::FieldDef; 1]> {
        match self {
            AstFragment::FieldDefs(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_variants(self) -> SmallVec<[ast::Variant; 1]> {
        match self {
            AstFragment::Variants(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_where_predicates(self) -> SmallVec<[ast::WherePredicate; 1]> {
        match self {
            AstFragment::WherePredicates(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    pub fn make_crate(self) -> ast::Crate {
        match self {
            AstFragment::Crate(ast) => ast,
            _ => {
                ::core::panicking::panic_fmt(format_args!("AstFragment::make_* called on the wrong kind of fragment"));
            }
        }
    }
    fn make_ast<T: InvocationCollectorNode>(self) -> T::OutputTy {
        T::fragment_to_output(self)
    }
    pub(crate) fn mut_visit_with(&mut self, vis: &mut impl MutVisitor) {
        match self {
            AstFragment::OptExpr(opt_expr) => {
                if let Some(expr) = opt_expr.take() {
                    *opt_expr = vis.filter_map_expr(expr)
                }
            }
            AstFragment::MethodReceiverExpr(expr) =>
                vis.visit_method_receiver_expr(expr),
            AstFragment::Expr(ast) => vis.visit_expr(ast),
            AstFragment::Pat(ast) => vis.visit_pat(ast),
            AstFragment::Ty(ast) => vis.visit_ty(ast),
            AstFragment::Crate(ast) => vis.visit_crate(ast),
            AstFragment::Stmts(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_stmt(ast)),
            AstFragment::Items(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_item(ast)),
            AstFragment::TraitItems(ast) =>
                ast.flat_map_in_place(|ast|
                        vis.flat_map_assoc_item(ast, AssocCtxt::Trait)),
            AstFragment::ImplItems(ast) =>
                ast.flat_map_in_place(|ast|
                        vis.flat_map_assoc_item(ast,
                            AssocCtxt::Impl { of_trait: false })),
            AstFragment::TraitImplItems(ast) =>
                ast.flat_map_in_place(|ast|
                        vis.flat_map_assoc_item(ast,
                            AssocCtxt::Impl { of_trait: true })),
            AstFragment::ForeignItems(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_foreign_item(ast)),
            AstFragment::Arms(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_arm(ast)),
            AstFragment::ExprFields(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_expr_field(ast)),
            AstFragment::PatFields(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_pat_field(ast)),
            AstFragment::GenericParams(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_generic_param(ast)),
            AstFragment::Params(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_param(ast)),
            AstFragment::FieldDefs(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_field_def(ast)),
            AstFragment::Variants(ast) =>
                ast.flat_map_in_place(|ast| vis.flat_map_variant(ast)),
            AstFragment::WherePredicates(ast) =>
                ast.flat_map_in_place(|ast|
                        vis.flat_map_where_predicate(ast)),
        }
    }
    pub fn visit_with<'a, V: Visitor<'a>>(&'a self, visitor: &mut V)
        -> V::Result {
        match self {
            AstFragment::OptExpr(Some(expr)) =>
                match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_expr(expr))
                    {
                    core::ops::ControlFlow::Continue(()) =>
                        (),
                        #[allow(unreachable_code)]
                        core::ops::ControlFlow::Break(r) => {
                        return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                    }
                },
            AstFragment::OptExpr(None) => {}
            AstFragment::MethodReceiverExpr(expr) =>
                match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_method_receiver_expr(expr))
                    {
                    core::ops::ControlFlow::Continue(()) =>
                        (),
                        #[allow(unreachable_code)]
                        core::ops::ControlFlow::Break(r) => {
                        return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                    }
                },
            AstFragment::Expr(ast) =>
                match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_expr(ast))
                    {
                    core::ops::ControlFlow::Continue(()) =>
                        (),
                        #[allow(unreachable_code)]
                        core::ops::ControlFlow::Break(r) => {
                        return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                    }
                },
            AstFragment::Pat(ast) =>
                match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_pat(ast))
                    {
                    core::ops::ControlFlow::Continue(()) =>
                        (),
                        #[allow(unreachable_code)]
                        core::ops::ControlFlow::Break(r) => {
                        return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                    }
                },
            AstFragment::Ty(ast) =>
                match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_ty(ast))
                    {
                    core::ops::ControlFlow::Continue(()) =>
                        (),
                        #[allow(unreachable_code)]
                        core::ops::ControlFlow::Break(r) => {
                        return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                    }
                },
            AstFragment::Crate(ast) =>
                match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_crate(ast))
                    {
                    core::ops::ControlFlow::Continue(()) =>
                        (),
                        #[allow(unreachable_code)]
                        core::ops::ControlFlow::Break(r) => {
                        return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                    }
                },
            AstFragment::Stmts(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_stmt(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::Items(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_item(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::TraitItems(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_assoc_item(elem,
                                AssocCtxt::Trait)) {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::ImplItems(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_assoc_item(elem,
                                AssocCtxt::Impl { of_trait: false })) {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::TraitImplItems(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_assoc_item(elem,
                                AssocCtxt::Impl { of_trait: true })) {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::ForeignItems(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_foreign_item(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::Arms(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_arm(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::ExprFields(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_expr_field(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::PatFields(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_pat_field(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::GenericParams(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_generic_param(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::Params(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_param(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::FieldDefs(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_field_def(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::Variants(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_variant(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
            AstFragment::WherePredicates(ast) =>
                for elem in &ast[..] {
                    match ::rustc_ast_ir::visit::VisitorResult::branch(visitor.visit_where_predicate(elem))
                        {
                        core::ops::ControlFlow::Continue(()) =>
                            (),
                            #[allow(unreachable_code)]
                            core::ops::ControlFlow::Break(r) => {
                            return ::rustc_ast_ir::visit::VisitorResult::from_residual(r);
                        }
                    };
                },
        }
        V::Result::output()
    }
    pub(crate) fn to_string(&self) -> String {
        match self {
            AstFragment::OptExpr(Some(expr)) => pprust::expr_to_string(expr),
            AstFragment::OptExpr(None) =>
                ::core::panicking::panic("internal error: entered unreachable code"),
            AstFragment::MethodReceiverExpr(expr) =>
                pprust::expr_to_string(expr),
            AstFragment::Expr(ast) => pprust::expr_to_string(ast),
            AstFragment::Pat(ast) => pprust::pat_to_string(ast),
            AstFragment::Ty(ast) => pprust::ty_to_string(ast),
            AstFragment::Crate(ast) => unreachable_to_string(ast),
            AstFragment::Stmts(ast) => {
                elems_to_string(&*ast, |ast| pprust::stmt_to_string(&*ast))
            }
            AstFragment::Items(ast) => {
                elems_to_string(&*ast, |ast| pprust::item_to_string(&*ast))
            }
            AstFragment::TraitItems(ast) => {
                elems_to_string(&*ast,
                    |ast| pprust::assoc_item_to_string(&*ast))
            }
            AstFragment::ImplItems(ast) => {
                elems_to_string(&*ast,
                    |ast| pprust::assoc_item_to_string(&*ast))
            }
            AstFragment::TraitImplItems(ast) => {
                elems_to_string(&*ast,
                    |ast| pprust::assoc_item_to_string(&*ast))
            }
            AstFragment::ForeignItems(ast) => {
                elems_to_string(&*ast,
                    |ast| pprust::foreign_item_to_string(&*ast))
            }
            AstFragment::Arms(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::ExprFields(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::PatFields(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::GenericParams(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::Params(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::FieldDefs(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::Variants(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
            AstFragment::WherePredicates(ast) => {
                elems_to_string(&*ast, |ast| unreachable_to_string(&*ast))
            }
        }
    }
}
impl<'a, 'b> MacResult for crate::mbe::macro_rules::ParserAnyMacro<'a, 'b> {
    fn make_expr(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<Box<ast::Expr>> {
        Some(self.make(AstFragmentKind::Expr).make_expr())
    }
    fn make_pat(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<Box<ast::Pat>> {
        Some(self.make(AstFragmentKind::Pat).make_pat())
    }
    fn make_ty(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<Box<ast::Ty>> {
        Some(self.make(AstFragmentKind::Ty).make_ty())
    }
    fn make_stmts(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::Stmt; 1]>> {
        Some(self.make(AstFragmentKind::Stmts).make_stmts())
    }
    fn make_items(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[Box<ast::Item>; 1]>> {
        Some(self.make(AstFragmentKind::Items).make_items())
    }
    fn make_trait_items(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
        Some(self.make(AstFragmentKind::TraitItems).make_trait_items())
    }
    fn make_impl_items(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
        Some(self.make(AstFragmentKind::ImplItems).make_impl_items())
    }
    fn make_trait_impl_items(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[Box<ast::AssocItem>; 1]>> {
        Some(self.make(AstFragmentKind::TraitImplItems).make_trait_impl_items())
    }
    fn make_foreign_items(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[Box<ast::ForeignItem>; 1]>> {
        Some(self.make(AstFragmentKind::ForeignItems).make_foreign_items())
    }
    fn make_arms(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::Arm; 1]>> {
        Some(self.make(AstFragmentKind::Arms).make_arms())
    }
    fn make_expr_fields(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::ExprField; 1]>> {
        Some(self.make(AstFragmentKind::ExprFields).make_expr_fields())
    }
    fn make_pat_fields(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::PatField; 1]>> {
        Some(self.make(AstFragmentKind::PatFields).make_pat_fields())
    }
    fn make_generic_params(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::GenericParam; 1]>> {
        Some(self.make(AstFragmentKind::GenericParams).make_generic_params())
    }
    fn make_params(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::Param; 1]>> {
        Some(self.make(AstFragmentKind::Params).make_params())
    }
    fn make_field_defs(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::FieldDef; 1]>> {
        Some(self.make(AstFragmentKind::FieldDefs).make_field_defs())
    }
    fn make_variants(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::Variant; 1]>> {
        Some(self.make(AstFragmentKind::Variants).make_variants())
    }
    fn make_where_predicates(self:
            Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<SmallVec<[ast::WherePredicate; 1]>> {
        Some(self.make(AstFragmentKind::WherePredicates).make_where_predicates())
    }
    fn make_crate(self: Box<crate::mbe::macro_rules::ParserAnyMacro<'a, 'b>>)
        -> Option<ast::Crate> {
        Some(self.make(AstFragmentKind::Crate).make_crate())
    }
}ast_fragments! {
196    Expr(Box<ast::Expr>) {
197        "expression";
198        one fn visit_expr; fn visit_expr; fn pprust::expr_to_string;
199        fn make_expr;
200    }
201    Pat(Box<ast::Pat>) {
202        "pattern";
203        one fn visit_pat; fn visit_pat; fn pprust::pat_to_string;
204        fn make_pat;
205    }
206    Ty(Box<ast::Ty>) {
207        "type";
208        one fn visit_ty; fn visit_ty; fn pprust::ty_to_string;
209        fn make_ty;
210    }
211    Stmts(SmallVec<[ast::Stmt; 1]>) {
212        "statement";
213        many fn flat_map_stmt; fn visit_stmt(); fn pprust::stmt_to_string;
214        fn make_stmts;
215    }
216    Items(SmallVec<[Box<ast::Item>; 1]>) {
217        "item";
218        many fn flat_map_item; fn visit_item(); fn pprust::item_to_string;
219        fn make_items;
220    }
221    TraitItems(SmallVec<[Box<ast::AssocItem>; 1]>) {
222        "trait item";
223        many fn flat_map_assoc_item; fn visit_assoc_item(AssocCtxt::Trait);
224            fn pprust::assoc_item_to_string;
225        fn make_trait_items;
226    }
227    ImplItems(SmallVec<[Box<ast::AssocItem>; 1]>) {
228        "impl item";
229        many fn flat_map_assoc_item; fn visit_assoc_item(AssocCtxt::Impl { of_trait: false });
230            fn pprust::assoc_item_to_string;
231        fn make_impl_items;
232    }
233    TraitImplItems(SmallVec<[Box<ast::AssocItem>; 1]>) {
234        "impl item";
235        many fn flat_map_assoc_item; fn visit_assoc_item(AssocCtxt::Impl { of_trait: true });
236            fn pprust::assoc_item_to_string;
237        fn make_trait_impl_items;
238    }
239    ForeignItems(SmallVec<[Box<ast::ForeignItem>; 1]>) {
240        "foreign item";
241        many fn flat_map_foreign_item; fn visit_foreign_item(); fn pprust::foreign_item_to_string;
242        fn make_foreign_items;
243    }
244    Arms(SmallVec<[ast::Arm; 1]>) {
245        "match arm";
246        many fn flat_map_arm; fn visit_arm(); fn unreachable_to_string;
247        fn make_arms;
248    }
249    ExprFields(SmallVec<[ast::ExprField; 1]>) {
250        "field expression";
251        many fn flat_map_expr_field; fn visit_expr_field(); fn unreachable_to_string;
252        fn make_expr_fields;
253    }
254    PatFields(SmallVec<[ast::PatField; 1]>) {
255        "field pattern";
256        many fn flat_map_pat_field; fn visit_pat_field(); fn unreachable_to_string;
257        fn make_pat_fields;
258    }
259    GenericParams(SmallVec<[ast::GenericParam; 1]>) {
260        "generic parameter";
261        many fn flat_map_generic_param; fn visit_generic_param(); fn unreachable_to_string;
262        fn make_generic_params;
263    }
264    Params(SmallVec<[ast::Param; 1]>) {
265        "function parameter";
266        many fn flat_map_param; fn visit_param(); fn unreachable_to_string;
267        fn make_params;
268    }
269    FieldDefs(SmallVec<[ast::FieldDef; 1]>) {
270        "field";
271        many fn flat_map_field_def; fn visit_field_def(); fn unreachable_to_string;
272        fn make_field_defs;
273    }
274    Variants(SmallVec<[ast::Variant; 1]>) {
275        "variant"; many fn flat_map_variant; fn visit_variant(); fn unreachable_to_string;
276        fn make_variants;
277    }
278    WherePredicates(SmallVec<[ast::WherePredicate; 1]>) {
279        "where predicate";
280        many fn flat_map_where_predicate; fn visit_where_predicate(); fn unreachable_to_string;
281        fn make_where_predicates;
282    }
283    Crate(ast::Crate) {
284        "crate";
285        one fn visit_crate; fn visit_crate; fn unreachable_to_string;
286        fn make_crate;
287    }
288}
289
290pub enum SupportsMacroExpansion {
291    No,
292    Yes { supports_inner_attrs: bool },
293}
294
295impl AstFragmentKind {
296    pub(crate) fn dummy(self, span: Span, guar: ErrorGuaranteed) -> AstFragment {
297        self.make_from(DummyResult::any(span, guar)).expect("couldn't create a dummy AST fragment")
298    }
299
300    pub fn supports_macro_expansion(self) -> SupportsMacroExpansion {
301        match self {
302            AstFragmentKind::OptExpr
303            | AstFragmentKind::Expr
304            | AstFragmentKind::MethodReceiverExpr
305            | AstFragmentKind::Stmts
306            | AstFragmentKind::Ty
307            | AstFragmentKind::Pat => SupportsMacroExpansion::Yes { supports_inner_attrs: false },
308            AstFragmentKind::Items
309            | AstFragmentKind::TraitItems
310            | AstFragmentKind::ImplItems
311            | AstFragmentKind::TraitImplItems
312            | AstFragmentKind::ForeignItems
313            | AstFragmentKind::Crate => SupportsMacroExpansion::Yes { supports_inner_attrs: true },
314            AstFragmentKind::Arms
315            | AstFragmentKind::ExprFields
316            | AstFragmentKind::PatFields
317            | AstFragmentKind::GenericParams
318            | AstFragmentKind::Params
319            | AstFragmentKind::FieldDefs
320            | AstFragmentKind::Variants
321            | AstFragmentKind::WherePredicates => SupportsMacroExpansion::No,
322        }
323    }
324
325    pub(crate) fn expect_from_annotatables(
326        self,
327        items: impl IntoIterator<Item = Annotatable>,
328    ) -> AstFragment {
329        let mut items = items.into_iter();
330        match self {
331            AstFragmentKind::Arms => {
332                AstFragment::Arms(items.map(Annotatable::expect_arm).collect())
333            }
334            AstFragmentKind::ExprFields => {
335                AstFragment::ExprFields(items.map(Annotatable::expect_expr_field).collect())
336            }
337            AstFragmentKind::PatFields => {
338                AstFragment::PatFields(items.map(Annotatable::expect_pat_field).collect())
339            }
340            AstFragmentKind::GenericParams => {
341                AstFragment::GenericParams(items.map(Annotatable::expect_generic_param).collect())
342            }
343            AstFragmentKind::Params => {
344                AstFragment::Params(items.map(Annotatable::expect_param).collect())
345            }
346            AstFragmentKind::FieldDefs => {
347                AstFragment::FieldDefs(items.map(Annotatable::expect_field_def).collect())
348            }
349            AstFragmentKind::Variants => {
350                AstFragment::Variants(items.map(Annotatable::expect_variant).collect())
351            }
352            AstFragmentKind::WherePredicates => AstFragment::WherePredicates(
353                items.map(Annotatable::expect_where_predicate).collect(),
354            ),
355            AstFragmentKind::Items => {
356                AstFragment::Items(items.map(Annotatable::expect_item).collect())
357            }
358            AstFragmentKind::ImplItems => {
359                AstFragment::ImplItems(items.map(Annotatable::expect_impl_item).collect())
360            }
361            AstFragmentKind::TraitImplItems => {
362                AstFragment::TraitImplItems(items.map(Annotatable::expect_impl_item).collect())
363            }
364            AstFragmentKind::TraitItems => {
365                AstFragment::TraitItems(items.map(Annotatable::expect_trait_item).collect())
366            }
367            AstFragmentKind::ForeignItems => {
368                AstFragment::ForeignItems(items.map(Annotatable::expect_foreign_item).collect())
369            }
370            AstFragmentKind::Stmts => {
371                AstFragment::Stmts(items.map(Annotatable::expect_stmt).collect())
372            }
373            AstFragmentKind::Expr => AstFragment::Expr(
374                items.next().expect("expected exactly one expression").expect_expr(),
375            ),
376            AstFragmentKind::MethodReceiverExpr => AstFragment::MethodReceiverExpr(
377                items.next().expect("expected exactly one expression").expect_expr(),
378            ),
379            AstFragmentKind::OptExpr => {
380                AstFragment::OptExpr(items.next().map(Annotatable::expect_expr))
381            }
382            AstFragmentKind::Crate => {
383                AstFragment::Crate(items.next().expect("expected exactly one crate").expect_crate())
384            }
385            AstFragmentKind::Pat | AstFragmentKind::Ty => {
386                {
    ::core::panicking::panic_fmt(format_args!("patterns and types aren\'t annotatable"));
}panic!("patterns and types aren't annotatable")
387            }
388        }
389    }
390}
391
392pub struct Invocation {
393    pub kind: InvocationKind,
394    pub fragment_kind: AstFragmentKind,
395    pub expansion_data: ExpansionData,
396}
397
398pub enum InvocationKind {
399    Bang {
400        mac: Box<ast::MacCall>,
401        span: Span,
402    },
403    Attr {
404        attr: ast::Attribute,
405        /// Re-insertion position for inert attributes.
406        pos: usize,
407        item: Annotatable,
408        /// Required for resolving derive helper attributes.
409        derives: Vec<ast::Path>,
410    },
411    Derive {
412        path: ast::Path,
413        is_const: bool,
414        item: Annotatable,
415    },
416    GlobDelegation {
417        item: Box<ast::AssocItem>,
418        /// Whether this is a trait impl or an inherent impl
419        of_trait: bool,
420    },
421}
422
423impl InvocationKind {
424    fn placeholder_visibility(&self) -> Option<ast::Visibility> {
425        // HACK: For unnamed fields placeholders should have the same visibility as the actual
426        // fields because for tuple structs/variants resolve determines visibilities of their
427        // constructor using these field visibilities before attributes on them are expanded.
428        // The assumption is that the attribute expansion cannot change field visibilities,
429        // and it holds because only inert attributes are supported in this position.
430        match self {
431            InvocationKind::Attr { item: Annotatable::FieldDef(field), .. }
432            | InvocationKind::Derive { item: Annotatable::FieldDef(field), .. }
433                if field.ident.is_none() =>
434            {
435                Some(field.vis.clone())
436            }
437            _ => None,
438        }
439    }
440}
441
442impl Invocation {
443    pub fn span(&self) -> Span {
444        match &self.kind {
445            InvocationKind::Bang { span, .. } => *span,
446            InvocationKind::Attr { attr, .. } => attr.span,
447            InvocationKind::Derive { path, .. } => path.span,
448            InvocationKind::GlobDelegation { item, .. } => item.span,
449        }
450    }
451
452    fn span_mut(&mut self) -> &mut Span {
453        match &mut self.kind {
454            InvocationKind::Bang { span, .. } => span,
455            InvocationKind::Attr { attr, .. } => &mut attr.span,
456            InvocationKind::Derive { path, .. } => &mut path.span,
457            InvocationKind::GlobDelegation { item, .. } => &mut item.span,
458        }
459    }
460}
461
462pub struct MacroExpander<'a, 'b> {
463    pub cx: &'a mut ExtCtxt<'b>,
464    monotonic: bool, // cf. `cx.monotonic_expander()`
465}
466
467impl<'a, 'b> MacroExpander<'a, 'b> {
468    pub fn new(cx: &'a mut ExtCtxt<'b>, monotonic: bool) -> Self {
469        MacroExpander { cx, monotonic }
470    }
471
472    pub fn expand_crate(&mut self, krate: ast::Crate) -> ast::Crate {
473        let file_path = match self.cx.source_map().span_to_filename(krate.spans.inner_span) {
474            FileName::Real(name) => name
475                .into_local_path()
476                .expect("attempting to resolve a file path in an external file"),
477            other => PathBuf::from(other.prefer_local_unconditionally().to_string()),
478        };
479        let dir_path = file_path.parent().unwrap_or(&file_path).to_owned();
480        self.cx.root_path = dir_path.clone();
481        self.cx.current_expansion.module = Rc::new(ModuleData {
482            mod_path: ::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
        [Ident::with_dummy_span(self.cx.ecfg.crate_name)]))vec![Ident::with_dummy_span(self.cx.ecfg.crate_name)],
483            file_path_stack: ::alloc::boxed::box_assume_init_into_vec_unsafe(::alloc::intrinsics::write_box_via_move(::alloc::boxed::Box::new_uninit(),
        [file_path]))vec![file_path],
484            dir_path,
485        });
486        let krate = self.fully_expand_fragment(AstFragment::Crate(krate)).make_crate();
487        match (&krate.id, &ast::CRATE_NODE_ID) {
    (left_val, right_val) => {
        if !(*left_val == *right_val) {
            let kind = ::core::panicking::AssertKind::Eq;
            ::core::panicking::assert_failed(kind, &*left_val, &*right_val,
                ::core::option::Option::None);
        }
    }
};assert_eq!(krate.id, ast::CRATE_NODE_ID);
488        self.cx.trace_macros_diag();
489        krate
490    }
491
492    /// Recursively expand all macro invocations in this AST fragment.
493    pub fn fully_expand_fragment(&mut self, input_fragment: AstFragment) -> AstFragment {
494        let orig_expansion_data = self.cx.current_expansion.clone();
495        let orig_force_mode = self.cx.force_mode;
496
497        // Collect all macro invocations and replace them with placeholders.
498        let (mut fragment_with_placeholders, mut invocations) =
499            self.collect_invocations(input_fragment, &[]);
500
501        // Optimization: if we resolve all imports now,
502        // we'll be able to immediately resolve most of imported macros.
503        self.resolve_imports();
504
505        // Resolve paths in all invocations and produce output expanded fragments for them, but
506        // do not insert them into our input AST fragment yet, only store in `expanded_fragments`.
507        // The output fragments also go through expansion recursively until no invocations are left.
508        // Unresolved macros produce dummy outputs as a recovery measure.
509        invocations.reverse();
510        let mut expanded_fragments = Vec::new();
511        let mut expanded_fragments_len = 0;
512        let mut undetermined_invocations = Vec::new();
513        let (mut progress, mut force) = (false, !self.monotonic);
514        loop {
515            let Some((invoc, ext)) = invocations.pop() else {
516                self.resolve_imports();
517                if undetermined_invocations.is_empty() {
518                    break;
519                }
520                invocations = mem::take(&mut undetermined_invocations);
521                force = !progress;
522                progress = false;
523                if force && self.monotonic {
524                    self.cx.dcx().span_delayed_bug(
525                        invocations.last().unwrap().0.span(),
526                        "expansion entered force mode without producing any errors",
527                    );
528                }
529                continue;
530            };
531
532            let ext = match ext {
533                Some(ext) => ext,
534                None => {
535                    let eager_expansion_root = if self.monotonic {
536                        invoc.expansion_data.id
537                    } else {
538                        orig_expansion_data.id
539                    };
540                    match self.cx.resolver.resolve_macro_invocation(
541                        &invoc,
542                        eager_expansion_root,
543                        force,
544                    ) {
545                        Ok(ext) => ext,
546                        Err(Indeterminate) => {
547                            // Cannot resolve, will retry this invocation later.
548                            undetermined_invocations.push((invoc, None));
549                            continue;
550                        }
551                    }
552                }
553            };
554
555            let ExpansionData { depth, id: expn_id, .. } = invoc.expansion_data;
556            let depth = depth - orig_expansion_data.depth;
557            self.cx.current_expansion = invoc.expansion_data.clone();
558            self.cx.force_mode = force;
559
560            let fragment_kind = invoc.fragment_kind;
561            match self.expand_invoc(invoc, &ext.kind) {
562                ExpandResult::Ready(fragment) => {
563                    let mut derive_invocations = Vec::new();
564                    let derive_placeholders = self
565                        .cx
566                        .resolver
567                        .take_derive_resolutions(expn_id)
568                        .map(|derives| {
569                            derive_invocations.reserve(derives.len());
570                            derives
571                                .into_iter()
572                                .map(|DeriveResolution { path, item, exts: _, is_const }| {
573                                    // FIXME: Consider using the derive resolutions (`_exts`)
574                                    // instead of enqueuing the derives to be resolved again later.
575                                    // Note that this can result in duplicate diagnostics.
576                                    let expn_id = LocalExpnId::fresh_empty();
577                                    derive_invocations.push((
578                                        Invocation {
579                                            kind: InvocationKind::Derive { path, item, is_const },
580                                            fragment_kind,
581                                            expansion_data: ExpansionData {
582                                                id: expn_id,
583                                                ..self.cx.current_expansion.clone()
584                                            },
585                                        },
586                                        None,
587                                    ));
588                                    NodeId::placeholder_from_expn_id(expn_id)
589                                })
590                                .collect::<Vec<_>>()
591                        })
592                        .unwrap_or_default();
593
594                    let (expanded_fragment, collected_invocations) =
595                        self.collect_invocations(fragment, &derive_placeholders);
596                    // We choose to expand any derive invocations associated with this macro
597                    // invocation *before* any macro invocations collected from the output
598                    // fragment.
599                    derive_invocations.extend(collected_invocations);
600
601                    progress = true;
602                    if expanded_fragments.len() < depth {
603                        expanded_fragments.push(Vec::new());
604                    }
605                    expanded_fragments[depth - 1].push((expn_id, expanded_fragment));
606                    expanded_fragments_len += 1;
607                    invocations.extend(derive_invocations.into_iter().rev());
608                }
609                ExpandResult::Retry(invoc) => {
610                    if force {
611                        self.cx.dcx().span_bug(
612                            invoc.span(),
613                            "expansion entered force mode but is still stuck",
614                        );
615                    } else {
616                        // Cannot expand, will retry this invocation later.
617                        undetermined_invocations.push((invoc, Some(ext)));
618                    }
619                }
620            }
621        }
622
623        self.cx.current_expansion = orig_expansion_data;
624        self.cx.force_mode = orig_force_mode;
625
626        // Finally incorporate all the expanded macros into the input AST fragment.
627        let mut placeholder_expander = PlaceholderExpander::with_capacity(expanded_fragments_len);
628        while let Some(expanded_fragments) = expanded_fragments.pop() {
629            for (expn_id, expanded_fragment) in expanded_fragments.into_iter().rev() {
630                placeholder_expander
631                    .add(NodeId::placeholder_from_expn_id(expn_id), expanded_fragment);
632            }
633        }
634        fragment_with_placeholders.mut_visit_with(&mut placeholder_expander);
635        fragment_with_placeholders
636    }
637
638    fn resolve_imports(&mut self) {
639        if self.monotonic {
640            self.cx.resolver.resolve_imports();
641        }
642    }
643
644    /// Collects all macro invocations reachable at this time in this AST fragment, and replace
645    /// them with "placeholders" - dummy macro invocations with specially crafted `NodeId`s.
646    /// Then call into resolver that builds a skeleton ("reduced graph") of the fragment and
647    /// prepares data for resolving paths of macro invocations.
648    fn collect_invocations(
649        &mut self,
650        mut fragment: AstFragment,
651        extra_placeholders: &[NodeId],
652    ) -> (AstFragment, Vec<(Invocation, Option<Arc<SyntaxExtension>>)>) {
653        // Resolve `$crate`s in the fragment for pretty-printing.
654        self.cx.resolver.resolve_dollar_crates();
655
656        let mut invocations = {
657            let mut collector = InvocationCollector {
658                // Non-derive macro invocations cannot see the results of cfg expansion - they
659                // will either be removed along with the item, or invoked before the cfg/cfg_attr
660                // attribute is expanded. Therefore, we don't need to configure the tokens
661                // Derive macros *can* see the results of cfg-expansion - they are handled
662                // specially in `fully_expand_fragment`
663                cx: self.cx,
664                invocations: Vec::new(),
665                monotonic: self.monotonic,
666            };
667            fragment.mut_visit_with(&mut collector);
668            fragment.add_placeholders(extra_placeholders);
669            collector.invocations
670        };
671
672        if self.monotonic {
673            self.cx
674                .resolver
675                .visit_ast_fragment_with_placeholders(self.cx.current_expansion.id, &fragment);
676
677            if self.cx.sess.opts.incremental.is_some() {
678                for (invoc, _) in invocations.iter_mut() {
679                    let expn_id = invoc.expansion_data.id;
680                    let parent_def = self.cx.resolver.invocation_parent(expn_id);
681                    let span = invoc.span_mut();
682                    *span = span.with_parent(Some(parent_def));
683                }
684            }
685        }
686
687        (fragment, invocations)
688    }
689
690    fn error_recursion_limit_reached(&mut self) -> ErrorGuaranteed {
691        let expn_data = self.cx.current_expansion.id.expn_data();
692        let suggested_limit = match self.cx.ecfg.recursion_limit {
693            Limit(0) => Limit(2),
694            limit => limit * 2,
695        };
696
697        let guar = self.cx.dcx().emit_err(RecursionLimitReached {
698            span: expn_data.call_site,
699            descr: expn_data.kind.descr(),
700            suggested_limit,
701            crate_name: self.cx.ecfg.crate_name,
702        });
703
704        self.cx.macro_error_and_trace_macros_diag();
705        guar
706    }
707
708    /// A macro's expansion does not fit in this fragment kind.
709    /// For example, a non-type macro in a type position.
710    fn error_wrong_fragment_kind(
711        &mut self,
712        kind: AstFragmentKind,
713        mac: &ast::MacCall,
714        span: Span,
715    ) -> ErrorGuaranteed {
716        let guar =
717            self.cx.dcx().emit_err(WrongFragmentKind { span, kind: kind.name(), name: &mac.path });
718        self.cx.macro_error_and_trace_macros_diag();
719        guar
720    }
721
722    fn expand_invoc(
723        &mut self,
724        invoc: Invocation,
725        ext: &SyntaxExtensionKind,
726    ) -> ExpandResult<AstFragment, Invocation> {
727        let recursion_limit = match self.cx.reduced_recursion_limit {
728            Some((limit, _)) => limit,
729            None => self.cx.ecfg.recursion_limit,
730        };
731
732        if !recursion_limit.value_within_limit(self.cx.current_expansion.depth) {
733            let guar = match self.cx.reduced_recursion_limit {
734                Some((_, guar)) => guar,
735                None => self.error_recursion_limit_reached(),
736            };
737
738            // Reduce the recursion limit by half each time it triggers.
739            self.cx.reduced_recursion_limit = Some((recursion_limit / 2, guar));
740
741            return ExpandResult::Ready(invoc.fragment_kind.dummy(invoc.span(), guar));
742        }
743
744        let macro_stats = self.cx.sess.opts.unstable_opts.macro_stats;
745
746        let (fragment_kind, span) = (invoc.fragment_kind, invoc.span());
747        ExpandResult::Ready(match invoc.kind {
748            InvocationKind::Bang { mac, span } => {
749                if let SyntaxExtensionKind::Bang(expander) = ext {
750                    match expander.expand(self.cx, span, mac.args.tokens.clone()) {
751                        Ok(tok_result) => {
752                            let fragment =
753                                self.parse_ast_fragment(tok_result, fragment_kind, &mac.path, span);
754                            if macro_stats {
755                                update_bang_macro_stats(
756                                    self.cx,
757                                    fragment_kind,
758                                    span,
759                                    mac,
760                                    &fragment,
761                                );
762                            }
763                            fragment
764                        }
765                        Err(guar) => return ExpandResult::Ready(fragment_kind.dummy(span, guar)),
766                    }
767                } else if let Some(expander) = ext.as_legacy_bang() {
768                    let tok_result = match expander.expand(self.cx, span, mac.args.tokens.clone()) {
769                        ExpandResult::Ready(tok_result) => tok_result,
770                        ExpandResult::Retry(_) => {
771                            // retry the original
772                            return ExpandResult::Retry(Invocation {
773                                kind: InvocationKind::Bang { mac, span },
774                                ..invoc
775                            });
776                        }
777                    };
778                    if let Some(fragment) = fragment_kind.make_from(tok_result) {
779                        if macro_stats {
780                            update_bang_macro_stats(self.cx, fragment_kind, span, mac, &fragment);
781                        }
782                        fragment
783                    } else {
784                        let guar = self.error_wrong_fragment_kind(fragment_kind, &mac, span);
785                        fragment_kind.dummy(span, guar)
786                    }
787                } else {
788                    ::core::panicking::panic("internal error: entered unreachable code");unreachable!();
789                }
790            }
791            InvocationKind::Attr { attr, pos, mut item, derives } => {
792                if let Some(expander) = ext.as_attr() {
793                    self.gate_proc_macro_input(&item);
794                    self.gate_proc_macro_attr_item(span, &item);
795                    let tokens = match &item {
796                        // FIXME: Collect tokens and use them instead of generating
797                        // fake ones. These are unstable, so it needs to be
798                        // fixed prior to stabilization
799                        // Fake tokens when we are invoking an inner attribute, and
800                        // we are invoking it on an out-of-line module or crate.
801                        Annotatable::Crate(krate) => {
802                            rustc_parse::fake_token_stream_for_crate(&self.cx.sess.psess, krate)
803                        }
804                        Annotatable::Item(item_inner)
805                            if #[allow(non_exhaustive_omitted_patterns)] match attr.style {
    AttrStyle::Inner => true,
    _ => false,
}matches!(attr.style, AttrStyle::Inner)
806                                && #[allow(non_exhaustive_omitted_patterns)] match item_inner.kind {
    ItemKind::Mod(_, _,
        ModKind::Unloaded | ModKind::Loaded(_, Inline::No { .. }, _)) => true,
    _ => false,
}matches!(
807                                    item_inner.kind,
808                                    ItemKind::Mod(
809                                        _,
810                                        _,
811                                        ModKind::Unloaded
812                                            | ModKind::Loaded(_, Inline::No { .. }, _),
813                                    )
814                                ) =>
815                        {
816                            rustc_parse::fake_token_stream_for_item(&self.cx.sess.psess, item_inner)
817                        }
818                        Annotatable::Item(item_inner) if item_inner.tokens.is_none() => {
819                            rustc_parse::fake_token_stream_for_item(&self.cx.sess.psess, item_inner)
820                        }
821                        // When a function has EII implementations attached (via `eii_impls`),
822                        // use fake tokens so the pretty-printer re-emits the EII attribute
823                        // (e.g. `#[hello]`) in the token stream. Without this, the EII
824                        // attribute is lost during the token roundtrip performed by
825                        // `AttrProcMacro` expanders like `contracts::requires/ensures`,
826                        // breaking the EII link on the resulting re-parsed item.
827                        Annotatable::Item(item_inner)
828                            if #[allow(non_exhaustive_omitted_patterns)] match &item_inner.kind {
    ItemKind::Fn(f) if !f.eii_impls.is_empty() => true,
    _ => false,
}matches!(&item_inner.kind,
829                                ItemKind::Fn(f) if !f.eii_impls.is_empty()) =>
830                        {
831                            rustc_parse::fake_token_stream_for_item(&self.cx.sess.psess, item_inner)
832                        }
833                        Annotatable::ForeignItem(item_inner) if item_inner.tokens.is_none() => {
834                            rustc_parse::fake_token_stream_for_foreign_item(
835                                &self.cx.sess.psess,
836                                item_inner,
837                            )
838                        }
839                        _ => item.to_tokens(),
840                    };
841                    let attr_item = attr.get_normal_item();
842                    let safety = attr_item.unsafety;
843                    if let AttrArgs::Eq { .. } = attr_item.args.unparsed_ref().unwrap() {
844                        self.cx.dcx().emit_err(UnsupportedKeyValue { span });
845                    }
846                    let inner_tokens = attr_item.args.unparsed_ref().unwrap().inner_tokens();
847                    match expander.expand_with_safety(self.cx, safety, span, inner_tokens, tokens) {
848                        Ok(tok_result) => {
849                            let fragment = self.parse_ast_fragment(
850                                tok_result,
851                                fragment_kind,
852                                &attr_item.path,
853                                span,
854                            );
855                            if macro_stats {
856                                update_attr_macro_stats(
857                                    self.cx,
858                                    fragment_kind,
859                                    span,
860                                    &attr_item.path,
861                                    &attr,
862                                    item,
863                                    &fragment,
864                                );
865                            }
866                            fragment
867                        }
868                        Err(guar) => return ExpandResult::Ready(fragment_kind.dummy(span, guar)),
869                    }
870                } else if let SyntaxExtensionKind::LegacyAttr(expander) = ext {
871                    // `LegacyAttr` is only used for builtin attribute macros, which have their
872                    // safety checked by `check_builtin_meta_item`, so we don't need to check
873                    // `unsafety` here.
874                    match validate_attr::parse_meta(&self.cx.sess.psess, &attr) {
875                        Ok(meta) => {
876                            let item_clone = macro_stats.then(|| item.clone());
877                            let items = match expander.expand(self.cx, span, &meta, item, false) {
878                                ExpandResult::Ready(items) => items,
879                                ExpandResult::Retry(item) => {
880                                    // Reassemble the original invocation for retrying.
881                                    return ExpandResult::Retry(Invocation {
882                                        kind: InvocationKind::Attr { attr, pos, item, derives },
883                                        ..invoc
884                                    });
885                                }
886                            };
887                            if #[allow(non_exhaustive_omitted_patterns)] match fragment_kind {
    AstFragmentKind::Expr | AstFragmentKind::MethodReceiverExpr => true,
    _ => false,
}matches!(
888                                fragment_kind,
889                                AstFragmentKind::Expr | AstFragmentKind::MethodReceiverExpr
890                            ) && items.is_empty()
891                            {
892                                let guar = self.cx.dcx().emit_err(RemoveExprNotSupported { span });
893                                fragment_kind.dummy(span, guar)
894                            } else {
895                                let fragment = fragment_kind.expect_from_annotatables(items);
896                                if macro_stats {
897                                    update_attr_macro_stats(
898                                        self.cx,
899                                        fragment_kind,
900                                        span,
901                                        &meta.path,
902                                        &attr,
903                                        item_clone.unwrap(),
904                                        &fragment,
905                                    );
906                                }
907                                fragment
908                            }
909                        }
910                        Err(err) => {
911                            let _guar = err.emit();
912                            fragment_kind.expect_from_annotatables(iter::once(item))
913                        }
914                    }
915                } else if let SyntaxExtensionKind::NonMacroAttr = ext {
916                    // `-Zmacro-stats` ignores these because they don't do any real expansion.
917                    self.cx.expanded_inert_attrs.mark(&attr);
918                    item.visit_attrs(|attrs| attrs.insert(pos, attr));
919                    fragment_kind.expect_from_annotatables(iter::once(item))
920                } else {
921                    ::core::panicking::panic("internal error: entered unreachable code");unreachable!();
922                }
923            }
924            InvocationKind::Derive { path, item, is_const } => match ext {
925                SyntaxExtensionKind::Derive(expander)
926                | SyntaxExtensionKind::LegacyDerive(expander) => {
927                    if let SyntaxExtensionKind::Derive(..) = ext {
928                        self.gate_proc_macro_input(&item);
929                    }
930                    // The `MetaItem` representing the trait to derive can't
931                    // have an unsafe around it (as of now).
932                    let meta = ast::MetaItem {
933                        unsafety: ast::Safety::Default,
934                        kind: MetaItemKind::Word,
935                        span,
936                        path,
937                    };
938                    let items = match expander.expand(self.cx, span, &meta, item, is_const) {
939                        ExpandResult::Ready(items) => items,
940                        ExpandResult::Retry(item) => {
941                            // Reassemble the original invocation for retrying.
942                            return ExpandResult::Retry(Invocation {
943                                kind: InvocationKind::Derive { path: meta.path, item, is_const },
944                                ..invoc
945                            });
946                        }
947                    };
948                    let fragment = fragment_kind.expect_from_annotatables(items);
949                    if macro_stats {
950                        update_derive_macro_stats(
951                            self.cx,
952                            fragment_kind,
953                            span,
954                            &meta.path,
955                            &fragment,
956                        );
957                    }
958                    fragment
959                }
960                SyntaxExtensionKind::MacroRules(expander)
961                    if expander.kinds().contains(MacroKinds::DERIVE) =>
962                {
963                    if is_const {
964                        let guar = self
965                            .cx
966                            .dcx()
967                            .span_err(span, "macro `derive` does not support const derives");
968                        return ExpandResult::Ready(fragment_kind.dummy(span, guar));
969                    }
970                    let body = item.to_tokens();
971                    match expander.expand_derive(self.cx, span, &body) {
972                        Ok(tok_result) => {
973                            let fragment =
974                                self.parse_ast_fragment(tok_result, fragment_kind, &path, span);
975                            if macro_stats {
976                                update_derive_macro_stats(
977                                    self.cx,
978                                    fragment_kind,
979                                    span,
980                                    &path,
981                                    &fragment,
982                                );
983                            }
984                            fragment
985                        }
986                        Err(guar) => return ExpandResult::Ready(fragment_kind.dummy(span, guar)),
987                    }
988                }
989                _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
990            },
991            InvocationKind::GlobDelegation { item, of_trait } => {
992                let AssocItemKind::DelegationMac(deleg) = &item.kind else { ::core::panicking::panic("internal error: entered unreachable code")unreachable!() };
993                let suffixes = match ext {
994                    SyntaxExtensionKind::GlobDelegation(expander) => match expander.expand(self.cx)
995                    {
996                        ExpandResult::Ready(suffixes) => suffixes,
997                        ExpandResult::Retry(()) => {
998                            // Reassemble the original invocation for retrying.
999                            return ExpandResult::Retry(Invocation {
1000                                kind: InvocationKind::GlobDelegation { item, of_trait },
1001                                ..invoc
1002                            });
1003                        }
1004                    },
1005                    SyntaxExtensionKind::Bang(..) => {
1006                        let msg = "expanded a dummy glob delegation";
1007                        let guar = self.cx.dcx().span_delayed_bug(span, msg);
1008                        return ExpandResult::Ready(fragment_kind.dummy(span, guar));
1009                    }
1010                    _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1011                };
1012
1013                type Node = AstNodeWrapper<Box<ast::AssocItem>, ImplItemTag>;
1014                let single_delegations = build_single_delegations::<Node>(
1015                    self.cx, deleg, &item, &suffixes, item.span, true,
1016                );
1017                // `-Zmacro-stats` ignores these because they don't seem important.
1018                fragment_kind.expect_from_annotatables(single_delegations.map(|item| {
1019                    Annotatable::AssocItem(Box::new(item), AssocCtxt::Impl { of_trait })
1020                }))
1021            }
1022        })
1023    }
1024
1025    fn gate_proc_macro_attr_item(&self, span: Span, item: &Annotatable) {
1026        let kind = match item {
1027            Annotatable::Item(_)
1028            | Annotatable::AssocItem(..)
1029            | Annotatable::ForeignItem(_)
1030            | Annotatable::Crate(..) => return,
1031            Annotatable::Stmt(stmt) => {
1032                // Attributes are stable on item statements,
1033                // but unstable on all other kinds of statements
1034                if stmt.is_item() {
1035                    return;
1036                }
1037                "statements"
1038            }
1039            Annotatable::Expr(_) => "expressions",
1040            Annotatable::Arm(..)
1041            | Annotatable::ExprField(..)
1042            | Annotatable::PatField(..)
1043            | Annotatable::GenericParam(..)
1044            | Annotatable::Param(..)
1045            | Annotatable::FieldDef(..)
1046            | Annotatable::Variant(..)
1047            | Annotatable::WherePredicate(..) => { ::core::panicking::panic_fmt(format_args!("unexpected annotatable")); }panic!("unexpected annotatable"),
1048        };
1049        if self.cx.ecfg.features.proc_macro_hygiene() {
1050            return;
1051        }
1052        feature_err(
1053            &self.cx.sess,
1054            sym::proc_macro_hygiene,
1055            span,
1056            ::alloc::__export::must_use({
        ::alloc::fmt::format(format_args!("custom attributes cannot be applied to {0}",
                kind))
    })format!("custom attributes cannot be applied to {kind}"),
1057        )
1058        .emit();
1059    }
1060
1061    fn gate_proc_macro_input(&self, annotatable: &Annotatable) {
1062        struct GateProcMacroInput<'a> {
1063            sess: &'a Session,
1064        }
1065
1066        impl<'ast, 'a> Visitor<'ast> for GateProcMacroInput<'a> {
1067            fn visit_item(&mut self, item: &'ast ast::Item) {
1068                match &item.kind {
1069                    ItemKind::Mod(_, _, mod_kind)
1070                        if !#[allow(non_exhaustive_omitted_patterns)] match mod_kind {
    ModKind::Loaded(_, Inline::Yes, _) => true,
    _ => false,
}matches!(mod_kind, ModKind::Loaded(_, Inline::Yes, _)) =>
1071                    {
1072                        feature_err(
1073                            self.sess,
1074                            sym::proc_macro_hygiene,
1075                            item.span,
1076                            rustc_errors::DiagMessage::Inline(std::borrow::Cow::Borrowed("file modules in proc macro input are unstable"))msg!("file modules in proc macro input are unstable"),
1077                        )
1078                        .emit();
1079                    }
1080                    _ => {}
1081                }
1082
1083                visit::walk_item(self, item);
1084            }
1085        }
1086
1087        if !self.cx.ecfg.features.proc_macro_hygiene() {
1088            annotatable.visit_with(&mut GateProcMacroInput { sess: &self.cx.sess });
1089        }
1090    }
1091
1092    fn parse_ast_fragment(
1093        &mut self,
1094        toks: TokenStream,
1095        kind: AstFragmentKind,
1096        path: &ast::Path,
1097        span: Span,
1098    ) -> AstFragment {
1099        let mut parser = self.cx.new_parser_from_tts(toks);
1100        match parse_ast_fragment(&mut parser, kind) {
1101            Ok(fragment) => {
1102                ensure_complete_parse(&parser, path, kind.name(), span);
1103                fragment
1104            }
1105            Err(mut err) => {
1106                if err.span.is_dummy() {
1107                    err.span(span);
1108                }
1109                annotate_err_with_kind(&mut err, kind, span);
1110                let guar = err.emit();
1111                self.cx.macro_error_and_trace_macros_diag();
1112                kind.dummy(span, guar)
1113            }
1114        }
1115    }
1116}
1117
1118pub fn parse_ast_fragment<'a>(
1119    this: &mut Parser<'a>,
1120    kind: AstFragmentKind,
1121) -> PResult<'a, AstFragment> {
1122    Ok(match kind {
1123        AstFragmentKind::Items => {
1124            let mut items = SmallVec::new();
1125            while let Some(item) = this.parse_item(ForceCollect::No, AllowConstBlockItems::Yes)? {
1126                items.push(item);
1127            }
1128            AstFragment::Items(items)
1129        }
1130        AstFragmentKind::TraitItems => {
1131            let mut items = SmallVec::new();
1132            while let Some(item) = this.parse_trait_item(ForceCollect::No)? {
1133                items.extend(item);
1134            }
1135            AstFragment::TraitItems(items)
1136        }
1137        AstFragmentKind::ImplItems => {
1138            let mut items = SmallVec::new();
1139            while let Some(item) = this.parse_impl_item(ForceCollect::No)? {
1140                items.extend(item);
1141            }
1142            AstFragment::ImplItems(items)
1143        }
1144        AstFragmentKind::TraitImplItems => {
1145            let mut items = SmallVec::new();
1146            while let Some(item) = this.parse_impl_item(ForceCollect::No)? {
1147                items.extend(item);
1148            }
1149            AstFragment::TraitImplItems(items)
1150        }
1151        AstFragmentKind::ForeignItems => {
1152            let mut items = SmallVec::new();
1153            while let Some(item) = this.parse_foreign_item(ForceCollect::No)? {
1154                items.extend(item);
1155            }
1156            AstFragment::ForeignItems(items)
1157        }
1158        AstFragmentKind::Stmts => {
1159            let mut stmts = SmallVec::new();
1160            // Won't make progress on a `}`.
1161            while this.token != token::Eof && this.token != token::CloseBrace {
1162                if let Some(stmt) = this.parse_full_stmt(AttemptLocalParseRecovery::Yes)? {
1163                    stmts.push(stmt);
1164                }
1165            }
1166            AstFragment::Stmts(stmts)
1167        }
1168        AstFragmentKind::Expr => AstFragment::Expr(this.parse_expr()?),
1169        AstFragmentKind::MethodReceiverExpr => AstFragment::MethodReceiverExpr(this.parse_expr()?),
1170        AstFragmentKind::OptExpr => {
1171            if this.token != token::Eof {
1172                AstFragment::OptExpr(Some(this.parse_expr()?))
1173            } else {
1174                AstFragment::OptExpr(None)
1175            }
1176        }
1177        AstFragmentKind::Ty => AstFragment::Ty(this.parse_ty()?),
1178        AstFragmentKind::Pat => AstFragment::Pat(Box::new(this.parse_pat_allow_top_guard(
1179            None,
1180            RecoverComma::No,
1181            RecoverColon::Yes,
1182            CommaRecoveryMode::LikelyTuple,
1183        )?)),
1184        AstFragmentKind::Crate => AstFragment::Crate(this.parse_crate_mod()?),
1185        AstFragmentKind::Arms
1186        | AstFragmentKind::ExprFields
1187        | AstFragmentKind::PatFields
1188        | AstFragmentKind::GenericParams
1189        | AstFragmentKind::Params
1190        | AstFragmentKind::FieldDefs
1191        | AstFragmentKind::Variants
1192        | AstFragmentKind::WherePredicates => {
    ::core::panicking::panic_fmt(format_args!("unexpected AST fragment kind"));
}panic!("unexpected AST fragment kind"),
1193    })
1194}
1195
1196pub(crate) fn ensure_complete_parse<'a>(
1197    parser: &Parser<'a>,
1198    macro_path: &ast::Path,
1199    kind_name: &str,
1200    span: Span,
1201) {
1202    if parser.token != token::Eof {
1203        let descr = token_descr(&parser.token);
1204        // Avoid emitting backtrace info twice.
1205        let def_site_span = parser.token.span.with_ctxt(SyntaxContext::root());
1206
1207        let semi_span = parser.psess.source_map().next_point(span);
1208        let add_semicolon = match &parser.psess.source_map().span_to_snippet(semi_span) {
1209            Ok(snippet) if &snippet[..] != ";" && kind_name == "expression" => {
1210                Some(span.shrink_to_hi())
1211            }
1212            _ => None,
1213        };
1214
1215        let expands_to_match_arm = kind_name == "pattern" && parser.token == token::FatArrow;
1216
1217        parser.dcx().emit_err(IncompleteParse {
1218            span: def_site_span,
1219            descr,
1220            label_span: span,
1221            macro_path,
1222            kind_name,
1223            expands_to_match_arm,
1224            add_semicolon,
1225        });
1226    }
1227}
1228
1229/// Wraps a call to `walk_*` / `walk_flat_map_*`
1230/// for an AST node that supports attributes
1231/// (see the `Annotatable` enum)
1232/// This method assigns a `NodeId`, and sets that `NodeId`
1233/// as our current 'lint node id'. If a macro call is found
1234/// inside this AST node, we will use this AST node's `NodeId`
1235/// to emit lints associated with that macro (allowing
1236/// `#[allow]` / `#[deny]` to be applied close to
1237/// the macro invocation).
1238///
1239/// Do *not* call this for a macro AST node
1240/// (e.g. `ExprKind::MacCall`) - we cannot emit lints
1241/// at these AST nodes, since they are removed and
1242/// replaced with the result of macro expansion.
1243///
1244/// All other `NodeId`s are assigned by `visit_id`.
1245/// * `self` is the 'self' parameter for the current method,
1246/// * `id` is a mutable reference to the `NodeId` field
1247///    of the current AST node.
1248/// * `closure` is a closure that executes the
1249///   `walk_*` / `walk_flat_map_*` method
1250///   for the current AST node.
1251macro_rules! assign_id {
1252    ($self:ident, $id:expr, $closure:expr) => {{
1253        let old_id = $self.cx.current_expansion.lint_node_id;
1254        if $self.monotonic {
1255            debug_assert_eq!(*$id, ast::DUMMY_NODE_ID);
1256            let new_id = $self.cx.resolver.next_node_id();
1257            *$id = new_id;
1258            $self.cx.current_expansion.lint_node_id = new_id;
1259        }
1260        let ret = ($closure)();
1261        $self.cx.current_expansion.lint_node_id = old_id;
1262        ret
1263    }};
1264}
1265
1266enum AddSemicolon {
1267    Yes,
1268    No,
1269}
1270
1271/// A trait implemented for all `AstFragment` nodes and providing all pieces
1272/// of functionality used by `InvocationCollector`.
1273trait InvocationCollectorNode: HasAttrs + HasNodeId + Sized {
1274    type OutputTy = SmallVec<[Self; 1]>;
1275    type ItemKind = ItemKind;
1276    const KIND: AstFragmentKind;
1277    fn to_annotatable(self) -> Annotatable;
1278    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy;
1279    fn descr() -> &'static str {
1280        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1281    }
1282    fn walk_flat_map(self, _collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1283        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1284    }
1285    fn walk(&mut self, _collector: &mut InvocationCollector<'_, '_>) {
1286        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1287    }
1288    fn is_mac_call(&self) -> bool {
1289        false
1290    }
1291    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1292        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1293    }
1294    fn delegation(&self) -> Option<(&ast::DelegationMac, &ast::Item<Self::ItemKind>)> {
1295        None
1296    }
1297    fn delegation_item_kind(_deleg: Box<ast::Delegation>) -> Self::ItemKind {
1298        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1299    }
1300    fn from_item(_item: ast::Item<Self::ItemKind>) -> Self {
1301        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1302    }
1303    fn flatten_outputs(_outputs: impl Iterator<Item = Self::OutputTy>) -> Self::OutputTy {
1304        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1305    }
1306    fn pre_flat_map_node_collect_attr(_cfg: &StripUnconfigured<'_>, _attr: &ast::Attribute) {}
1307    fn post_flat_map_node_collect_bang(_output: &mut Self::OutputTy, _add_semicolon: AddSemicolon) {
1308    }
1309    fn wrap_flat_map_node_walk_flat_map(
1310        node: Self,
1311        collector: &mut InvocationCollector<'_, '_>,
1312        walk_flat_map: impl FnOnce(Self, &mut InvocationCollector<'_, '_>) -> Self::OutputTy,
1313    ) -> Result<Self::OutputTy, Self> {
1314        Ok(walk_flat_map(node, collector))
1315    }
1316    fn expand_cfg_false(
1317        &mut self,
1318        collector: &mut InvocationCollector<'_, '_>,
1319        _pos: usize,
1320        span: Span,
1321    ) {
1322        collector.cx.dcx().emit_err(RemoveNodeNotSupported { span, descr: Self::descr() });
1323    }
1324
1325    /// All of the identifiers (items) declared by this node.
1326    /// This is an approximation and should only be used for diagnostics.
1327    fn declared_idents(&self) -> Vec<Ident> {
1328        ::alloc::vec::Vec::new()vec![]
1329    }
1330
1331    fn as_target(&self) -> Target;
1332}
1333
1334impl InvocationCollectorNode for Box<ast::Item> {
1335    const KIND: AstFragmentKind = AstFragmentKind::Items;
1336    fn to_annotatable(self) -> Annotatable {
1337        Annotatable::Item(self)
1338    }
1339    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1340        fragment.make_items()
1341    }
1342    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1343        walk_flat_map_item(collector, self)
1344    }
1345    fn is_mac_call(&self) -> bool {
1346        #[allow(non_exhaustive_omitted_patterns)] match self.kind {
    ItemKind::MacCall(..) => true,
    _ => false,
}matches!(self.kind, ItemKind::MacCall(..))
1347    }
1348    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1349        match self.kind {
1350            ItemKind::MacCall(mac) => (mac, self.attrs, AddSemicolon::No),
1351            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1352        }
1353    }
1354    fn delegation(&self) -> Option<(&ast::DelegationMac, &ast::Item<Self::ItemKind>)> {
1355        match &self.kind {
1356            ItemKind::DelegationMac(deleg) => Some((deleg, self)),
1357            _ => None,
1358        }
1359    }
1360    fn delegation_item_kind(deleg: Box<ast::Delegation>) -> Self::ItemKind {
1361        ItemKind::Delegation(deleg)
1362    }
1363    fn from_item(item: ast::Item<Self::ItemKind>) -> Self {
1364        Box::new(item)
1365    }
1366    fn flatten_outputs(items: impl Iterator<Item = Self::OutputTy>) -> Self::OutputTy {
1367        items.flatten().collect()
1368    }
1369    fn wrap_flat_map_node_walk_flat_map(
1370        mut node: Self,
1371        collector: &mut InvocationCollector<'_, '_>,
1372        walk_flat_map: impl FnOnce(Self, &mut InvocationCollector<'_, '_>) -> Self::OutputTy,
1373    ) -> Result<Self::OutputTy, Self> {
1374        if !#[allow(non_exhaustive_omitted_patterns)] match node.kind {
    ItemKind::Mod(..) => true,
    _ => false,
}matches!(node.kind, ItemKind::Mod(..)) {
1375            return Ok(walk_flat_map(node, collector));
1376        }
1377
1378        // Work around borrow checker not seeing through `P`'s deref.
1379        let (span, mut attrs) = (node.span, mem::take(&mut node.attrs));
1380        let ItemKind::Mod(_, ident, ref mut mod_kind) = node.kind else { ::core::panicking::panic("internal error: entered unreachable code")unreachable!() };
1381        let ecx = &mut collector.cx;
1382        let (file_path, dir_path, dir_ownership) = match mod_kind {
1383            ModKind::Loaded(_, inline, _) => {
1384                // Inline `mod foo { ... }`, but we still need to push directories.
1385                let (dir_path, dir_ownership) = mod_dir_path(
1386                    ecx.sess,
1387                    ident,
1388                    &attrs,
1389                    &ecx.current_expansion.module,
1390                    ecx.current_expansion.dir_ownership,
1391                    *inline,
1392                );
1393                // If the module was parsed from an external file, recover its path.
1394                // This lets `parse_external_mod` catch cycles if it's self-referential.
1395                let file_path = match inline {
1396                    Inline::Yes => None,
1397                    Inline::No { .. } => mod_file_path_from_attr(ecx.sess, &attrs, &dir_path),
1398                };
1399                node.attrs = attrs;
1400                (file_path, dir_path, dir_ownership)
1401            }
1402            ModKind::Unloaded => {
1403                // We have an outline `mod foo;` so we need to parse the file.
1404                let old_attrs_len = attrs.len();
1405                let ParsedExternalMod {
1406                    items,
1407                    spans,
1408                    file_path,
1409                    dir_path,
1410                    dir_ownership,
1411                    had_parse_error,
1412                } = parse_external_mod(
1413                    ecx.sess,
1414                    ident,
1415                    span,
1416                    &ecx.current_expansion.module,
1417                    ecx.current_expansion.dir_ownership,
1418                    &mut attrs,
1419                );
1420
1421                if let Some(lint_store) = ecx.lint_store {
1422                    lint_store.pre_expansion_lint(
1423                        ecx.sess,
1424                        ecx.ecfg.features,
1425                        ecx.resolver.registered_tools(),
1426                        ecx.current_expansion.lint_node_id,
1427                        &attrs,
1428                        &items,
1429                        ident.name,
1430                    );
1431                }
1432
1433                *mod_kind = ModKind::Loaded(items, Inline::No { had_parse_error }, spans);
1434                node.attrs = attrs;
1435                if node.attrs.len() > old_attrs_len {
1436                    // If we loaded an out-of-line module and added some inner attributes,
1437                    // then we need to re-configure it and re-collect attributes for
1438                    // resolution and expansion.
1439                    return Err(node);
1440                }
1441                (Some(file_path), dir_path, dir_ownership)
1442            }
1443        };
1444
1445        // Set the module info before we flat map.
1446        let mut module = ecx.current_expansion.module.with_dir_path(dir_path);
1447        module.mod_path.push(ident);
1448        if let Some(file_path) = file_path {
1449            module.file_path_stack.push(file_path);
1450        }
1451
1452        let orig_module = mem::replace(&mut ecx.current_expansion.module, Rc::new(module));
1453        let orig_dir_ownership =
1454            mem::replace(&mut ecx.current_expansion.dir_ownership, dir_ownership);
1455
1456        let res = Ok(walk_flat_map(node, collector));
1457
1458        collector.cx.current_expansion.dir_ownership = orig_dir_ownership;
1459        collector.cx.current_expansion.module = orig_module;
1460        res
1461    }
1462
1463    fn declared_idents(&self) -> Vec<Ident> {
1464        if let ItemKind::Use(ut) = &self.kind {
1465            fn collect_use_tree_leaves(ut: &ast::UseTree, idents: &mut Vec<Ident>) {
1466                match &ut.kind {
1467                    ast::UseTreeKind::Glob(_) => {}
1468                    ast::UseTreeKind::Simple(_) => idents.push(ut.ident()),
1469                    ast::UseTreeKind::Nested { items, .. } => {
1470                        for (ut, _) in items {
1471                            collect_use_tree_leaves(ut, idents);
1472                        }
1473                    }
1474                }
1475            }
1476            let mut idents = Vec::new();
1477            collect_use_tree_leaves(&ut, &mut idents);
1478            idents
1479        } else {
1480            self.kind.ident().into_iter().collect()
1481        }
1482    }
1483
1484    fn as_target(&self) -> Target {
1485        Target::from_ast_item(&*self)
1486    }
1487}
1488
1489struct TraitItemTag;
1490impl InvocationCollectorNode for AstNodeWrapper<Box<ast::AssocItem>, TraitItemTag> {
1491    type OutputTy = SmallVec<[Box<ast::AssocItem>; 1]>;
1492    type ItemKind = AssocItemKind;
1493    const KIND: AstFragmentKind = AstFragmentKind::TraitItems;
1494    fn to_annotatable(self) -> Annotatable {
1495        Annotatable::AssocItem(self.wrapped, AssocCtxt::Trait)
1496    }
1497    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1498        fragment.make_trait_items()
1499    }
1500    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1501        walk_flat_map_assoc_item(collector, self.wrapped, AssocCtxt::Trait)
1502    }
1503    fn is_mac_call(&self) -> bool {
1504        #[allow(non_exhaustive_omitted_patterns)] match self.wrapped.kind {
    AssocItemKind::MacCall(..) => true,
    _ => false,
}matches!(self.wrapped.kind, AssocItemKind::MacCall(..))
1505    }
1506    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1507        let item = self.wrapped;
1508        match item.kind {
1509            AssocItemKind::MacCall(mac) => (mac, item.attrs, AddSemicolon::No),
1510            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1511        }
1512    }
1513    fn delegation(&self) -> Option<(&ast::DelegationMac, &ast::Item<Self::ItemKind>)> {
1514        match &self.wrapped.kind {
1515            AssocItemKind::DelegationMac(deleg) => Some((deleg, &self.wrapped)),
1516            _ => None,
1517        }
1518    }
1519    fn delegation_item_kind(deleg: Box<ast::Delegation>) -> Self::ItemKind {
1520        AssocItemKind::Delegation(deleg)
1521    }
1522    fn from_item(item: ast::Item<Self::ItemKind>) -> Self {
1523        AstNodeWrapper::new(Box::new(item), TraitItemTag)
1524    }
1525    fn flatten_outputs(items: impl Iterator<Item = Self::OutputTy>) -> Self::OutputTy {
1526        items.flatten().collect()
1527    }
1528    fn as_target(&self) -> Target {
1529        Target::from_assoc_item_kind(&self.wrapped.kind, AssocCtxt::Trait)
1530    }
1531}
1532
1533struct ImplItemTag;
1534impl InvocationCollectorNode for AstNodeWrapper<Box<ast::AssocItem>, ImplItemTag> {
1535    type OutputTy = SmallVec<[Box<ast::AssocItem>; 1]>;
1536    type ItemKind = AssocItemKind;
1537    const KIND: AstFragmentKind = AstFragmentKind::ImplItems;
1538    fn to_annotatable(self) -> Annotatable {
1539        Annotatable::AssocItem(self.wrapped, AssocCtxt::Impl { of_trait: false })
1540    }
1541    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1542        fragment.make_impl_items()
1543    }
1544    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1545        walk_flat_map_assoc_item(collector, self.wrapped, AssocCtxt::Impl { of_trait: false })
1546    }
1547    fn is_mac_call(&self) -> bool {
1548        #[allow(non_exhaustive_omitted_patterns)] match self.wrapped.kind {
    AssocItemKind::MacCall(..) => true,
    _ => false,
}matches!(self.wrapped.kind, AssocItemKind::MacCall(..))
1549    }
1550    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1551        let item = self.wrapped;
1552        match item.kind {
1553            AssocItemKind::MacCall(mac) => (mac, item.attrs, AddSemicolon::No),
1554            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1555        }
1556    }
1557    fn delegation(&self) -> Option<(&ast::DelegationMac, &ast::Item<Self::ItemKind>)> {
1558        match &self.wrapped.kind {
1559            AssocItemKind::DelegationMac(deleg) => Some((deleg, &self.wrapped)),
1560            _ => None,
1561        }
1562    }
1563    fn delegation_item_kind(deleg: Box<ast::Delegation>) -> Self::ItemKind {
1564        AssocItemKind::Delegation(deleg)
1565    }
1566    fn from_item(item: ast::Item<Self::ItemKind>) -> Self {
1567        AstNodeWrapper::new(Box::new(item), ImplItemTag)
1568    }
1569    fn flatten_outputs(items: impl Iterator<Item = Self::OutputTy>) -> Self::OutputTy {
1570        items.flatten().collect()
1571    }
1572    fn as_target(&self) -> Target {
1573        Target::from_assoc_item_kind(&self.wrapped.kind, AssocCtxt::Impl { of_trait: false })
1574    }
1575}
1576
1577struct TraitImplItemTag;
1578impl InvocationCollectorNode for AstNodeWrapper<Box<ast::AssocItem>, TraitImplItemTag> {
1579    type OutputTy = SmallVec<[Box<ast::AssocItem>; 1]>;
1580    type ItemKind = AssocItemKind;
1581    const KIND: AstFragmentKind = AstFragmentKind::TraitImplItems;
1582    fn to_annotatable(self) -> Annotatable {
1583        Annotatable::AssocItem(self.wrapped, AssocCtxt::Impl { of_trait: true })
1584    }
1585    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1586        fragment.make_trait_impl_items()
1587    }
1588    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1589        walk_flat_map_assoc_item(collector, self.wrapped, AssocCtxt::Impl { of_trait: true })
1590    }
1591    fn is_mac_call(&self) -> bool {
1592        #[allow(non_exhaustive_omitted_patterns)] match self.wrapped.kind {
    AssocItemKind::MacCall(..) => true,
    _ => false,
}matches!(self.wrapped.kind, AssocItemKind::MacCall(..))
1593    }
1594    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1595        let item = self.wrapped;
1596        match item.kind {
1597            AssocItemKind::MacCall(mac) => (mac, item.attrs, AddSemicolon::No),
1598            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1599        }
1600    }
1601    fn delegation(&self) -> Option<(&ast::DelegationMac, &ast::Item<Self::ItemKind>)> {
1602        match &self.wrapped.kind {
1603            AssocItemKind::DelegationMac(deleg) => Some((deleg, &self.wrapped)),
1604            _ => None,
1605        }
1606    }
1607    fn delegation_item_kind(deleg: Box<ast::Delegation>) -> Self::ItemKind {
1608        AssocItemKind::Delegation(deleg)
1609    }
1610    fn from_item(item: ast::Item<Self::ItemKind>) -> Self {
1611        AstNodeWrapper::new(Box::new(item), TraitImplItemTag)
1612    }
1613    fn flatten_outputs(items: impl Iterator<Item = Self::OutputTy>) -> Self::OutputTy {
1614        items.flatten().collect()
1615    }
1616    fn as_target(&self) -> Target {
1617        Target::from_assoc_item_kind(&self.wrapped.kind, AssocCtxt::Impl { of_trait: true })
1618    }
1619}
1620
1621impl InvocationCollectorNode for Box<ast::ForeignItem> {
1622    const KIND: AstFragmentKind = AstFragmentKind::ForeignItems;
1623    fn to_annotatable(self) -> Annotatable {
1624        Annotatable::ForeignItem(self)
1625    }
1626    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1627        fragment.make_foreign_items()
1628    }
1629    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1630        walk_flat_map_foreign_item(collector, self)
1631    }
1632    fn is_mac_call(&self) -> bool {
1633        #[allow(non_exhaustive_omitted_patterns)] match self.kind {
    ForeignItemKind::MacCall(..) => true,
    _ => false,
}matches!(self.kind, ForeignItemKind::MacCall(..))
1634    }
1635    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1636        match self.kind {
1637            ForeignItemKind::MacCall(mac) => (mac, self.attrs, AddSemicolon::No),
1638            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1639        }
1640    }
1641    fn as_target(&self) -> Target {
1642        match &self.kind {
1643            ForeignItemKind::Static(_) => Target::ForeignStatic,
1644            ForeignItemKind::Fn(_) => Target::ForeignFn,
1645            ForeignItemKind::TyAlias(_) => Target::ForeignTy,
1646            ForeignItemKind::MacCall(_) => Target::MacroCall,
1647        }
1648    }
1649}
1650
1651impl InvocationCollectorNode for ast::Variant {
1652    const KIND: AstFragmentKind = AstFragmentKind::Variants;
1653    fn to_annotatable(self) -> Annotatable {
1654        Annotatable::Variant(self)
1655    }
1656    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1657        fragment.make_variants()
1658    }
1659    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1660        walk_flat_map_variant(collector, self)
1661    }
1662    fn as_target(&self) -> Target {
1663        Target::Variant
1664    }
1665}
1666
1667impl InvocationCollectorNode for ast::WherePredicate {
1668    const KIND: AstFragmentKind = AstFragmentKind::WherePredicates;
1669    fn to_annotatable(self) -> Annotatable {
1670        Annotatable::WherePredicate(self)
1671    }
1672    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1673        fragment.make_where_predicates()
1674    }
1675    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1676        walk_flat_map_where_predicate(collector, self)
1677    }
1678    fn as_target(&self) -> Target {
1679        Target::WherePredicate
1680    }
1681}
1682
1683impl InvocationCollectorNode for ast::FieldDef {
1684    const KIND: AstFragmentKind = AstFragmentKind::FieldDefs;
1685    fn to_annotatable(self) -> Annotatable {
1686        Annotatable::FieldDef(self)
1687    }
1688    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1689        fragment.make_field_defs()
1690    }
1691    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1692        walk_flat_map_field_def(collector, self)
1693    }
1694    fn as_target(&self) -> Target {
1695        Target::Field
1696    }
1697}
1698
1699impl InvocationCollectorNode for ast::PatField {
1700    const KIND: AstFragmentKind = AstFragmentKind::PatFields;
1701    fn to_annotatable(self) -> Annotatable {
1702        Annotatable::PatField(self)
1703    }
1704    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1705        fragment.make_pat_fields()
1706    }
1707    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1708        walk_flat_map_pat_field(collector, self)
1709    }
1710    fn as_target(&self) -> Target {
1711        Target::PatField
1712    }
1713}
1714
1715impl InvocationCollectorNode for ast::ExprField {
1716    const KIND: AstFragmentKind = AstFragmentKind::ExprFields;
1717    fn to_annotatable(self) -> Annotatable {
1718        Annotatable::ExprField(self)
1719    }
1720    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1721        fragment.make_expr_fields()
1722    }
1723    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1724        walk_flat_map_expr_field(collector, self)
1725    }
1726    fn as_target(&self) -> Target {
1727        Target::ExprField
1728    }
1729}
1730
1731impl InvocationCollectorNode for ast::Param {
1732    const KIND: AstFragmentKind = AstFragmentKind::Params;
1733    fn to_annotatable(self) -> Annotatable {
1734        Annotatable::Param(self)
1735    }
1736    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1737        fragment.make_params()
1738    }
1739    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1740        walk_flat_map_param(collector, self)
1741    }
1742    fn as_target(&self) -> Target {
1743        Target::Param
1744    }
1745}
1746
1747impl InvocationCollectorNode for ast::GenericParam {
1748    const KIND: AstFragmentKind = AstFragmentKind::GenericParams;
1749    fn to_annotatable(self) -> Annotatable {
1750        Annotatable::GenericParam(self)
1751    }
1752    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1753        fragment.make_generic_params()
1754    }
1755    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1756        walk_flat_map_generic_param(collector, self)
1757    }
1758    fn as_target(&self) -> Target {
1759        let mut has_default = false;
1760        Target::GenericParam {
1761            kind: match &self.kind {
1762                rustc_ast::GenericParamKind::Lifetime => {
1763                    rustc_hir::target::GenericParamKind::Lifetime
1764                }
1765                rustc_ast::GenericParamKind::Type { default } => {
1766                    has_default = default.is_some();
1767                    rustc_hir::target::GenericParamKind::Type
1768                }
1769                rustc_ast::GenericParamKind::Const { default, .. } => {
1770                    has_default = default.is_some();
1771                    rustc_hir::target::GenericParamKind::Const
1772                }
1773            },
1774            has_default,
1775        }
1776    }
1777}
1778
1779impl InvocationCollectorNode for ast::Arm {
1780    const KIND: AstFragmentKind = AstFragmentKind::Arms;
1781    fn to_annotatable(self) -> Annotatable {
1782        Annotatable::Arm(self)
1783    }
1784    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1785        fragment.make_arms()
1786    }
1787    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1788        walk_flat_map_arm(collector, self)
1789    }
1790    fn as_target(&self) -> Target {
1791        Target::Arm
1792    }
1793}
1794
1795impl InvocationCollectorNode for ast::Stmt {
1796    const KIND: AstFragmentKind = AstFragmentKind::Stmts;
1797    fn to_annotatable(self) -> Annotatable {
1798        Annotatable::Stmt(Box::new(self))
1799    }
1800    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1801        fragment.make_stmts()
1802    }
1803    fn walk_flat_map(self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
1804        walk_flat_map_stmt(collector, self)
1805    }
1806    fn is_mac_call(&self) -> bool {
1807        match &self.kind {
1808            StmtKind::MacCall(..) => true,
1809            StmtKind::Item(item) => #[allow(non_exhaustive_omitted_patterns)] match item.kind {
    ItemKind::MacCall(..) => true,
    _ => false,
}matches!(item.kind, ItemKind::MacCall(..)),
1810            StmtKind::Semi(expr) => #[allow(non_exhaustive_omitted_patterns)] match expr.kind {
    ExprKind::MacCall(..) => true,
    _ => false,
}matches!(expr.kind, ExprKind::MacCall(..)),
1811            StmtKind::Expr(..) => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1812            StmtKind::Let(..) | StmtKind::Empty => false,
1813        }
1814    }
1815    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1816        // We pull macro invocations (both attributes and fn-like macro calls) out of their
1817        // `StmtKind`s and treat them as statement macro invocations, not as items or expressions.
1818        let (add_semicolon, mac, attrs) = match self.kind {
1819            StmtKind::MacCall(mac) => {
1820                let ast::MacCallStmt { mac, style, attrs, .. } = *mac;
1821                (style == MacStmtStyle::Semicolon, mac, attrs)
1822            }
1823            StmtKind::Item(item) => match *item {
1824                ast::Item { kind: ItemKind::MacCall(mac), attrs, .. } => {
1825                    (mac.args.need_semicolon(), mac, attrs)
1826                }
1827                _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1828            },
1829            StmtKind::Semi(expr) => match *expr {
1830                ast::Expr { kind: ExprKind::MacCall(mac), attrs, .. } => {
1831                    (mac.args.need_semicolon(), mac, attrs)
1832                }
1833                _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1834            },
1835            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1836        };
1837        (mac, attrs, if add_semicolon { AddSemicolon::Yes } else { AddSemicolon::No })
1838    }
1839    fn delegation(&self) -> Option<(&ast::DelegationMac, &ast::Item<Self::ItemKind>)> {
1840        match &self.kind {
1841            StmtKind::Item(item) => match &item.kind {
1842                ItemKind::DelegationMac(deleg) => Some((deleg, item)),
1843                _ => None,
1844            },
1845            _ => None,
1846        }
1847    }
1848    fn delegation_item_kind(deleg: Box<ast::Delegation>) -> Self::ItemKind {
1849        ItemKind::Delegation(deleg)
1850    }
1851    fn from_item(item: ast::Item<Self::ItemKind>) -> Self {
1852        ast::Stmt { id: ast::DUMMY_NODE_ID, span: item.span, kind: StmtKind::Item(Box::new(item)) }
1853    }
1854    fn flatten_outputs(items: impl Iterator<Item = Self::OutputTy>) -> Self::OutputTy {
1855        items.flatten().collect()
1856    }
1857    fn post_flat_map_node_collect_bang(stmts: &mut Self::OutputTy, add_semicolon: AddSemicolon) {
1858        // If this is a macro invocation with a semicolon, then apply that
1859        // semicolon to the final statement produced by expansion.
1860        if #[allow(non_exhaustive_omitted_patterns)] match add_semicolon {
    AddSemicolon::Yes => true,
    _ => false,
}matches!(add_semicolon, AddSemicolon::Yes) {
1861            if let Some(stmt) = stmts.pop() {
1862                stmts.push(stmt.add_trailing_semicolon());
1863            }
1864        }
1865    }
1866    fn as_target(&self) -> Target {
1867        Target::Statement
1868    }
1869}
1870
1871impl InvocationCollectorNode for ast::Crate {
1872    type OutputTy = ast::Crate;
1873    const KIND: AstFragmentKind = AstFragmentKind::Crate;
1874    fn to_annotatable(self) -> Annotatable {
1875        Annotatable::Crate(self)
1876    }
1877    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1878        fragment.make_crate()
1879    }
1880    fn walk(&mut self, collector: &mut InvocationCollector<'_, '_>) {
1881        walk_crate(collector, self)
1882    }
1883    fn expand_cfg_false(
1884        &mut self,
1885        collector: &mut InvocationCollector<'_, '_>,
1886        pos: usize,
1887        _span: Span,
1888    ) {
1889        // Attributes above `cfg(FALSE)` are left in place, because we may want to configure
1890        // some global crate properties even on fully unconfigured crates.
1891        self.attrs.truncate(pos);
1892        // Standard prelude imports are left in the crate for backward compatibility.
1893        self.items.truncate(collector.cx.num_standard_library_imports);
1894    }
1895    fn as_target(&self) -> Target {
1896        Target::Crate
1897    }
1898}
1899
1900impl InvocationCollectorNode for ast::Ty {
1901    type OutputTy = Box<ast::Ty>;
1902    const KIND: AstFragmentKind = AstFragmentKind::Ty;
1903    fn to_annotatable(self) -> Annotatable {
1904        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1905    }
1906    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1907        fragment.make_ty()
1908    }
1909    fn walk(&mut self, collector: &mut InvocationCollector<'_, '_>) {
1910        // Save the pre-expanded name of this `ImplTrait`, so that later when defining
1911        // an APIT we use a name that doesn't have any placeholder fragments in it.
1912        if let ast::TyKind::ImplTrait(..) = self.kind {
1913            // HACK: pprust breaks strings with newlines when the type
1914            // gets too long. We don't want these to show up in compiler
1915            // output or built artifacts, so replace them here...
1916            // Perhaps we should instead format APITs more robustly.
1917            let name = Symbol::intern(&pprust::ty_to_string(self).replace('\n', " "));
1918            collector.cx.resolver.insert_impl_trait_name(self.id, name);
1919        }
1920        walk_ty(collector, self)
1921    }
1922    fn is_mac_call(&self) -> bool {
1923        #[allow(non_exhaustive_omitted_patterns)] match self.kind {
    ast::TyKind::MacCall(..) => true,
    _ => false,
}matches!(self.kind, ast::TyKind::MacCall(..))
1924    }
1925    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1926        match self.kind {
1927            TyKind::MacCall(mac) => (mac, AttrVec::new(), AddSemicolon::No),
1928            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1929        }
1930    }
1931    fn as_target(&self) -> Target {
1932        // This is only used for attribute parsing, which are not allowed on types.
1933        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1934    }
1935}
1936
1937impl InvocationCollectorNode for ast::Pat {
1938    type OutputTy = Box<ast::Pat>;
1939    const KIND: AstFragmentKind = AstFragmentKind::Pat;
1940    fn to_annotatable(self) -> Annotatable {
1941        ::core::panicking::panic("internal error: entered unreachable code")unreachable!()
1942    }
1943    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1944        fragment.make_pat()
1945    }
1946    fn walk(&mut self, collector: &mut InvocationCollector<'_, '_>) {
1947        walk_pat(collector, self)
1948    }
1949    fn is_mac_call(&self) -> bool {
1950        #[allow(non_exhaustive_omitted_patterns)] match self.kind {
    PatKind::MacCall(..) => true,
    _ => false,
}matches!(self.kind, PatKind::MacCall(..))
1951    }
1952    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1953        match self.kind {
1954            PatKind::MacCall(mac) => (mac, AttrVec::new(), AddSemicolon::No),
1955            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1956        }
1957    }
1958    fn as_target(&self) -> Target {
1959        ::core::panicking::panic("not yet implemented");todo!();
1960    }
1961}
1962
1963impl InvocationCollectorNode for ast::Expr {
1964    type OutputTy = Box<ast::Expr>;
1965    const KIND: AstFragmentKind = AstFragmentKind::Expr;
1966    fn to_annotatable(self) -> Annotatable {
1967        Annotatable::Expr(Box::new(self))
1968    }
1969    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
1970        fragment.make_expr()
1971    }
1972    fn descr() -> &'static str {
1973        "an expression"
1974    }
1975    fn walk(&mut self, collector: &mut InvocationCollector<'_, '_>) {
1976        walk_expr(collector, self)
1977    }
1978    fn is_mac_call(&self) -> bool {
1979        #[allow(non_exhaustive_omitted_patterns)] match self.kind {
    ExprKind::MacCall(..) => true,
    _ => false,
}matches!(self.kind, ExprKind::MacCall(..))
1980    }
1981    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
1982        match self.kind {
1983            ExprKind::MacCall(mac) => (mac, self.attrs, AddSemicolon::No),
1984            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
1985        }
1986    }
1987    fn as_target(&self) -> Target {
1988        Target::Expression
1989    }
1990}
1991
1992struct OptExprTag;
1993impl InvocationCollectorNode for AstNodeWrapper<Box<ast::Expr>, OptExprTag> {
1994    type OutputTy = Option<Box<ast::Expr>>;
1995    const KIND: AstFragmentKind = AstFragmentKind::OptExpr;
1996    fn to_annotatable(self) -> Annotatable {
1997        Annotatable::Expr(self.wrapped)
1998    }
1999    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
2000        fragment.make_opt_expr()
2001    }
2002    fn walk_flat_map(mut self, collector: &mut InvocationCollector<'_, '_>) -> Self::OutputTy {
2003        walk_expr(collector, &mut self.wrapped);
2004        Some(self.wrapped)
2005    }
2006    fn is_mac_call(&self) -> bool {
2007        #[allow(non_exhaustive_omitted_patterns)] match self.wrapped.kind {
    ast::ExprKind::MacCall(..) => true,
    _ => false,
}matches!(self.wrapped.kind, ast::ExprKind::MacCall(..))
2008    }
2009    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
2010        let node = self.wrapped;
2011        match node.kind {
2012            ExprKind::MacCall(mac) => (mac, node.attrs, AddSemicolon::No),
2013            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2014        }
2015    }
2016    fn pre_flat_map_node_collect_attr(cfg: &StripUnconfigured<'_>, attr: &ast::Attribute) {
2017        cfg.maybe_emit_expr_attr_err(attr);
2018    }
2019    fn as_target(&self) -> Target {
2020        Target::Expression
2021    }
2022}
2023
2024/// This struct is a hack to workaround unstable of `stmt_expr_attributes`.
2025/// It can be removed once that feature is stabilized.
2026struct MethodReceiverTag;
2027
2028impl InvocationCollectorNode for AstNodeWrapper<ast::Expr, MethodReceiverTag> {
2029    type OutputTy = AstNodeWrapper<Box<ast::Expr>, MethodReceiverTag>;
2030    const KIND: AstFragmentKind = AstFragmentKind::MethodReceiverExpr;
2031    fn descr() -> &'static str {
2032        "an expression"
2033    }
2034    fn to_annotatable(self) -> Annotatable {
2035        Annotatable::Expr(Box::new(self.wrapped))
2036    }
2037    fn fragment_to_output(fragment: AstFragment) -> Self::OutputTy {
2038        AstNodeWrapper::new(fragment.make_method_receiver_expr(), MethodReceiverTag)
2039    }
2040    fn walk(&mut self, collector: &mut InvocationCollector<'_, '_>) {
2041        walk_expr(collector, &mut self.wrapped)
2042    }
2043    fn is_mac_call(&self) -> bool {
2044        #[allow(non_exhaustive_omitted_patterns)] match self.wrapped.kind {
    ast::ExprKind::MacCall(..) => true,
    _ => false,
}matches!(self.wrapped.kind, ast::ExprKind::MacCall(..))
2045    }
2046    fn take_mac_call(self) -> (Box<ast::MacCall>, ast::AttrVec, AddSemicolon) {
2047        let node = self.wrapped;
2048        match node.kind {
2049            ExprKind::MacCall(mac) => (mac, node.attrs, AddSemicolon::No),
2050            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2051        }
2052    }
2053    fn as_target(&self) -> Target {
2054        Target::Expression
2055    }
2056}
2057
2058fn build_single_delegations<'a, Node: InvocationCollectorNode>(
2059    ecx: &ExtCtxt<'_>,
2060    deleg: &'a ast::DelegationMac,
2061    item: &'a ast::Item<Node::ItemKind>,
2062    suffixes: &'a [(Ident, Option<Ident>)],
2063    item_span: Span,
2064    from_glob: bool,
2065) -> impl Iterator<Item = ast::Item<Node::ItemKind>> + 'a {
2066    if suffixes.is_empty() {
2067        // Report an error for now, to avoid keeping stem for resolution and
2068        // stability checks.
2069        let kind = String::from(if from_glob { "glob" } else { "list" });
2070        ecx.dcx().emit_err(EmptyDelegationMac { span: item.span, kind });
2071    }
2072
2073    suffixes.iter().map(move |&(ident, rename)| {
2074        let mut path = deleg.prefix.clone();
2075        path.segments.push(ast::PathSegment { ident, id: ast::DUMMY_NODE_ID, args: None });
2076
2077        ast::Item {
2078            attrs: item.attrs.clone(),
2079            id: ast::DUMMY_NODE_ID,
2080            span: if from_glob { item_span } else { ident.span },
2081            vis: item.vis.clone(),
2082            kind: Node::delegation_item_kind(Box::new(ast::Delegation {
2083                id: ast::DUMMY_NODE_ID,
2084                qself: deleg.qself.clone(),
2085                path,
2086                ident: rename.unwrap_or(ident),
2087                rename,
2088                body: deleg.body.clone(),
2089                from_glob,
2090            })),
2091            tokens: None,
2092        }
2093    })
2094}
2095
2096/// Required for `visit_node` obtained an owned `Node` from `&mut Node`.
2097trait DummyAstNode {
2098    fn dummy() -> Self;
2099}
2100
2101impl DummyAstNode for ast::Crate {
2102    fn dummy() -> Self {
2103        ast::Crate {
2104            attrs: Default::default(),
2105            items: Default::default(),
2106            spans: Default::default(),
2107            id: DUMMY_NODE_ID,
2108            is_placeholder: Default::default(),
2109        }
2110    }
2111}
2112
2113impl DummyAstNode for ast::Ty {
2114    fn dummy() -> Self {
2115        ast::Ty {
2116            id: DUMMY_NODE_ID,
2117            kind: TyKind::Dummy,
2118            span: Default::default(),
2119            tokens: Default::default(),
2120        }
2121    }
2122}
2123
2124impl DummyAstNode for ast::Pat {
2125    fn dummy() -> Self {
2126        ast::Pat {
2127            id: DUMMY_NODE_ID,
2128            kind: PatKind::Wild,
2129            span: Default::default(),
2130            tokens: Default::default(),
2131        }
2132    }
2133}
2134
2135impl DummyAstNode for ast::Expr {
2136    fn dummy() -> Self {
2137        ast::Expr::dummy()
2138    }
2139}
2140
2141impl DummyAstNode for AstNodeWrapper<ast::Expr, MethodReceiverTag> {
2142    fn dummy() -> Self {
2143        AstNodeWrapper::new(ast::Expr::dummy(), MethodReceiverTag)
2144    }
2145}
2146
2147struct InvocationCollector<'a, 'b> {
2148    cx: &'a mut ExtCtxt<'b>,
2149    invocations: Vec<(Invocation, Option<Arc<SyntaxExtension>>)>,
2150    monotonic: bool,
2151}
2152
2153impl<'a, 'b> InvocationCollector<'a, 'b> {
2154    fn cfg(&self) -> StripUnconfigured<'_> {
2155        StripUnconfigured {
2156            sess: self.cx.sess,
2157            features: Some(self.cx.ecfg.features),
2158            config_tokens: false,
2159            lint_node_id: self.cx.current_expansion.lint_node_id,
2160        }
2161    }
2162
2163    fn collect(&mut self, fragment_kind: AstFragmentKind, kind: InvocationKind) -> AstFragment {
2164        let expn_id = LocalExpnId::fresh_empty();
2165        if #[allow(non_exhaustive_omitted_patterns)] match kind {
    InvocationKind::GlobDelegation { .. } => true,
    _ => false,
}matches!(kind, InvocationKind::GlobDelegation { .. }) {
2166            // In resolver we need to know which invocation ids are delegations early,
2167            // before their `ExpnData` is filled.
2168            self.cx.resolver.register_glob_delegation(expn_id);
2169        }
2170        let vis = kind.placeholder_visibility();
2171        self.invocations.push((
2172            Invocation {
2173                kind,
2174                fragment_kind,
2175                expansion_data: ExpansionData {
2176                    id: expn_id,
2177                    depth: self.cx.current_expansion.depth + 1,
2178                    ..self.cx.current_expansion.clone()
2179                },
2180            },
2181            None,
2182        ));
2183        placeholder(fragment_kind, NodeId::placeholder_from_expn_id(expn_id), vis)
2184    }
2185
2186    fn collect_bang(&mut self, mac: Box<ast::MacCall>, kind: AstFragmentKind) -> AstFragment {
2187        // cache the macro call span so that it can be
2188        // easily adjusted for incremental compilation
2189        let span = mac.span();
2190        self.collect(kind, InvocationKind::Bang { mac, span })
2191    }
2192
2193    fn collect_attr(
2194        &mut self,
2195        (attr, pos, derives): (ast::Attribute, usize, Vec<ast::Path>),
2196        item: Annotatable,
2197        kind: AstFragmentKind,
2198    ) -> AstFragment {
2199        self.collect(kind, InvocationKind::Attr { attr, pos, item, derives })
2200    }
2201
2202    fn collect_glob_delegation(
2203        &mut self,
2204        item: Box<ast::AssocItem>,
2205        of_trait: bool,
2206        kind: AstFragmentKind,
2207    ) -> AstFragment {
2208        self.collect(kind, InvocationKind::GlobDelegation { item, of_trait })
2209    }
2210
2211    /// If `item` is an attribute invocation, remove the attribute and return it together with
2212    /// its position and derives following it. We have to collect the derives in order to resolve
2213    /// legacy derive helpers (helpers written before derives that introduce them).
2214    fn take_first_attr(
2215        &self,
2216        item: &mut impl HasAttrs,
2217    ) -> Option<(ast::Attribute, usize, Vec<ast::Path>)> {
2218        let mut attr = None;
2219
2220        let mut cfg_pos = None;
2221        let mut attr_pos = None;
2222        for (pos, attr) in item.attrs().iter().enumerate() {
2223            if !attr.is_doc_comment() && !self.cx.expanded_inert_attrs.is_marked(attr) {
2224                let name = attr.name();
2225                if name == Some(sym::cfg) || name == Some(sym::cfg_attr) {
2226                    cfg_pos = Some(pos); // a cfg attr found, no need to search anymore
2227                    break;
2228                } else if attr_pos.is_none()
2229                    && !name.is_some_and(rustc_feature::is_builtin_attr_name)
2230                {
2231                    attr_pos = Some(pos); // a non-cfg attr found, still may find a cfg attr
2232                }
2233            }
2234        }
2235
2236        item.visit_attrs(|attrs| {
2237            attr = Some(match (cfg_pos, attr_pos) {
2238                (Some(pos), _) => (attrs.remove(pos), pos, Vec::new()),
2239                (_, Some(pos)) => {
2240                    let attr = attrs.remove(pos);
2241                    let following_derives = attrs[pos..]
2242                        .iter()
2243                        .filter(|a| a.has_name(sym::derive))
2244                        .flat_map(|a| a.meta_item_list().unwrap_or_default())
2245                        .filter_map(|meta_item_inner| match meta_item_inner {
2246                            MetaItemInner::MetaItem(ast::MetaItem {
2247                                kind: MetaItemKind::Word,
2248                                path,
2249                                ..
2250                            }) => Some(path),
2251                            _ => None,
2252                        })
2253                        .collect();
2254
2255                    (attr, pos, following_derives)
2256                }
2257                _ => return,
2258            });
2259        });
2260
2261        attr
2262    }
2263
2264    // Detect use of feature-gated or invalid attributes on macro invocations
2265    // since they will not be detected after macro expansion.
2266    fn check_attributes(&self, attrs: &[ast::Attribute], call: &ast::MacCall) {
2267        let features = self.cx.ecfg.features;
2268        let mut attrs = attrs.iter().peekable();
2269        let mut span: Option<Span> = None;
2270        while let Some(attr) = attrs.next() {
2271            rustc_ast_passes::feature_gate::check_attribute(attr, self.cx.sess, features);
2272            validate_attr::check_attr(&self.cx.sess.psess, attr);
2273            AttributeParser::parse_limited_all(
2274                self.cx.sess,
2275                slice::from_ref(attr),
2276                None,
2277                Target::MacroCall,
2278                call.span(),
2279                self.cx.current_expansion.lint_node_id,
2280                Some(self.cx.ecfg.features),
2281                ShouldEmit::ErrorsAndLints { recovery: Recovery::Allowed },
2282            );
2283
2284            let current_span = if let Some(sp) = span { sp.to(attr.span) } else { attr.span };
2285            span = Some(current_span);
2286
2287            if attrs.peek().is_some_and(|next_attr| next_attr.doc_str().is_some()) {
2288                continue;
2289            }
2290
2291            if attr.doc_str_and_fragment_kind().is_some() {
2292                self.cx.sess.psess.buffer_lint(
2293                    UNUSED_DOC_COMMENTS,
2294                    current_span,
2295                    self.cx.current_expansion.lint_node_id,
2296                    crate::errors::MacroCallUnusedDocComment { span: attr.span },
2297                );
2298            } else if rustc_attr_parsing::is_builtin_attr(attr)
2299                && !AttributeParser::<Early>::is_parsed_attribute(&attr.path())
2300            {
2301                let attr_name = attr.name().unwrap();
2302                self.cx.sess.psess.buffer_lint(
2303                    UNUSED_ATTRIBUTES,
2304                    attr.span,
2305                    self.cx.current_expansion.lint_node_id,
2306                    crate::errors::UnusedBuiltinAttribute {
2307                        attr_name,
2308                        macro_name: pprust::path_to_string(&call.path),
2309                        invoc_span: call.path.span,
2310                        attr_span: attr.span,
2311                    },
2312                );
2313            }
2314        }
2315    }
2316
2317    fn expand_cfg_true(
2318        &mut self,
2319        node: &mut impl InvocationCollectorNode,
2320        attr: ast::Attribute,
2321        pos: usize,
2322    ) -> EvalConfigResult {
2323        let Some(cfg) = AttributeParser::parse_single(
2324            self.cfg().sess,
2325            &attr,
2326            attr.span,
2327            self.cfg().lint_node_id,
2328            node.as_target(),
2329            self.cfg().features,
2330            ShouldEmit::ErrorsAndLints { recovery: Recovery::Allowed },
2331            parse_cfg,
2332            &CFG_TEMPLATE,
2333            AllowExprMetavar::Yes,
2334        ) else {
2335            // Cfg attribute was not parsable, give up
2336            return EvalConfigResult::True;
2337        };
2338
2339        let res = eval_config_entry(self.cfg().sess, &cfg);
2340        if res.as_bool() {
2341            // A trace attribute left in AST in place of the original `cfg` attribute.
2342            // It can later be used by lints or other diagnostics.
2343            let mut trace_attr = attr_into_trace(attr, sym::cfg_trace);
2344            trace_attr.replace_args(AttrItemKind::Parsed(EarlyParsedAttribute::CfgTrace(cfg)));
2345            node.visit_attrs(|attrs| attrs.insert(pos, trace_attr));
2346        }
2347
2348        res
2349    }
2350
2351    fn expand_cfg_attr(&self, node: &mut impl HasAttrs, attr: &ast::Attribute, pos: usize) {
2352        node.visit_attrs(|attrs| {
2353            // Repeated `insert` calls is inefficient, but the number of
2354            // insertions is almost always 0 or 1 in practice.
2355            for cfg in self.cfg().expand_cfg_attr(attr, false).into_iter().rev() {
2356                attrs.insert(pos, cfg)
2357            }
2358        });
2359    }
2360
2361    fn flat_map_node<Node: InvocationCollectorNode<OutputTy: Default>>(
2362        &mut self,
2363        mut node: Node,
2364    ) -> Node::OutputTy {
2365        loop {
2366            return match self.take_first_attr(&mut node) {
2367                Some((attr, pos, derives)) => match attr.name() {
2368                    Some(sym::cfg) => {
2369                        let res = self.expand_cfg_true(&mut node, attr, pos);
2370                        match res {
2371                            EvalConfigResult::True => continue,
2372                            EvalConfigResult::False { reason, reason_span } => {
2373                                for ident in node.declared_idents() {
2374                                    self.cx.resolver.append_stripped_cfg_item(
2375                                        self.cx.current_expansion.lint_node_id,
2376                                        ident,
2377                                        reason.clone(),
2378                                        reason_span,
2379                                    )
2380                                }
2381                            }
2382                        }
2383
2384                        Default::default()
2385                    }
2386                    Some(sym::cfg_attr) => {
2387                        self.expand_cfg_attr(&mut node, &attr, pos);
2388                        continue;
2389                    }
2390                    _ => {
2391                        Node::pre_flat_map_node_collect_attr(&self.cfg(), &attr);
2392                        self.collect_attr((attr, pos, derives), node.to_annotatable(), Node::KIND)
2393                            .make_ast::<Node>()
2394                    }
2395                },
2396                None if node.is_mac_call() => {
2397                    let (mac, attrs, add_semicolon) = node.take_mac_call();
2398                    self.check_attributes(&attrs, &mac);
2399                    let mut res = self.collect_bang(mac, Node::KIND).make_ast::<Node>();
2400                    Node::post_flat_map_node_collect_bang(&mut res, add_semicolon);
2401                    res
2402                }
2403                None if let Some((deleg, item)) = node.delegation() => {
2404                    let Some(suffixes) = &deleg.suffixes else {
2405                        let traitless_qself =
2406                            #[allow(non_exhaustive_omitted_patterns)] match &deleg.qself {
    Some(qself) if qself.position == 0 => true,
    _ => false,
}matches!(&deleg.qself, Some(qself) if qself.position == 0);
2407                        let (item, of_trait) = match node.to_annotatable() {
2408                            Annotatable::AssocItem(item, AssocCtxt::Impl { of_trait }) => {
2409                                (item, of_trait)
2410                            }
2411                            ann @ (Annotatable::Item(_)
2412                            | Annotatable::AssocItem(..)
2413                            | Annotatable::Stmt(_)) => {
2414                                let span = ann.span();
2415                                self.cx.dcx().emit_err(GlobDelegationOutsideImpls { span });
2416                                return Default::default();
2417                            }
2418                            _ => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2419                        };
2420                        if traitless_qself {
2421                            let span = item.span;
2422                            self.cx.dcx().emit_err(GlobDelegationTraitlessQpath { span });
2423                            return Default::default();
2424                        }
2425                        return self
2426                            .collect_glob_delegation(item, of_trait, Node::KIND)
2427                            .make_ast::<Node>();
2428                    };
2429
2430                    let single_delegations = build_single_delegations::<Node>(
2431                        self.cx, deleg, item, suffixes, item.span, false,
2432                    );
2433                    Node::flatten_outputs(single_delegations.map(|item| {
2434                        let mut item = Node::from_item(item);
2435                        {
    let old_id = self.cx.current_expansion.lint_node_id;
    if self.monotonic {
        if true {
            match (&*item.node_id_mut(), &ast::DUMMY_NODE_ID) {
                (left_val, right_val) => {
                    if !(*left_val == *right_val) {
                        let kind = ::core::panicking::AssertKind::Eq;
                        ::core::panicking::assert_failed(kind, &*left_val,
                            &*right_val, ::core::option::Option::None);
                    }
                }
            };
        };
        let new_id = self.cx.resolver.next_node_id();
        *item.node_id_mut() = new_id;
        self.cx.current_expansion.lint_node_id = new_id;
    }
    let ret = (|| item.walk_flat_map(self))();
    self.cx.current_expansion.lint_node_id = old_id;
    ret
}assign_id!(self, item.node_id_mut(), || item.walk_flat_map(self))
2436                    }))
2437                }
2438                None => {
2439                    match Node::wrap_flat_map_node_walk_flat_map(node, self, |mut node, this| {
2440                        {
    let old_id = this.cx.current_expansion.lint_node_id;
    if this.monotonic {
        if true {
            match (&*node.node_id_mut(), &ast::DUMMY_NODE_ID) {
                (left_val, right_val) => {
                    if !(*left_val == *right_val) {
                        let kind = ::core::panicking::AssertKind::Eq;
                        ::core::panicking::assert_failed(kind, &*left_val,
                            &*right_val, ::core::option::Option::None);
                    }
                }
            };
        };
        let new_id = this.cx.resolver.next_node_id();
        *node.node_id_mut() = new_id;
        this.cx.current_expansion.lint_node_id = new_id;
    }
    let ret = (|| node.walk_flat_map(this))();
    this.cx.current_expansion.lint_node_id = old_id;
    ret
}assign_id!(this, node.node_id_mut(), || node.walk_flat_map(this))
2441                    }) {
2442                        Ok(output) => output,
2443                        Err(returned_node) => {
2444                            node = returned_node;
2445                            continue;
2446                        }
2447                    }
2448                }
2449            };
2450        }
2451    }
2452
2453    fn visit_node<Node: InvocationCollectorNode<OutputTy: Into<Node>> + DummyAstNode>(
2454        &mut self,
2455        node: &mut Node,
2456    ) {
2457        loop {
2458            return match self.take_first_attr(node) {
2459                Some((attr, pos, derives)) => match attr.name() {
2460                    Some(sym::cfg) => {
2461                        let span = attr.span;
2462                        if self.expand_cfg_true(node, attr, pos).as_bool() {
2463                            continue;
2464                        }
2465
2466                        node.expand_cfg_false(self, pos, span);
2467                        continue;
2468                    }
2469                    Some(sym::cfg_attr) => {
2470                        self.expand_cfg_attr(node, &attr, pos);
2471                        continue;
2472                    }
2473                    _ => {
2474                        let n = mem::replace(node, Node::dummy());
2475                        *node = self
2476                            .collect_attr((attr, pos, derives), n.to_annotatable(), Node::KIND)
2477                            .make_ast::<Node>()
2478                            .into()
2479                    }
2480                },
2481                None if node.is_mac_call() => {
2482                    let n = mem::replace(node, Node::dummy());
2483                    let (mac, attrs, _) = n.take_mac_call();
2484                    self.check_attributes(&attrs, &mac);
2485
2486                    *node = self.collect_bang(mac, Node::KIND).make_ast::<Node>().into()
2487                }
2488                None if node.delegation().is_some() => ::core::panicking::panic("internal error: entered unreachable code")unreachable!(),
2489                None => {
2490                    {
    let old_id = self.cx.current_expansion.lint_node_id;
    if self.monotonic {
        if true {
            match (&*node.node_id_mut(), &ast::DUMMY_NODE_ID) {
                (left_val, right_val) => {
                    if !(*left_val == *right_val) {
                        let kind = ::core::panicking::AssertKind::Eq;
                        ::core::panicking::assert_failed(kind, &*left_val,
                            &*right_val, ::core::option::Option::None);
                    }
                }
            };
        };
        let new_id = self.cx.resolver.next_node_id();
        *node.node_id_mut() = new_id;
        self.cx.current_expansion.lint_node_id = new_id;
    }
    let ret = (|| node.walk(self))();
    self.cx.current_expansion.lint_node_id = old_id;
    ret
}assign_id!(self, node.node_id_mut(), || node.walk(self))
2491                }
2492            };
2493        }
2494    }
2495}
2496
2497impl<'a, 'b> MutVisitor for InvocationCollector<'a, 'b> {
2498    fn flat_map_item(&mut self, node: Box<ast::Item>) -> SmallVec<[Box<ast::Item>; 1]> {
2499        self.flat_map_node(node)
2500    }
2501
2502    fn flat_map_assoc_item(
2503        &mut self,
2504        node: Box<ast::AssocItem>,
2505        ctxt: AssocCtxt,
2506    ) -> SmallVec<[Box<ast::AssocItem>; 1]> {
2507        match ctxt {
2508            AssocCtxt::Trait => self.flat_map_node(AstNodeWrapper::new(node, TraitItemTag)),
2509            AssocCtxt::Impl { of_trait: false, .. } => {
2510                self.flat_map_node(AstNodeWrapper::new(node, ImplItemTag))
2511            }
2512            AssocCtxt::Impl { of_trait: true, .. } => {
2513                self.flat_map_node(AstNodeWrapper::new(node, TraitImplItemTag))
2514            }
2515        }
2516    }
2517
2518    fn flat_map_foreign_item(
2519        &mut self,
2520        node: Box<ast::ForeignItem>,
2521    ) -> SmallVec<[Box<ast::ForeignItem>; 1]> {
2522        self.flat_map_node(node)
2523    }
2524
2525    fn flat_map_variant(&mut self, node: ast::Variant) -> SmallVec<[ast::Variant; 1]> {
2526        self.flat_map_node(node)
2527    }
2528
2529    fn flat_map_where_predicate(
2530        &mut self,
2531        node: ast::WherePredicate,
2532    ) -> SmallVec<[ast::WherePredicate; 1]> {
2533        self.flat_map_node(node)
2534    }
2535
2536    fn flat_map_field_def(&mut self, node: ast::FieldDef) -> SmallVec<[ast::FieldDef; 1]> {
2537        self.flat_map_node(node)
2538    }
2539
2540    fn flat_map_pat_field(&mut self, node: ast::PatField) -> SmallVec<[ast::PatField; 1]> {
2541        self.flat_map_node(node)
2542    }
2543
2544    fn flat_map_expr_field(&mut self, node: ast::ExprField) -> SmallVec<[ast::ExprField; 1]> {
2545        self.flat_map_node(node)
2546    }
2547
2548    fn flat_map_param(&mut self, node: ast::Param) -> SmallVec<[ast::Param; 1]> {
2549        self.flat_map_node(node)
2550    }
2551
2552    fn flat_map_generic_param(
2553        &mut self,
2554        node: ast::GenericParam,
2555    ) -> SmallVec<[ast::GenericParam; 1]> {
2556        self.flat_map_node(node)
2557    }
2558
2559    fn flat_map_arm(&mut self, node: ast::Arm) -> SmallVec<[ast::Arm; 1]> {
2560        self.flat_map_node(node)
2561    }
2562
2563    fn flat_map_stmt(&mut self, node: ast::Stmt) -> SmallVec<[ast::Stmt; 1]> {
2564        // FIXME: invocations in semicolon-less expressions positions are expanded as expressions,
2565        // changing that requires some compatibility measures.
2566        if node.is_expr() {
2567            // The only way that we can end up with a `MacCall` expression statement,
2568            // (as opposed to a `StmtKind::MacCall`) is if we have a macro as the
2569            // trailing expression in a block (e.g. `fn foo() { my_macro!() }`).
2570            // Record this information, so that we can report a more specific
2571            // `SEMICOLON_IN_EXPRESSIONS_FROM_MACROS` lint if needed.
2572            // See #78991 for an investigation of treating macros in this position
2573            // as statements, rather than expressions, during parsing.
2574            return match &node.kind {
2575                StmtKind::Expr(expr)
2576                    if #[allow(non_exhaustive_omitted_patterns)] match **expr {
    ast::Expr { kind: ExprKind::MacCall(..), .. } => true,
    _ => false,
}matches!(**expr, ast::Expr { kind: ExprKind::MacCall(..), .. }) =>
2577                {
2578                    self.cx.current_expansion.is_trailing_mac = true;
2579                    // Don't use `assign_id` for this statement - it may get removed
2580                    // entirely due to a `#[cfg]` on the contained expression
2581                    let res = walk_flat_map_stmt(self, node);
2582                    self.cx.current_expansion.is_trailing_mac = false;
2583                    res
2584                }
2585                _ => walk_flat_map_stmt(self, node),
2586            };
2587        }
2588
2589        self.flat_map_node(node)
2590    }
2591
2592    fn visit_crate(&mut self, node: &mut ast::Crate) {
2593        self.visit_node(node)
2594    }
2595
2596    fn visit_ty(&mut self, node: &mut ast::Ty) {
2597        self.visit_node(node)
2598    }
2599
2600    fn visit_pat(&mut self, node: &mut ast::Pat) {
2601        self.visit_node(node)
2602    }
2603
2604    fn visit_expr(&mut self, node: &mut ast::Expr) {
2605        // FIXME: Feature gating is performed inconsistently between `Expr` and `OptExpr`.
2606        if let Some(attr) = node.attrs.first() {
2607            self.cfg().maybe_emit_expr_attr_err(attr);
2608        }
2609        ensure_sufficient_stack(|| self.visit_node(node))
2610    }
2611
2612    fn visit_method_receiver_expr(&mut self, node: &mut ast::Expr) {
2613        self.visit_node(AstNodeWrapper::from_mut(node, MethodReceiverTag))
2614    }
2615
2616    fn filter_map_expr(&mut self, node: Box<ast::Expr>) -> Option<Box<ast::Expr>> {
2617        self.flat_map_node(AstNodeWrapper::new(node, OptExprTag))
2618    }
2619
2620    fn visit_block(&mut self, node: &mut ast::Block) {
2621        let orig_dir_ownership = mem::replace(
2622            &mut self.cx.current_expansion.dir_ownership,
2623            DirOwnership::UnownedViaBlock,
2624        );
2625        walk_block(self, node);
2626        self.cx.current_expansion.dir_ownership = orig_dir_ownership;
2627    }
2628
2629    fn visit_id(&mut self, id: &mut NodeId) {
2630        // We may have already assigned a `NodeId`
2631        // by calling `assign_id`
2632        if self.monotonic && *id == ast::DUMMY_NODE_ID {
2633            *id = self.cx.resolver.next_node_id();
2634        }
2635    }
2636}
2637
2638pub struct ExpansionConfig<'feat> {
2639    pub crate_name: Symbol,
2640    pub features: &'feat Features,
2641    pub recursion_limit: Limit,
2642    pub trace_mac: bool,
2643    /// If false, strip `#[test]` nodes
2644    pub should_test: bool,
2645    /// If true, use verbose debugging for `proc_macro::Span`
2646    pub span_debug: bool,
2647    /// If true, show backtraces for proc-macro panics
2648    pub proc_macro_backtrace: bool,
2649}
2650
2651impl ExpansionConfig<'_> {
2652    pub fn default(crate_name: Symbol, features: &Features) -> ExpansionConfig<'_> {
2653        ExpansionConfig {
2654            crate_name,
2655            features,
2656            // FIXME should this limit be configurable?
2657            recursion_limit: Limit::new(1024),
2658            trace_mac: false,
2659            should_test: false,
2660            span_debug: false,
2661            proc_macro_backtrace: false,
2662        }
2663    }
2664}