diff options
| author | Jacob Kiesel <kieseljake@live.com> | 2022-09-14 13:28:54 -0600 |
|---|---|---|
| committer | Jacob Kiesel <kieseljake@live.com> | 2022-09-14 13:35:35 -0600 |
| commit | 4ffdce09b6bbbeb6462f13cd2a3da311578a679f (patch) | |
| tree | 18696b427d8680e35b53d5e12df576cec5fe3365 | |
| parent | 992560087025370e2a0397a67106b0852e238b4e (diff) | |
| download | rust-4ffdce09b6bbbeb6462f13cd2a3da311578a679f.tar.gz rust-4ffdce09b6bbbeb6462f13cd2a3da311578a679f.zip | |
refactor: use clippy_utils::Sugg instead of direct string ops
| -rw-r--r-- | clippy_lints/src/bool_to_int_with_if.rs | 37 | ||||
| -rw-r--r-- | tests/ui/bool_to_int_with_if.fixed | 4 | ||||
| -rw-r--r-- | tests/ui/bool_to_int_with_if.stderr | 6 |
3 files changed, 21 insertions, 26 deletions
diff --git a/clippy_lints/src/bool_to_int_with_if.rs b/clippy_lints/src/bool_to_int_with_if.rs index bea74279e03..51e98cda845 100644 --- a/clippy_lints/src/bool_to_int_with_if.rs +++ b/clippy_lints/src/bool_to_int_with_if.rs @@ -1,9 +1,9 @@ -use rustc_ast::{ExprPrecedence, LitKind}; +use rustc_ast::LitKind; use rustc_hir::{Block, ExprKind}; use rustc_lint::{LateContext, LateLintPass}; use rustc_session::{declare_lint_pass, declare_tool_lint}; -use clippy_utils::{diagnostics::span_lint_and_then, is_else_clause, source::snippet_block_with_applicability}; +use clippy_utils::{diagnostics::span_lint_and_then, is_else_clause, sugg::Sugg}; use rustc_errors::Applicability; declare_clippy_lint! { @@ -69,28 +69,27 @@ fn check_if_else<'tcx>(ctx: &LateContext<'tcx>, expr: &'tcx rustc_hir::Expr<'tcx return; }; let mut applicability = Applicability::MachineApplicable; - let snippet = snippet_block_with_applicability(ctx, check.span, "..", None, &mut applicability); - - let invert = if inverted { "!" } else { "" }; - let need_parens = should_have_parentheses(check); - - let snippet_with_braces = { - let (left_paren, right_paren) = if need_parens {("(", ")")} else {("", "")}; - format!("{invert}{left_paren}{snippet}{right_paren}") + let snippet = { + let mut sugg = Sugg::hir_with_applicability(ctx, check, "..", &mut applicability); + if inverted { + sugg = !sugg; + } + sugg }; let ty = ctx.typeck_results().expr_ty(then_lit); // then and else must be of same type let suggestion = { let wrap_in_curly = is_else_clause(ctx.tcx, expr); - let (left_curly, right_curly) = if wrap_in_curly {("{", "}")} else {("", "")}; - let (left_paren, right_paren) = if inverted && need_parens {("(", ")")} else {("", "")}; - format!( - "{left_curly}{ty}::from({invert}{left_paren}{snippet}{right_paren}){right_curly}" - ) + let mut s = Sugg::NonParen(format!("{ty}::from({snippet})").into()); + if wrap_in_curly { + s = s.blockify(); + } + s }; // when used in else clause if statement should be wrapped in curly braces - let (inverted_left_paren, inverted_right_paren) = if inverted {("(", ")")} else {("", "")}; + let into_snippet = snippet.clone().maybe_par(); + let as_snippet = snippet.as_ty(ty); span_lint_and_then(ctx, BOOL_TO_INT_WITH_IF, @@ -103,7 +102,7 @@ fn check_if_else<'tcx>(ctx: &LateContext<'tcx>, expr: &'tcx rustc_hir::Expr<'tcx suggestion, applicability, ); - diag.note(format!("`{snippet_with_braces} as {ty}` or `{inverted_left_paren}{snippet_with_braces}{inverted_right_paren}.into()` can also be valid options")); + diag.note(format!("`{as_snippet}` or `{into_snippet}.into()` can also be valid options")); }); }; } @@ -135,7 +134,3 @@ fn check_int_literal_equals_val<'tcx>(expr: &'tcx rustc_hir::Expr<'tcx>, expecte false } } - -fn should_have_parentheses<'tcx>(check: &'tcx rustc_hir::Expr<'tcx>) -> bool { - check.precedence().order() < ExprPrecedence::Cast.order() -} diff --git a/tests/ui/bool_to_int_with_if.fixed b/tests/ui/bool_to_int_with_if.fixed index c48dc941b84..2c8339cdd7f 100644 --- a/tests/ui/bool_to_int_with_if.fixed +++ b/tests/ui/bool_to_int_with_if.fixed @@ -22,12 +22,12 @@ fn main() { // if else if if a { 123 - } else {i32::from(b)}; + } else { i32::from(b) }; // if else if inverted if a { 123 - } else {i32::from(!b)}; + } else { i32::from(!b) }; // Shouldn't lint diff --git a/tests/ui/bool_to_int_with_if.stderr b/tests/ui/bool_to_int_with_if.stderr index cc3e0395aa4..e695440f668 100644 --- a/tests/ui/bool_to_int_with_if.stderr +++ b/tests/ui/bool_to_int_with_if.stderr @@ -33,7 +33,7 @@ LL | | 0 LL | | }; | |_____^ help: replace with from: `i32::from(!a)` | - = note: `!a as i32` or `!a.into()` can also be valid options + = note: `!a as i32` or `(!a).into()` can also be valid options error: boolean to int conversion using if --> $DIR/bool_to_int_with_if.rs:30:5 @@ -80,7 +80,7 @@ LL | | 1 LL | | } else { LL | | 0 LL | | }; - | |_____^ help: replace with from: `{i32::from(b)}` + | |_____^ help: replace with from: `{ i32::from(b) }` | = note: `b as i32` or `b.into()` can also be valid options @@ -93,7 +93,7 @@ LL | | 0 LL | | } else { LL | | 1 LL | | }; - | |_____^ help: replace with from: `{i32::from(!b)}` + | |_____^ help: replace with from: `{ i32::from(!b) }` | = note: `!b as i32` or `(!b).into()` can also be valid options |
