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 pub enum AstFragment {
73 OptExpr(Option<Box<ast::Expr>>),
74 MethodReceiverExpr(Box<ast::Expr>),
75 $($Kind($AstTy),)*
76 }
77
78 #[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 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
195pub 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),
}
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 pos: usize,
407 item: Annotatable,
408 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 of_trait: bool,
420 },
421}
422
423impl InvocationKind {
424 fn placeholder_visibility(&self) -> Option<ast::Visibility> {
425 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, }
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 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 let (mut fragment_with_placeholders, mut invocations) =
499 self.collect_invocations(input_fragment, &[]);
500
501 self.resolve_imports();
504
505 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 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 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 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 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 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 fn collect_invocations(
649 &mut self,
650 mut fragment: AstFragment,
651 extra_placeholders: &[NodeId],
652 ) -> (AstFragment, Vec<(Invocation, Option<Arc<SyntaxExtension>>)>) {
653 self.cx.resolver.resolve_dollar_crates();
655
656 let mut invocations = {
657 let mut collector = InvocationCollector {
658 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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 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
1229macro_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
1271trait 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 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 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 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 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 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 return Err(node);
1440 }
1441 (Some(file_path), dir_path, dir_ownership)
1442 }
1443 };
1444
1445 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 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 #[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 self.attrs.truncate(pos);
1892 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 if let ast::TyKind::ImplTrait(..) = self.kind {
1913 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 ::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
2024struct 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 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
2096trait 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 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 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 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); 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); }
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 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 return EvalConfigResult::True;
2337 };
2338
2339 let res = eval_config_entry(self.cfg().sess, &cfg);
2340 if res.as_bool() {
2341 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 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 if node.is_expr() {
2567 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 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 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 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 pub should_test: bool,
2645 pub span_debug: bool,
2647 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 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}