fold_file wild attribute

Summary: Fold over wild attributes Reviewed By: perehonchuk Differential Revision: D49821521 fbshipit-source-id: 3a83ab84f272352838feb1c096950b80f6e2806c
WhatsApp · Oct 3, 2023 · bd5f4e8 · bd5f4e8
1 parent 08c811e
commit bd5f4e8
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 2 deletions.
diff --git a/crates/hir/src/fold.rs b/crates/hir/src/fold.rs
@@ -80,8 +80,10 @@ pub fn fold_file<'a, T>(
                     callback(acc, ctx)
                 })
             }
-            FormIdx::Attribute(_attribute_id) => {
-                todo!()
+            FormIdx::Attribute(attribute_id) => {
+                sema.fold_attribute(InFile::new(file_id, attribute_id), r, &mut |acc, ctx| {
+                    callback(acc, ctx)
+                })
             }
             FormIdx::CompileOption(_attribute_id) => {
                 todo!()
@@ -1578,6 +1580,13 @@ bar() ->
                     acc
                 }
             }
+            AnyExpr::Term(Term::Literal(Literal::Atom(atom))) => {
+                if atom == hir_atom {
+                    acc + 1
+                } else {
+                    acc
+                }
+            }
             _ => acc,
         });
 
@@ -1638,4 +1647,13 @@ bar() ->
         // Note: fold does not look into field names
         count_atom_foo(fixture_str, 1);
     }
+
+    #[test]
+    fn traverse_attribute() {
+        let fixture_str = r#"
+               -module(foo).
+               -wild(r1, {f1, f~oo}).
+               "#;
+        count_atom_foo(fixture_str, 1);
+    }
 }
diff --git a/crates/hir/src/sema.rs b/crates/hir/src/sema.rs
@@ -48,6 +48,7 @@ pub use crate::intern::MinInternDatabaseStorage;
 use crate::resolver::Resolution;
 use crate::resolver::Resolver;
 use crate::AnyExprId;
+use crate::AttributeId;
 use crate::Body;
 use crate::BodySourceMap;
 use crate::CRClause;
@@ -637,6 +638,23 @@ impl<'db> Semantic<'db> {
         })
     }
 
+    pub fn fold_attribute<'a, T>(
+        &self,
+        attribute_id: InFile<AttributeId>,
+        initial: T,
+        callback: AnyCallBack<'a, T>,
+    ) -> T {
+        let body = self.db.attribute_body(attribute_id);
+        FoldCtx::fold_term(
+            &body.body,
+            Strategy::TopDown,
+            FormIdx::Attribute(attribute_id.value),
+            body.value,
+            initial,
+            callback,
+        )
+    }
+
     pub fn bound_vars_in_pattern_diagnostic(
         &self,
         file_id: FileId,