bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-completion/src/completions/item_list.rs')
-rw-r--r--crates/ide-completion/src/completions/item_list.rs126
1 files changed, 75 insertions, 51 deletions
diff --git a/crates/ide-completion/src/completions/item_list.rs b/crates/ide-completion/src/completions/item_list.rs
index 1883377408..3ab341e4ed 100644
--- a/crates/ide-completion/src/completions/item_list.rs
+++ b/crates/ide-completion/src/completions/item_list.rs
@@ -29,7 +29,9 @@ pub(crate) fn complete_item_list(
kind: &ItemListKind,
) {
let _p = tracing::info_span!("complete_item_list").entered();
- if path_ctx.is_trivial_path() {
+
+ // We handle completions for trait-impls in [`item_list::trait_impl`]
+ if path_ctx.is_trivial_path() && !matches!(kind, ItemListKind::TraitImpl(_)) {
add_keywords(acc, ctx, Some(kind));
}
@@ -75,73 +77,95 @@ fn add_keywords(acc: &mut Completions, ctx: &CompletionContext<'_>, kind: Option
let in_item_list = matches!(kind, Some(ItemListKind::SourceFile | ItemListKind::Module) | None);
let in_assoc_non_trait_impl = matches!(kind, Some(ItemListKind::Impl | ItemListKind::Trait));
- let in_extern_block = matches!(kind, Some(ItemListKind::ExternBlock));
+
+ let in_extern_block = matches!(kind, Some(ItemListKind::ExternBlock { .. }));
+ let in_unsafe_extern_block =
+ matches!(kind, Some(ItemListKind::ExternBlock { is_unsafe: true }));
+
let in_trait = matches!(kind, Some(ItemListKind::Trait));
- let in_trait_impl = matches!(kind, Some(ItemListKind::TraitImpl(_)));
let in_inherent_impl = matches!(kind, Some(ItemListKind::Impl));
- let no_vis_qualifiers = ctx.qualifier_ctx.vis_node.is_none();
let in_block = kind.is_none();
- let missing_qualifiers = [
- ctx.qualifier_ctx.unsafe_tok.is_none().then_some(("unsafe", "unsafe $0")),
- ctx.qualifier_ctx.async_tok.is_none().then_some(("async", "async $0")),
- ];
-
- if !in_trait_impl {
- // handle qualifier tokens
- if missing_qualifiers.iter().any(Option::is_none) {
- // only complete missing qualifiers
- missing_qualifiers.iter().filter_map(|x| *x).for_each(|(kw, snippet)| {
- add_keyword(kw, snippet);
- });
-
- if in_item_list || in_assoc_non_trait_impl {
- add_keyword("fn", "fn $1($2) {\n $0\n}");
- }
+ let no_vis_qualifiers = ctx.qualifier_ctx.vis_node.is_none();
+ let has_unsafe_kw = ctx.qualifier_ctx.unsafe_tok.is_some();
+ let has_async_kw = ctx.qualifier_ctx.async_tok.is_some();
+ let has_safe_kw = ctx.qualifier_ctx.safe_tok.is_some();
+
+ // Some keywords are invalid after non-vis qualifiers, so we handle them first.
+ if (has_unsafe_kw || has_safe_kw) && in_extern_block {
+ add_keyword("fn", "fn $1($2);");
+ add_keyword("static", "static $1: $2;");
+ return;
+ }
- if ctx.qualifier_ctx.unsafe_tok.is_some() && in_item_list {
- add_keyword("trait", "trait $1 {\n $0\n}");
- if no_vis_qualifiers {
- add_keyword("impl", "impl $1 {\n $0\n}");
- }
- }
+ if has_unsafe_kw || has_async_kw {
+ if !has_unsafe_kw {
+ add_keyword("unsafe", "unsafe $0");
+ }
+ if !has_async_kw {
+ add_keyword("async", "async $0");
+ }
- return;
+ if in_item_list || in_assoc_non_trait_impl {
+ add_keyword("fn", "fn $1($2) {\n $0\n}");
}
- if in_item_list {
- add_keyword("enum", "enum $1 {\n $0\n}");
- add_keyword("mod", "mod $0");
- add_keyword("static", "static $0");
- add_keyword("struct", "struct $0");
+ if has_unsafe_kw && in_item_list {
add_keyword("trait", "trait $1 {\n $0\n}");
- add_keyword("union", "union $1 {\n $0\n}");
- add_keyword("use", "use $0");
if no_vis_qualifiers {
add_keyword("impl", "impl $1 {\n $0\n}");
}
}
- if !in_trait && !in_block && no_vis_qualifiers {
- add_keyword("pub(crate)", "pub(crate) $0");
- add_keyword("pub(super)", "pub(super) $0");
- add_keyword("pub", "pub $0");
+ if !has_async_kw && no_vis_qualifiers && in_item_list {
+ add_keyword("extern", "extern $0");
}
- if in_extern_block {
- add_keyword("fn", "fn $1($2);");
- } else {
- if !in_inherent_impl {
- if !in_trait {
- add_keyword("extern", "extern $0");
- }
- add_keyword("type", "type $0");
- }
+ return;
+ }
- add_keyword("fn", "fn $1($2) {\n $0\n}");
- add_keyword("unsafe", "unsafe $0");
- add_keyword("const", "const $0");
- add_keyword("async", "async $0");
+ // ...and the rest deals with cases without any non-vis qualifiers.
+
+ // Visibility qualifiers
+ if !in_trait && !in_block && no_vis_qualifiers {
+ add_keyword("pub(crate)", "pub(crate) $0");
+ add_keyword("pub(super)", "pub(super) $0");
+ add_keyword("pub", "pub $0");
+ }
+
+ // Keywords that are valid in `item_list`
+ if in_item_list {
+ add_keyword("enum", "enum $1 {\n $0\n}");
+ add_keyword("mod", "mod $0");
+ add_keyword("static", "static $0");
+ add_keyword("struct", "struct $0");
+ add_keyword("trait", "trait $1 {\n $0\n}");
+ add_keyword("union", "union $1 {\n $0\n}");
+ add_keyword("use", "use $0");
+ if no_vis_qualifiers {
+ add_keyword("impl", "impl $1 {\n $0\n}");
+ }
+ }
+
+ if in_extern_block {
+ add_keyword("unsafe", "unsafe $0");
+ if in_unsafe_extern_block {
+ add_keyword("safe", "safe $0");
}
+
+ add_keyword("fn", "fn $1($2);");
+ add_keyword("static", "static $1: $2;");
+ } else {
+ if !in_inherent_impl {
+ if !in_trait {
+ add_keyword("extern", "extern $0");
+ }
+ add_keyword("type", "type $0");
+ }
+
+ add_keyword("fn", "fn $1($2) {\n $0\n}");
+ add_keyword("unsafe", "unsafe $0");
+ add_keyword("const", "const $0");
+ add_keyword("async", "async $0");
}
}