]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_ast/src/mut_visit.rs
Actually infer args in visitors
[rust.git] / compiler / rustc_ast / src / mut_visit.rs
1 //! A `MutVisitor` represents an AST modification; it accepts an AST piece and
2 //! and mutates it in place. So, for instance, macro expansion is a `MutVisitor`
3 //! that walks over an AST and modifies it.
4 //!
5 //! Note: using a `MutVisitor` (other than the `MacroExpander` `MutVisitor`) on
6 //! an AST before macro expansion is probably a bad idea. For instance,
7 //! a `MutVisitor` renaming item names in a module will miss all of those
8 //! that are created by the expansion of a macro.
9
10 use crate::ast::*;
11 use crate::ptr::P;
12 use crate::token::{self, Token};
13 use crate::tokenstream::*;
14
15 use rustc_data_structures::map_in_place::MapInPlace;
16 use rustc_data_structures::sync::Lrc;
17 use rustc_span::source_map::Spanned;
18 use rustc_span::symbol::Ident;
19 use rustc_span::Span;
20
21 use smallvec::{smallvec, Array, SmallVec};
22 use std::ops::DerefMut;
23 use std::{panic, process, ptr};
24
25 pub trait ExpectOne<A: Array> {
26     fn expect_one(self, err: &'static str) -> A::Item;
27 }
28
29 impl<A: Array> ExpectOne<A> for SmallVec<A> {
30     fn expect_one(self, err: &'static str) -> A::Item {
31         assert!(self.len() == 1, "{}", err);
32         self.into_iter().next().unwrap()
33     }
34 }
35
36 pub trait MutVisitor: Sized {
37     /// Mutable token visiting only exists for the `macro_rules` token marker and should not be
38     /// used otherwise. Token visitor would be entirely separate from the regular visitor if
39     /// the marker didn't have to visit AST fragments in nonterminal tokens.
40     fn token_visiting_enabled(&self) -> bool {
41         false
42     }
43
44     // Methods in this trait have one of three forms:
45     //
46     //   fn visit_t(&mut self, t: &mut T);                      // common
47     //   fn flat_map_t(&mut self, t: T) -> SmallVec<[T; 1]>;    // rare
48     //   fn filter_map_t(&mut self, t: T) -> Option<T>;         // rarest
49     //
50     // Any additions to this trait should happen in form of a call to a public
51     // `noop_*` function that only calls out to the visitor again, not other
52     // `noop_*` functions. This is a necessary API workaround to the problem of
53     // not being able to call out to the super default method in an overridden
54     // default method.
55     //
56     // When writing these methods, it is better to use destructuring like this:
57     //
58     //   fn visit_abc(&mut self, ABC { a, b, c: _ }: &mut ABC) {
59     //       visit_a(a);
60     //       visit_b(b);
61     //   }
62     //
63     // than to use field access like this:
64     //
65     //   fn visit_abc(&mut self, abc: &mut ABC) {
66     //       visit_a(&mut abc.a);
67     //       visit_b(&mut abc.b);
68     //       // ignore abc.c
69     //   }
70     //
71     // As well as being more concise, the former is explicit about which fields
72     // are skipped. Furthermore, if a new field is added, the destructuring
73     // version will cause a compile error, which is good. In comparison, the
74     // field access version will continue working and it would be easy to
75     // forget to add handling for it.
76
77     fn visit_crate(&mut self, c: &mut Crate) {
78         noop_visit_crate(c, self)
79     }
80
81     fn visit_meta_list_item(&mut self, list_item: &mut NestedMetaItem) {
82         noop_visit_meta_list_item(list_item, self);
83     }
84
85     fn visit_meta_item(&mut self, meta_item: &mut MetaItem) {
86         noop_visit_meta_item(meta_item, self);
87     }
88
89     fn visit_use_tree(&mut self, use_tree: &mut UseTree) {
90         noop_visit_use_tree(use_tree, self);
91     }
92
93     fn flat_map_foreign_item(&mut self, ni: P<ForeignItem>) -> SmallVec<[P<ForeignItem>; 1]> {
94         noop_flat_map_foreign_item(ni, self)
95     }
96
97     fn flat_map_item(&mut self, i: P<Item>) -> SmallVec<[P<Item>; 1]> {
98         noop_flat_map_item(i, self)
99     }
100
101     fn visit_fn_header(&mut self, header: &mut FnHeader) {
102         noop_visit_fn_header(header, self);
103     }
104
105     fn flat_map_field_def(&mut self, fd: FieldDef) -> SmallVec<[FieldDef; 1]> {
106         noop_flat_map_field_def(fd, self)
107     }
108
109     fn visit_item_kind(&mut self, i: &mut ItemKind) {
110         noop_visit_item_kind(i, self);
111     }
112
113     fn flat_map_trait_item(&mut self, i: P<AssocItem>) -> SmallVec<[P<AssocItem>; 1]> {
114         noop_flat_map_assoc_item(i, self)
115     }
116
117     fn flat_map_impl_item(&mut self, i: P<AssocItem>) -> SmallVec<[P<AssocItem>; 1]> {
118         noop_flat_map_assoc_item(i, self)
119     }
120
121     fn visit_fn_decl(&mut self, d: &mut P<FnDecl>) {
122         noop_visit_fn_decl(d, self);
123     }
124
125     fn visit_asyncness(&mut self, a: &mut Async) {
126         noop_visit_asyncness(a, self);
127     }
128
129     fn visit_block(&mut self, b: &mut P<Block>) {
130         noop_visit_block(b, self);
131     }
132
133     fn flat_map_stmt(&mut self, s: Stmt) -> SmallVec<[Stmt; 1]> {
134         noop_flat_map_stmt(s, self)
135     }
136
137     fn flat_map_arm(&mut self, arm: Arm) -> SmallVec<[Arm; 1]> {
138         noop_flat_map_arm(arm, self)
139     }
140
141     fn visit_pat(&mut self, p: &mut P<Pat>) {
142         noop_visit_pat(p, self);
143     }
144
145     fn visit_anon_const(&mut self, c: &mut AnonConst) {
146         noop_visit_anon_const(c, self);
147     }
148
149     fn visit_expr(&mut self, e: &mut P<Expr>) {
150         noop_visit_expr(e, self);
151     }
152
153     fn filter_map_expr(&mut self, e: P<Expr>) -> Option<P<Expr>> {
154         noop_filter_map_expr(e, self)
155     }
156
157     fn visit_generic_arg(&mut self, arg: &mut GenericArg) {
158         noop_visit_generic_arg(arg, self);
159     }
160
161     fn visit_ty(&mut self, t: &mut P<Ty>) {
162         noop_visit_ty(t, self);
163     }
164
165     fn visit_lifetime(&mut self, l: &mut Lifetime) {
166         noop_visit_lifetime(l, self);
167     }
168
169     fn visit_ty_constraint(&mut self, t: &mut AssocTyConstraint) {
170         noop_visit_ty_constraint(t, self);
171     }
172
173     fn visit_foreign_mod(&mut self, nm: &mut ForeignMod) {
174         noop_visit_foreign_mod(nm, self);
175     }
176
177     fn flat_map_variant(&mut self, v: Variant) -> SmallVec<[Variant; 1]> {
178         noop_flat_map_variant(v, self)
179     }
180
181     fn visit_ident(&mut self, i: &mut Ident) {
182         noop_visit_ident(i, self);
183     }
184
185     fn visit_path(&mut self, p: &mut Path) {
186         noop_visit_path(p, self);
187     }
188
189     fn visit_qself(&mut self, qs: &mut Option<QSelf>) {
190         noop_visit_qself(qs, self);
191     }
192
193     fn visit_generic_args(&mut self, p: &mut GenericArgs) {
194         noop_visit_generic_args(p, self);
195     }
196
197     fn visit_angle_bracketed_parameter_data(&mut self, p: &mut AngleBracketedArgs) {
198         noop_visit_angle_bracketed_parameter_data(p, self);
199     }
200
201     fn visit_parenthesized_parameter_data(&mut self, p: &mut ParenthesizedArgs) {
202         noop_visit_parenthesized_parameter_data(p, self);
203     }
204
205     fn visit_local(&mut self, l: &mut P<Local>) {
206         noop_visit_local(l, self);
207     }
208
209     fn visit_mac_call(&mut self, mac: &mut MacCall) {
210         noop_visit_mac(mac, self);
211     }
212
213     fn visit_macro_def(&mut self, def: &mut MacroDef) {
214         noop_visit_macro_def(def, self);
215     }
216
217     fn visit_label(&mut self, label: &mut Label) {
218         noop_visit_label(label, self);
219     }
220
221     fn visit_attribute(&mut self, at: &mut Attribute) {
222         noop_visit_attribute(at, self);
223     }
224
225     fn flat_map_param(&mut self, param: Param) -> SmallVec<[Param; 1]> {
226         noop_flat_map_param(param, self)
227     }
228
229     fn visit_generics(&mut self, generics: &mut Generics) {
230         noop_visit_generics(generics, self);
231     }
232
233     fn visit_trait_ref(&mut self, tr: &mut TraitRef) {
234         noop_visit_trait_ref(tr, self);
235     }
236
237     fn visit_poly_trait_ref(&mut self, p: &mut PolyTraitRef) {
238         noop_visit_poly_trait_ref(p, self);
239     }
240
241     fn visit_variant_data(&mut self, vdata: &mut VariantData) {
242         noop_visit_variant_data(vdata, self);
243     }
244
245     fn flat_map_generic_param(&mut self, param: GenericParam) -> SmallVec<[GenericParam; 1]> {
246         noop_flat_map_generic_param(param, self)
247     }
248
249     fn visit_param_bound(&mut self, tpb: &mut GenericBound) {
250         noop_visit_param_bound(tpb, self);
251     }
252
253     fn visit_mt(&mut self, mt: &mut MutTy) {
254         noop_visit_mt(mt, self);
255     }
256
257     fn flat_map_expr_field(&mut self, f: ExprField) -> SmallVec<[ExprField; 1]> {
258         noop_flat_map_expr_field(f, self)
259     }
260
261     fn visit_where_clause(&mut self, where_clause: &mut WhereClause) {
262         noop_visit_where_clause(where_clause, self);
263     }
264
265     fn visit_where_predicate(&mut self, where_predicate: &mut WherePredicate) {
266         noop_visit_where_predicate(where_predicate, self);
267     }
268
269     fn visit_vis(&mut self, vis: &mut Visibility) {
270         noop_visit_vis(vis, self);
271     }
272
273     fn visit_id(&mut self, _id: &mut NodeId) {
274         // Do nothing.
275     }
276
277     fn visit_span(&mut self, _sp: &mut Span) {
278         // Do nothing.
279     }
280
281     fn flat_map_pat_field(&mut self, fp: PatField) -> SmallVec<[PatField; 1]> {
282         noop_flat_map_pat_field(fp, self)
283     }
284 }
285
286 /// Use a map-style function (`FnOnce(T) -> T`) to overwrite a `&mut T`. Useful
287 /// when using a `flat_map_*` or `filter_map_*` method within a `visit_`
288 /// method. Abort the program if the closure panics.
289 //
290 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
291 pub fn visit_clobber<T, F>(t: &mut T, f: F)
292 where
293     F: FnOnce(T) -> T,
294 {
295     unsafe {
296         // Safe because `t` is used in a read-only fashion by `read()` before
297         // being overwritten by `write()`.
298         let old_t = ptr::read(t);
299         let new_t = panic::catch_unwind(panic::AssertUnwindSafe(|| f(old_t)))
300             .unwrap_or_else(|_| process::abort());
301         ptr::write(t, new_t);
302     }
303 }
304
305 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
306 #[inline]
307 pub fn visit_vec<T, F>(elems: &mut Vec<T>, mut visit_elem: F)
308 where
309     F: FnMut(&mut T),
310 {
311     for elem in elems {
312         visit_elem(elem);
313     }
314 }
315
316 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
317 #[inline]
318 pub fn visit_opt<T, F>(opt: &mut Option<T>, mut visit_elem: F)
319 where
320     F: FnMut(&mut T),
321 {
322     if let Some(elem) = opt {
323         visit_elem(elem);
324     }
325 }
326
327 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
328 pub fn visit_attrs<T: MutVisitor>(attrs: &mut Vec<Attribute>, vis: &mut T) {
329     visit_vec(attrs, |attr| vis.visit_attribute(attr));
330 }
331
332 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
333 pub fn visit_thin_attrs<T: MutVisitor>(attrs: &mut AttrVec, vis: &mut T) {
334     for attr in attrs.iter_mut() {
335         vis.visit_attribute(attr);
336     }
337 }
338
339 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
340 pub fn visit_exprs<T: MutVisitor>(exprs: &mut Vec<P<Expr>>, vis: &mut T) {
341     exprs.flat_map_in_place(|expr| vis.filter_map_expr(expr))
342 }
343
344 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
345 pub fn visit_bounds<T: MutVisitor>(bounds: &mut GenericBounds, vis: &mut T) {
346     visit_vec(bounds, |bound| vis.visit_param_bound(bound));
347 }
348
349 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
350 pub fn visit_fn_sig<T: MutVisitor>(FnSig { header, decl, span }: &mut FnSig, vis: &mut T) {
351     vis.visit_fn_header(header);
352     vis.visit_fn_decl(decl);
353     vis.visit_span(span);
354 }
355
356 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
357 pub fn visit_mac_args<T: MutVisitor>(args: &mut MacArgs, vis: &mut T) {
358     match args {
359         MacArgs::Empty => {}
360         MacArgs::Delimited(dspan, _delim, tokens) => {
361             visit_delim_span(dspan, vis);
362             visit_tts(tokens, vis);
363         }
364         MacArgs::Eq(eq_span, token) => {
365             vis.visit_span(eq_span);
366             if vis.token_visiting_enabled() {
367                 visit_token(token, vis);
368             } else {
369                 // The value in `#[key = VALUE]` must be visited as an expression for backward
370                 // compatibility, so that macros can be expanded in that position.
371                 match &mut token.kind {
372                     token::Interpolated(nt) => match Lrc::make_mut(nt) {
373                         token::NtExpr(expr) => vis.visit_expr(expr),
374                         t => panic!("unexpected token in key-value attribute: {:?}", t),
375                     },
376                     t => panic!("unexpected token in key-value attribute: {:?}", t),
377                 }
378             }
379         }
380     }
381 }
382
383 pub fn visit_delim_span<T: MutVisitor>(dspan: &mut DelimSpan, vis: &mut T) {
384     vis.visit_span(&mut dspan.open);
385     vis.visit_span(&mut dspan.close);
386 }
387
388 pub fn noop_flat_map_pat_field<T: MutVisitor>(
389     mut fp: PatField,
390     vis: &mut T,
391 ) -> SmallVec<[PatField; 1]> {
392     let PatField { attrs, id, ident, is_placeholder: _, is_shorthand: _, pat, span } = &mut fp;
393     vis.visit_id(id);
394     vis.visit_ident(ident);
395     vis.visit_pat(pat);
396     vis.visit_span(span);
397     visit_thin_attrs(attrs, vis);
398     smallvec![fp]
399 }
400
401 pub fn noop_visit_use_tree<T: MutVisitor>(use_tree: &mut UseTree, vis: &mut T) {
402     let UseTree { prefix, kind, span } = use_tree;
403     vis.visit_path(prefix);
404     match kind {
405         UseTreeKind::Simple(rename, id1, id2) => {
406             visit_opt(rename, |rename| vis.visit_ident(rename));
407             vis.visit_id(id1);
408             vis.visit_id(id2);
409         }
410         UseTreeKind::Nested(items) => {
411             for (tree, id) in items {
412                 vis.visit_use_tree(tree);
413                 vis.visit_id(id);
414             }
415         }
416         UseTreeKind::Glob => {}
417     }
418     vis.visit_span(span);
419 }
420
421 pub fn noop_flat_map_arm<T: MutVisitor>(mut arm: Arm, vis: &mut T) -> SmallVec<[Arm; 1]> {
422     let Arm { attrs, pat, guard, body, span, id, is_placeholder: _ } = &mut arm;
423     visit_thin_attrs(attrs, vis);
424     vis.visit_id(id);
425     vis.visit_pat(pat);
426     visit_opt(guard, |guard| vis.visit_expr(guard));
427     vis.visit_expr(body);
428     vis.visit_span(span);
429     smallvec![arm]
430 }
431
432 pub fn noop_visit_ty_constraint<T: MutVisitor>(
433     AssocTyConstraint { id, ident, gen_args, kind, span }: &mut AssocTyConstraint,
434     vis: &mut T,
435 ) {
436     vis.visit_id(id);
437     vis.visit_ident(ident);
438     if let Some(ref mut gen_args) = gen_args {
439         vis.visit_generic_args(gen_args);
440     }
441     match kind {
442         AssocTyConstraintKind::Equality { ref mut ty } => {
443             vis.visit_ty(ty);
444         }
445         AssocTyConstraintKind::Bound { ref mut bounds } => {
446             visit_bounds(bounds, vis);
447         }
448     }
449     vis.visit_span(span);
450 }
451
452 pub fn noop_visit_ty<T: MutVisitor>(ty: &mut P<Ty>, vis: &mut T) {
453     let Ty { id, kind, span, tokens } = ty.deref_mut();
454     vis.visit_id(id);
455     match kind {
456         TyKind::Infer | TyKind::ImplicitSelf | TyKind::Err | TyKind::Never | TyKind::CVarArgs => {}
457         TyKind::Slice(ty) => vis.visit_ty(ty),
458         TyKind::Ptr(mt) => vis.visit_mt(mt),
459         TyKind::Rptr(lt, mt) => {
460             visit_opt(lt, |lt| noop_visit_lifetime(lt, vis));
461             vis.visit_mt(mt);
462         }
463         TyKind::BareFn(bft) => {
464             let BareFnTy { unsafety: _, ext: _, generic_params, decl } = bft.deref_mut();
465             generic_params.flat_map_in_place(|param| vis.flat_map_generic_param(param));
466             vis.visit_fn_decl(decl);
467         }
468         TyKind::Tup(tys) => visit_vec(tys, |ty| vis.visit_ty(ty)),
469         TyKind::Paren(ty) => vis.visit_ty(ty),
470         TyKind::Path(qself, path) => {
471             vis.visit_qself(qself);
472             vis.visit_path(path);
473         }
474         TyKind::Array(ty, length) => {
475             vis.visit_ty(ty);
476             vis.visit_anon_const(length);
477         }
478         TyKind::Typeof(expr) => vis.visit_anon_const(expr),
479         TyKind::TraitObject(bounds, _syntax) => {
480             visit_vec(bounds, |bound| vis.visit_param_bound(bound))
481         }
482         TyKind::ImplTrait(id, bounds) => {
483             vis.visit_id(id);
484             visit_vec(bounds, |bound| vis.visit_param_bound(bound));
485         }
486         TyKind::MacCall(mac) => vis.visit_mac_call(mac),
487         TyKind::AnonymousStruct(fields, ..) | TyKind::AnonymousUnion(fields, ..) => {
488             fields.flat_map_in_place(|field| vis.flat_map_field_def(field));
489         }
490     }
491     vis.visit_span(span);
492     visit_lazy_tts(tokens, vis);
493 }
494
495 pub fn noop_visit_foreign_mod<T: MutVisitor>(foreign_mod: &mut ForeignMod, vis: &mut T) {
496     let ForeignMod { unsafety: _, abi: _, items } = foreign_mod;
497     items.flat_map_in_place(|item| vis.flat_map_foreign_item(item));
498 }
499
500 pub fn noop_flat_map_variant<T: MutVisitor>(
501     mut variant: Variant,
502     visitor: &mut T,
503 ) -> SmallVec<[Variant; 1]> {
504     let Variant { ident, vis, attrs, id, data, disr_expr, span, is_placeholder: _ } = &mut variant;
505     visitor.visit_ident(ident);
506     visitor.visit_vis(vis);
507     visit_thin_attrs(attrs, visitor);
508     visitor.visit_id(id);
509     visitor.visit_variant_data(data);
510     visit_opt(disr_expr, |disr_expr| visitor.visit_anon_const(disr_expr));
511     visitor.visit_span(span);
512     smallvec![variant]
513 }
514
515 pub fn noop_visit_ident<T: MutVisitor>(Ident { name: _, span }: &mut Ident, vis: &mut T) {
516     vis.visit_span(span);
517 }
518
519 pub fn noop_visit_path<T: MutVisitor>(Path { segments, span, tokens }: &mut Path, vis: &mut T) {
520     vis.visit_span(span);
521     for PathSegment { ident, id, args } in segments {
522         vis.visit_ident(ident);
523         vis.visit_id(id);
524         visit_opt(args, |args| vis.visit_generic_args(args));
525     }
526     visit_lazy_tts(tokens, vis);
527 }
528
529 pub fn noop_visit_qself<T: MutVisitor>(qself: &mut Option<QSelf>, vis: &mut T) {
530     visit_opt(qself, |QSelf { ty, path_span, position: _ }| {
531         vis.visit_ty(ty);
532         vis.visit_span(path_span);
533     })
534 }
535
536 pub fn noop_visit_generic_args<T: MutVisitor>(generic_args: &mut GenericArgs, vis: &mut T) {
537     match generic_args {
538         GenericArgs::AngleBracketed(data) => vis.visit_angle_bracketed_parameter_data(data),
539         GenericArgs::Parenthesized(data) => vis.visit_parenthesized_parameter_data(data),
540     }
541 }
542
543 pub fn noop_visit_generic_arg<T: MutVisitor>(arg: &mut GenericArg, vis: &mut T) {
544     match arg {
545         GenericArg::Lifetime(lt) => vis.visit_lifetime(lt),
546         GenericArg::Type(ty) => vis.visit_ty(ty),
547         GenericArg::Const(ct) => vis.visit_anon_const(ct),
548     }
549 }
550
551 pub fn noop_visit_angle_bracketed_parameter_data<T: MutVisitor>(
552     data: &mut AngleBracketedArgs,
553     vis: &mut T,
554 ) {
555     let AngleBracketedArgs { args, span } = data;
556     visit_vec(args, |arg| match arg {
557         AngleBracketedArg::Arg(arg) => vis.visit_generic_arg(arg),
558         AngleBracketedArg::Constraint(constraint) => vis.visit_ty_constraint(constraint),
559     });
560     vis.visit_span(span);
561 }
562
563 pub fn noop_visit_parenthesized_parameter_data<T: MutVisitor>(
564     args: &mut ParenthesizedArgs,
565     vis: &mut T,
566 ) {
567     let ParenthesizedArgs { inputs, output, span, .. } = args;
568     visit_vec(inputs, |input| vis.visit_ty(input));
569     noop_visit_fn_ret_ty(output, vis);
570     vis.visit_span(span);
571 }
572
573 pub fn noop_visit_local<T: MutVisitor>(local: &mut P<Local>, vis: &mut T) {
574     let Local { id, pat, ty, init, span, attrs, tokens } = local.deref_mut();
575     vis.visit_id(id);
576     vis.visit_pat(pat);
577     visit_opt(ty, |ty| vis.visit_ty(ty));
578     visit_opt(init, |init| vis.visit_expr(init));
579     vis.visit_span(span);
580     visit_thin_attrs(attrs, vis);
581     visit_lazy_tts(tokens, vis);
582 }
583
584 pub fn noop_visit_attribute<T: MutVisitor>(attr: &mut Attribute, vis: &mut T) {
585     let Attribute { kind, id: _, style: _, span } = attr;
586     match kind {
587         AttrKind::Normal(AttrItem { path, args, tokens }, attr_tokens) => {
588             vis.visit_path(path);
589             visit_mac_args(args, vis);
590             visit_lazy_tts(tokens, vis);
591             visit_lazy_tts(attr_tokens, vis);
592         }
593         AttrKind::DocComment(..) => {}
594     }
595     vis.visit_span(span);
596 }
597
598 pub fn noop_visit_mac<T: MutVisitor>(mac: &mut MacCall, vis: &mut T) {
599     let MacCall { path, args, prior_type_ascription: _ } = mac;
600     vis.visit_path(path);
601     visit_mac_args(args, vis);
602 }
603
604 pub fn noop_visit_macro_def<T: MutVisitor>(macro_def: &mut MacroDef, vis: &mut T) {
605     let MacroDef { body, macro_rules: _ } = macro_def;
606     visit_mac_args(body, vis);
607 }
608
609 pub fn noop_visit_meta_list_item<T: MutVisitor>(li: &mut NestedMetaItem, vis: &mut T) {
610     match li {
611         NestedMetaItem::MetaItem(mi) => vis.visit_meta_item(mi),
612         NestedMetaItem::Literal(_lit) => {}
613     }
614 }
615
616 pub fn noop_visit_meta_item<T: MutVisitor>(mi: &mut MetaItem, vis: &mut T) {
617     let MetaItem { path: _, kind, span } = mi;
618     match kind {
619         MetaItemKind::Word => {}
620         MetaItemKind::List(mis) => visit_vec(mis, |mi| vis.visit_meta_list_item(mi)),
621         MetaItemKind::NameValue(_s) => {}
622     }
623     vis.visit_span(span);
624 }
625
626 pub fn noop_flat_map_param<T: MutVisitor>(mut param: Param, vis: &mut T) -> SmallVec<[Param; 1]> {
627     let Param { attrs, id, pat, span, ty, is_placeholder: _ } = &mut param;
628     vis.visit_id(id);
629     visit_thin_attrs(attrs, vis);
630     vis.visit_pat(pat);
631     vis.visit_span(span);
632     vis.visit_ty(ty);
633     smallvec![param]
634 }
635
636 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
637 pub fn visit_attr_annotated_tt<T: MutVisitor>(tt: &mut AttrAnnotatedTokenTree, vis: &mut T) {
638     match tt {
639         AttrAnnotatedTokenTree::Token(token) => {
640             visit_token(token, vis);
641         }
642         AttrAnnotatedTokenTree::Delimited(DelimSpan { open, close }, _delim, tts) => {
643             vis.visit_span(open);
644             vis.visit_span(close);
645             visit_attr_annotated_tts(tts, vis);
646         }
647         AttrAnnotatedTokenTree::Attributes(data) => {
648             for attr in &mut *data.attrs {
649                 match &mut attr.kind {
650                     AttrKind::Normal(_, attr_tokens) => {
651                         visit_lazy_tts(attr_tokens, vis);
652                     }
653                     AttrKind::DocComment(..) => {
654                         vis.visit_span(&mut attr.span);
655                     }
656                 }
657             }
658             visit_lazy_tts_opt_mut(Some(&mut data.tokens), vis);
659         }
660     }
661 }
662
663 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
664 pub fn visit_tt<T: MutVisitor>(tt: &mut TokenTree, vis: &mut T) {
665     match tt {
666         TokenTree::Token(token) => {
667             visit_token(token, vis);
668         }
669         TokenTree::Delimited(DelimSpan { open, close }, _delim, tts) => {
670             vis.visit_span(open);
671             vis.visit_span(close);
672             visit_tts(tts, vis);
673         }
674     }
675 }
676
677 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
678 pub fn visit_tts<T: MutVisitor>(TokenStream(tts): &mut TokenStream, vis: &mut T) {
679     if vis.token_visiting_enabled() && !tts.is_empty() {
680         let tts = Lrc::make_mut(tts);
681         visit_vec(tts, |(tree, _is_joint)| visit_tt(tree, vis));
682     }
683 }
684
685 pub fn visit_attr_annotated_tts<T: MutVisitor>(
686     AttrAnnotatedTokenStream(tts): &mut AttrAnnotatedTokenStream,
687     vis: &mut T,
688 ) {
689     if vis.token_visiting_enabled() && !tts.is_empty() {
690         let tts = Lrc::make_mut(tts);
691         visit_vec(tts, |(tree, _is_joint)| visit_attr_annotated_tt(tree, vis));
692     }
693 }
694
695 pub fn visit_lazy_tts_opt_mut<T: MutVisitor>(lazy_tts: Option<&mut LazyTokenStream>, vis: &mut T) {
696     if vis.token_visiting_enabled() {
697         if let Some(lazy_tts) = lazy_tts {
698             let mut tts = lazy_tts.create_token_stream();
699             visit_attr_annotated_tts(&mut tts, vis);
700             *lazy_tts = LazyTokenStream::new(tts);
701         }
702     }
703 }
704
705 pub fn visit_lazy_tts<T: MutVisitor>(lazy_tts: &mut Option<LazyTokenStream>, vis: &mut T) {
706     visit_lazy_tts_opt_mut(lazy_tts.as_mut(), vis);
707 }
708
709 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
710 // Applies ident visitor if it's an ident; applies other visits to interpolated nodes.
711 // In practice the ident part is not actually used by specific visitors right now,
712 // but there's a test below checking that it works.
713 pub fn visit_token<T: MutVisitor>(t: &mut Token, vis: &mut T) {
714     let Token { kind, span } = t;
715     match kind {
716         token::Ident(name, _) | token::Lifetime(name) => {
717             let mut ident = Ident::new(*name, *span);
718             vis.visit_ident(&mut ident);
719             *name = ident.name;
720             *span = ident.span;
721             return; // Avoid visiting the span for the second time.
722         }
723         token::Interpolated(nt) => {
724             let mut nt = Lrc::make_mut(nt);
725             visit_interpolated(&mut nt, vis);
726         }
727         _ => {}
728     }
729     vis.visit_span(span);
730 }
731
732 // No `noop_` prefix because there isn't a corresponding method in `MutVisitor`.
733 /// Applies the visitor to elements of interpolated nodes.
734 //
735 // N.B., this can occur only when applying a visitor to partially expanded
736 // code, where parsed pieces have gotten implanted ito *other* macro
737 // invocations. This is relevant for macro hygiene, but possibly not elsewhere.
738 //
739 // One problem here occurs because the types for flat_map_item, flat_map_stmt,
740 // etc., allow the visitor to return *multiple* items; this is a problem for the
741 // nodes here, because they insist on having exactly one piece. One solution
742 // would be to mangle the MutVisitor trait to include one-to-many and
743 // one-to-one versions of these entry points, but that would probably confuse a
744 // lot of people and help very few. Instead, I'm just going to put in dynamic
745 // checks. I think the performance impact of this will be pretty much
746 // nonexistent. The danger is that someone will apply a `MutVisitor` to a
747 // partially expanded node, and will be confused by the fact that their
748 // `flat_map_item` or `flat_map_stmt` isn't getting called on `NtItem` or `NtStmt`
749 // nodes. Hopefully they'll wind up reading this comment, and doing something
750 // appropriate.
751 //
752 // BTW, design choice: I considered just changing the type of, e.g., `NtItem` to
753 // contain multiple items, but decided against it when I looked at
754 // `parse_item_or_view_item` and tried to figure out what I would do with
755 // multiple items there....
756 pub fn visit_interpolated<T: MutVisitor>(nt: &mut token::Nonterminal, vis: &mut T) {
757     match nt {
758         token::NtItem(item) => visit_clobber(item, |item| {
759             // This is probably okay, because the only visitors likely to
760             // peek inside interpolated nodes will be renamings/markings,
761             // which map single items to single items.
762             vis.flat_map_item(item).expect_one("expected visitor to produce exactly one item")
763         }),
764         token::NtBlock(block) => vis.visit_block(block),
765         token::NtStmt(stmt) => visit_clobber(stmt, |stmt| {
766             // See reasoning above.
767             vis.flat_map_stmt(stmt).expect_one("expected visitor to produce exactly one item")
768         }),
769         token::NtPat(pat) => vis.visit_pat(pat),
770         token::NtExpr(expr) => vis.visit_expr(expr),
771         token::NtTy(ty) => vis.visit_ty(ty),
772         token::NtIdent(ident, _is_raw) => vis.visit_ident(ident),
773         token::NtLifetime(ident) => vis.visit_ident(ident),
774         token::NtLiteral(expr) => vis.visit_expr(expr),
775         token::NtMeta(item) => {
776             let AttrItem { path, args, tokens } = item.deref_mut();
777             vis.visit_path(path);
778             visit_mac_args(args, vis);
779             visit_lazy_tts(tokens, vis);
780         }
781         token::NtPath(path) => vis.visit_path(path),
782         token::NtTT(tt) => visit_tt(tt, vis),
783         token::NtVis(visib) => vis.visit_vis(visib),
784     }
785 }
786
787 pub fn noop_visit_asyncness<T: MutVisitor>(asyncness: &mut Async, vis: &mut T) {
788     match asyncness {
789         Async::Yes { span: _, closure_id, return_impl_trait_id } => {
790             vis.visit_id(closure_id);
791             vis.visit_id(return_impl_trait_id);
792         }
793         Async::No => {}
794     }
795 }
796
797 pub fn noop_visit_fn_decl<T: MutVisitor>(decl: &mut P<FnDecl>, vis: &mut T) {
798     let FnDecl { inputs, output } = decl.deref_mut();
799     inputs.flat_map_in_place(|param| vis.flat_map_param(param));
800     noop_visit_fn_ret_ty(output, vis);
801 }
802
803 pub fn noop_visit_fn_ret_ty<T: MutVisitor>(fn_ret_ty: &mut FnRetTy, vis: &mut T) {
804     match fn_ret_ty {
805         FnRetTy::Default(span) => vis.visit_span(span),
806         FnRetTy::Ty(ty) => vis.visit_ty(ty),
807     }
808 }
809
810 pub fn noop_visit_param_bound<T: MutVisitor>(pb: &mut GenericBound, vis: &mut T) {
811     match pb {
812         GenericBound::Trait(ty, _modifier) => vis.visit_poly_trait_ref(ty),
813         GenericBound::Outlives(lifetime) => noop_visit_lifetime(lifetime, vis),
814     }
815 }
816
817 pub fn noop_flat_map_generic_param<T: MutVisitor>(
818     mut param: GenericParam,
819     vis: &mut T,
820 ) -> SmallVec<[GenericParam; 1]> {
821     let GenericParam { id, ident, attrs, bounds, kind, is_placeholder: _ } = &mut param;
822     vis.visit_id(id);
823     vis.visit_ident(ident);
824     visit_thin_attrs(attrs, vis);
825     visit_vec(bounds, |bound| noop_visit_param_bound(bound, vis));
826     match kind {
827         GenericParamKind::Lifetime => {}
828         GenericParamKind::Type { default } => {
829             visit_opt(default, |default| vis.visit_ty(default));
830         }
831         GenericParamKind::Const { ty, kw_span: _, default } => {
832             vis.visit_ty(ty);
833             visit_opt(default, |default| vis.visit_anon_const(default));
834         }
835     }
836     smallvec![param]
837 }
838
839 pub fn noop_visit_label<T: MutVisitor>(Label { ident }: &mut Label, vis: &mut T) {
840     vis.visit_ident(ident);
841 }
842
843 fn noop_visit_lifetime<T: MutVisitor>(Lifetime { id, ident }: &mut Lifetime, vis: &mut T) {
844     vis.visit_id(id);
845     vis.visit_ident(ident);
846 }
847
848 pub fn noop_visit_generics<T: MutVisitor>(generics: &mut Generics, vis: &mut T) {
849     let Generics { params, where_clause, span } = generics;
850     params.flat_map_in_place(|param| vis.flat_map_generic_param(param));
851     vis.visit_where_clause(where_clause);
852     vis.visit_span(span);
853 }
854
855 pub fn noop_visit_where_clause<T: MutVisitor>(wc: &mut WhereClause, vis: &mut T) {
856     let WhereClause { has_where_token: _, predicates, span } = wc;
857     visit_vec(predicates, |predicate| vis.visit_where_predicate(predicate));
858     vis.visit_span(span);
859 }
860
861 pub fn noop_visit_where_predicate<T: MutVisitor>(pred: &mut WherePredicate, vis: &mut T) {
862     match pred {
863         WherePredicate::BoundPredicate(bp) => {
864             let WhereBoundPredicate { span, bound_generic_params, bounded_ty, bounds } = bp;
865             vis.visit_span(span);
866             bound_generic_params.flat_map_in_place(|param| vis.flat_map_generic_param(param));
867             vis.visit_ty(bounded_ty);
868             visit_vec(bounds, |bound| vis.visit_param_bound(bound));
869         }
870         WherePredicate::RegionPredicate(rp) => {
871             let WhereRegionPredicate { span, lifetime, bounds } = rp;
872             vis.visit_span(span);
873             noop_visit_lifetime(lifetime, vis);
874             visit_vec(bounds, |bound| noop_visit_param_bound(bound, vis));
875         }
876         WherePredicate::EqPredicate(ep) => {
877             let WhereEqPredicate { id, span, lhs_ty, rhs_ty } = ep;
878             vis.visit_id(id);
879             vis.visit_span(span);
880             vis.visit_ty(lhs_ty);
881             vis.visit_ty(rhs_ty);
882         }
883     }
884 }
885
886 pub fn noop_visit_variant_data<T: MutVisitor>(vdata: &mut VariantData, vis: &mut T) {
887     match vdata {
888         VariantData::Struct(fields, ..) => {
889             fields.flat_map_in_place(|field| vis.flat_map_field_def(field));
890         }
891         VariantData::Tuple(fields, id) => {
892             fields.flat_map_in_place(|field| vis.flat_map_field_def(field));
893             vis.visit_id(id);
894         }
895         VariantData::Unit(id) => vis.visit_id(id),
896     }
897 }
898
899 pub fn noop_visit_trait_ref<T: MutVisitor>(TraitRef { path, ref_id }: &mut TraitRef, vis: &mut T) {
900     vis.visit_path(path);
901     vis.visit_id(ref_id);
902 }
903
904 pub fn noop_visit_poly_trait_ref<T: MutVisitor>(p: &mut PolyTraitRef, vis: &mut T) {
905     let PolyTraitRef { bound_generic_params, trait_ref, span } = p;
906     bound_generic_params.flat_map_in_place(|param| vis.flat_map_generic_param(param));
907     vis.visit_trait_ref(trait_ref);
908     vis.visit_span(span);
909 }
910
911 pub fn noop_flat_map_field_def<T: MutVisitor>(
912     mut fd: FieldDef,
913     visitor: &mut T,
914 ) -> SmallVec<[FieldDef; 1]> {
915     let FieldDef { span, ident, vis, id, ty, attrs, is_placeholder: _ } = &mut fd;
916     visitor.visit_span(span);
917     visit_opt(ident, |ident| visitor.visit_ident(ident));
918     visitor.visit_vis(vis);
919     visitor.visit_id(id);
920     visitor.visit_ty(ty);
921     visit_thin_attrs(attrs, visitor);
922     smallvec![fd]
923 }
924
925 pub fn noop_flat_map_expr_field<T: MutVisitor>(
926     mut f: ExprField,
927     vis: &mut T,
928 ) -> SmallVec<[ExprField; 1]> {
929     let ExprField { ident, expr, span, is_shorthand: _, attrs, id, is_placeholder: _ } = &mut f;
930     vis.visit_ident(ident);
931     vis.visit_expr(expr);
932     vis.visit_id(id);
933     vis.visit_span(span);
934     visit_thin_attrs(attrs, vis);
935     smallvec![f]
936 }
937
938 pub fn noop_visit_mt<T: MutVisitor>(MutTy { ty, mutbl: _ }: &mut MutTy, vis: &mut T) {
939     vis.visit_ty(ty);
940 }
941
942 pub fn noop_visit_block<T: MutVisitor>(block: &mut P<Block>, vis: &mut T) {
943     let Block { id, stmts, rules: _, span, tokens } = block.deref_mut();
944     vis.visit_id(id);
945     stmts.flat_map_in_place(|stmt| vis.flat_map_stmt(stmt));
946     vis.visit_span(span);
947     visit_lazy_tts(tokens, vis);
948 }
949
950 pub fn noop_visit_item_kind<T: MutVisitor>(kind: &mut ItemKind, vis: &mut T) {
951     match kind {
952         ItemKind::ExternCrate(_orig_name) => {}
953         ItemKind::Use(use_tree) => vis.visit_use_tree(use_tree),
954         ItemKind::Static(ty, _, expr) | ItemKind::Const(_, ty, expr) => {
955             vis.visit_ty(ty);
956             visit_opt(expr, |expr| vis.visit_expr(expr));
957         }
958         ItemKind::Fn(box FnKind(_, sig, generics, body)) => {
959             visit_fn_sig(sig, vis);
960             vis.visit_generics(generics);
961             visit_opt(body, |body| vis.visit_block(body));
962         }
963         ItemKind::Mod(_unsafety, mod_kind) => match mod_kind {
964             ModKind::Loaded(items, _inline, inner_span) => {
965                 vis.visit_span(inner_span);
966                 items.flat_map_in_place(|item| vis.flat_map_item(item));
967             }
968             ModKind::Unloaded => {}
969         },
970         ItemKind::ForeignMod(nm) => vis.visit_foreign_mod(nm),
971         ItemKind::GlobalAsm(asm) => noop_visit_inline_asm(asm, vis),
972         ItemKind::TyAlias(box TyAliasKind(_, generics, bounds, ty)) => {
973             vis.visit_generics(generics);
974             visit_bounds(bounds, vis);
975             visit_opt(ty, |ty| vis.visit_ty(ty));
976         }
977         ItemKind::Enum(EnumDef { variants }, generics) => {
978             variants.flat_map_in_place(|variant| vis.flat_map_variant(variant));
979             vis.visit_generics(generics);
980         }
981         ItemKind::Struct(variant_data, generics) | ItemKind::Union(variant_data, generics) => {
982             vis.visit_variant_data(variant_data);
983             vis.visit_generics(generics);
984         }
985         ItemKind::Impl(box ImplKind {
986             unsafety: _,
987             polarity: _,
988             defaultness: _,
989             constness: _,
990             generics,
991             of_trait,
992             self_ty,
993             items,
994         }) => {
995             vis.visit_generics(generics);
996             visit_opt(of_trait, |trait_ref| vis.visit_trait_ref(trait_ref));
997             vis.visit_ty(self_ty);
998             items.flat_map_in_place(|item| vis.flat_map_impl_item(item));
999         }
1000         ItemKind::Trait(box TraitKind(.., generics, bounds, items)) => {
1001             vis.visit_generics(generics);
1002             visit_bounds(bounds, vis);
1003             items.flat_map_in_place(|item| vis.flat_map_trait_item(item));
1004         }
1005         ItemKind::TraitAlias(generics, bounds) => {
1006             vis.visit_generics(generics);
1007             visit_bounds(bounds, vis);
1008         }
1009         ItemKind::MacCall(m) => vis.visit_mac_call(m),
1010         ItemKind::MacroDef(def) => vis.visit_macro_def(def),
1011     }
1012 }
1013
1014 pub fn noop_flat_map_assoc_item<T: MutVisitor>(
1015     mut item: P<AssocItem>,
1016     visitor: &mut T,
1017 ) -> SmallVec<[P<AssocItem>; 1]> {
1018     let Item { id, ident, vis, attrs, kind, span, tokens } = item.deref_mut();
1019     visitor.visit_id(id);
1020     visitor.visit_ident(ident);
1021     visitor.visit_vis(vis);
1022     visit_attrs(attrs, visitor);
1023     match kind {
1024         AssocItemKind::Const(_, ty, expr) => {
1025             visitor.visit_ty(ty);
1026             visit_opt(expr, |expr| visitor.visit_expr(expr));
1027         }
1028         AssocItemKind::Fn(box FnKind(_, sig, generics, body)) => {
1029             visitor.visit_generics(generics);
1030             visit_fn_sig(sig, visitor);
1031             visit_opt(body, |body| visitor.visit_block(body));
1032         }
1033         AssocItemKind::TyAlias(box TyAliasKind(_, generics, bounds, ty)) => {
1034             visitor.visit_generics(generics);
1035             visit_bounds(bounds, visitor);
1036             visit_opt(ty, |ty| visitor.visit_ty(ty));
1037         }
1038         AssocItemKind::MacCall(mac) => visitor.visit_mac_call(mac),
1039     }
1040     visitor.visit_span(span);
1041     visit_lazy_tts(tokens, visitor);
1042     smallvec![item]
1043 }
1044
1045 pub fn noop_visit_fn_header<T: MutVisitor>(header: &mut FnHeader, vis: &mut T) {
1046     let FnHeader { unsafety: _, asyncness, constness: _, ext: _ } = header;
1047     vis.visit_asyncness(asyncness);
1048 }
1049
1050 // FIXME: Avoid visiting the crate as a `Mod` item, flat map only the inner items if possible,
1051 // or make crate visiting first class if necessary.
1052 pub fn noop_visit_crate<T: MutVisitor>(krate: &mut Crate, vis: &mut T) {
1053     visit_clobber(krate, |Crate { attrs, items, span, proc_macros }| {
1054         let item_vis =
1055             Visibility { kind: VisibilityKind::Public, span: span.shrink_to_lo(), tokens: None };
1056         let item = P(Item {
1057             ident: Ident::invalid(),
1058             attrs,
1059             id: DUMMY_NODE_ID,
1060             vis: item_vis,
1061             span,
1062             kind: ItemKind::Mod(Unsafe::No, ModKind::Loaded(items, Inline::Yes, span)),
1063             tokens: None,
1064         });
1065         let items = vis.flat_map_item(item);
1066
1067         let len = items.len();
1068         if len == 0 {
1069             Crate { attrs: vec![], items: vec![], span, proc_macros }
1070         } else if len == 1 {
1071             let Item { attrs, span, kind, .. } = items.into_iter().next().unwrap().into_inner();
1072             match kind {
1073                 ItemKind::Mod(_, ModKind::Loaded(items, ..)) => {
1074                     Crate { attrs, items, span, proc_macros }
1075                 }
1076                 _ => panic!("visitor converted a module to not a module"),
1077             }
1078         } else {
1079             panic!("a crate cannot expand to more than one item");
1080         }
1081     });
1082 }
1083
1084 // Mutates one item into possibly many items.
1085 pub fn noop_flat_map_item<T: MutVisitor>(
1086     mut item: P<Item>,
1087     visitor: &mut T,
1088 ) -> SmallVec<[P<Item>; 1]> {
1089     let Item { ident, attrs, id, kind, vis, span, tokens } = item.deref_mut();
1090     visitor.visit_ident(ident);
1091     visit_attrs(attrs, visitor);
1092     visitor.visit_id(id);
1093     visitor.visit_item_kind(kind);
1094     visitor.visit_vis(vis);
1095     visitor.visit_span(span);
1096     visit_lazy_tts(tokens, visitor);
1097
1098     smallvec![item]
1099 }
1100
1101 pub fn noop_flat_map_foreign_item<T: MutVisitor>(
1102     mut item: P<ForeignItem>,
1103     visitor: &mut T,
1104 ) -> SmallVec<[P<ForeignItem>; 1]> {
1105     let Item { ident, attrs, id, kind, vis, span, tokens } = item.deref_mut();
1106     visitor.visit_id(id);
1107     visitor.visit_ident(ident);
1108     visitor.visit_vis(vis);
1109     visit_attrs(attrs, visitor);
1110     match kind {
1111         ForeignItemKind::Static(ty, _, expr) => {
1112             visitor.visit_ty(ty);
1113             visit_opt(expr, |expr| visitor.visit_expr(expr));
1114         }
1115         ForeignItemKind::Fn(box FnKind(_, sig, generics, body)) => {
1116             visitor.visit_generics(generics);
1117             visit_fn_sig(sig, visitor);
1118             visit_opt(body, |body| visitor.visit_block(body));
1119         }
1120         ForeignItemKind::TyAlias(box TyAliasKind(_, generics, bounds, ty)) => {
1121             visitor.visit_generics(generics);
1122             visit_bounds(bounds, visitor);
1123             visit_opt(ty, |ty| visitor.visit_ty(ty));
1124         }
1125         ForeignItemKind::MacCall(mac) => visitor.visit_mac_call(mac),
1126     }
1127     visitor.visit_span(span);
1128     visit_lazy_tts(tokens, visitor);
1129     smallvec![item]
1130 }
1131
1132 pub fn noop_visit_pat<T: MutVisitor>(pat: &mut P<Pat>, vis: &mut T) {
1133     let Pat { id, kind, span, tokens } = pat.deref_mut();
1134     vis.visit_id(id);
1135     match kind {
1136         PatKind::Wild | PatKind::Rest => {}
1137         PatKind::Ident(_binding_mode, ident, sub) => {
1138             vis.visit_ident(ident);
1139             visit_opt(sub, |sub| vis.visit_pat(sub));
1140         }
1141         PatKind::Lit(e) => vis.visit_expr(e),
1142         PatKind::TupleStruct(qself, path, elems) => {
1143             vis.visit_qself(qself);
1144             vis.visit_path(path);
1145             visit_vec(elems, |elem| vis.visit_pat(elem));
1146         }
1147         PatKind::Path(qself, path) => {
1148             vis.visit_qself(qself);
1149             vis.visit_path(path);
1150         }
1151         PatKind::Struct(qself, path, fields, _etc) => {
1152             vis.visit_qself(qself);
1153             vis.visit_path(path);
1154             fields.flat_map_in_place(|field| vis.flat_map_pat_field(field));
1155         }
1156         PatKind::Box(inner) => vis.visit_pat(inner),
1157         PatKind::Ref(inner, _mutbl) => vis.visit_pat(inner),
1158         PatKind::Range(e1, e2, Spanned { span: _, node: _ }) => {
1159             visit_opt(e1, |e| vis.visit_expr(e));
1160             visit_opt(e2, |e| vis.visit_expr(e));
1161             vis.visit_span(span);
1162         }
1163         PatKind::Tuple(elems) | PatKind::Slice(elems) | PatKind::Or(elems) => {
1164             visit_vec(elems, |elem| vis.visit_pat(elem))
1165         }
1166         PatKind::Paren(inner) => vis.visit_pat(inner),
1167         PatKind::MacCall(mac) => vis.visit_mac_call(mac),
1168     }
1169     vis.visit_span(span);
1170     visit_lazy_tts(tokens, vis);
1171 }
1172
1173 pub fn noop_visit_anon_const<T: MutVisitor>(AnonConst { id, value }: &mut AnonConst, vis: &mut T) {
1174     vis.visit_id(id);
1175     vis.visit_expr(value);
1176 }
1177
1178 fn noop_visit_inline_asm<T: MutVisitor>(asm: &mut InlineAsm, vis: &mut T) {
1179     for (op, _) in &mut asm.operands {
1180         match op {
1181             InlineAsmOperand::In { expr, .. }
1182             | InlineAsmOperand::InOut { expr, .. }
1183             | InlineAsmOperand::Sym { expr, .. } => vis.visit_expr(expr),
1184             InlineAsmOperand::Out { expr, .. } => {
1185                 if let Some(expr) = expr {
1186                     vis.visit_expr(expr);
1187                 }
1188             }
1189             InlineAsmOperand::SplitInOut { in_expr, out_expr, .. } => {
1190                 vis.visit_expr(in_expr);
1191                 if let Some(out_expr) = out_expr {
1192                     vis.visit_expr(out_expr);
1193                 }
1194             }
1195             InlineAsmOperand::Const { anon_const, .. } => vis.visit_anon_const(anon_const),
1196         }
1197     }
1198 }
1199
1200 pub fn noop_visit_expr<T: MutVisitor>(
1201     Expr { kind, id, span, attrs, tokens }: &mut Expr,
1202     vis: &mut T,
1203 ) {
1204     match kind {
1205         ExprKind::Box(expr) => vis.visit_expr(expr),
1206         ExprKind::Array(exprs) => visit_exprs(exprs, vis),
1207         ExprKind::ConstBlock(anon_const) => {
1208             vis.visit_anon_const(anon_const);
1209         }
1210         ExprKind::Repeat(expr, count) => {
1211             vis.visit_expr(expr);
1212             vis.visit_anon_const(count);
1213         }
1214         ExprKind::Tup(exprs) => visit_exprs(exprs, vis),
1215         ExprKind::Call(f, args) => {
1216             vis.visit_expr(f);
1217             visit_exprs(args, vis);
1218         }
1219         ExprKind::MethodCall(PathSegment { ident, id, args }, exprs, span) => {
1220             vis.visit_ident(ident);
1221             vis.visit_id(id);
1222             visit_opt(args, |args| vis.visit_generic_args(args));
1223             visit_exprs(exprs, vis);
1224             vis.visit_span(span);
1225         }
1226         ExprKind::Binary(_binop, lhs, rhs) => {
1227             vis.visit_expr(lhs);
1228             vis.visit_expr(rhs);
1229         }
1230         ExprKind::Unary(_unop, ohs) => vis.visit_expr(ohs),
1231         ExprKind::Cast(expr, ty) => {
1232             vis.visit_expr(expr);
1233             vis.visit_ty(ty);
1234         }
1235         ExprKind::Type(expr, ty) => {
1236             vis.visit_expr(expr);
1237             vis.visit_ty(ty);
1238         }
1239         ExprKind::AddrOf(_, _, ohs) => vis.visit_expr(ohs),
1240         ExprKind::Let(pat, scrutinee) => {
1241             vis.visit_pat(pat);
1242             vis.visit_expr(scrutinee);
1243         }
1244         ExprKind::If(cond, tr, fl) => {
1245             vis.visit_expr(cond);
1246             vis.visit_block(tr);
1247             visit_opt(fl, |fl| vis.visit_expr(fl));
1248         }
1249         ExprKind::While(cond, body, label) => {
1250             vis.visit_expr(cond);
1251             vis.visit_block(body);
1252             visit_opt(label, |label| vis.visit_label(label));
1253         }
1254         ExprKind::ForLoop(pat, iter, body, label) => {
1255             vis.visit_pat(pat);
1256             vis.visit_expr(iter);
1257             vis.visit_block(body);
1258             visit_opt(label, |label| vis.visit_label(label));
1259         }
1260         ExprKind::Loop(body, label) => {
1261             vis.visit_block(body);
1262             visit_opt(label, |label| vis.visit_label(label));
1263         }
1264         ExprKind::Match(expr, arms) => {
1265             vis.visit_expr(expr);
1266             arms.flat_map_in_place(|arm| vis.flat_map_arm(arm));
1267         }
1268         ExprKind::Closure(_capture_by, asyncness, _movability, decl, body, span) => {
1269             vis.visit_asyncness(asyncness);
1270             vis.visit_fn_decl(decl);
1271             vis.visit_expr(body);
1272             vis.visit_span(span);
1273         }
1274         ExprKind::Block(blk, label) => {
1275             vis.visit_block(blk);
1276             visit_opt(label, |label| vis.visit_label(label));
1277         }
1278         ExprKind::Async(_capture_by, node_id, body) => {
1279             vis.visit_id(node_id);
1280             vis.visit_block(body);
1281         }
1282         ExprKind::Await(expr) => vis.visit_expr(expr),
1283         ExprKind::Assign(el, er, _) => {
1284             vis.visit_expr(el);
1285             vis.visit_expr(er);
1286         }
1287         ExprKind::AssignOp(_op, el, er) => {
1288             vis.visit_expr(el);
1289             vis.visit_expr(er);
1290         }
1291         ExprKind::Field(el, ident) => {
1292             vis.visit_expr(el);
1293             vis.visit_ident(ident);
1294         }
1295         ExprKind::Index(el, er) => {
1296             vis.visit_expr(el);
1297             vis.visit_expr(er);
1298         }
1299         ExprKind::Range(e1, e2, _lim) => {
1300             visit_opt(e1, |e1| vis.visit_expr(e1));
1301             visit_opt(e2, |e2| vis.visit_expr(e2));
1302         }
1303         ExprKind::Underscore => {}
1304         ExprKind::Path(qself, path) => {
1305             vis.visit_qself(qself);
1306             vis.visit_path(path);
1307         }
1308         ExprKind::Break(label, expr) => {
1309             visit_opt(label, |label| vis.visit_label(label));
1310             visit_opt(expr, |expr| vis.visit_expr(expr));
1311         }
1312         ExprKind::Continue(label) => {
1313             visit_opt(label, |label| vis.visit_label(label));
1314         }
1315         ExprKind::Ret(expr) => {
1316             visit_opt(expr, |expr| vis.visit_expr(expr));
1317         }
1318         ExprKind::InlineAsm(asm) => noop_visit_inline_asm(asm, vis),
1319         ExprKind::LlvmInlineAsm(asm) => {
1320             let LlvmInlineAsm {
1321                 asm: _,
1322                 asm_str_style: _,
1323                 outputs,
1324                 inputs,
1325                 clobbers: _,
1326                 volatile: _,
1327                 alignstack: _,
1328                 dialect: _,
1329             } = asm.deref_mut();
1330             for out in outputs {
1331                 let LlvmInlineAsmOutput { constraint: _, expr, is_rw: _, is_indirect: _ } = out;
1332                 vis.visit_expr(expr);
1333             }
1334             visit_vec(inputs, |(_c, expr)| vis.visit_expr(expr));
1335         }
1336         ExprKind::MacCall(mac) => vis.visit_mac_call(mac),
1337         ExprKind::Struct(se) => {
1338             let StructExpr { qself, path, fields, rest } = se.deref_mut();
1339             vis.visit_qself(qself);
1340             vis.visit_path(path);
1341             fields.flat_map_in_place(|field| vis.flat_map_expr_field(field));
1342             match rest {
1343                 StructRest::Base(expr) => vis.visit_expr(expr),
1344                 StructRest::Rest(_span) => {}
1345                 StructRest::None => {}
1346             }
1347         }
1348         ExprKind::Paren(expr) => {
1349             vis.visit_expr(expr);
1350         }
1351         ExprKind::Yield(expr) => {
1352             visit_opt(expr, |expr| vis.visit_expr(expr));
1353         }
1354         ExprKind::Try(expr) => vis.visit_expr(expr),
1355         ExprKind::TryBlock(body) => vis.visit_block(body),
1356         ExprKind::Lit(_) | ExprKind::Err => {}
1357     }
1358     vis.visit_id(id);
1359     vis.visit_span(span);
1360     visit_thin_attrs(attrs, vis);
1361     visit_lazy_tts(tokens, vis);
1362 }
1363
1364 pub fn noop_filter_map_expr<T: MutVisitor>(mut e: P<Expr>, vis: &mut T) -> Option<P<Expr>> {
1365     Some({
1366         vis.visit_expr(&mut e);
1367         e
1368     })
1369 }
1370
1371 pub fn noop_flat_map_stmt<T: MutVisitor>(
1372     Stmt { kind, mut span, mut id }: Stmt,
1373     vis: &mut T,
1374 ) -> SmallVec<[Stmt; 1]> {
1375     vis.visit_id(&mut id);
1376     vis.visit_span(&mut span);
1377     noop_flat_map_stmt_kind(kind, vis).into_iter().map(|kind| Stmt { id, kind, span }).collect()
1378 }
1379
1380 pub fn noop_flat_map_stmt_kind<T: MutVisitor>(
1381     kind: StmtKind,
1382     vis: &mut T,
1383 ) -> SmallVec<[StmtKind; 1]> {
1384     match kind {
1385         StmtKind::Local(mut local) => smallvec![StmtKind::Local({
1386             vis.visit_local(&mut local);
1387             local
1388         })],
1389         StmtKind::Item(item) => vis.flat_map_item(item).into_iter().map(StmtKind::Item).collect(),
1390         StmtKind::Expr(expr) => vis.filter_map_expr(expr).into_iter().map(StmtKind::Expr).collect(),
1391         StmtKind::Semi(expr) => vis.filter_map_expr(expr).into_iter().map(StmtKind::Semi).collect(),
1392         StmtKind::Empty => smallvec![StmtKind::Empty],
1393         StmtKind::MacCall(mut mac) => {
1394             let MacCallStmt { mac: mac_, style: _, attrs, tokens } = mac.deref_mut();
1395             vis.visit_mac_call(mac_);
1396             visit_thin_attrs(attrs, vis);
1397             visit_lazy_tts(tokens, vis);
1398             smallvec![StmtKind::MacCall(mac)]
1399         }
1400     }
1401 }
1402
1403 pub fn noop_visit_vis<T: MutVisitor>(visibility: &mut Visibility, vis: &mut T) {
1404     match &mut visibility.kind {
1405         VisibilityKind::Public | VisibilityKind::Crate(_) | VisibilityKind::Inherited => {}
1406         VisibilityKind::Restricted { path, id } => {
1407             vis.visit_path(path);
1408             vis.visit_id(id);
1409         }
1410     }
1411     vis.visit_span(&mut visibility.span);
1412 }