From c20afbdfe017a904180f01499139adfafe296bbf Mon Sep 17 00:00:00 2001
From: Marcin Serwin <toxyxer@gmail.com>
Date: Wed, 15 Apr 2020 18:37:37 +0200
Subject: [PATCH] Lint map_flatten if caller is an Option
---
clippy_lints/src/methods/mod.rs | 20 +++++++++++++++++++-
1 file changed, 19 insertions(+), 1 deletion(-)
diff --git a/clippy_lints/src/methods/mod.rs b/clippy_lints/src/methods/mod.rs
index 5255fec2cec..608b351c3e1 100644
--- a/clippy_lints/src/methods/mod.rs
+++ b/clippy_lints/src/methods/mod.rs
@@ -2484,7 +2484,7 @@ fn lint_ok_expect(cx: &LateContext<'_, '_>, expr: &hir::Expr<'_>, ok_args: &[hir
}
}
-/// lint use of `map().flatten()` for `Iterators`
+/// lint use of `map().flatten()` for `Iterators` and 'Options'
fn lint_map_flatten<'a, 'tcx>(cx: &LateContext<'a, 'tcx>, expr: &'tcx hir::Expr<'_>, map_args: &'tcx [hir::Expr<'_>]) {
// lint if caller of `.map().flatten()` is an Iterator
if match_trait_method(cx, expr, &paths::ITERATOR) {
@@ -2503,6 +2503,24 @@ fn lint_map_flatten<'a, 'tcx>(cx: &LateContext<'a, 'tcx>, expr: &'tcx hir::Expr<
Applicability::MachineApplicable,
);
}
+
+ // lint if caller of `.map().flatten()` is an Option
+ if match_type(cx, cx.tables.expr_ty(&map_args[0]), &paths::OPTION) {
+ let msg = "called `map(..).flatten()` on an `Option`. \
+ This is more succinctly expressed by calling `.and_then(..)`";
+ let self_snippet = snippet(cx, map_args[0].span, "..");
+ let func_snippet = snippet(cx, map_args[1].span, "..");
+ let hint = format!("{0}.and_then({1})", self_snippet, func_snippet);
+ span_lint_and_sugg(
+ cx,
+ MAP_FLATTEN,
+ expr.span,
+ msg,
+ "try using `and_then` instead",
+ hint,
+ Applicability::MachineApplicable,
+ );
+ }
}
/// lint use of `map().unwrap_or_else()` for `Option`s and `Result`s