aboutsummaryrefslogtreecommitdiff
path: root/crates/ide_completion/src
diff options
context:
space:
mode:
Diffstat (limited to 'crates/ide_completion/src')
-rw-r--r--crates/ide_completion/src/item.rs123
-rw-r--r--crates/ide_completion/src/lib.rs5
-rw-r--r--crates/ide_completion/src/render.rs22
3 files changed, 117 insertions, 33 deletions
diff --git a/crates/ide_completion/src/item.rs b/crates/ide_completion/src/item.rs
index cf1aaa131..3febab32b 100644
--- a/crates/ide_completion/src/item.rs
+++ b/crates/ide_completion/src/item.rs
@@ -70,7 +70,7 @@ pub struct CompletionItem {
70 /// Note that Relevance ignores fuzzy match score. We compute Relevance for 70 /// Note that Relevance ignores fuzzy match score. We compute Relevance for
71 /// all possible items, and then separately build an ordered completion list 71 /// all possible items, and then separately build an ordered completion list
72 /// based on relevance and fuzzy matching with the already typed identifier. 72 /// based on relevance and fuzzy matching with the already typed identifier.
73 relevance: Relevance, 73 relevance: CompletionRelevance,
74 74
75 /// Indicates that a reference or mutable reference to this variable is a 75 /// Indicates that a reference or mutable reference to this variable is a
76 /// possible match. 76 /// possible match.
@@ -107,9 +107,11 @@ impl fmt::Debug for CompletionItem {
107 if self.deprecated { 107 if self.deprecated {
108 s.field("deprecated", &true); 108 s.field("deprecated", &true);
109 } 109 }
110 if self.relevance.is_relevant() { 110
111 if self.relevance != CompletionRelevance::default() {
111 s.field("relevance", &self.relevance); 112 s.field("relevance", &self.relevance);
112 } 113 }
114
113 if let Some(mutability) = &self.ref_match { 115 if let Some(mutability) = &self.ref_match {
114 s.field("ref_match", &format!("&{}", mutability.as_keyword_for_ref())); 116 s.field("ref_match", &format!("&{}", mutability.as_keyword_for_ref()));
115 } 117 }
@@ -120,16 +122,8 @@ impl fmt::Debug for CompletionItem {
120 } 122 }
121} 123}
122 124
123#[derive(Debug, Clone, Copy, Ord, PartialOrd, Eq, PartialEq)]
124pub enum CompletionScore {
125 /// If only type match
126 TypeMatch,
127 /// If type and name match
128 TypeAndNameMatch,
129}
130
131#[derive(Debug, Clone, Copy, Ord, PartialOrd, Eq, PartialEq, Default)] 125#[derive(Debug, Clone, Copy, Ord, PartialOrd, Eq, PartialEq, Default)]
132pub struct Relevance { 126pub struct CompletionRelevance {
133 /// This is set in cases like these: 127 /// This is set in cases like these:
134 /// 128 ///
135 /// ``` 129 /// ```
@@ -152,9 +146,34 @@ pub struct Relevance {
152 pub exact_type_match: bool, 146 pub exact_type_match: bool,
153} 147}
154 148
155impl Relevance { 149impl CompletionRelevance {
150 /// Provides a relevance score. Higher values are more relevant.
151 ///
152 /// The absolute value of the relevance score is not meaningful, for
153 /// example a value of 0 doesn't mean "not relevant", rather
154 /// it means "least relevant". The score value should only be used
155 /// for relative ordering.
156 ///
157 /// See is_relevant if you need to make some judgement about score
158 /// in an absolute sense.
159 pub fn score(&self) -> u32 {
160 let mut score = 0;
161
162 if self.exact_name_match {
163 score += 1;
164 }
165 if self.exact_type_match {
166 score += 1;
167 }
168
169 score
170 }
171
172 /// Returns true when the score for this threshold is above
173 /// some threshold such that we think it is especially likely
174 /// to be relevant.
156 pub fn is_relevant(&self) -> bool { 175 pub fn is_relevant(&self) -> bool {
157 self != &Relevance::default() 176 self.score() > 0
158 } 177 }
159} 178}
160 179
@@ -249,7 +268,7 @@ impl CompletionItem {
249 text_edit: None, 268 text_edit: None,
250 deprecated: false, 269 deprecated: false,
251 trigger_call_info: None, 270 trigger_call_info: None,
252 relevance: Relevance::default(), 271 relevance: CompletionRelevance::default(),
253 ref_match: None, 272 ref_match: None,
254 import_to_add: None, 273 import_to_add: None,
255 } 274 }
@@ -292,7 +311,7 @@ impl CompletionItem {
292 self.deprecated 311 self.deprecated
293 } 312 }
294 313
295 pub fn relevance(&self) -> Relevance { 314 pub fn relevance(&self) -> CompletionRelevance {
296 self.relevance 315 self.relevance
297 } 316 }
298 317
@@ -300,8 +319,14 @@ impl CompletionItem {
300 self.trigger_call_info 319 self.trigger_call_info
301 } 320 }
302 321
303 pub fn ref_match(&self) -> Option<Mutability> { 322 pub fn ref_match(&self) -> Option<(Mutability, CompletionRelevance)> {
304 self.ref_match 323 // Relevance of the ref match should be the same as the original
324 // match, but with exact type match set because self.ref_match
325 // is only set if there is an exact type match.
326 let mut relevance = self.relevance;
327 relevance.exact_type_match = true;
328
329 self.ref_match.map(|mutability| (mutability, relevance))
305 } 330 }
306 331
307 pub fn import_to_add(&self) -> Option<&ImportEdit> { 332 pub fn import_to_add(&self) -> Option<&ImportEdit> {
@@ -349,7 +374,7 @@ pub(crate) struct Builder {
349 text_edit: Option<TextEdit>, 374 text_edit: Option<TextEdit>,
350 deprecated: bool, 375 deprecated: bool,
351 trigger_call_info: Option<bool>, 376 trigger_call_info: Option<bool>,
352 relevance: Relevance, 377 relevance: CompletionRelevance,
353 ref_match: Option<Mutability>, 378 ref_match: Option<Mutability>,
354} 379}
355 380
@@ -457,7 +482,7 @@ impl Builder {
457 self.deprecated = deprecated; 482 self.deprecated = deprecated;
458 self 483 self
459 } 484 }
460 pub(crate) fn set_relevance(&mut self, relevance: Relevance) -> &mut Builder { 485 pub(crate) fn set_relevance(&mut self, relevance: CompletionRelevance) -> &mut Builder {
461 self.relevance = relevance; 486 self.relevance = relevance;
462 self 487 self
463 } 488 }
@@ -474,3 +499,63 @@ impl Builder {
474 self 499 self
475 } 500 }
476} 501}
502
503#[cfg(test)]
504mod tests {
505 use itertools::Itertools;
506 use test_utils::assert_eq_text;
507
508 use super::CompletionRelevance;
509
510 /// Check that these are CompletionRelevance are sorted in ascending order
511 /// by their relevance score.
512 ///
513 /// We want to avoid making assertions about the absolute score of any
514 /// item, but we do want to assert whether each is >, <, or == to the
515 /// others.
516 ///
517 /// If provided vec![vec![a], vec![b, c], vec![d]], then this will assert:
518 /// a.score < b.score == c.score < d.score
519 fn check_relevance_score_ordered(expected_relevance_order: Vec<Vec<CompletionRelevance>>) {
520 let expected = format!("{:#?}", &expected_relevance_order);
521
522 let actual_relevance_order = expected_relevance_order
523 .into_iter()
524 .flatten()
525 .map(|r| (r.score(), r))
526 .sorted_by_key(|(score, _r)| *score)
527 .fold(
528 (u32::MIN, vec![vec![]]),
529 |(mut currently_collecting_score, mut out), (score, r)| {
530 if currently_collecting_score == score {
531 out.last_mut().unwrap().push(r);
532 } else {
533 currently_collecting_score = score;
534 out.push(vec![r]);
535 }
536 (currently_collecting_score, out)
537 },
538 )
539 .1;
540
541 let actual = format!("{:#?}", &actual_relevance_order);
542
543 assert_eq_text!(&expected, &actual);
544 }
545
546 #[test]
547 fn relevance_score() {
548 // This test asserts that the relevance score for these items is ascending, and
549 // that any items in the same vec have the same score.
550 let expected_relevance_order = vec![
551 vec![CompletionRelevance::default()],
552 vec![
553 CompletionRelevance { exact_name_match: true, ..CompletionRelevance::default() },
554 CompletionRelevance { exact_type_match: true, ..CompletionRelevance::default() },
555 ],
556 vec![CompletionRelevance { exact_name_match: true, exact_type_match: true }],
557 ];
558
559 check_relevance_score_ordered(expected_relevance_order);
560 }
561}
diff --git a/crates/ide_completion/src/lib.rs b/crates/ide_completion/src/lib.rs
index d46f521a0..263554ecf 100644
--- a/crates/ide_completion/src/lib.rs
+++ b/crates/ide_completion/src/lib.rs
@@ -23,10 +23,7 @@ use crate::{completions::Completions, context::CompletionContext, item::Completi
23 23
24pub use crate::{ 24pub use crate::{
25 config::CompletionConfig, 25 config::CompletionConfig,
26 item::{ 26 item::{CompletionItem, CompletionItemKind, CompletionRelevance, ImportEdit, InsertTextFormat},
27 CompletionItem, CompletionItemKind, CompletionScore, ImportEdit, InsertTextFormat,
28 Relevance,
29 },
30}; 27};
31 28
32//FIXME: split the following feature into fine-grained features. 29//FIXME: split the following feature into fine-grained features.
diff --git a/crates/ide_completion/src/render.rs b/crates/ide_completion/src/render.rs
index f7f9084d9..db31896e5 100644
--- a/crates/ide_completion/src/render.rs
+++ b/crates/ide_completion/src/render.rs
@@ -20,7 +20,7 @@ use ide_db::{
20use syntax::TextRange; 20use syntax::TextRange;
21 21
22use crate::{ 22use crate::{
23 item::{ImportEdit, Relevance}, 23 item::{CompletionRelevance, ImportEdit},
24 CompletionContext, CompletionItem, CompletionItemKind, CompletionKind, 24 CompletionContext, CompletionItem, CompletionItemKind, CompletionKind,
25}; 25};
26 26
@@ -322,9 +322,9 @@ impl<'a> Render<'a> {
322 } 322 }
323} 323}
324 324
325fn compute_relevance(ctx: &RenderContext, ty: &Type, name: &str) -> Option<Relevance> { 325fn compute_relevance(ctx: &RenderContext, ty: &Type, name: &str) -> Option<CompletionRelevance> {
326 let (expected_name, expected_type) = ctx.expected_name_and_type()?; 326 let (expected_name, expected_type) = ctx.expected_name_and_type()?;
327 let mut res = Relevance::default(); 327 let mut res = CompletionRelevance::default();
328 res.exact_type_match = ty == &expected_type; 328 res.exact_type_match = ty == &expected_type;
329 res.exact_name_match = name == &expected_name; 329 res.exact_name_match = name == &expected_name;
330 Some(res) 330 Some(res)
@@ -338,7 +338,7 @@ mod tests {
338 338
339 use crate::{ 339 use crate::{
340 test_utils::{check_edit, do_completion, get_all_items, TEST_CONFIG}, 340 test_utils::{check_edit, do_completion, get_all_items, TEST_CONFIG},
341 CompletionKind, Relevance, 341 CompletionKind, CompletionRelevance,
342 }; 342 };
343 343
344 fn check(ra_fixture: &str, expect: Expect) { 344 fn check(ra_fixture: &str, expect: Expect) {
@@ -347,12 +347,14 @@ mod tests {
347 } 347 }
348 348
349 fn check_relevance(ra_fixture: &str, expect: Expect) { 349 fn check_relevance(ra_fixture: &str, expect: Expect) {
350 fn display_relevance(relevance: Relevance) -> &'static str { 350 fn display_relevance(relevance: CompletionRelevance) -> &'static str {
351 match relevance { 351 match relevance {
352 Relevance { exact_type_match: true, exact_name_match: true } => "[type+name]", 352 CompletionRelevance { exact_type_match: true, exact_name_match: true } => {
353 Relevance { exact_type_match: true, exact_name_match: false } => "[type]", 353 "[type+name]"
354 Relevance { exact_type_match: false, exact_name_match: true } => "[name]", 354 }
355 Relevance { exact_type_match: false, exact_name_match: false } => "[]", 355 CompletionRelevance { exact_type_match: true, exact_name_match: false } => "[type]",
356 CompletionRelevance { exact_type_match: false, exact_name_match: true } => "[name]",
357 CompletionRelevance { exact_type_match: false, exact_name_match: false } => "[]",
356 } 358 }
357 } 359 }
358 360
@@ -975,7 +977,7 @@ fn main() {
975 Local, 977 Local,
976 ), 978 ),
977 detail: "S", 979 detail: "S",
978 relevance: Relevance { 980 relevance: CompletionRelevance {
979 exact_name_match: true, 981 exact_name_match: true,
980 exact_type_match: false, 982 exact_type_match: false,
981 }, 983 },