aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--crates/ra_ide/src/completion/complete_record_literal.rs61
1 files changed, 60 insertions, 1 deletions
diff --git a/crates/ra_ide/src/completion/complete_record_literal.rs b/crates/ra_ide/src/completion/complete_record_literal.rs
index 83ed1d52c..a0bc3ee0a 100644
--- a/crates/ra_ide/src/completion/complete_record_literal.rs
+++ b/crates/ra_ide/src/completion/complete_record_literal.rs
@@ -11,8 +11,24 @@ pub(super) fn complete_record_literal(acc: &mut Completions, ctx: &CompletionCon
11 _ => return, 11 _ => return,
12 }; 12 };
13 13
14 let already_present_names: Vec<String> = ctx
15 .record_lit_syntax
16 .as_ref()
17 .and_then(|record_literal| record_literal.record_field_list())
18 .map(|field_list| field_list.fields())
19 .map(|fields| {
20 fields
21 .into_iter()
22 .filter_map(|field| field.name_ref())
23 .map(|name_ref| name_ref.to_string())
24 .collect()
25 })
26 .unwrap_or_default();
27
14 for (field, field_ty) in ty.variant_fields(ctx.db, variant) { 28 for (field, field_ty) in ty.variant_fields(ctx.db, variant) {
15 acc.add_field(ctx, field, &field_ty); 29 if !already_present_names.contains(&field.name(ctx.db).to_string()) {
30 acc.add_field(ctx, field, &field_ty);
31 }
16 } 32 }
17} 33}
18 34
@@ -178,4 +194,47 @@ mod tests {
178 ] 194 ]
179 "###); 195 "###);
180 } 196 }
197
198 #[test]
199 fn only_missing_fields_are_completed() {
200 let completions = complete(
201 r"
202 struct S {
203 foo1: u32,
204 foo2: u32,
205 bar: u32,
206 baz: u32,
207 }
208
209 fn main() {
210 let foo1 = 1;
211 let s = S {
212 foo1,
213 foo2: 5,
214 <|>
215 }
216 }
217 ",
218 );
219 assert_debug_snapshot!(completions, @r###"
220 [
221 CompletionItem {
222 label: "bar",
223 source_range: [302; 302),
224 delete: [302; 302),
225 insert: "bar",
226 kind: Field,
227 detail: "u32",
228 },
229 CompletionItem {
230 label: "baz",
231 source_range: [302; 302),
232 delete: [302; 302),
233 insert: "baz",
234 kind: Field,
235 detail: "u32",
236 },
237 ]
238 "###);
239 }
181} 240}