feat(rust): Scope identifiers

alexpovel · Aug 16, 2024 · 007f0be · 007f0be
1 parent d51290c
commit 007f0be
Show file tree

Hide file tree

Showing 4 changed files with 688 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1547,6 +1547,7 @@ Language scopes:
           - mod:              `mod` blocks
           - mod-tests:        `mod tests` blocks
           - type-def:         Type definitions (`struct`, `enum`, `union`)
+          - identifier:       Identifiers
 
       --rust-query <TREE-SITTER-QUERY>
           Scope Rust code using a custom tree-sitter query.

diff --git a/src/scoping/langs/rust.rs b/src/scoping/langs/rust.rs
@@ -88,6 +88,8 @@ pub enum PreparedRustQuery {
     ModTests,
     /// Type definitions (`struct`, `enum`, `union`).
     TypeDef,
+    /// Identifiers.
+    Identifier,
 }
 
 impl From<PreparedRustQuery> for TSQuery {
@@ -293,6 +295,7 @@ impl From<PreparedRustQuery> for TSQuery {
                     @typedef
                     "
                 }
+                PreparedRustQuery::Identifier => "(identifier) @identifier",
             },
         )
         .expect("Prepared queries to be valid")

diff --git a/tests/langs/mod.rs b/tests/langs/mod.rs
@@ -357,6 +357,11 @@ impl InScopeLinePart {
     include_str!("rust/base.rs"),
     Rust::new(CodeQuery::Prepared(PreparedRustQuery::TypeDef)),
 )]
+#[case(
+    "base.rs_identifier",
+    include_str!("rust/base.rs"),
+    Rust::new(CodeQuery::Prepared(PreparedRustQuery::Identifier)),
+)]
 #[case(
     "base.tf_variable-block",
     include_str!("hcl/base.tf"),