Use multi-span suggestions

clippy_lints/src/semicolon_block.rs

@@ -1,10 +1,9 @@
-use clippy_utils::diagnostics::span_lint_and_sugg;
+use clippy_utils::diagnostics::{multispan_sugg_with_applicability, span_lint_and_then};
-use clippy_utils::source::snippet_with_macro_callsite;
-use clippy_utils::{get_parent_expr_for_hir, get_parent_node};
 use rustc_errors::Applicability;
-use rustc_hir::{Block, Expr, ExprKind, Node, Stmt, StmtKind};
+use rustc_hir::{Block, Expr, ExprKind, Stmt, StmtKind};
 use rustc_lint::{LateContext, LateLintPass};
 use rustc_session::{declare_lint_pass, declare_tool_lint};
+use rustc_span::Span;
 
 declare_clippy_lint! {
     /// ### What it does
@@ -65,69 +64,73 @@ declare_clippy_lint! {
 declare_lint_pass!(SemicolonBlock => [SEMICOLON_INSIDE_BLOCK, SEMICOLON_OUTSIDE_BLOCK]);
 
 impl LateLintPass<'_> for SemicolonBlock {
-    fn check_block(&mut self, cx: &LateContext<'_>, block: &Block<'_>) {
+    fn check_stmt(&mut self, cx: &LateContext<'_>, stmt: &Stmt<'_>) {
-        semicolon_inside_block(cx, block);
+        match stmt.kind {
-        semicolon_outside_block(cx, block);
+            StmtKind::Expr(Expr {
-    }
+                kind:
-}
+                    ExprKind::Block(
-
+                        block @ Block {
-fn semicolon_inside_block(cx: &LateContext<'_>, block: &Block<'_>) {
+                            expr: None,
-    if !block.span.from_expansion()
+                            stmts:
-        && let Some(tail) = block.expr
+                                &[
-        && let Some(block_expr @ Expr { kind: ExprKind::Block(_, _), ..}) = get_parent_expr_for_hir(cx, block.hir_id)
+                                    ..,
-        && let Some(Node::Stmt(Stmt { kind: StmtKind::Semi(_), span, .. })) = get_parent_node(cx.tcx, block_expr.hir_id)
+                                    Stmt {
-    {
+                                        kind: StmtKind::Semi(expr),
-        let expr_snip = snippet_with_macro_callsite(cx, tail.span, "..");
+                                        span,
-
+                                        ..
-        let mut suggestion: String = snippet_with_macro_callsite(cx, block.span, "..").to_string();
+                                    },
-
+                                ],
-        if let Some((expr_offset, _)) = suggestion.rmatch_indices(&*expr_snip).next() {
+                            ..
-            suggestion.insert(expr_offset + expr_snip.len(), ';');
+                        },
-        } else {
+                        _,
-            return;
+                    ),
+                ..
+            }) if !block.span.from_expansion() => semicolon_outside_block(cx, block, expr, span),
+            StmtKind::Semi(Expr {
+                kind: ExprKind::Block(block @ Block { expr: Some(tail), .. }, _),
+                ..
+            }) if !block.span.from_expansion() => semicolon_inside_block(cx, block, tail, stmt.span),
+            _ => (),
         }
-
-        span_lint_and_sugg(
-            cx,
-            SEMICOLON_INSIDE_BLOCK,
-            *span,
-            "consider moving the `;` inside the block for consistent formatting",
-            "put the `;` here",
-            suggestion,
-            Applicability::MaybeIncorrect,
-        );
     }
 }
 
-fn semicolon_outside_block(cx: &LateContext<'_>, block: &Block<'_>) {
+fn semicolon_inside_block(cx: &LateContext<'_>, block: &Block<'_>, tail: &Expr<'_>, semi_span: Span) {
-    if !block.span.from_expansion()
+    let insert_span = tail.span.with_lo(tail.span.hi());
-        && block.expr.is_none()
+    let remove_span = semi_span.with_lo(block.span.hi());
-        && let [.., Stmt { kind: StmtKind::Semi(expr), .. }] = block.stmts
-        && let Some(block_expr @ Expr { kind: ExprKind::Block(_, _), ..}) = get_parent_expr_for_hir(cx,block.hir_id)
-        && let Some(Node::Stmt(Stmt { kind: StmtKind::Expr(_), .. })) = get_parent_node(cx.tcx, block_expr.hir_id)
-    {
-        let expr_snip = snippet_with_macro_callsite(cx, expr.span, "..");
 
-        let mut suggestion: String = snippet_with_macro_callsite(cx, block.span, "..").to_string();
+    span_lint_and_then(
-
+        cx,
-        if let Some((expr_offset, _)) = suggestion.rmatch_indices(&*expr_snip).next()
+        SEMICOLON_INSIDE_BLOCK,
-            && let Some(semi_offset) = suggestion[expr_offset + expr_snip.len()..].find(';')
+        semi_span,
-        {
+        "consider moving the `;` inside the block for consistent formatting",
-            suggestion.remove(expr_offset +  expr_snip.len() + semi_offset);
+        |diag| {
-        } else {
+            multispan_sugg_with_applicability(
-            return;
+                diag,
-        }
+                "put the `;` here",
-
+                Applicability::MachineApplicable,
-        suggestion.push(';');
+                [(remove_span, String::new()), (insert_span, ";".to_owned())],
-
+            );
-        span_lint_and_sugg(
+        },
-            cx,
+    );
-            SEMICOLON_OUTSIDE_BLOCK,
+}
-            block.span,
+
-            "consider moving the `;` outside the block for consistent formatting",
+fn semicolon_outside_block(cx: &LateContext<'_>, block: &Block<'_>, tail_stmt_expr: &Expr<'_>, semi_span: Span) {
-            "put the `;` outside the block",
+    let insert_span = block.span.with_lo(block.span.hi());
-            suggestion,
+    let remove_span = semi_span.with_lo(tail_stmt_expr.span.source_callsite().hi());
-            Applicability::MaybeIncorrect,
+
-        );
+    span_lint_and_then(
-    }
+        cx,
+        SEMICOLON_OUTSIDE_BLOCK,
+        block.span,
+        "consider moving the `;` outside the block for consistent formatting",
+        |diag| {
+            multispan_sugg_with_applicability(
+                diag,
+                "put the `;` here",
+                Applicability::MachineApplicable,
+                [(remove_span, String::new()), (insert_span, ";".to_owned())],
+            );
+        },
+    );
 }

tests/ui/semicolon_inside_block.fixed

@@ -10,7 +10,7 @@
 
 macro_rules! m {
     (()) => {
-        ()
+        ();
     };
     (0) => {{
         0
@@ -58,7 +58,7 @@ fn main() {
         unit_fn_block();
     };
 
-    { m!(()); }
+    { m!(()) }
     { m!(()); }
     { m!(()); };
     m!(0);

tests/ui/semicolon_inside_block.stderr

@@ -2,15 +2,26 @@ error: consider moving the `;` inside the block for consistent formatting
   --> $DIR/semicolon_inside_block.rs:39:5
    |
 LL |     { unit_fn_block() };
-   |     ^^^^^^^^^^^^^^^^^^^^ help: put the `;` here: `{ unit_fn_block(); }`
+   |     ^^^^^^^^^^^^^^^^^^^^
    |
    = note: `-D clippy::semicolon-inside-block` implied by `-D warnings`
+help: put the `;` here
+   |
+LL -     { unit_fn_block() };
+LL +     { unit_fn_block(); }
+   |
 
 error: consider moving the `;` inside the block for consistent formatting
   --> $DIR/semicolon_inside_block.rs:40:5
    |
 LL |     unsafe { unit_fn_block() };
-   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^ help: put the `;` here: `unsafe { unit_fn_block(); }`
+   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^
+   |
+help: put the `;` here
+   |
+LL -     unsafe { unit_fn_block() };
+LL +     unsafe { unit_fn_block(); }
+   |
 
 error: consider moving the `;` inside the block for consistent formatting
   --> $DIR/semicolon_inside_block.rs:48:5
@@ -23,17 +34,24 @@ LL | |     };
    |
 help: put the `;` here
    |
-LL ~     {
+LL ~         unit_fn_block();
-LL +         unit_fn_block();
+LL ~     }
-LL +         unit_fn_block();
-LL +     }
    |
 
 error: consider moving the `;` inside the block for consistent formatting
   --> $DIR/semicolon_inside_block.rs:61:5
    |
 LL |     { m!(()) };
-   |     ^^^^^^^^^^^ help: put the `;` here: `{ m!(()); }`
+   |     ^^^^^^^^^^^
+   |
+help: put the `;` here
+   |
+LL ~         ();
+LL |     };
+ ...
+LL | 
+LL ~     { m!(()) }
+   |
 
 error: aborting due to 4 previous errors
 

tests/ui/semicolon_outside_block.fixed

@@ -21,6 +21,9 @@ macro_rules! m {
     (2) => {{
         2;
     }};
+    (stmt) => {
+        stmt;
+    };
 }
 
 fn unit_fn_block() {
@@ -39,8 +42,8 @@ fn main() {
     { unit_fn_block() };
     unsafe { unit_fn_block() };
 
-    { unit_fn_block() };
+    { unit_fn_block(); }
-    unsafe { unit_fn_block() };
+    unsafe { unit_fn_block(); }
 
     { unit_fn_block(); };
     unsafe { unit_fn_block(); };
@@ -51,19 +54,20 @@ fn main() {
     };
     {
         unit_fn_block();
-        unit_fn_block()
+        unit_fn_block();
-    };
+    }
     {
         unit_fn_block();
         unit_fn_block();
     };
 
     { m!(()) };
-    { m!(()) };
+    { m!(()); }
     { m!(()); };
     m!(0);
     m!(1);
     m!(2);
+    { m!(stmt) };
 
     for _ in [()] {
         unit_fn_block();

tests/ui/semicolon_outside_block.stderr

@@ -2,15 +2,26 @@ error: consider moving the `;` outside the block for consistent formatting
   --> $DIR/semicolon_outside_block.rs:42:5
    |
 LL |     { unit_fn_block(); }
-   |     ^^^^^^^^^^^^^^^^^^^^ help: put the `;` outside the block: `{ unit_fn_block() };`
+   |     ^^^^^^^^^^^^^^^^^^^^
    |
    = note: `-D clippy::semicolon-outside-block` implied by `-D warnings`
+help: put the `;` here
+   |
+LL -     { unit_fn_block(); }
+LL +     { unit_fn_block() };
+   |
 
 error: consider moving the `;` outside the block for consistent formatting
   --> $DIR/semicolon_outside_block.rs:43:5
    |
 LL |     unsafe { unit_fn_block(); }
-   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^ help: put the `;` outside the block: `unsafe { unit_fn_block() };`
+   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^
+   |
+help: put the `;` here
+   |
+LL -     unsafe { unit_fn_block(); }
+LL +     unsafe { unit_fn_block() };
+   |
 
 error: consider moving the `;` outside the block for consistent formatting
   --> $DIR/semicolon_outside_block.rs:52:5
@@ -21,19 +32,23 @@ LL | |         unit_fn_block();
 LL | |     }
    | |_____^
    |
-help: put the `;` outside the block
+help: put the `;` here
    |
-LL ~     {
+LL ~         unit_fn_block()
-LL +         unit_fn_block();
+LL ~     };
-LL +         unit_fn_block()
-LL +     };
    |
 
 error: consider moving the `;` outside the block for consistent formatting
   --> $DIR/semicolon_outside_block.rs:62:5
    |
 LL |     { m!(()); }
-   |     ^^^^^^^^^^^ help: put the `;` outside the block: `{ m!(()) };`
+   |     ^^^^^^^^^^^
+   |
+help: put the `;` here
+   |
+LL -         ()
+LL +         (); };
+   |
 
 error: aborting due to 4 previous errors