Allow searching with prefix

jkelleyrtp · jkelleyrtp · commit 69e9d75678d8 · 2024-05-22T13:47:05.000-07:00
diff --git a/src/tools/rust-analyzer/crates/ide-db/src/symbol_index.rs b/src/tools/rust-analyzer/crates/ide-db/src/symbol_index.rs
@@ -381,7 +381,7 @@ impl Query {
                     if non_type_for_type_only_query || !self.matches_assoc_mode(symbol.is_assoc) {
                         continue;
                     }
-                    if !self.include_hidden && symbol.name.starts_with("__") {
+                    if self.should_hide_query(&symbol) {
                         continue;
                     }
                     if self.mode.check(&self.query, self.case_sensitive, &symbol.name) {
@@ -392,6 +392,11 @@ impl Query {
         }
     }
 
+    fn should_hide_query(&self, symbol: &FileSymbol) -> bool {
+        // Hide symbols that start with `__` unless the query starts with `__`
+        !self.include_hidden && symbol.name.starts_with("__") && !self.query.starts_with("__")
+    }
+
     fn matches_assoc_mode(&self, is_trait_assoc_item: bool) -> bool {
         !matches!(
             (is_trait_assoc_item, self.assoc_mode),

Original file line number	Diff line number	Diff line change
`@@ -381,7 +381,7 @@ impl Query {`
`381`	`381`	`if non_type_for_type_only_query \|\| !self.matches_assoc_mode(symbol.is_assoc) {`
`382`	`382`	`continue;`
`383`	`383`	`}`
`384`		`- if !self.include_hidden && symbol.name.starts_with("__") {`
	`384`	`+ if self.should_hide_query(&symbol) {`
`385`	`385`	`continue;`
`386`	`386`	`}`
`387`	`387`	`if self.mode.check(&self.query, self.case_sensitive, &symbol.name) {`
`@@ -392,6 +392,11 @@ impl Query {`
`392`	`392`	`}`
`393`	`393`	`}`
`394`	`394`
	`395`	`+ fn should_hide_query(&self, symbol: &FileSymbol) -> bool {`
	`396`	+ // Hide symbols that start with `__` unless the query starts with `__`
	`397`	`+ !self.include_hidden && symbol.name.starts_with("__") && !self.query.starts_with("__")`
	`398`	`+ }`
	`399`	`+`
`395`	`400`	`fn matches_assoc_mode(&self, is_trait_assoc_item: bool) -> bool {`
`396`	`401`	`!matches!(`
`397`	`402`	`(is_trait_assoc_item, self.assoc_mode),`