feat: add non-exhaustive-let diagnostic

rosefromthedead · rosefromthedead · commit cb8733fb9e1c · 2024-01-14T15:39:07.000Z
diff --git a/crates/hir-ty/src/diagnostics/expr.rs b/crates/hir-ty/src/diagnostics/expr.rs
@@ -5,6 +5,7 @@
 use std::fmt;
 
 use either::Either;
+use hir_def::hir::Statement;
 use hir_def::lang_item::LangItem;
 use hir_def::{resolver::HasResolver, AdtId, AssocItemId, DefWithBodyId, HasModule};
 use hir_def::{ItemContainerId, Lookup};
@@ -44,6 +45,10 @@ pub enum BodyValidationDiagnostic {
         match_expr: ExprId,
         uncovered_patterns: String,
     },
+    NonExhaustiveLet {
+        pat: PatId,
+        uncovered_patterns: String,
+    },
 }
 
 impl BodyValidationDiagnostic {
@@ -61,7 +66,7 @@ struct ExprValidator {
     owner: DefWithBodyId,
     body: Arc<Body>,
     infer: Arc<InferenceResult>,
-    pub(super) diagnostics: Vec<BodyValidationDiagnostic>,
+    diagnostics: Vec<BodyValidationDiagnostic>,
 }
 
 impl ExprValidator {
@@ -88,6 +93,9 @@ impl ExprValidator {
                 Expr::Call { .. } | Expr::MethodCall { .. } => {
                     self.validate_call(db, id, expr, &mut filter_map_next_checker);
                 }
+                Expr::Block { .. } => {
+                    self.validate_block(db, expr);
+                }
                 _ => {}
             }
         }
@@ -211,7 +219,7 @@ impl ExprValidator {
         if !witnesses.is_empty() {
             self.diagnostics.push(BodyValidationDiagnostic::MissingMatchArms {
                 match_expr,
-                uncovered_patterns: missing_match_arms(&cx, scrut_ty, witnesses, arms),
+                uncovered_patterns: missing_match_arms(&cx, scrut_ty, witnesses, &m_arms),
             });
         }
     }
@@ -231,6 +239,35 @@ impl ExprValidator {
         }
         pattern
     }
+
+    fn validate_block(&mut self, db: &dyn HirDatabase, expr: &Expr) {
+        let Expr::Block { statements, .. } = expr else { return };
+        let pattern_arena = Arena::new();
+        let cx = MatchCheckCtx::new(self.owner.module(db.upcast()), self.owner, db, &pattern_arena);
+        for stmt in &**statements {
+            let Statement::Let { pat, initializer, else_branch: None, .. } = stmt else { continue };
+            let Some(initializer) = initializer else { continue };
+            let init_ty = &self.infer[*initializer];
+
+            let mut have_errors = false;
+            let match_arm = match_check::MatchArm {
+                pat: self.lower_pattern(&cx, *pat, db, &mut have_errors),
+                has_guard: false,
+            };
+            if have_errors {
+                continue;
+            }
+
+            let report = compute_match_usefulness(&cx, &[match_arm], init_ty);
+            let witnesses = report.non_exhaustiveness_witnesses;
+            if !witnesses.is_empty() {
+                self.diagnostics.push(BodyValidationDiagnostic::NonExhaustiveLet {
+                    pat: *pat,
+                    uncovered_patterns: missing_match_arms(&cx, init_ty, witnesses, &[match_arm]),
+                });
+            }
+        }
+    }
 }
 
 struct FilterMapNextChecker {
@@ -371,7 +408,7 @@ fn missing_match_arms<'p>(
     cx: &MatchCheckCtx<'_, 'p>,
     scrut_ty: &Ty,
     witnesses: Vec<DeconstructedPat<'p>>,
-    arms: &[MatchArm],
+    arms: &[match_check::MatchArm<'_>],
 ) -> String {
     struct DisplayWitness<'a, 'p>(&'a DeconstructedPat<'p>, &'a MatchCheckCtx<'a, 'p>);
     impl fmt::Display for DisplayWitness<'_, '_> {
diff --git a/crates/hir/src/diagnostics.rs b/crates/hir/src/diagnostics.rs
@@ -49,6 +49,7 @@ diagnostics![
     MissingUnsafe,
     MovedOutOfRef,
     NeedMut,
+    NonExhaustiveLet,
     NoSuchField,
     PrivateAssocItem,
     PrivateField,
@@ -263,6 +264,12 @@ pub struct MissingMatchArms {
     pub uncovered_patterns: String,
 }
 
+#[derive(Debug)]
+pub struct NonExhaustiveLet {
+    pub pat: InFile<AstPtr<ast::Pat>>,
+    pub uncovered_patterns: String,
+}
+
 #[derive(Debug)]
 pub struct TypeMismatch {
     pub expr_or_pat: InFile<AstPtr<Either<ast::Expr, ast::Pat>>>,
diff --git a/crates/hir/src/lib.rs b/crates/hir/src/lib.rs
@@ -1994,6 +1994,22 @@ impl DefWithBody {
                         Err(SyntheticSyntax) => (),
                     }
                 }
+                BodyValidationDiagnostic::NonExhaustiveLet { pat, uncovered_patterns } => {
+                    match source_map.pat_syntax(pat) {
+                        Ok(source_ptr) => {
+                            if let Some(ast_pat) = source_ptr.value.clone().cast::<ast::Pat>() {
+                                acc.push(
+                                    NonExhaustiveLet {
+                                        pat: InFile::new(source_ptr.file_id, ast_pat),
+                                        uncovered_patterns,
+                                    }
+                                    .into(),
+                                );
+                            }
+                        }
+                        Err(SyntheticSyntax) => {}
+                    }
+                }
             }
         }
 
diff --git a/crates/ide-diagnostics/src/handlers/mutability_errors.rs b/crates/ide-diagnostics/src/handlers/mutability_errors.rs
@@ -815,7 +815,7 @@ fn f() {
 //- minicore: option
 fn f(_: i32) {}
 fn main() {
-    let ((Some(mut x), None) | (_, Some(mut x))) = (None, Some(7));
+    let ((Some(mut x), None) | (_, Some(mut x))) = (None, Some(7)) else { return };
              //^^^^^ 💡 warn: variable does not need to be mutable
     f(x);
 }
diff --git a/crates/ide-diagnostics/src/handlers/non_exhaustive_let.rs b/crates/ide-diagnostics/src/handlers/non_exhaustive_let.rs
@@ -0,0 +1,47 @@
+use crate::{Diagnostic, DiagnosticCode, DiagnosticsContext};
+
+// Diagnostic: non-exhaustive-let
+//
+// This diagnostic is triggered if a `let` statement without an `else` branch has a non-exhaustive
+// pattern.
+pub(crate) fn non_exhaustive_let(
+    ctx: &DiagnosticsContext<'_>,
+    d: &hir::NonExhaustiveLet,
+) -> Diagnostic {
+    Diagnostic::new_with_syntax_node_ptr(
+        ctx,
+        DiagnosticCode::RustcHardError("E0005"),
+        format!("non-exhaustive pattern: {}", d.uncovered_patterns),
+        d.pat.clone().map(Into::into),
+    )
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::tests::check_diagnostics;
+
+    #[test]
+    fn option_nonexhaustive() {
+        check_diagnostics(
+            r#"
+//- minicore: option
+fn main() {
+    let None = Some(5);
+      //^^^^ error: non-exhaustive pattern: `Some(_)` not covered
+}
+"#,
+        );
+    }
+
+    #[test]
+    fn option_exhaustive() {
+        check_diagnostics(
+            r#"
+//- minicore: option
+fn main() {
+    let Some(_) | None = Some(5);
+}
+"#,
+        );
+    }
+}
diff --git a/crates/ide-diagnostics/src/lib.rs b/crates/ide-diagnostics/src/lib.rs
@@ -41,6 +41,7 @@ mod handlers {
     pub(crate) mod moved_out_of_ref;
     pub(crate) mod mutability_errors;
     pub(crate) mod no_such_field;
+    pub(crate) mod non_exhaustive_let;
     pub(crate) mod private_assoc_item;
     pub(crate) mod private_field;
     pub(crate) mod replace_filter_map_next_with_find_map;
@@ -360,6 +361,7 @@ pub fn diagnostics(
             AnyDiagnostic::MissingUnsafe(d) => handlers::missing_unsafe::missing_unsafe(&ctx, &d),
             AnyDiagnostic::MovedOutOfRef(d) => handlers::moved_out_of_ref::moved_out_of_ref(&ctx, &d),
             AnyDiagnostic::NeedMut(d) => handlers::mutability_errors::need_mut(&ctx, &d),
+            AnyDiagnostic::NonExhaustiveLet(d) => handlers::non_exhaustive_let::non_exhaustive_let(&ctx, &d),
             AnyDiagnostic::NoSuchField(d) => handlers::no_such_field::no_such_field(&ctx, &d),
             AnyDiagnostic::PrivateAssocItem(d) => handlers::private_assoc_item::private_assoc_item(&ctx, &d),
             AnyDiagnostic::PrivateField(d) => handlers::private_field::private_field(&ctx, &d),

Original file line number	Diff line number	Diff line change
`@@ -1994,6 +1994,22 @@ impl DefWithBody {`
`1994`	`1994`	`Err(SyntheticSyntax) => (),`
`1995`	`1995`	`}`
`1996`	`1996`	`}`
	`1997`	`+ BodyValidationDiagnostic::NonExhaustiveLet { pat, uncovered_patterns } => {`
	`1998`	`+ match source_map.pat_syntax(pat) {`
	`1999`	`+ Ok(source_ptr) => {`
	`2000`	`+ if let Some(ast_pat) = source_ptr.value.clone().cast::<ast::Pat>() {`
	`2001`	`+ acc.push(`
	`2002`	`+ NonExhaustiveLet {`
	`2003`	`+ pat: InFile::new(source_ptr.file_id, ast_pat),`
	`2004`	`+ uncovered_patterns,`
	`2005`	`+ }`
	`2006`	`+ .into(),`
	`2007`	`+ );`
	`2008`	`+ }`
	`2009`	`+ }`
	`2010`	`+ Err(SyntheticSyntax) => {}`
	`2011`	`+ }`
	`2012`	`+ }`
`1997`	`2013`	`}`
`1998`	`2014`	`}`
`1999`	`2015`
Original file line number	Diff line number	Diff line change
`@@ -815,7 +815,7 @@ fn f() {`
`815`	`815`	`//- minicore: option`
`816`	`816`	`fn f(_: i32) {}`
`817`	`817`	`fn main() {`
`818`		`- let ((Some(mut x), None) \| (_, Some(mut x))) = (None, Some(7));`
	`818`	`+ let ((Some(mut x), None) \| (_, Some(mut x))) = (None, Some(7)) else { return };`
`819`	`819`	`//^^^^^ 💡 warn: variable does not need to be mutable`
`820`	`820`	`f(x);`
`821`	`821`	`}`