Use approximate_suggestion for non-reducible closures

This commit is contained in:
Philipp Hansch 2018-04-15 11:35:20 +02:00
parent 7de707fdba
commit d87385b406
No known key found for this signature in database
GPG Key ID: B6FA06A6E0E2665B
3 changed files with 67 additions and 17 deletions

View File

@ -89,18 +89,27 @@ fn reduce_unit_expression<'a>(cx: &LateContext, expr: &'a hir::Expr) -> Option<S
hir::ExprBlock(ref block) => {
match (&block.stmts[..], block.expr.as_ref()) {
(&[], Some(inner_expr)) => {
// Reduce `{ X }` to `X`
// If block only contains an expression,
// reduce `{ X }` to `X`
reduce_unit_expression(cx, inner_expr)
},
(&[ref inner_stmt], None) => {
// Reduce `{ X; }` to `X` or `X;`
// If block only contains statements,
// reduce `{ X; }` to `X` or `X;`
match inner_stmt.node {
hir::StmtDecl(ref d, _) => Some(d.span),
hir::StmtExpr(ref e, _) => Some(e.span),
hir::StmtSemi(_, _) => Some(inner_stmt.span),
}
},
_ => None,
_ => {
// For closures that contain multiple statements
// it's difficult to get a correct suggestion span
// for all cases (multi-line closures specifically)
//
// We do not attempt to build a suggestion for those right now.
None
}
}
},
_ => None,
@ -142,25 +151,36 @@ fn lint_map_unit_fn(cx: &LateContext, stmt: &hir::Stmt, expr: &hir::Expr, map_ar
OPTION_MAP_UNIT_FN,
expr.span,
msg,
|db| { db.span_suggestion(stmt.span, "try this", suggestion); });
|db| { db.span_approximate_suggestion(stmt.span, "try this", suggestion); });
} else if let Some((binding, closure_expr)) = unit_closure(cx, fn_arg) {
let msg = "called `map(f)` on an Option value where `f` is a unit closure";
enum Suggestion {
Full(String),
Approx(String)
}
let suggestion = if let Some(expr_span) = reduce_unit_expression(cx, closure_expr) {
format!("if let Some({0}) = {1} {{ {2} }}",
snippet(cx, binding.pat.span, "_"),
snippet(cx, var_arg.span, "_"),
snippet(cx, expr_span, "_"))
Suggestion::Full(
format!("if let Some({0}) = {1} {{ {2} }}",
snippet(cx, binding.pat.span, "_"),
snippet(cx, var_arg.span, "_"),
snippet(cx, expr_span, "_"))
)
} else {
format!("if let Some({0}) = {1} {{ ... }}",
snippet(cx, binding.pat.span, "_"),
snippet(cx, var_arg.span, "_"))
Suggestion::Approx(
format!("if let Some({0}) = {1} {{ ... }}",
snippet(cx, binding.pat.span, "_"),
snippet(cx, var_arg.span, "_"))
)
};
span_lint_and_then(cx,
OPTION_MAP_UNIT_FN,
expr.span,
msg,
|db| { db.span_suggestion(stmt.span, "try this", suggestion); });
span_lint_and_then(cx, OPTION_MAP_UNIT_FN, expr.span, msg, |db| {
match suggestion {
Suggestion::Full(sugg) => db.span_suggestion(stmt.span, "try this", sugg),
Suggestion::Approx(sugg) => db.span_approximate_suggestion(stmt.span, "try this", sugg),
};
});
}
}

View File

@ -78,4 +78,12 @@ fn main() {
x.field.map(|value| { do_nothing(value); do_nothing(value) });
x.field.map(|value| if value > 0 { do_nothing(value); do_nothing(value) });
// Suggestion for the let block should be `{ ... }` as it's too difficult to build a
// proper suggestion for these cases
x.field.map(|value| {
do_nothing(value);
do_nothing(value)
});
x.field.map(|value| { do_nothing(value); do_nothing(value); });
}

View File

@ -152,5 +152,27 @@ error: called `map(f)` on an Option value where `f` is a unit closure
| |
| help: try this: `if let Some(value) = x.field { ... }`
error: aborting due to 19 previous errors
error: called `map(f)` on an Option value where `f` is a unit closure
--> $DIR/option_map_unit_fn.rs:84:5
|
84 | x.field.map(|value| {
| _____^
| |_____|
| ||
85 | || do_nothing(value);
86 | || do_nothing(value)
87 | || });
| ||______^- help: try this: `if let Some(value) = x.field { ... }`
| |_______|
|
error: called `map(f)` on an Option value where `f` is a unit closure
--> $DIR/option_map_unit_fn.rs:88:5
|
88 | x.field.map(|value| { do_nothing(value); do_nothing(value); });
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^-
| |
| help: try this: `if let Some(value) = x.field { ... }`
error: aborting due to 21 previous errors