Different suggestions for when associated functions are referred to

This commit is contained in:
clubby789 2022-10-21 14:38:44 +01:00
parent 0940040c04
commit 717bf35366

View file

@ -38,8 +38,8 @@ type Res = def::Res<ast::NodeId>;
/// A field or associated item from self type suggested in case of resolution failure. /// A field or associated item from self type suggested in case of resolution failure.
enum AssocSuggestion { enum AssocSuggestion {
Field, Field,
MethodWithSelf, MethodWithSelf { called: bool },
AssocFn, AssocFn { called: bool },
AssocType, AssocType,
AssocConst, AssocConst,
} }
@ -48,8 +48,14 @@ impl AssocSuggestion {
fn action(&self) -> &'static str { fn action(&self) -> &'static str {
match self { match self {
AssocSuggestion::Field => "use the available field", AssocSuggestion::Field => "use the available field",
AssocSuggestion::MethodWithSelf => "call the method with the fully-qualified path", AssocSuggestion::MethodWithSelf { called: true } => {
AssocSuggestion::AssocFn => "call the associated function", "call the method with the fully-qualified path"
}
AssocSuggestion::MethodWithSelf { called: false } => {
"refer to the method with the fully-qualified path"
}
AssocSuggestion::AssocFn { called: true } => "call the associated function",
AssocSuggestion::AssocFn { called: false } => "refer to the associated function",
AssocSuggestion::AssocConst => "use the associated `const`", AssocSuggestion::AssocConst => "use the associated `const`",
AssocSuggestion::AssocType => "use the associated type", AssocSuggestion::AssocType => "use the associated type",
} }
@ -498,7 +504,9 @@ impl<'a: 'ast, 'ast> LateResolutionVisitor<'a, '_, 'ast> {
// Try Levenshtein algorithm. // Try Levenshtein algorithm.
let typo_sugg = self.lookup_typo_candidate(path, source.namespace(), is_expected); let typo_sugg = self.lookup_typo_candidate(path, source.namespace(), is_expected);
if path.len() == 1 && self.self_type_is_available() { if path.len() == 1 && self.self_type_is_available() {
if let Some(candidate) = self.lookup_assoc_candidate(ident, ns, is_expected) { if let Some(candidate) =
self.lookup_assoc_candidate(ident, ns, is_expected, source.is_call())
{
let self_is_available = self.self_value_is_available(path[0].ident.span); let self_is_available = self.self_value_is_available(path[0].ident.span);
match candidate { match candidate {
AssocSuggestion::Field => { AssocSuggestion::Field => {
@ -513,16 +521,21 @@ impl<'a: 'ast, 'ast> LateResolutionVisitor<'a, '_, 'ast> {
err.span_label(span, "a field by this name exists in `Self`"); err.span_label(span, "a field by this name exists in `Self`");
} }
} }
AssocSuggestion::MethodWithSelf if self_is_available => { AssocSuggestion::MethodWithSelf { called } if self_is_available => {
let msg = if called {
"you might have meant to call the method"
} else {
"you might have meant to refer to the method"
};
err.span_suggestion( err.span_suggestion(
span, span,
"you might have meant to call the method", msg,
format!("self.{path_str}"), format!("self.{path_str}"),
Applicability::MachineApplicable, Applicability::MachineApplicable,
); );
} }
AssocSuggestion::MethodWithSelf AssocSuggestion::MethodWithSelf { .. }
| AssocSuggestion::AssocFn | AssocSuggestion::AssocFn { .. }
| AssocSuggestion::AssocConst | AssocSuggestion::AssocConst
| AssocSuggestion::AssocType => { | AssocSuggestion::AssocType => {
err.span_suggestion( err.span_suggestion(
@ -1494,6 +1507,7 @@ impl<'a: 'ast, 'ast> LateResolutionVisitor<'a, '_, 'ast> {
ident: Ident, ident: Ident,
ns: Namespace, ns: Namespace,
filter_fn: FilterFn, filter_fn: FilterFn,
called: bool,
) -> Option<AssocSuggestion> ) -> Option<AssocSuggestion>
where where
FilterFn: Fn(Res) -> bool, FilterFn: Fn(Res) -> bool,
@ -1535,9 +1549,9 @@ impl<'a: 'ast, 'ast> LateResolutionVisitor<'a, '_, 'ast> {
return Some(match &assoc_item.kind { return Some(match &assoc_item.kind {
ast::AssocItemKind::Const(..) => AssocSuggestion::AssocConst, ast::AssocItemKind::Const(..) => AssocSuggestion::AssocConst,
ast::AssocItemKind::Fn(box ast::Fn { sig, .. }) if sig.decl.has_self() => { ast::AssocItemKind::Fn(box ast::Fn { sig, .. }) if sig.decl.has_self() => {
AssocSuggestion::MethodWithSelf AssocSuggestion::MethodWithSelf { called }
} }
ast::AssocItemKind::Fn(..) => AssocSuggestion::AssocFn, ast::AssocItemKind::Fn(..) => AssocSuggestion::AssocFn { called },
ast::AssocItemKind::Type(..) => AssocSuggestion::AssocType, ast::AssocItemKind::Type(..) => AssocSuggestion::AssocType,
ast::AssocItemKind::MacCall(_) => continue, ast::AssocItemKind::MacCall(_) => continue,
}); });
@ -1556,10 +1570,12 @@ impl<'a: 'ast, 'ast> LateResolutionVisitor<'a, '_, 'ast> {
let res = binding.res(); let res = binding.res();
if filter_fn(res) { if filter_fn(res) {
if self.r.has_self.contains(&res.def_id()) { if self.r.has_self.contains(&res.def_id()) {
return Some(AssocSuggestion::MethodWithSelf); return Some(AssocSuggestion::MethodWithSelf { called });
} else { } else {
match res { match res {
Res::Def(DefKind::AssocFn, _) => return Some(AssocSuggestion::AssocFn), Res::Def(DefKind::AssocFn, _) => {
return Some(AssocSuggestion::AssocFn { called });
}
Res::Def(DefKind::AssocConst, _) => { Res::Def(DefKind::AssocConst, _) => {
return Some(AssocSuggestion::AssocConst); return Some(AssocSuggestion::AssocConst);
} }