From a195711df068a58069236c2f8d99211f6f231987 Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Wed, 10 Feb 2021 09:29:10 +0100 Subject: Resolve TupleStructPat in SourceAnalyzer::resolve_path --- crates/hir/src/source_analyzer.rs | 17 +++++++++++------ crates/ide/src/references.rs | 23 +++++++++++++++++++++++ 2 files changed, 34 insertions(+), 6 deletions(-) diff --git a/crates/hir/src/source_analyzer.rs b/crates/hir/src/source_analyzer.rs index 626c3078a..bed3fa50f 100644 --- a/crates/hir/src/source_analyzer.rs +++ b/crates/hir/src/source_analyzer.rs @@ -222,8 +222,9 @@ impl SourceAnalyzer { db: &dyn HirDatabase, path: &ast::Path, ) -> Option { + let parent = || path.syntax().parent(); let mut prefer_value_ns = false; - if let Some(path_expr) = path.syntax().parent().and_then(ast::PathExpr::cast) { + if let Some(path_expr) = parent().and_then(ast::PathExpr::cast) { let expr_id = self.expr_id(db, &path_expr.into())?; let infer = self.infer.as_ref()?; if let Some(assoc) = infer.assoc_resolutions_for_expr(expr_id) { @@ -237,7 +238,7 @@ impl SourceAnalyzer { prefer_value_ns = true; } - if let Some(path_pat) = path.syntax().parent().and_then(ast::PathPat::cast) { + if let Some(path_pat) = parent().and_then(ast::PathPat::cast) { let pat_id = self.pat_id(&path_pat.into())?; if let Some(assoc) = self.infer.as_ref()?.assoc_resolutions_for_pat(pat_id) { return Some(PathResolution::AssocItem(assoc.into())); @@ -249,7 +250,7 @@ impl SourceAnalyzer { } } - if let Some(rec_lit) = path.syntax().parent().and_then(ast::RecordExpr::cast) { + if let Some(rec_lit) = parent().and_then(ast::RecordExpr::cast) { let expr_id = self.expr_id(db, &rec_lit.into())?; if let Some(VariantId::EnumVariantId(variant)) = self.infer.as_ref()?.variant_resolution_for_expr(expr_id) @@ -258,8 +259,12 @@ impl SourceAnalyzer { } } - if let Some(rec_pat) = path.syntax().parent().and_then(ast::RecordPat::cast) { - let pat_id = self.pat_id(&rec_pat.into())?; + if let Some(pat) = parent() + .and_then(ast::RecordPat::cast) + .map(ast::Pat::from) + .or_else(|| parent().and_then(ast::TupleStructPat::cast).map(ast::Pat::from)) + { + let pat_id = self.pat_id(&pat)?; if let Some(VariantId::EnumVariantId(variant)) = self.infer.as_ref()?.variant_resolution_for_pat(pat_id) { @@ -272,7 +277,7 @@ impl SourceAnalyzer { // Case where path is a qualifier of another path, e.g. foo::bar::Baz where we // trying to resolve foo::bar. - if let Some(outer_path) = path.syntax().parent().and_then(ast::Path::cast) { + if let Some(outer_path) = parent().and_then(ast::Path::cast) { if let Some(qualifier) = outer_path.qualifier() { if path == &qualifier { return resolve_hir_path_qualifier(db, &self.resolver, &hir_path); diff --git a/crates/ide/src/references.rs b/crates/ide/src/references.rs index 40d9487eb..6999dacee 100644 --- a/crates/ide/src/references.rs +++ b/crates/ide/src/references.rs @@ -1114,4 +1114,27 @@ trait Foo { "#]], ); } + + #[test] + fn test_self_variant_with_payload() { + check( + r#" +enum Foo { Bar() } + +impl Foo { + fn foo(self) { + match self { + Self::Bar$0() => (), + } + } +} + +"#, + expect![[r#" + Bar Variant FileId(0) 11..16 11..14 Other + + FileId(0) 89..92 Other + "#]], + ); + } } -- cgit v1.2.3