Add Fold and VisitMut methods for Vec<Attribute>

dtolnay · Oct 20, 2024 · 7681951 · 7681951
1 parent 4c7f82e
commit 7681951
Show file tree

Hide file tree

Showing 4 changed files with 235 additions and 383 deletions.
diff --git a/codegen/src/fold.rs b/codegen/src/fold.rs
@@ -32,10 +32,16 @@ fn visit(
         }
         Type::Vec(t) => {
             let Type::Syn(t) = &**t else { unimplemented!() };
-            let method = method_name(t);
-            Some(quote! {
-                fold_vec(#name, f, F::#method)
-            })
+            if t == "Attribute" {
+                Some(quote! {
+                    f.fold_attributes(#name)
+                })
+            } else {
+                let method = method_name(t);
+                Some(quote! {
+                    fold_vec(#name, f, F::#method)
+                })
+            }
         }
         Type::Punctuated(p) => {
             let t = &*p.element;
@@ -214,6 +220,14 @@ fn node(traits: &mut TokenStream, impls: &mut TokenStream, s: &Node, defs: &Defi
             #fold_impl
         }
     });
+
+    if s.ident == "Attribute" {
+        traits.extend(quote! {
+            fn fold_attributes(&mut self, i: Vec<crate::Attribute>) -> Vec<crate::Attribute> {
+                fold_vec(i, self, Self::fold_attribute)
+            }
+        });
+    }
 }
 
 pub fn generate(defs: &Definitions) -> Result<()> {

diff --git a/codegen/src/visit_mut.rs b/codegen/src/visit_mut.rs
@@ -36,14 +36,20 @@ fn visit(
             visit(t, features, defs, &Owned(quote!(*#name)))
         }
         Type::Vec(t) => {
-            let operand = Borrowed(quote!(it));
-            let val = visit(t, features, defs, &operand)?;
             let name = name.ref_mut_tokens();
-            Some(quote! {
-                for it in #name {
-                    #val;
-                }
-            })
+            if matches!(&**t, Type::Syn(t) if t == "Attribute") {
+                Some(quote! {
+                    v.visit_attributes_mut(#name);
+                })
+            } else {
+                let operand = Borrowed(quote!(it));
+                let val = visit(t, features, defs, &operand)?;
+                Some(quote! {
+                    for it in #name {
+                        #val;
+                    }
+                })
+            }
         }
         Type::Punctuated(p) => {
             let operand = Borrowed(quote!(it));
@@ -193,6 +199,16 @@ fn node(traits: &mut TokenStream, impls: &mut TokenStream, s: &Node, defs: &Defi
             #visit_mut_impl
         }
     });
+
+    if s.ident == "Attribute" {
+        traits.extend(quote! {
+            fn visit_attributes_mut(&mut self, i: &mut Vec<crate::Attribute>) {
+                for attr in i {
+                    self.visit_attribute_mut(attr);
+                }
+            }
+        });
+    }
 }
 
 pub fn generate(defs: &Definitions) -> Result<()> {