c3a4c4429d
This introduces the new type -- Semantics. Semantics maps SyntaxNodes to various semantic info, such as type, name resolution or macro expansions. To do so, Semantics maintains a HashMap which maps every node it saw to the file from which the node originated. This is enough to get all the necessary hir bits just from syntax.
65 lines
2.2 KiB
Rust
65 lines
2.2 KiB
Rust
//! This module contains an import search funcionality that is provided to the ra_assists module.
|
|
//! Later, this should be moved away to a separate crate that is accessible from the ra_assists module.
|
|
|
|
use hir::{ModuleDef, Semantics};
|
|
use ra_prof::profile;
|
|
use ra_syntax::{ast, AstNode, SyntaxKind::NAME};
|
|
|
|
use crate::{
|
|
defs::{classify_name, NameDefinition},
|
|
symbol_index::{self, FileSymbol, Query},
|
|
RootDatabase,
|
|
};
|
|
|
|
pub struct ImportsLocator<'a> {
|
|
sema: Semantics<'a, RootDatabase>,
|
|
}
|
|
|
|
impl<'a> ImportsLocator<'a> {
|
|
pub fn new(db: &'a RootDatabase) -> Self {
|
|
Self { sema: Semantics::new(db) }
|
|
}
|
|
|
|
pub fn find_imports(&mut self, name_to_import: &str) -> Vec<ModuleDef> {
|
|
let _p = profile("search_for_imports");
|
|
let db = self.sema.db;
|
|
|
|
let project_results = {
|
|
let mut query = Query::new(name_to_import.to_string());
|
|
query.exact();
|
|
query.limit(40);
|
|
symbol_index::world_symbols(db, query)
|
|
};
|
|
let lib_results = {
|
|
let mut query = Query::new(name_to_import.to_string());
|
|
query.libs();
|
|
query.exact();
|
|
query.limit(40);
|
|
symbol_index::world_symbols(db, query)
|
|
};
|
|
|
|
project_results
|
|
.into_iter()
|
|
.chain(lib_results.into_iter())
|
|
.filter_map(|import_candidate| self.get_name_definition(&import_candidate))
|
|
.filter_map(|name_definition_to_import| match name_definition_to_import {
|
|
NameDefinition::ModuleDef(module_def) => Some(module_def),
|
|
_ => None,
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
fn get_name_definition(&mut self, import_candidate: &FileSymbol) -> Option<NameDefinition> {
|
|
let _p = profile("get_name_definition");
|
|
let file_id = import_candidate.file_id;
|
|
|
|
let candidate_node = import_candidate.ptr.to_node(self.sema.parse(file_id).syntax());
|
|
let candidate_name_node = if candidate_node.kind() != NAME {
|
|
candidate_node.children().find(|it| it.kind() == NAME)?
|
|
} else {
|
|
candidate_node
|
|
};
|
|
classify_name(&self.sema, &ast::Name::cast(candidate_name_node)?)
|
|
}
|
|
}
|