move path completion to a separate component

This commit is contained in:
Aleksey Kladov 2018-12-21 23:04:56 +03:00
parent cbe67339df
commit 2136e75c0b
3 changed files with 127 additions and 44 deletions

View file

@ -0,0 +1,95 @@
use crate::{
completion::{CompletionItem, Completions, CompletionKind::*, SyntaxContext},
Cancelable,
};
pub(super) fn complete_path(acc: &mut Completions, ctx: &SyntaxContext) -> Cancelable<()> {
let (path, module) = match (&ctx.path_prefix, &ctx.module) {
(Some(path), Some(module)) => (path.clone(), module),
_ => return Ok(()),
};
let def_id = match module.resolve_path(ctx.db, path)? {
None => return Ok(()),
Some(it) => it,
};
let target_module = match def_id.resolve(ctx.db)? {
hir::Def::Module(it) => it,
_ => return Ok(()),
};
let module_scope = target_module.scope(ctx.db)?;
module_scope.entries().for_each(|(name, _res)| {
CompletionItem::new(name.to_string())
.kind(Reference)
.add_to(acc)
});
Ok(())
}
#[cfg(test)]
mod tests {
use crate::completion::{CompletionKind, check_completion};
fn check_reference_completion(code: &str, expected_completions: &str) {
check_completion(code, expected_completions, CompletionKind::Reference);
}
#[test]
fn completes_use_item_starting_with_self() {
check_reference_completion(
r"
use self::m::<|>;
mod m {
struct Bar;
}
",
"Bar",
);
}
#[test]
fn completes_use_item_starting_with_crate() {
check_reference_completion(
"
//- /lib.rs
mod foo;
struct Spam;
//- /foo.rs
use crate::Sp<|>
",
"Spam;foo",
);
}
#[test]
fn completes_nested_use_tree() {
check_reference_completion(
"
//- /lib.rs
mod foo;
struct Spam;
//- /foo.rs
use crate::{Sp<|>};
",
"Spam;foo",
);
}
#[test]
fn completes_deeply_nested_use_tree() {
check_reference_completion(
"
//- /lib.rs
mod foo;
pub mod bar {
pub mod baz {
pub struct Spam;
}
}
//- /foo.rs
use crate::{bar::{baz::Sp<|>}};
",
"Spam",
);
}
}

View file

@ -1,4 +1,4 @@
use rustc_hash::{FxHashSet};
use rustc_hash::FxHashSet;
use ra_syntax::{
SourceFileNode, AstNode,
ast,
@ -6,7 +6,7 @@ use ra_syntax::{
};
use hir::{
self,
FnScopes, Def, Path
FnScopes, Path
};
use crate::{
@ -53,7 +53,7 @@ pub(super) fn completions(
.add_to(acc)
});
}
NameRefKind::Path(path) => complete_path(acc, db, module, path)?,
NameRefKind::Path(_) => (),
NameRefKind::BareIdentInMod => (),
}
Ok(())
@ -121,33 +121,6 @@ fn complete_fn(name_ref: ast::NameRef, scopes: &FnScopes, acc: &mut Completions)
}
}
fn complete_path(
acc: &mut Completions,
db: &RootDatabase,
module: &hir::Module,
mut path: Path,
) -> Cancelable<()> {
if path.segments.is_empty() {
return Ok(());
}
path.segments.pop();
let def_id = match module.resolve_path(db, path)? {
None => return Ok(()),
Some(it) => it,
};
let target_module = match def_id.resolve(db)? {
Def::Module(it) => it,
_ => return Ok(()),
};
let module_scope = target_module.scope(db)?;
module_scope.entries().for_each(|(name, _res)| {
CompletionItem::new(name.to_string())
.kind(Reference)
.add_to(acc)
});
Ok(())
}
#[cfg(test)]
mod tests {
use crate::completion::{CompletionKind, check_completion};