From b184bfad7a2dc6a9bf6654a7eec6c68a27c49f70 Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Sun, 20 Dec 2020 18:19:23 +0100 Subject: Add completions for patterns --- crates/completion/src/completions.rs | 33 +++++- crates/completion/src/completions/pattern.rs | 154 ++++++++++++++++++++++++--- crates/completion/src/context.rs | 21 ++-- crates/completion/src/render.rs | 1 + crates/completion/src/render/pattern.rs | 128 ++++++++++++++++++++++ crates/hir/src/code_model.rs | 4 + 6 files changed, 315 insertions(+), 26 deletions(-) create mode 100644 crates/completion/src/render/pattern.rs diff --git a/crates/completion/src/completions.rs b/crates/completion/src/completions.rs index 1ef6b5f48..d9fe13485 100644 --- a/crates/completion/src/completions.rs +++ b/crates/completion/src/completions.rs @@ -19,9 +19,14 @@ use hir::{ModPath, ScopeDef, Type}; use crate::{ item::Builder, render::{ - const_::render_const, enum_variant::render_variant, function::render_fn, - macro_::render_macro, render_field, render_resolution, render_tuple_field, - type_alias::render_type_alias, RenderContext, + const_::render_const, + enum_variant::render_variant, + function::render_fn, + macro_::render_macro, + pattern::{render_struct_pat, render_variant_pat}, + render_field, render_resolution, render_tuple_field, + type_alias::render_type_alias, + RenderContext, }, CompletionContext, CompletionItem, }; @@ -105,6 +110,28 @@ impl Completions { self.add(item) } + pub(crate) fn add_variant_pat( + &mut self, + ctx: &CompletionContext, + variant: hir::Variant, + local_name: Option, + ) { + if let Some(item) = render_variant_pat(RenderContext::new(ctx), variant, local_name) { + self.add(item); + } + } + + pub(crate) fn add_struct_pat( + &mut self, + ctx: &CompletionContext, + strukt: hir::Struct, + local_name: Option, + ) { + if let Some(item) = render_struct_pat(RenderContext::new(ctx), strukt, local_name) { + self.add(item); + } + } + pub(crate) fn add_const(&mut self, ctx: &CompletionContext, constant: hir::Const) { if let Some(item) = render_const(RenderContext::new(ctx), constant) { self.add(item); diff --git a/crates/completion/src/completions/pattern.rs b/crates/completion/src/completions/pattern.rs index 4d56731ec..496f0b040 100644 --- a/crates/completion/src/completions/pattern.rs +++ b/crates/completion/src/completions/pattern.rs @@ -1,10 +1,12 @@ //! Completes constats and paths in patterns. +use hir::StructKind; + use crate::{CompletionContext, Completions}; -/// Completes constats and paths in patterns. +/// Completes constants and paths in patterns. pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) { - if !(ctx.is_pat_binding_or_const || ctx.is_irrefutable_let_pat_binding) { + if !(ctx.is_pat_binding_or_const || ctx.is_irrefutable_pat_binding) { return; } if ctx.record_pat_syntax.is_some() { @@ -15,20 +17,25 @@ pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) { // suggest variants + auto-imports ctx.scope.process_all_names(&mut |name, res| { let add_resolution = match &res { - hir::ScopeDef::ModuleDef(def) => { - if ctx.is_irrefutable_let_pat_binding { - matches!(def, hir::ModuleDef::Adt(hir::Adt::Struct(_))) - } else { - matches!( - def, - hir::ModuleDef::Adt(hir::Adt::Enum(..)) - | hir::ModuleDef::Adt(hir::Adt::Struct(..)) - | hir::ModuleDef::Variant(..) - | hir::ModuleDef::Const(..) - | hir::ModuleDef::Module(..) - ) + hir::ScopeDef::ModuleDef(def) => match def { + hir::ModuleDef::Adt(hir::Adt::Struct(strukt)) => { + acc.add_struct_pat(ctx, strukt.clone(), Some(name.clone())); + true + } + hir::ModuleDef::Variant(variant) + if !ctx.is_irrefutable_pat_binding + // render_resolution already does some pattern completion tricks for tuple variants + && variant.kind(ctx.db) == StructKind::Record => + { + acc.add_variant_pat(ctx, variant.clone(), Some(name.clone())); + true } - } + hir::ModuleDef::Adt(hir::Adt::Enum(..)) + | hir::ModuleDef::Variant(..) + | hir::ModuleDef::Const(..) + | hir::ModuleDef::Module(..) => !ctx.is_irrefutable_pat_binding, + _ => false, + }, hir::ScopeDef::MacroDef(_) => true, _ => false, }; @@ -49,6 +56,11 @@ mod tests { expect.assert_eq(&actual) } + fn check_snippet(ra_fixture: &str, expect: Expect) { + let actual = completion_list(ra_fixture, CompletionKind::Snippet); + expect.assert_eq(&actual) + } + #[test] fn completes_enum_variants_and_modules() { check( @@ -114,4 +126,116 @@ fn foo() { "#]], ); } + + #[test] + fn completes_in_param() { + check( + r#" +enum E { X } + +static FOO: E = E::X; +struct Bar { f: u32 } + +fn foo(<|>) { +} +"#, + expect![[r#" + st Bar + "#]], + ); + } + + #[test] + fn completes_pat_in_let() { + check_snippet( + r#" +struct Bar { f: u32 } + +fn foo() { + let <|> +} +"#, + expect![[r#" + bn Bar Bar { f }$0 + "#]], + ); + } + + #[test] + fn completes_param_pattern() { + check_snippet( + r#" +struct Foo { bar: String, baz: String } +struct Bar(String, String); +struct Baz; +fn outer(<|>) {} +"#, + expect![[r#" + bn Foo Foo { bar, baz }: Foo$0 + bn Bar Bar($1, $2): Bar$0 + "#]], + ) + } + + #[test] + fn completes_let_pattern() { + check_snippet( + r#" +struct Foo { bar: String, baz: String } +struct Bar(String, String); +struct Baz; +fn outer() { + let <|> +} +"#, + expect![[r#" + bn Foo Foo { bar, baz }$0 + bn Bar Bar($1, $2)$0 + "#]], + ) + } + + #[test] + fn completes_refutable_pattern() { + check_snippet( + r#" +struct Foo { bar: i32, baz: i32 } +struct Bar(String, String); +struct Baz; +fn outer() { + match () { + <|> + } +} +"#, + expect![[r#" + bn Foo Foo { bar, baz }$0 + bn Bar Bar($1, $2)$0 + "#]], + ) + } + + #[test] + fn omits_private_fields_pat() { + check_snippet( + r#" +mod foo { + pub struct Foo { pub bar: i32, baz: i32 } + pub struct Bar(pub String, String); + pub struct Invisible(String, String); +} +use foo::*; + +fn outer() { + match () { + <|> + } +} +"#, + expect![[r#" + bn Foo Foo { bar, .. }$0 + bn Bar Bar($1, ..)$0 + "#]], + ) + } } diff --git a/crates/completion/src/context.rs b/crates/completion/src/context.rs index 5cd11cf77..41de324d8 100644 --- a/crates/completion/src/context.rs +++ b/crates/completion/src/context.rs @@ -51,7 +51,7 @@ pub(crate) struct CompletionContext<'a> { /// If a name-binding or reference to a const in a pattern. /// Irrefutable patterns (like let) are excluded. pub(super) is_pat_binding_or_const: bool, - pub(super) is_irrefutable_let_pat_binding: bool, + pub(super) is_irrefutable_pat_binding: bool, /// A single-indent path, like `foo`. `::foo` should not be considered a trivial path. pub(super) is_trivial_path: bool, /// If not a trivial path, the prefix (qualifier). @@ -147,7 +147,7 @@ impl<'a> CompletionContext<'a> { active_parameter: ActiveParameter::at(db, position), is_param: false, is_pat_binding_or_const: false, - is_irrefutable_let_pat_binding: false, + is_irrefutable_pat_binding: false, is_trivial_path: false, path_qual: None, after_if: false, @@ -327,14 +327,19 @@ impl<'a> CompletionContext<'a> { if bind_pat.syntax().parent().and_then(ast::RecordPatFieldList::cast).is_some() { self.is_pat_binding_or_const = false; } - if let Some(let_stmt) = bind_pat.syntax().ancestors().find_map(ast::LetStmt::cast) { - if let Some(pat) = let_stmt.pat() { - if pat.syntax().text_range().contains_range(bind_pat.syntax().text_range()) - { - self.is_pat_binding_or_const = false; - self.is_irrefutable_let_pat_binding = true; + if let Some(Some(pat)) = bind_pat.syntax().ancestors().find_map(|node| { + match_ast! { + match node { + ast::LetStmt(it) => Some(it.pat()), + ast::Param(it) => Some(it.pat()), + _ => None, } } + }) { + if pat.syntax().text_range().contains_range(bind_pat.syntax().text_range()) { + self.is_pat_binding_or_const = false; + self.is_irrefutable_pat_binding = true; + } } } if is_node::(name.syntax()) { diff --git a/crates/completion/src/render.rs b/crates/completion/src/render.rs index 1092a4825..945158d10 100644 --- a/crates/completion/src/render.rs +++ b/crates/completion/src/render.rs @@ -5,6 +5,7 @@ pub(crate) mod macro_; pub(crate) mod function; pub(crate) mod enum_variant; pub(crate) mod const_; +pub(crate) mod pattern; pub(crate) mod type_alias; mod builder_ext; diff --git a/crates/completion/src/render/pattern.rs b/crates/completion/src/render/pattern.rs new file mode 100644 index 000000000..e20b0027b --- /dev/null +++ b/crates/completion/src/render/pattern.rs @@ -0,0 +1,128 @@ +//! Renderer for patterns. + +use hir::{db::HirDatabase, HasVisibility, Name, StructKind}; +use itertools::Itertools; + +use crate::{item::CompletionKind, render::RenderContext, CompletionItem, CompletionItemKind}; + +pub(crate) fn render_struct_pat<'a>( + ctx: RenderContext<'a>, + strukt: hir::Struct, + local_name: Option, +) -> Option { + let _p = profile::span("render_struct_pat"); + + let module = ctx.completion.scope.module()?; + let fields = strukt.fields(ctx.db()); + let n_fields = fields.len(); + let fields = fields + .into_iter() + .filter(|field| field.is_visible_from(ctx.db(), module)) + .collect::>(); + + if fields.is_empty() { + // Matching a struct without matching its fields is pointless, unlike matching a Variant without its fields + return None; + } + let fields_omitted = n_fields - fields.len() > 0; + + let name = local_name.unwrap_or_else(|| strukt.name(ctx.db())).to_string(); + let mut pat = match strukt.kind(ctx.db()) { + StructKind::Tuple if ctx.snippet_cap().is_some() => { + render_tuple_as_pat(&fields, &name, fields_omitted) + } + StructKind::Record => render_record_as_pat(ctx.db(), &fields, &name, fields_omitted), + _ => return None, + }; + + if ctx.completion.is_param { + pat.push(':'); + pat.push(' '); + pat.push_str(&name); + } + if ctx.snippet_cap().is_some() { + pat.push_str("$0"); + } + + let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) + .kind(CompletionItemKind::Binding) + .set_documentation(ctx.docs(strukt)) + .set_deprecated(ctx.is_deprecated(strukt)) + .detail(&pat); + if let Some(snippet_cap) = ctx.snippet_cap() { + completion = completion.insert_snippet(snippet_cap, pat); + } else { + completion = completion.insert_text(pat); + } + Some(completion.build()) +} + +pub(crate) fn render_variant_pat<'a>( + ctx: RenderContext<'a>, + variant: hir::Variant, + local_name: Option, +) -> Option { + let _p = profile::span("render_variant_pat"); + + let module = ctx.completion.scope.module()?; + let fields = variant.fields(ctx.db()); + let n_fields = fields.len(); + let fields = fields + .into_iter() + .filter(|field| field.is_visible_from(ctx.db(), module)) + .collect::>(); + + let fields_omitted = n_fields - fields.len() > 0; + + let name = local_name.unwrap_or_else(|| variant.name(ctx.db())).to_string(); + let mut pat = match variant.kind(ctx.db()) { + StructKind::Tuple if ctx.snippet_cap().is_some() => { + render_tuple_as_pat(&fields, &name, fields_omitted) + } + StructKind::Record => render_record_as_pat(ctx.db(), &fields, &name, fields_omitted), + _ => return None, + }; + + if ctx.completion.is_param { + pat.push(':'); + pat.push(' '); + pat.push_str(&name); + } + if ctx.snippet_cap().is_some() { + pat.push_str("$0"); + } + let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) + .kind(CompletionItemKind::Binding) + .set_documentation(ctx.docs(variant)) + .set_deprecated(ctx.is_deprecated(variant)) + .detail(&pat); + if let Some(snippet_cap) = ctx.snippet_cap() { + completion = completion.insert_snippet(snippet_cap, pat); + } else { + completion = completion.insert_text(pat); + } + Some(completion.build()) +} + +fn render_record_as_pat( + db: &dyn HirDatabase, + fields: &[hir::Field], + name: &str, + fields_omitted: bool, +) -> String { + format!( + "{name} {{ {}{} }}", + fields.into_iter().map(|field| field.name(db)).format(", "), + if fields_omitted { ", .." } else { "" }, + name = name + ) +} + +fn render_tuple_as_pat(fields: &[hir::Field], name: &str, fields_omitted: bool) -> String { + format!( + "{name}({}{})", + fields.into_iter().enumerate().map(|(idx, _)| format!("${}", idx + 1)).format(", "), + if fields_omitted { ", .." } else { "" }, + name = name + ) +} diff --git a/crates/hir/src/code_model.rs b/crates/hir/src/code_model.rs index d6c7e71ea..dbc937e4f 100644 --- a/crates/hir/src/code_model.rs +++ b/crates/hir/src/code_model.rs @@ -511,6 +511,10 @@ impl Struct { db.struct_data(self.id).repr.clone() } + pub fn kind(self, db: &dyn HirDatabase) -> StructKind { + self.variant_data(db).kind() + } + fn variant_data(self, db: &dyn HirDatabase) -> Arc { db.struct_data(self.id).variant_data.clone() } -- cgit v1.2.3 From 2f6a24950a555bdfecbda7a50735d643f9d0e7f9 Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Sun, 20 Dec 2020 19:16:28 +0100 Subject: Emit snippets for struct pattern completion if enabled --- crates/completion/src/completions/pattern.rs | 10 +++---- crates/completion/src/render/pattern.rs | 41 +++++++++++++++++++++------- 2 files changed, 36 insertions(+), 15 deletions(-) diff --git a/crates/completion/src/completions/pattern.rs b/crates/completion/src/completions/pattern.rs index 496f0b040..23a00803c 100644 --- a/crates/completion/src/completions/pattern.rs +++ b/crates/completion/src/completions/pattern.rs @@ -156,7 +156,7 @@ fn foo() { } "#, expect![[r#" - bn Bar Bar { f }$0 + bn Bar Bar { ${1:f} }$0 "#]], ); } @@ -171,7 +171,7 @@ struct Baz; fn outer(<|>) {} "#, expect![[r#" - bn Foo Foo { bar, baz }: Foo$0 + bn Foo Foo { ${1:bar}, ${2:baz} }: Foo$0 bn Bar Bar($1, $2): Bar$0 "#]], ) @@ -189,7 +189,7 @@ fn outer() { } "#, expect![[r#" - bn Foo Foo { bar, baz }$0 + bn Foo Foo { ${1:bar}, ${2:baz} }$0 bn Bar Bar($1, $2)$0 "#]], ) @@ -209,7 +209,7 @@ fn outer() { } "#, expect![[r#" - bn Foo Foo { bar, baz }$0 + bn Foo Foo { ${1:bar}, ${2:baz} }$0 bn Bar Bar($1, $2)$0 "#]], ) @@ -233,7 +233,7 @@ fn outer() { } "#, expect![[r#" - bn Foo Foo { bar, .. }$0 + bn Foo Foo { ${1:bar}, .. }$0 bn Bar Bar($1, ..)$0 "#]], ) diff --git a/crates/completion/src/render/pattern.rs b/crates/completion/src/render/pattern.rs index e20b0027b..2327bf363 100644 --- a/crates/completion/src/render/pattern.rs +++ b/crates/completion/src/render/pattern.rs @@ -3,7 +3,10 @@ use hir::{db::HirDatabase, HasVisibility, Name, StructKind}; use itertools::Itertools; -use crate::{item::CompletionKind, render::RenderContext, CompletionItem, CompletionItemKind}; +use crate::{ + config::SnippetCap, item::CompletionKind, render::RenderContext, CompletionItem, + CompletionItemKind, +}; pub(crate) fn render_struct_pat<'a>( ctx: RenderContext<'a>, @@ -31,7 +34,9 @@ pub(crate) fn render_struct_pat<'a>( StructKind::Tuple if ctx.snippet_cap().is_some() => { render_tuple_as_pat(&fields, &name, fields_omitted) } - StructKind::Record => render_record_as_pat(ctx.db(), &fields, &name, fields_omitted), + StructKind::Record => { + render_record_as_pat(ctx.db(), ctx.snippet_cap(), &fields, &name, fields_omitted) + } _ => return None, }; @@ -79,7 +84,9 @@ pub(crate) fn render_variant_pat<'a>( StructKind::Tuple if ctx.snippet_cap().is_some() => { render_tuple_as_pat(&fields, &name, fields_omitted) } - StructKind::Record => render_record_as_pat(ctx.db(), &fields, &name, fields_omitted), + StructKind::Record => { + render_record_as_pat(ctx.db(), ctx.snippet_cap(), &fields, &name, fields_omitted) + } _ => return None, }; @@ -106,22 +113,36 @@ pub(crate) fn render_variant_pat<'a>( fn render_record_as_pat( db: &dyn HirDatabase, + snippet_cap: Option, fields: &[hir::Field], name: &str, fields_omitted: bool, ) -> String { - format!( - "{name} {{ {}{} }}", - fields.into_iter().map(|field| field.name(db)).format(", "), - if fields_omitted { ", .." } else { "" }, - name = name - ) + let fields = fields.iter(); + if snippet_cap.is_some() { + format!( + "{name} {{ {}{} }}", + fields + .enumerate() + .map(|(idx, field)| format!("${{{}:{}}}", idx + 1, field.name(db))) + .format(", "), + if fields_omitted { ", .." } else { "" }, + name = name + ) + } else { + format!( + "{name} {{ {}{} }}", + fields.map(|field| field.name(db)).format(", "), + if fields_omitted { ", .." } else { "" }, + name = name + ) + } } fn render_tuple_as_pat(fields: &[hir::Field], name: &str, fields_omitted: bool) -> String { format!( "{name}({}{})", - fields.into_iter().enumerate().map(|(idx, _)| format!("${}", idx + 1)).format(", "), + fields.iter().enumerate().map(|(idx, _)| format!("${}", idx + 1)).format(", "), if fields_omitted { ", .." } else { "" }, name = name ) -- cgit v1.2.3 From a5874a38cb1bbaee685ff92a40b66f9d5bcf34f6 Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Sun, 20 Dec 2020 19:20:55 +0100 Subject: Reduce code duplication in pattern completion --- crates/completion/src/render/pattern.rs | 63 +++++++++++++++------------------ 1 file changed, 29 insertions(+), 34 deletions(-) diff --git a/crates/completion/src/render/pattern.rs b/crates/completion/src/render/pattern.rs index 2327bf363..cfe2116bd 100644 --- a/crates/completion/src/render/pattern.rs +++ b/crates/completion/src/render/pattern.rs @@ -8,8 +8,8 @@ use crate::{ CompletionItemKind, }; -pub(crate) fn render_struct_pat<'a>( - ctx: RenderContext<'a>, +pub(crate) fn render_struct_pat( + ctx: RenderContext<'_>, strukt: hir::Struct, local_name: Option, ) -> Option { @@ -30,24 +30,7 @@ pub(crate) fn render_struct_pat<'a>( let fields_omitted = n_fields - fields.len() > 0; let name = local_name.unwrap_or_else(|| strukt.name(ctx.db())).to_string(); - let mut pat = match strukt.kind(ctx.db()) { - StructKind::Tuple if ctx.snippet_cap().is_some() => { - render_tuple_as_pat(&fields, &name, fields_omitted) - } - StructKind::Record => { - render_record_as_pat(ctx.db(), ctx.snippet_cap(), &fields, &name, fields_omitted) - } - _ => return None, - }; - - if ctx.completion.is_param { - pat.push(':'); - pat.push(' '); - pat.push_str(&name); - } - if ctx.snippet_cap().is_some() { - pat.push_str("$0"); - } + let pat = render_pat(&ctx, &name, strukt.kind(ctx.db()), &fields, fields_omitted)?; let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) .kind(CompletionItemKind::Binding) @@ -62,8 +45,8 @@ pub(crate) fn render_struct_pat<'a>( Some(completion.build()) } -pub(crate) fn render_variant_pat<'a>( - ctx: RenderContext<'a>, +pub(crate) fn render_variant_pat( + ctx: RenderContext<'_>, variant: hir::Variant, local_name: Option, ) -> Option { @@ -80,7 +63,29 @@ pub(crate) fn render_variant_pat<'a>( let fields_omitted = n_fields - fields.len() > 0; let name = local_name.unwrap_or_else(|| variant.name(ctx.db())).to_string(); - let mut pat = match variant.kind(ctx.db()) { + let pat = render_pat(&ctx, &name, variant.kind(ctx.db()), &fields, fields_omitted)?; + + let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) + .kind(CompletionItemKind::Binding) + .set_documentation(ctx.docs(variant)) + .set_deprecated(ctx.is_deprecated(variant)) + .detail(&pat); + if let Some(snippet_cap) = ctx.snippet_cap() { + completion = completion.insert_snippet(snippet_cap, pat); + } else { + completion = completion.insert_text(pat); + } + Some(completion.build()) +} + +fn render_pat( + ctx: &RenderContext<'_>, + name: &str, + kind: StructKind, + fields: &[hir::Field], + fields_omitted: bool, +) -> Option { + let mut pat = match kind { StructKind::Tuple if ctx.snippet_cap().is_some() => { render_tuple_as_pat(&fields, &name, fields_omitted) } @@ -98,17 +103,7 @@ pub(crate) fn render_variant_pat<'a>( if ctx.snippet_cap().is_some() { pat.push_str("$0"); } - let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) - .kind(CompletionItemKind::Binding) - .set_documentation(ctx.docs(variant)) - .set_deprecated(ctx.is_deprecated(variant)) - .detail(&pat); - if let Some(snippet_cap) = ctx.snippet_cap() { - completion = completion.insert_snippet(snippet_cap, pat); - } else { - completion = completion.insert_text(pat); - } - Some(completion.build()) + Some(pat) } fn render_record_as_pat( -- cgit v1.2.3 From 33ecad4407e1345cd662f535c2ae7a04e4bee52b Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Sun, 20 Dec 2020 20:13:06 +0100 Subject: Create non-exhaustive patterns for non_exhaustive attributed items --- crates/completion/src/render/pattern.rs | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/crates/completion/src/render/pattern.rs b/crates/completion/src/render/pattern.rs index cfe2116bd..2d1600105 100644 --- a/crates/completion/src/render/pattern.rs +++ b/crates/completion/src/render/pattern.rs @@ -1,6 +1,6 @@ //! Renderer for patterns. -use hir::{db::HirDatabase, HasVisibility, Name, StructKind}; +use hir::{db::HirDatabase, HasAttrs, HasVisibility, Name, StructKind}; use itertools::Itertools; use crate::{ @@ -27,7 +27,8 @@ pub(crate) fn render_struct_pat( // Matching a struct without matching its fields is pointless, unlike matching a Variant without its fields return None; } - let fields_omitted = n_fields - fields.len() > 0; + let fields_omitted = + n_fields - fields.len() > 0 || strukt.attrs(ctx.db()).by_key("non_exhaustive").exists(); let name = local_name.unwrap_or_else(|| strukt.name(ctx.db())).to_string(); let pat = render_pat(&ctx, &name, strukt.kind(ctx.db()), &fields, fields_omitted)?; @@ -60,7 +61,8 @@ pub(crate) fn render_variant_pat( .filter(|field| field.is_visible_from(ctx.db(), module)) .collect::>(); - let fields_omitted = n_fields - fields.len() > 0; + let fields_omitted = + n_fields - fields.len() > 0 || variant.attrs(ctx.db()).by_key("non_exhaustive").exists(); let name = local_name.unwrap_or_else(|| variant.name(ctx.db())).to_string(); let pat = render_pat(&ctx, &name, variant.kind(ctx.db()), &fields, fields_omitted)?; -- cgit v1.2.3 From 2cd2947bf8cb7abfb06ceb00804447def899d37d Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Mon, 21 Dec 2020 15:10:27 +0100 Subject: Insert snippet positions after fields names in record patterns --- crates/completion/src/completions/pattern.rs | 10 +++++----- crates/completion/src/render/pattern.rs | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/crates/completion/src/completions/pattern.rs b/crates/completion/src/completions/pattern.rs index 23a00803c..ba9921a44 100644 --- a/crates/completion/src/completions/pattern.rs +++ b/crates/completion/src/completions/pattern.rs @@ -156,7 +156,7 @@ fn foo() { } "#, expect![[r#" - bn Bar Bar { ${1:f} }$0 + bn Bar Bar { f$1 }$0 "#]], ); } @@ -171,7 +171,7 @@ struct Baz; fn outer(<|>) {} "#, expect![[r#" - bn Foo Foo { ${1:bar}, ${2:baz} }: Foo$0 + bn Foo Foo { bar$1, baz$2 }: Foo$0 bn Bar Bar($1, $2): Bar$0 "#]], ) @@ -189,7 +189,7 @@ fn outer() { } "#, expect![[r#" - bn Foo Foo { ${1:bar}, ${2:baz} }$0 + bn Foo Foo { bar$1, baz$2 }$0 bn Bar Bar($1, $2)$0 "#]], ) @@ -209,7 +209,7 @@ fn outer() { } "#, expect![[r#" - bn Foo Foo { ${1:bar}, ${2:baz} }$0 + bn Foo Foo { bar$1, baz$2 }$0 bn Bar Bar($1, $2)$0 "#]], ) @@ -233,7 +233,7 @@ fn outer() { } "#, expect![[r#" - bn Foo Foo { ${1:bar}, .. }$0 + bn Foo Foo { bar$1, .. }$0 bn Bar Bar($1, ..)$0 "#]], ) diff --git a/crates/completion/src/render/pattern.rs b/crates/completion/src/render/pattern.rs index 2d1600105..073d50a11 100644 --- a/crates/completion/src/render/pattern.rs +++ b/crates/completion/src/render/pattern.rs @@ -121,7 +121,7 @@ fn render_record_as_pat( "{name} {{ {}{} }}", fields .enumerate() - .map(|(idx, field)| format!("${{{}:{}}}", idx + 1, field.name(db))) + .map(|(idx, field)| format!("{}${}", field.name(db), idx + 1)) .format(", "), if fields_omitted { ", .." } else { "" }, name = name -- cgit v1.2.3 From 83121efcd577124a992dc8bd304690b36bda2931 Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Tue, 22 Dec 2020 19:00:38 +0100 Subject: Reduce some more code duplication --- crates/completion/src/completions/pattern.rs | 38 +++++++++---- crates/completion/src/render.rs | 6 +++ crates/completion/src/render/builder_ext.rs | 1 - crates/completion/src/render/enum_variant.rs | 45 ---------------- crates/completion/src/render/pattern.rs | 80 ++++++++++++++-------------- 5 files changed, 76 insertions(+), 94 deletions(-) diff --git a/crates/completion/src/completions/pattern.rs b/crates/completion/src/completions/pattern.rs index ba9921a44..eee31098d 100644 --- a/crates/completion/src/completions/pattern.rs +++ b/crates/completion/src/completions/pattern.rs @@ -1,7 +1,5 @@ //! Completes constats and paths in patterns. -use hir::StructKind; - use crate::{CompletionContext, Completions}; /// Completes constants and paths in patterns. @@ -22,11 +20,7 @@ pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) { acc.add_struct_pat(ctx, strukt.clone(), Some(name.clone())); true } - hir::ModuleDef::Variant(variant) - if !ctx.is_irrefutable_pat_binding - // render_resolution already does some pattern completion tricks for tuple variants - && variant.kind(ctx.db) == StructKind::Record => - { + hir::ModuleDef::Variant(variant) if !ctx.is_irrefutable_pat_binding => { acc.add_variant_pat(ctx, variant.clone(), Some(name.clone())); true } @@ -49,7 +43,10 @@ pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) { mod tests { use expect_test::{expect, Expect}; - use crate::{test_utils::completion_list, CompletionKind}; + use crate::{ + test_utils::{check_edit, completion_list}, + CompletionKind, + }; fn check(ra_fixture: &str, expect: Expect) { let actual = completion_list(ra_fixture, CompletionKind::Reference); @@ -81,7 +78,7 @@ fn foo() { en E ct Z st Bar - ev X () + ev X md m "#]], ); @@ -238,4 +235,27 @@ fn outer() { "#]], ) } + + #[test] + fn only_shows_ident_completion() { + check_edit( + "Foo", + r#" +struct Foo(i32); +fn main() { + match Foo(92) { + <|>(92) => (), + } +} +"#, + r#" +struct Foo(i32); +fn main() { + match Foo(92) { + Foo(92) => (), + } +} +"#, + ); + } } diff --git a/crates/completion/src/render.rs b/crates/completion/src/render.rs index 945158d10..1ba7201a1 100644 --- a/crates/completion/src/render.rs +++ b/crates/completion/src/render.rs @@ -160,6 +160,12 @@ impl<'a> Render<'a> { let item = render_fn(self.ctx, import_to_add, Some(local_name), *func); return Some(item); } + ScopeDef::ModuleDef(Variant(_)) + if self.ctx.completion.is_pat_binding_or_const + | self.ctx.completion.is_irrefutable_pat_binding => + { + CompletionItemKind::EnumVariant + } ScopeDef::ModuleDef(Variant(var)) => { let item = render_variant(self.ctx, import_to_add, Some(local_name), *var, None); return Some(item); diff --git a/crates/completion/src/render/builder_ext.rs b/crates/completion/src/render/builder_ext.rs index ce8718bd5..d053a988b 100644 --- a/crates/completion/src/render/builder_ext.rs +++ b/crates/completion/src/render/builder_ext.rs @@ -34,7 +34,6 @@ impl Builder { return false; } if ctx.is_pattern_call { - mark::hit!(dont_duplicate_pattern_parens); return false; } if ctx.is_call { diff --git a/crates/completion/src/render/enum_variant.rs b/crates/completion/src/render/enum_variant.rs index 7176fd9b3..732e139ec 100644 --- a/crates/completion/src/render/enum_variant.rs +++ b/crates/completion/src/render/enum_variant.rs @@ -124,51 +124,6 @@ use Option::*; fn main() -> Option { Some($0) } -"#, - ); - check_edit( - "Some", - r#" -enum Option { Some(T), None } -use Option::*; -fn main(value: Option) { - match value { - Som<|> - } -} -"#, - r#" -enum Option { Some(T), None } -use Option::*; -fn main(value: Option) { - match value { - Some($0) - } -} -"#, - ); - } - - #[test] - fn dont_duplicate_pattern_parens() { - mark::check!(dont_duplicate_pattern_parens); - check_edit( - "Var", - r#" -enum E { Var(i32) } -fn main() { - match E::Var(92) { - E::<|>(92) => (), - } -} -"#, - r#" -enum E { Var(i32) } -fn main() { - match E::Var(92) { - E::Var(92) => (), - } -} "#, ); } diff --git a/crates/completion/src/render/pattern.rs b/crates/completion/src/render/pattern.rs index 073d50a11..a3b6a3cac 100644 --- a/crates/completion/src/render/pattern.rs +++ b/crates/completion/src/render/pattern.rs @@ -8,6 +8,24 @@ use crate::{ CompletionItemKind, }; +fn visible_fields( + ctx: &RenderContext<'_>, + fields: &[hir::Field], + item: impl HasAttrs, +) -> Option<(Vec, bool)> { + let module = ctx.completion.scope.module()?; + let n_fields = fields.len(); + let fields = fields + .into_iter() + .filter(|field| field.is_visible_from(ctx.db(), module)) + .copied() + .collect::>(); + + let fields_omitted = + n_fields - fields.len() > 0 || item.attrs(ctx.db()).by_key("non_exhaustive").exists(); + Some((fields, fields_omitted)) +} + pub(crate) fn render_struct_pat( ctx: RenderContext<'_>, strukt: hir::Struct, @@ -15,35 +33,18 @@ pub(crate) fn render_struct_pat( ) -> Option { let _p = profile::span("render_struct_pat"); - let module = ctx.completion.scope.module()?; let fields = strukt.fields(ctx.db()); - let n_fields = fields.len(); - let fields = fields - .into_iter() - .filter(|field| field.is_visible_from(ctx.db(), module)) - .collect::>(); + let (visible_fields, fields_omitted) = visible_fields(&ctx, &fields, strukt)?; - if fields.is_empty() { + if visible_fields.is_empty() { // Matching a struct without matching its fields is pointless, unlike matching a Variant without its fields return None; } - let fields_omitted = - n_fields - fields.len() > 0 || strukt.attrs(ctx.db()).by_key("non_exhaustive").exists(); let name = local_name.unwrap_or_else(|| strukt.name(ctx.db())).to_string(); - let pat = render_pat(&ctx, &name, strukt.kind(ctx.db()), &fields, fields_omitted)?; + let pat = render_pat(&ctx, &name, strukt.kind(ctx.db()), &visible_fields, fields_omitted)?; - let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) - .kind(CompletionItemKind::Binding) - .set_documentation(ctx.docs(strukt)) - .set_deprecated(ctx.is_deprecated(strukt)) - .detail(&pat); - if let Some(snippet_cap) = ctx.snippet_cap() { - completion = completion.insert_snippet(snippet_cap, pat); - } else { - completion = completion.insert_text(pat); - } - Some(completion.build()) + Some(build_completion(ctx, name, pat, strukt)) } pub(crate) fn render_variant_pat( @@ -53,31 +54,32 @@ pub(crate) fn render_variant_pat( ) -> Option { let _p = profile::span("render_variant_pat"); - let module = ctx.completion.scope.module()?; let fields = variant.fields(ctx.db()); - let n_fields = fields.len(); - let fields = fields - .into_iter() - .filter(|field| field.is_visible_from(ctx.db(), module)) - .collect::>(); - - let fields_omitted = - n_fields - fields.len() > 0 || variant.attrs(ctx.db()).by_key("non_exhaustive").exists(); + let (visible_fields, fields_omitted) = visible_fields(&ctx, &fields, variant)?; let name = local_name.unwrap_or_else(|| variant.name(ctx.db())).to_string(); - let pat = render_pat(&ctx, &name, variant.kind(ctx.db()), &fields, fields_omitted)?; + let pat = render_pat(&ctx, &name, variant.kind(ctx.db()), &visible_fields, fields_omitted)?; - let mut completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) + Some(build_completion(ctx, name, pat, variant)) +} + +fn build_completion( + ctx: RenderContext<'_>, + name: String, + pat: String, + item: impl HasAttrs + Copy, +) -> CompletionItem { + let completion = CompletionItem::new(CompletionKind::Snippet, ctx.source_range(), name) .kind(CompletionItemKind::Binding) - .set_documentation(ctx.docs(variant)) - .set_deprecated(ctx.is_deprecated(variant)) + .set_documentation(ctx.docs(item)) + .set_deprecated(ctx.is_deprecated(item)) .detail(&pat); - if let Some(snippet_cap) = ctx.snippet_cap() { - completion = completion.insert_snippet(snippet_cap, pat); + let completion = if let Some(snippet_cap) = ctx.snippet_cap() { + completion.insert_snippet(snippet_cap, pat) } else { - completion = completion.insert_text(pat); - } - Some(completion.build()) + completion.insert_text(pat) + }; + completion.build() } fn render_pat( -- cgit v1.2.3