From 7e956ac7c4a38c862e344e5e60243b3995bc5831 Mon Sep 17 00:00:00 2001
From: mcarton <cartonmartin+git@gmail.com>
Date: Fri, 29 Sep 2017 19:13:21 +0200
Subject: [PATCH] Fix regression with `print!`

---
 clippy_lints/src/format.rs         | 29 -----------------------------
 clippy_lints/src/print.rs          | 26 ++++++++++++++++++++------
 tests/ui/print_with_newline.stderr | 28 ++++++++++++++++++++++++++++
 3 files changed, 48 insertions(+), 35 deletions(-)

diff --git a/clippy_lints/src/format.rs b/clippy_lints/src/format.rs
index 26b500d554635..6a6cbadb6fa3e 100644
--- a/clippy_lints/src/format.rs
+++ b/clippy_lints/src/format.rs
@@ -1,9 +1,7 @@
 use rustc::hir::*;
-use rustc::hir::map::Node::NodeItem;
 use rustc::lint::*;
 use rustc::ty;
 use syntax::ast::LitKind;
-use syntax::symbol::InternedString;
 use utils::paths;
 use utils::{is_expn_of, match_def_path, match_type, resolve_node, span_lint, walk_ptrs_ty, opt_def_id};
 
@@ -70,33 +68,6 @@ impl<'a, 'tcx> LateLintPass<'a, 'tcx> for Pass {
     }
 }
 
-/// Returns the slice of format string parts in an `Arguments::new_v1` call.
-/// Public because it's shared with a lint in print.rs.
-pub fn get_argument_fmtstr_parts<'a, 'b>(cx: &LateContext<'a, 'b>, expr: &'a Expr) -> Option<Vec<InternedString>> {
-    if_let_chain! {[
-        let ExprBlock(ref block) = expr.node,
-        block.stmts.len() == 1,
-        let StmtDecl(ref decl, _) = block.stmts[0].node,
-        let DeclItem(ref decl) = decl.node,
-        let Some(NodeItem(decl)) = cx.tcx.hir.find(decl.id),
-        decl.name == "__STATIC_FMTSTR",
-        let ItemStatic(_, _, ref expr) = decl.node,
-        let ExprAddrOf(_, ref expr) = cx.tcx.hir.body(*expr).value.node, // &["…", "…", …]
-        let ExprArray(ref exprs) = expr.node,
-    ], {
-        let mut result = Vec::new();
-        for expr in exprs {
-            if let ExprLit(ref lit) = expr.node {
-                if let LitKind::Str(ref lit, _) = lit.node {
-                    result.push(lit.as_str());
-                }
-            }
-        }
-        return Some(result);
-    }}
-    None
-}
-
 /// Checks if the expressions matches `&[""]`
 fn check_static_str(expr: &Expr) -> bool {
     if_let_chain! {[
diff --git a/clippy_lints/src/print.rs b/clippy_lints/src/print.rs
index 078e208467aa5..96557b8b0cb4a 100644
--- a/clippy_lints/src/print.rs
+++ b/clippy_lints/src/print.rs
@@ -1,9 +1,10 @@
 use rustc::hir::*;
 use rustc::hir::map::Node::{NodeImplItem, NodeItem};
 use rustc::lint::*;
-use utils::{paths, opt_def_id};
+use syntax::ast::LitKind;
+use syntax::symbol::InternedString;
 use utils::{is_expn_of, match_def_path, match_path, resolve_node, span_lint};
-use format::get_argument_fmtstr_parts;
+use utils::{paths, opt_def_id};
 
 /// **What it does:** This lint warns when you using `print!()` with a format
 /// string that
@@ -103,15 +104,14 @@ impl<'a, 'tcx> LateLintPass<'a, 'tcx> for Pass {
                         let ExprTup(ref args) = args.node,
 
                         // collect the format string parts and check the last one
-                        let Some(fmtstrs) = get_argument_fmtstr_parts(cx, &args_args[0]),
-                        let Some(last_str) = fmtstrs.last(),
-                        let Some('\n') = last_str.chars().last(),
+                        let Some((fmtstr, fmtlen)) = get_argument_fmtstr_parts(&args_args[0]),
+                        let Some('\n') = fmtstr.chars().last(),
 
                         // "foo{}bar" is made into two strings + one argument,
                         // if the format string starts with `{}` (eg. "{}foo"),
                         // the string array is prepended an empty string "".
                         // We only want to check the last string after any `{}`:
-                        args.len() < fmtstrs.len(),
+                        args.len() < fmtlen,
                     ], {
                         span_lint(cx, PRINT_WITH_NEWLINE, span,
                                   "using `print!()` with a format string that ends in a \
@@ -150,3 +150,17 @@ fn is_in_debug_impl(cx: &LateContext, expr: &Expr) -> bool {
 
     false
 }
+
+/// Returns the slice of format string parts in an `Arguments::new_v1` call.
+fn get_argument_fmtstr_parts(expr: &Expr) -> Option<(InternedString, usize)> {
+    if_let_chain! {[
+        let ExprAddrOf(_, ref expr) = expr.node, // &["…", "…", …]
+        let ExprArray(ref exprs) = expr.node,
+        let Some(expr) = exprs.last(),
+        let ExprLit(ref lit) = expr.node,
+        let LitKind::Str(ref lit, _) = lit.node,
+    ], {
+        return Some((lit.as_str(), exprs.len()));
+    }}
+    None
+}
diff --git a/tests/ui/print_with_newline.stderr b/tests/ui/print_with_newline.stderr
index e69de29bb2d1d..1bacc40bfb4fd 100644
--- a/tests/ui/print_with_newline.stderr
+++ b/tests/ui/print_with_newline.stderr
@@ -0,0 +1,28 @@
+error: using `print!()` with a format string that ends in a newline, consider using `println!()` instead
+ --> $DIR/print_with_newline.rs:6:5
+  |
+6 |     print!("Hello/n");
+  |     ^^^^^^^^^^^^^^^^^^
+  |
+  = note: `-D print-with-newline` implied by `-D warnings`
+
+error: using `print!()` with a format string that ends in a newline, consider using `println!()` instead
+ --> $DIR/print_with_newline.rs:7:5
+  |
+7 |     print!("Hello {}/n", "world");
+  |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+
+error: using `print!()` with a format string that ends in a newline, consider using `println!()` instead
+ --> $DIR/print_with_newline.rs:8:5
+  |
+8 |     print!("Hello {} {}/n/n", "world", "#2");
+  |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+
+error: using `print!()` with a format string that ends in a newline, consider using `println!()` instead
+ --> $DIR/print_with_newline.rs:9:5
+  |
+9 |     print!("{}/n", 1265);
+  |     ^^^^^^^^^^^^^^^^^^^^^
+
+error: aborting due to 4 previous errors
+