68 lines
2.3 KiB
Rust
68 lines
2.3 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::{MacroDef, ModuleDef, Semantics};
|
|
use ra_prof::profile;
|
|
use ra_syntax::{ast, AstNode, SyntaxKind::NAME};
|
|
|
|
use crate::{
|
|
defs::{classify_name, Definition},
|
|
symbol_index::{self, FileSymbol, Query},
|
|
RootDatabase,
|
|
};
|
|
use either::Either;
|
|
|
|
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<Either<ModuleDef, MacroDef>> {
|
|
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 {
|
|
Definition::ModuleDef(module_def) => Some(Either::Left(module_def)),
|
|
Definition::Macro(macro_def) => Some(Either::Right(macro_def)),
|
|
_ => None,
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
fn get_name_definition(&mut self, import_candidate: &FileSymbol) -> Option<Definition> {
|
|
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
|
|
};
|
|
let name = ast::Name::cast(candidate_name_node)?;
|
|
classify_name(&self.sema, &name)?.into_definition()
|
|
}
|
|
}
|