From ee06c07f50fe590c5185807daa19de1d5540e66b Mon Sep 17 00:00:00 2001
From: Lukas Wirth <lukastw97@gmail.com>
Date: Wed, 25 Nov 2020 23:25:10 +0100
Subject: Complete struct in irrefutable let

---
 crates/completion/src/completions/pattern.rs | 57 +++++++++++++++++++++-------
 1 file changed, 43 insertions(+), 14 deletions(-)

(limited to 'crates/completion/src/completions')

diff --git a/crates/completion/src/completions/pattern.rs b/crates/completion/src/completions/pattern.rs
index 7ab7f09fe..4f63ff0ef 100644
--- a/crates/completion/src/completions/pattern.rs
+++ b/crates/completion/src/completions/pattern.rs
@@ -4,7 +4,7 @@ use crate::{CompletionContext, Completions};
 
 /// Completes constats and paths in patterns.
 pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) {
-    if !ctx.is_pat_binding_or_const {
+    if !(ctx.is_pat_binding_or_const || ctx.is_irrefutable_let_pat_binding) {
         return;
     }
     if ctx.record_pat_syntax.is_some() {
@@ -14,20 +14,27 @@ pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) {
     // FIXME: ideally, we should look at the type we are matching against and
     // suggest variants + auto-imports
     ctx.scope.process_all_names(&mut |name, res| {
-        match &res {
-            hir::ScopeDef::ModuleDef(def) => match def {
-                hir::ModuleDef::Adt(hir::Adt::Enum(..))
-                | hir::ModuleDef::Adt(hir::Adt::Struct(..))
-                | hir::ModuleDef::EnumVariant(..)
-                | hir::ModuleDef::Const(..)
-                | hir::ModuleDef::Module(..) => (),
-                _ => return,
-            },
-            hir::ScopeDef::MacroDef(_) => (),
-            _ => return,
+        let add_resolution = match &res {
+            hir::ScopeDef::ModuleDef(def) => {
+                if ctx.is_irrefutable_let_pat_binding {
+                    matches!(def, hir::ModuleDef::Adt(hir::Adt::Struct(_)))
+                } else {
+                    matches!(
+                        def,
+                        hir::ModuleDef::Adt(hir::Adt::Enum(..))
+                            | hir::ModuleDef::Adt(hir::Adt::Struct(..))
+                            | hir::ModuleDef::EnumVariant(..)
+                            | hir::ModuleDef::Const(..)
+                            | hir::ModuleDef::Module(..)
+                    )
+                }
+            }
+            hir::ScopeDef::MacroDef(_) => true,
+            _ => false,
         };
-
-        acc.add_resolution(ctx, name.to_string(), &res)
+        if add_resolution {
+            acc.add_resolution(ctx, name.to_string(), &res);
+        }
     });
 }
 
@@ -85,4 +92,26 @@ fn foo() {
             "#]],
         );
     }
+
+    #[test]
+    fn completes_in_irrefutable_let() {
+        check(
+            r#"
+enum E { X }
+use self::E::X;
+const Z: E = E::X;
+mod m {}
+
+static FOO: E = E::X;
+struct Bar { f: u32 }
+
+fn foo() {
+   let <|>
+}
+"#,
+            expect![[r#"
+                st Bar
+            "#]],
+        );
+    }
 }
-- 
cgit v1.2.3