135 lines
3.4 KiB
Rust
135 lines
3.4 KiB
Rust
use crate::{
|
|
Cancelable,
|
|
completion::{CompletionItem, CompletionItemKind, Completions, CompletionKind, CompletionContext},
|
|
};
|
|
|
|
pub(super) fn complete_path(acc: &mut Completions, ctx: &CompletionContext) -> 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)?.take_types() {
|
|
Some(it) => it,
|
|
None => return Ok(()),
|
|
};
|
|
match def_id.resolve(ctx.db)? {
|
|
hir::Def::Module(module) => {
|
|
let module_scope = module.scope(ctx.db)?;
|
|
module_scope.entries().for_each(|(name, res)| {
|
|
CompletionItem::new(CompletionKind::Reference, name.to_string())
|
|
.from_resolution(ctx, res)
|
|
.add_to(acc)
|
|
});
|
|
}
|
|
hir::Def::Enum(e) => {
|
|
e.variants(ctx.db)?
|
|
.unwrap_or(vec![])
|
|
.into_iter()
|
|
.for_each(|variant| {
|
|
let variant_name = variant.name(ctx.db);
|
|
|
|
if let Ok(Some(name)) = variant_name {
|
|
CompletionItem::new(CompletionKind::Reference, name.to_string())
|
|
.kind(CompletionItemKind::EnumVariant)
|
|
.add_to(acc)
|
|
}
|
|
})
|
|
}
|
|
_ => return Ok(()),
|
|
};
|
|
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",
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn completes_enum_variant() {
|
|
check_reference_completion(
|
|
"
|
|
//- /lib.rs
|
|
enum E { Foo, Bar(i32) }
|
|
fn foo() { let _ = E::<|> }
|
|
",
|
|
"Foo;Bar",
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn dont_render_function_parens_in_use_item() {
|
|
check_reference_completion(
|
|
"
|
|
//- /lib.rs
|
|
mod m { pub fn foo() {} }
|
|
use crate::m::f<|>;
|
|
",
|
|
"foo",
|
|
)
|
|
}
|
|
}
|