1 //! `render` module provides utilities for rendering completion suggestions
2 //! into code pieces that will be presented to user.
5 pub(crate) mod function;
6 pub(crate) mod enum_variant;
8 pub(crate) mod pattern;
9 pub(crate) mod type_alias;
10 pub(crate) mod struct_literal;
14 use hir::{AsAssocItem, HasAttrs, HirDisplay, ScopeDef};
16 helpers::{item_name, SnippetCap},
17 RootDatabase, SymbolKind,
19 use syntax::{SmolStr, SyntaxKind, TextRange};
22 context::{PathCompletionContext, PathKind},
23 item::{CompletionRelevanceTypeMatch, ImportEdit},
24 render::{enum_variant::render_variant, function::render_fn, macro_::render_macro},
25 CompletionContext, CompletionItem, CompletionItemKind, CompletionRelevance,
27 /// Interface for data and methods required for items rendering.
29 pub(crate) struct RenderContext<'a> {
30 completion: &'a CompletionContext<'a>,
33 impl<'a> RenderContext<'a> {
34 pub(crate) fn new(completion: &'a CompletionContext<'a>) -> RenderContext<'a> {
35 RenderContext { completion }
38 fn snippet_cap(&self) -> Option<SnippetCap> {
39 self.completion.config.snippet_cap
42 fn db(&self) -> &'a RootDatabase {
46 fn source_range(&self) -> TextRange {
47 self.completion.source_range()
50 fn is_deprecated(&self, def: impl HasAttrs) -> bool {
51 let attrs = def.attrs(self.db());
52 attrs.by_key("deprecated").exists() || attrs.by_key("rustc_deprecated").exists()
55 fn is_deprecated_assoc_item(&self, as_assoc_item: impl AsAssocItem) -> bool {
57 let assoc = match as_assoc_item.as_assoc_item(db) {
62 let is_assoc_deprecated = match assoc {
63 hir::AssocItem::Function(it) => self.is_deprecated(it),
64 hir::AssocItem::Const(it) => self.is_deprecated(it),
65 hir::AssocItem::TypeAlias(it) => self.is_deprecated(it),
69 .containing_trait_or_trait_impl(db)
70 .map(|trait_| self.is_deprecated(trait_))
75 fn docs(&self, def: impl HasAttrs) -> Option<hir::Documentation> {
80 pub(crate) fn render_field(
81 ctx: RenderContext<'_>,
82 receiver: Option<hir::Name>,
86 let is_deprecated = ctx.is_deprecated(field);
87 let name = field.name(ctx.db()).to_smol_str();
88 let mut item = CompletionItem::new(
91 receiver.map_or_else(|| name.clone(), |receiver| format!("{}.{}", receiver, name).into()),
93 item.set_relevance(CompletionRelevance {
94 type_match: compute_type_match(ctx.completion, ty),
95 exact_name_match: compute_exact_name_match(ctx.completion, name.as_str()),
96 ..CompletionRelevance::default()
98 item.detail(ty.display(ctx.db()).to_string())
99 .set_documentation(field.docs(ctx.db()))
100 .set_deprecated(is_deprecated)
101 .lookup_by(name.clone());
102 let is_keyword = SyntaxKind::from_keyword(name.as_str()).is_some();
103 if is_keyword && !matches!(name.as_str(), "self" | "crate" | "super" | "Self") {
104 item.insert_text(format!("r#{}", name));
106 if let Some(_ref_match) = compute_ref_match(ctx.completion, ty) {
108 // For now we don't properly calculate the edits for ref match
109 // completions on struct fields, so we've disabled them. See #8058.
114 pub(crate) fn render_tuple_field(
115 ctx: RenderContext<'_>,
116 receiver: Option<hir::Name>,
119 ) -> CompletionItem {
120 let mut item = CompletionItem::new(
123 receiver.map_or_else(|| field.to_string(), |receiver| format!("{}.{}", receiver, field)),
125 item.detail(ty.display(ctx.db()).to_string()).lookup_by(field.to_string());
129 pub(crate) fn render_resolution(
130 ctx: RenderContext<'_>,
131 local_name: hir::Name,
132 resolution: ScopeDef,
133 ) -> Option<CompletionItem> {
134 render_resolution_(ctx, local_name, None, resolution)
137 pub(crate) fn render_resolution_with_import(
138 ctx: RenderContext<'_>,
139 import_edit: ImportEdit,
140 ) -> Option<CompletionItem> {
141 let resolution = ScopeDef::from(import_edit.import.original_item);
142 let local_name = match resolution {
143 ScopeDef::ModuleDef(hir::ModuleDef::Function(f)) => f.name(ctx.completion.db),
144 ScopeDef::ModuleDef(hir::ModuleDef::Const(c)) => c.name(ctx.completion.db)?,
145 ScopeDef::ModuleDef(hir::ModuleDef::TypeAlias(t)) => t.name(ctx.completion.db),
146 _ => item_name(ctx.db(), import_edit.import.original_item)?,
148 render_resolution_(ctx, local_name, Some(import_edit), resolution)
151 fn render_resolution_(
152 ctx: RenderContext<'_>,
153 local_name: hir::Name,
154 import_to_add: Option<ImportEdit>,
155 resolution: ScopeDef,
156 ) -> Option<CompletionItem> {
157 let _p = profile::span("render_resolution");
158 use hir::ModuleDef::*;
162 let kind = match resolution {
163 ScopeDef::ModuleDef(Function(func)) => {
164 return Some(render_fn(ctx, import_to_add, Some(local_name), func));
166 ScopeDef::ModuleDef(Variant(var)) if ctx.completion.pattern_ctx.is_none() => {
167 return Some(render_variant(ctx, import_to_add, Some(local_name), var, None));
169 ScopeDef::MacroDef(mac) => return render_macro(ctx, import_to_add, local_name, mac),
170 ScopeDef::Unknown => {
171 let mut item = CompletionItem::new(
172 CompletionItemKind::UnresolvedReference,
174 local_name.to_smol_str(),
176 if let Some(import_to_add) = import_to_add {
177 item.add_import(import_to_add);
179 return Some(item.build());
182 ScopeDef::ModuleDef(Variant(_)) => CompletionItemKind::SymbolKind(SymbolKind::Variant),
183 ScopeDef::ModuleDef(Module(..)) => CompletionItemKind::SymbolKind(SymbolKind::Module),
184 ScopeDef::ModuleDef(Adt(adt)) => CompletionItemKind::SymbolKind(match adt {
185 hir::Adt::Struct(_) => SymbolKind::Struct,
186 hir::Adt::Union(_) => SymbolKind::Union,
187 hir::Adt::Enum(_) => SymbolKind::Enum,
189 ScopeDef::ModuleDef(Const(..)) => CompletionItemKind::SymbolKind(SymbolKind::Const),
190 ScopeDef::ModuleDef(Static(..)) => CompletionItemKind::SymbolKind(SymbolKind::Static),
191 ScopeDef::ModuleDef(Trait(..)) => CompletionItemKind::SymbolKind(SymbolKind::Trait),
192 ScopeDef::ModuleDef(TypeAlias(..)) => CompletionItemKind::SymbolKind(SymbolKind::TypeAlias),
193 ScopeDef::ModuleDef(BuiltinType(..)) => CompletionItemKind::BuiltinType,
194 ScopeDef::GenericParam(param) => CompletionItemKind::SymbolKind(match param {
195 hir::GenericParam::TypeParam(_) => SymbolKind::TypeParam,
196 hir::GenericParam::LifetimeParam(_) => SymbolKind::LifetimeParam,
197 hir::GenericParam::ConstParam(_) => SymbolKind::ConstParam,
199 ScopeDef::Local(..) => CompletionItemKind::SymbolKind(SymbolKind::Local),
200 ScopeDef::Label(..) => CompletionItemKind::SymbolKind(SymbolKind::Label),
201 ScopeDef::AdtSelfType(..) | ScopeDef::ImplSelfType(..) => {
202 CompletionItemKind::SymbolKind(SymbolKind::SelfParam)
206 let local_name = local_name.to_smol_str();
207 let mut item = CompletionItem::new(kind, ctx.source_range(), local_name.clone());
208 if let ScopeDef::Local(local) = resolution {
209 let ty = local.ty(db);
210 if !ty.is_unknown() {
211 item.detail(ty.display(db).to_string());
214 item.set_relevance(CompletionRelevance {
215 type_match: compute_type_match(ctx.completion, &ty),
216 exact_name_match: compute_exact_name_match(ctx.completion, &local_name),
218 ..CompletionRelevance::default()
221 if let Some(ref_match) = compute_ref_match(ctx.completion, &ty) {
222 item.ref_match(ref_match);
226 // Add `<>` for generic types
227 let type_path_no_ty_args = matches!(
228 ctx.completion.path_context,
229 Some(PathCompletionContext { kind: Some(PathKind::Type), has_type_args: false, .. })
230 ) && ctx.completion.config.add_call_parenthesis;
231 if type_path_no_ty_args {
232 if let Some(cap) = ctx.snippet_cap() {
233 let has_non_default_type_params = match resolution {
234 ScopeDef::ModuleDef(Adt(it)) => it.has_non_default_type_params(db),
235 ScopeDef::ModuleDef(TypeAlias(it)) => it.has_non_default_type_params(db),
238 if has_non_default_type_params {
239 cov_mark::hit!(inserts_angle_brackets_for_generics);
240 item.lookup_by(local_name.clone())
241 .label(SmolStr::from_iter([&local_name, "<…>"]))
242 .insert_snippet(cap, format!("{}<$0>", local_name));
246 item.set_documentation(scope_def_docs(db, resolution))
247 .set_deprecated(scope_def_is_deprecated(&ctx, resolution));
249 if let Some(import_to_add) = import_to_add {
250 item.add_import(import_to_add);
255 fn scope_def_docs(db: &RootDatabase, resolution: ScopeDef) -> Option<hir::Documentation> {
256 use hir::ModuleDef::*;
258 ScopeDef::ModuleDef(Module(it)) => it.docs(db),
259 ScopeDef::ModuleDef(Adt(it)) => it.docs(db),
260 ScopeDef::ModuleDef(Variant(it)) => it.docs(db),
261 ScopeDef::ModuleDef(Const(it)) => it.docs(db),
262 ScopeDef::ModuleDef(Static(it)) => it.docs(db),
263 ScopeDef::ModuleDef(Trait(it)) => it.docs(db),
264 ScopeDef::ModuleDef(TypeAlias(it)) => it.docs(db),
269 fn scope_def_is_deprecated(ctx: &RenderContext<'_>, resolution: ScopeDef) -> bool {
271 ScopeDef::ModuleDef(it) => ctx.is_deprecated_assoc_item(it),
272 ScopeDef::MacroDef(it) => ctx.is_deprecated(it),
273 ScopeDef::GenericParam(it) => ctx.is_deprecated(it),
274 ScopeDef::AdtSelfType(it) => ctx.is_deprecated(it),
279 fn compute_type_match(
280 ctx: &CompletionContext,
281 completion_ty: &hir::Type,
282 ) -> Option<CompletionRelevanceTypeMatch> {
283 let expected_type = ctx.expected_type.as_ref()?;
285 // We don't ever consider unit type to be an exact type match, since
286 // nearly always this is not meaningful to the user.
287 if expected_type.is_unit() {
291 if completion_ty == expected_type {
292 Some(CompletionRelevanceTypeMatch::Exact)
293 } else if expected_type.could_unify_with(ctx.db, completion_ty) {
294 Some(CompletionRelevanceTypeMatch::CouldUnify)
300 fn compute_exact_name_match(ctx: &CompletionContext, completion_name: &str) -> bool {
301 ctx.expected_name.as_ref().map_or(false, |name| name.text() == completion_name)
304 fn compute_ref_match(
305 ctx: &CompletionContext,
306 completion_ty: &hir::Type,
307 ) -> Option<hir::Mutability> {
308 let expected_type = ctx.expected_type.as_ref()?;
309 if completion_ty != expected_type {
310 let expected_type_without_ref = expected_type.remove_ref()?;
311 if completion_ty.autoderef(ctx.db).any(|deref_ty| deref_ty == expected_type_without_ref) {
312 cov_mark::hit!(suggest_ref);
313 let mutability = if expected_type.is_mutable_reference() {
316 hir::Mutability::Shared
318 return Some(mutability);
328 use expect_test::{expect, Expect};
329 use ide_db::SymbolKind;
330 use itertools::Itertools;
333 item::CompletionRelevanceTypeMatch,
334 tests::{check_edit, do_completion, get_all_items, TEST_CONFIG},
335 CompletionItem, CompletionItemKind, CompletionRelevance,
339 fn check(ra_fixture: &str, kind: impl Into<CompletionItemKind>, expect: Expect) {
340 let actual = do_completion(ra_fixture, kind.into());
341 expect.assert_debug_eq(&actual);
345 fn check_kinds(ra_fixture: &str, kinds: &[CompletionItemKind], expect: Expect) {
347 kinds.iter().flat_map(|&kind| do_completion(ra_fixture, kind)).collect();
348 expect.assert_debug_eq(&actual);
352 fn check_relevance_for_kinds(ra_fixture: &str, kinds: &[CompletionItemKind], expect: Expect) {
353 let mut actual = get_all_items(TEST_CONFIG, ra_fixture);
354 actual.retain(|it| kinds.contains(&it.kind()));
355 actual.sort_by_key(|it| cmp::Reverse(it.relevance().score()));
356 check_relevance_(actual, expect);
360 fn check_relevance(ra_fixture: &str, expect: Expect) {
361 let mut actual = get_all_items(TEST_CONFIG, ra_fixture);
362 actual.retain(|it| it.kind() != CompletionItemKind::Snippet);
363 actual.retain(|it| it.kind() != CompletionItemKind::Keyword);
364 actual.retain(|it| it.kind() != CompletionItemKind::BuiltinType);
365 actual.sort_by_key(|it| cmp::Reverse(it.relevance().score()));
366 check_relevance_(actual, expect);
370 fn check_relevance_(actual: Vec<CompletionItem>, expect: Expect) {
374 let mut items = vec![];
376 let tag = it.kind().tag();
377 let relevance = display_relevance(it.relevance());
378 items.push(format!("{} {} {}\n", tag, it.label(), relevance));
380 if let Some((mutability, relevance)) = it.ref_match() {
381 let label = format!("&{}{}", mutability.as_keyword_for_ref(), it.label());
382 let relevance = display_relevance(relevance);
384 items.push(format!("{} {} {}\n", tag, label, relevance));
389 .collect::<String>();
391 expect.assert_eq(&actual);
393 fn display_relevance(relevance: CompletionRelevance) -> String {
394 let relevance_factors = vec![
395 (relevance.type_match == Some(CompletionRelevanceTypeMatch::Exact), "type"),
397 relevance.type_match == Some(CompletionRelevanceTypeMatch::CouldUnify),
400 (relevance.exact_name_match, "name"),
401 (relevance.is_local, "local"),
402 (relevance.exact_postfix_snippet_match, "snippet"),
405 .filter_map(|(cond, desc)| if cond { Some(desc) } else { None })
408 format!("[{}]", relevance_factors)
413 fn enum_detail_includes_record_fields() {
416 enum Foo { Foo { x: i32, y: i32 } }
418 fn main() { Foo::Fo$0 }
425 source_range: 54..56,
431 detail: "{x: i32, y: i32}",
439 fn enum_detail_doesnt_include_tuple_fields() {
442 enum Foo { Foo (i32, i32) }
444 fn main() { Foo::Fo$0 }
451 source_range: 46..48,
458 detail: "(i32, i32)",
459 trigger_call_info: true,
467 fn fn_detail_includes_args_and_return_type() {
470 fn foo<T>(a: u32, b: u32, t: T) -> (u32, T) { (a, t) }
474 SymbolKind::Function,
479 source_range: 68..70,
481 insert: "foo(${1:a}, ${2:b}, ${3:t})$0",
486 detail: "fn(u32, u32, T) -> (u32, T)",
487 trigger_call_info: true,
491 source_range: 68..70,
506 fn enum_detail_just_parentheses_for_unit() {
511 fn main() { Foo::Fo$0 }
518 source_range: 35..37,
532 fn lookup_enums_by_two_qualifiers() {
536 pub enum Spam { Foo, Bar(i32) }
538 fn main() { let _: m::Spam = S$0 }
541 CompletionItemKind::SymbolKind(SymbolKind::Function),
542 CompletionItemKind::SymbolKind(SymbolKind::Module),
543 CompletionItemKind::SymbolKind(SymbolKind::Variant),
549 source_range: 75..76,
560 source_range: 75..76,
568 label: "Spam::Bar(…)",
569 source_range: 75..76,
571 insert: "Spam::Bar($0)",
577 relevance: CompletionRelevance {
578 exact_name_match: false,
583 exact_postfix_snippet_match: false,
585 trigger_call_info: true,
588 label: "m::Spam::Foo",
589 source_range: 75..76,
591 insert: "m::Spam::Foo",
597 relevance: CompletionRelevance {
598 exact_name_match: false,
603 exact_postfix_snippet_match: false,
612 fn sets_deprecated_flag_in_items() {
616 fn something_deprecated() {}
617 #[rustc_deprecated(since = "1.0.0")]
618 fn something_else_deprecated() {}
622 SymbolKind::Function,
627 source_range: 127..130,
637 label: "something_deprecated()",
638 source_range: 127..130,
640 insert: "something_deprecated()$0",
644 lookup: "something_deprecated",
649 label: "something_else_deprecated()",
650 source_range: 127..130,
652 insert: "something_else_deprecated()$0",
656 lookup: "something_else_deprecated",
666 struct A { #[deprecated] the_field: u32 }
667 fn foo() { A { the$0 } }
674 source_range: 57..60,
682 relevance: CompletionRelevance {
683 exact_name_match: false,
688 exact_postfix_snippet_match: false,
706 fn bar(self) { self.$0 }
708 &[CompletionItemKind::Method, CompletionItemKind::SymbolKind(SymbolKind::Field)],
713 source_range: 94..94,
719 documentation: Documentation(
725 source_range: 94..94,
732 documentation: Documentation(
755 CompletionItemKind::SymbolKind(SymbolKind::Module),
756 CompletionItemKind::SymbolKind(SymbolKind::Variant),
757 CompletionItemKind::SymbolKind(SymbolKind::Enum),
763 source_range: 10..12,
769 documentation: Documentation(
775 source_range: 10..12,
782 documentation: Documentation(
788 source_range: 10..12,
794 documentation: Documentation(
804 fn dont_render_attrs() {
810 fn the_method(&self) { }
812 fn foo(s: S) { s.$0 }
814 CompletionItemKind::Method,
818 label: "the_method()",
819 source_range: 81..81,
821 insert: "the_method()$0",
823 lookup: "the_method",
832 fn no_call_parens_if_fn_ptr_needed() {
833 cov_mark::check!(no_call_parens_if_fn_ptr_needed);
837 fn foo(foo: u8, bar: u8) {}
838 struct ManualVtable { f: fn(u8, u8) }
840 fn main() -> ManualVtable {
841 ManualVtable { f: f$0 }
845 fn foo(foo: u8, bar: u8) {}
846 struct ManualVtable { f: fn(u8, u8) }
848 fn main() -> ManualVtable {
849 ManualVtable { f: foo }
856 struct RawIdentTable { r#type: u32 }
858 fn main() -> RawIdentTable {
859 RawIdentTable { t$0: 42 }
863 struct RawIdentTable { r#type: u32 }
865 fn main() -> RawIdentTable {
866 RawIdentTable { r#type: 42 }
873 fn no_parens_in_use_item() {
874 cov_mark::check!(no_parens_in_use_item);
878 mod m { pub fn foo() {} }
882 mod m { pub fn foo() {} }
889 fn no_parens_in_call() {
905 impl Foo { fn foo(&self){} }
906 fn f(foo: &Foo) { foo.f$0(); }
910 impl Foo { fn foo(&self){} }
911 fn f(foo: &Foo) { foo.foo(); }
917 fn inserts_angle_brackets_for_generics() {
918 cov_mark::check!(inserts_angle_brackets_for_generics);
944 struct Vec<T = i128> {}
948 struct Vec<T = i128> {}
956 fn foo(xs: Ve$0<i128>)
960 fn foo(xs: Vec<i128>)
966 fn active_param_relevance() {
969 struct S { foo: i64, bar: u32, baz: u32 }
970 fn test(bar: u32) { }
971 fn foo(s: S) { test(s.$0) }
982 fn record_field_relevances() {
985 struct A { foo: i64, bar: u32, baz: u32 }
986 struct B { x: (), y: f32, bar: u32 }
987 fn foo(a: A) { B { bar: a.$0 }; }
998 fn record_field_and_call_relevances() {
1001 struct A { foo: i64, bar: u32, baz: u32 }
1002 struct B { x: (), y: f32, bar: u32 }
1004 fn foo(a: A) { B { bar: f(a.$0) }; }
1014 struct A { foo: i64, bar: u32, baz: u32 }
1015 struct B { x: (), y: f32, bar: u32 }
1017 fn foo(a: A) { f(B { bar: a.$0 }); }
1028 fn prioritize_exact_ref_match() {
1031 struct WorldSnapshot { _f: () };
1032 fn go(world: &WorldSnapshot) { go(w$0) }
1035 lc world [type+name+local]
1043 fn too_many_arguments() {
1044 cov_mark::check!(too_many_arguments);
1048 fn f(foo: &Foo) { f(foo, w$0) }
1059 fn score_fn_type_and_name_match() {
1062 struct A { bar: u8 }
1063 fn baz() -> u8 { 0 }
1064 fn bar() -> u8 { 0 }
1065 fn f() { A { bar: b$0 }; }
1068 fn bar() [type+name]
1077 fn score_method_type_and_name_match() {
1083 fn aaa(&self) -> u32 { 0 }
1084 fn bbb(&self) -> u32 { 0 }
1085 fn ccc(&self) -> u64 { 0 }
1092 me aaa() [type+name]
1100 fn score_method_name_match_only() {
1106 fn aaa(&self) -> u64 { 0 }
1119 fn suggest_ref_mut() {
1120 cov_mark::check!(suggest_ref);
1124 fn foo(s: &mut S) {}
1132 lc &mut s [type+name+local]
1141 fn foo(s: &mut S) {}
1148 lc s [type+name+local]
1157 fn suggest_deref() {
1164 impl core::ops::Deref for T {
1167 fn deref(&self) -> &Self::Target {
1196 fn suggest_deref_mut() {
1199 //- minicore: deref_mut
1203 impl core::ops::Deref for T {
1206 fn deref(&self) -> &Self::Target {
1211 impl core::ops::DerefMut for T {
1212 fn deref_mut(&mut self) -> &mut Self::Target {
1217 fn foo(s: &mut S) {}
1229 lc &mut t [type+local]
1301 fn suggest_deref_fn_ret() {
1308 impl core::ops::Deref for T {
1311 fn deref(&self) -> &Self::Target {
1337 fn struct_field_method_ref() {
1340 struct Foo { bar: u32 }
1341 impl Foo { fn baz(&self) -> u32 { 0 } }
1343 fn foo(f: Foo) { let _: &u32 = f.b$0 }
1345 &[CompletionItemKind::Method, CompletionItemKind::SymbolKind(SymbolKind::Field)],
1347 // Ideally we'd also suggest &f.bar and &f.baz() as exact
1348 // type matches. See #8058.
1353 source_range: 98..99,
1358 detail: "fn(&self) -> u32",
1362 source_range: 98..99,
1379 enum Foo<T> { A(T), B }
1380 // bar() should not be an exact type match
1381 // because the generic parameters are different
1382 fn bar() -> Foo<u8> { Foo::B }
1383 // FIXME baz() should be an exact type match
1384 // because the types could unify, but it currently
1385 // is not. This is due to the T here being
1386 // TyKind::Placeholder rather than TyKind::Missing.
1387 fn baz<T>() -> Foo<T> { Foo::B }
1389 let foo: Foo<u32> = Foo::B;
1390 let _: Foo<u32> = f$0;
1395 ev Foo::A(…) [type_could_unify]
1396 ev Foo::B [type_could_unify]
1406 fn postfix_completion_relevance() {
1407 check_relevance_for_kinds(
1412 fn not(self) -> Self::Output;
1417 fn not(self) -> bool { if self { false } else { true }}
1422 let _: bool = (9 > 2).not$0;
1425 &[CompletionItemKind::Snippet, CompletionItemKind::Method],
1428 me not() (use ops::Not) [type_could_unify]