feat: add identifier extractos for annotations

jpedroh · Sep 13, 2024 · dfc4bb4 · dfc4bb4
1 parent 2ecb58c
commit dfc4bb4
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/parsing/src/tree_sitter_parser.rs b/parsing/src/tree_sitter_parser.rs
@@ -99,6 +99,22 @@ impl From<Language> for ParserConfiguration {
                             tree_sitter_java::language(),
                         )),
                     );
+
+                    map.insert(
+                        "marker_annotation",
+                        Box::new(TreeSitterQuery::new(
+                            r#"(marker_annotation name: _ @name)"#,
+                            tree_sitter_java::language(),
+                        )),
+                    );
+
+                    map.insert(
+                        "annotation",
+                        Box::new(TreeSitterQuery::new(
+                            r#"(annotation name: _ @name)"#,
+                            tree_sitter_java::language(),
+                        )),
+                    );
                     map
                 },
             },