Skip to content

Commit 9462322

Browse files
committed
Fix scip symbols for implicit modules
1 parent a329b6a commit 9462322

File tree

2 files changed

+30
-13
lines changed

2 files changed

+30
-13
lines changed

crates/ide/src/static_index.rs

+26-11
Original file line numberDiff line numberDiff line change
@@ -169,10 +169,10 @@ impl StaticIndex<'_> {
169169
.unwrap();
170170
// hovers
171171
let sema = hir::Semantics::new(self.db);
172-
let tokens_or_nodes = sema.parse_guess_edition(file_id).syntax().clone();
172+
let root = sema.parse_guess_edition(file_id).syntax().clone();
173173
let edition =
174174
sema.attach_first_edition(file_id).map(|it| it.edition()).unwrap_or(Edition::CURRENT);
175-
let tokens = tokens_or_nodes.descendants_with_tokens().filter_map(|it| match it {
175+
let tokens = root.descendants_with_tokens().filter_map(|it| match it {
176176
syntax::NodeOrToken::Node(_) => None,
177177
syntax::NodeOrToken::Token(it) => Some(it),
178178
});
@@ -194,24 +194,19 @@ impl StaticIndex<'_> {
194194
)
195195
});
196196
let mut result = StaticIndexedFile { file_id, inlay_hints, folds, tokens: vec![] };
197-
for token in tokens {
198-
let range = token.text_range();
199-
let node = token.parent().unwrap();
200-
let def = match get_definition(&sema, token.clone()) {
201-
Some(it) => it,
202-
None => continue,
203-
};
197+
198+
let mut add_token = |def: Definition, range: TextRange, scope_node: &SyntaxNode| {
204199
let id = if let Some(it) = self.def_map.get(&def) {
205200
*it
206201
} else {
207202
let it = self.tokens.insert(TokenStaticData {
208-
documentation: documentation_for_definition(&sema, def, &node),
203+
documentation: documentation_for_definition(&sema, def, scope_node),
209204
hover: Some(hover_for_definition(
210205
&sema,
211206
file_id,
212207
def,
213208
None,
214-
&node,
209+
scope_node,
215210
None,
216211
false,
217212
&hover_config,
@@ -240,6 +235,22 @@ impl StaticIndex<'_> {
240235
},
241236
});
242237
result.tokens.push((range, id));
238+
};
239+
240+
if let Some(module) = sema.file_to_module_def(file_id) {
241+
let def = Definition::Module(module);
242+
let range = root.text_range();
243+
add_token(def, range, &root);
244+
}
245+
246+
for token in tokens {
247+
let range = token.text_range();
248+
let node = token.parent().unwrap();
249+
let def = match get_definition(&sema, token.clone()) {
250+
Some(it) => it,
251+
None => continue,
252+
};
253+
add_token(def, range, &node);
243254
}
244255
self.files.push(result);
245256
}
@@ -300,6 +311,10 @@ mod tests {
300311
let mut range_set: FxHashSet<_> = ranges.iter().map(|it| it.0).collect();
301312
for f in s.files {
302313
for (range, _) in f.tokens {
314+
if range.start() == TextSize::from(0) {
315+
// ignore whole file range corresponding to module definition
316+
continue;
317+
}
303318
let it = FileRange { file_id: f.file_id, range };
304319
if !range_set.contains(&it) {
305320
panic!("additional range {it:?}");

crates/rust-analyzer/src/cli/scip.rs

+4-2
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,9 @@ mod test {
549549
continue;
550550
}
551551
for &(range, id) in &file.tokens {
552-
if range.contains(offset - TextSize::from(1)) {
552+
// check if cursor is within token, ignoring token for the module defined by the file (whose range is the whole file)
553+
if range.start() != TextSize::from(0) && range.contains(offset - TextSize::from(1))
554+
{
553555
let token = si.tokens.get(id).unwrap();
554556
found_symbol = match token.moniker.as_ref() {
555557
None => None,
@@ -885,7 +887,7 @@ pub mod example_mod {
885887
);
886888

887889
let file = si.files.first().unwrap();
888-
let (_, token_id) = file.tokens.first().unwrap();
890+
let (_, token_id) = file.tokens.get(1).unwrap(); // first token is file module, second is `bar`
889891
let token = si.tokens.get(*token_id).unwrap();
890892

891893
assert_eq!(token.documentation.as_ref().map(|d| d.as_str()), Some("foo"));

0 commit comments

Comments
 (0)