clippy_lints/src/strings.rs

   1 use rustc::declare_lint_pass;
   2 use rustc::hir::*;
   3 use rustc::lint::{LateContext, LateLintPass, LintArray, LintPass};
   4 use rustc_errors::Applicability;
   5 use rustc_session::declare_tool_lint;
   6 use syntax::source_map::Spanned;
   7
   8 use if_chain::if_chain;
   9
  10 use crate::utils::SpanlessEq;
  11 use crate::utils::{get_parent_expr, is_allowed, match_type, paths, span_lint, span_lint_and_sugg, walk_ptrs_ty};
  12
  13 declare_clippy_lint! {
  14     /// **What it does:** Checks for string appends of the form `x = x + y` (without
  15     /// `let`!).
  16     ///
  17     /// **Why is this bad?** It's not really bad, but some people think that the
  18     /// `.push_str(_)` method is more readable.
  19     ///
  20     /// **Known problems:** None.
  21     ///
  22     /// **Example:**
  23     ///
  24     /// ```rust
  25     /// let mut x = "Hello".to_owned();
  26     /// x = x + ", World";
  27     /// ```
  28     pub STRING_ADD_ASSIGN,
  29     pedantic,
  30     "using `x = x + ..` where x is a `String` instead of `push_str()`"
  31 }
  32
  33 declare_clippy_lint! {
  34     /// **What it does:** Checks for all instances of `x + _` where `x` is of type
  35     /// `String`, but only if [`string_add_assign`](#string_add_assign) does *not*
  36     /// match.
  37     ///
  38     /// **Why is this bad?** It's not bad in and of itself. However, this particular
  39     /// `Add` implementation is asymmetric (the other operand need not be `String`,
  40     /// but `x` does), while addition as mathematically defined is symmetric, also
  41     /// the `String::push_str(_)` function is a perfectly good replacement.
  42     /// Therefore, some dislike it and wish not to have it in their code.
  43     ///
  44     /// That said, other people think that string addition, having a long tradition
  45     /// in other languages is actually fine, which is why we decided to make this
  46     /// particular lint `allow` by default.
  47     ///
  48     /// **Known problems:** None.
  49     ///
  50     /// **Example:**
  51     ///
  52     /// ```rust
  53     /// let x = "Hello".to_owned();
  54     /// x + ", World";
  55     /// ```
  56     pub STRING_ADD,
  57     restriction,
  58     "using `x + ..` where x is a `String` instead of `push_str()`"
  59 }
  60
  61 declare_clippy_lint! {
  62     /// **What it does:** Checks for the `as_bytes` method called on string literals
  63     /// that contain only ASCII characters.
  64     ///
  65     /// **Why is this bad?** Byte string literals (e.g., `b"foo"`) can be used
  66     /// instead. They are shorter but less discoverable than `as_bytes()`.
  67     ///
  68     /// **Known Problems:** None.
  69     ///
  70     /// **Example:**
  71     /// ```rust
  72     /// let bs = "a byte string".as_bytes();
  73     /// ```
  74     pub STRING_LIT_AS_BYTES,
  75     style,
  76     "calling `as_bytes` on a string literal instead of using a byte string literal"
  77 }
  78
  79 declare_lint_pass!(StringAdd => [STRING_ADD, STRING_ADD_ASSIGN]);
  80
  81 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for StringAdd {
  82     fn check_expr(&mut self, cx: &LateContext<'a, 'tcx>, e: &'tcx Expr) {
  83         if let ExprKind::Binary(
  84             Spanned {
  85                 node: BinOpKind::Add, ..
  86             },
  87             ref left,
  88             _,
  89         ) = e.kind
  90         {
  91             if is_string(cx, left) {
  92                 if !is_allowed(cx, STRING_ADD_ASSIGN, e.hir_id) {
  93                     let parent = get_parent_expr(cx, e);
  94                     if let Some(p) = parent {
  95                         if let ExprKind::Assign(ref target, _) = p.kind {
  96                             // avoid duplicate matches
  97                             if SpanlessEq::new(cx).eq_expr(target, left) {
  98                                 return;
  99                             }
 100                         }
 101                     }
 102                 }
 103                 span_lint(
 104                     cx,
 105                     STRING_ADD,
 106                     e.span,
 107                     "you added something to a string. Consider using `String::push_str()` instead",
 108                 );
 109             }
 110         } else if let ExprKind::Assign(ref target, ref src) = e.kind {
 111             if is_string(cx, target) && is_add(cx, src, target) {
 112                 span_lint(
 113                     cx,
 114                     STRING_ADD_ASSIGN,
 115                     e.span,
 116                     "you assigned the result of adding something to this string. Consider using \
 117                      `String::push_str()` instead",
 118                 );
 119             }
 120         }
 121     }
 122 }
 123
 124 fn is_string(cx: &LateContext<'_, '_>, e: &Expr) -> bool {
 125     match_type(cx, walk_ptrs_ty(cx.tables.expr_ty(e)), &paths::STRING)
 126 }
 127
 128 fn is_add(cx: &LateContext<'_, '_>, src: &Expr, target: &Expr) -> bool {
 129     match src.kind {
 130         ExprKind::Binary(
 131             Spanned {
 132                 node: BinOpKind::Add, ..
 133             },
 134             ref left,
 135             _,
 136         ) => SpanlessEq::new(cx).eq_expr(target, left),
 137         ExprKind::Block(ref block, _) => {
 138             block.stmts.is_empty() && block.expr.as_ref().map_or(false, |expr| is_add(cx, expr, target))
 139         },
 140         _ => false,
 141     }
 142 }
 143
 144 // Max length a b"foo" string can take
 145 const MAX_LENGTH_BYTE_STRING_LIT: usize = 32;
 146
 147 declare_lint_pass!(StringLitAsBytes => [STRING_LIT_AS_BYTES]);
 148
 149 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for StringLitAsBytes {
 150     fn check_expr(&mut self, cx: &LateContext<'a, 'tcx>, e: &'tcx Expr) {
 151         use crate::utils::{snippet, snippet_with_applicability};
 152         use syntax::ast::LitKind;
 153
 154         if_chain! {
 155             if let ExprKind::MethodCall(path, _, args) = &e.kind;
 156             if path.ident.name == sym!(as_bytes);
 157             if let ExprKind::Lit(lit) = &args[0].kind;
 158             if let LitKind::Str(lit_content, _) = &lit.node;
 159             then {
 160                 let callsite = snippet(cx, args[0].span.source_callsite(), r#""foo""#);
 161                 let mut applicability = Applicability::MachineApplicable;
 162                 if callsite.starts_with("include_str!") {
 163                     span_lint_and_sugg(
 164                         cx,
 165                         STRING_LIT_AS_BYTES,
 166                         e.span,
 167                         "calling `as_bytes()` on `include_str!(..)`",
 168                         "consider using `include_bytes!(..)` instead",
 169                         snippet_with_applicability(cx, args[0].span, r#""foo""#, &mut applicability).replacen(
 170                             "include_str",
 171                             "include_bytes",
 172                             1,
 173                         ),
 174                         applicability,
 175                     );
 176                 } else if lit_content.as_str().is_ascii()
 177                     && lit_content.as_str().len() <= MAX_LENGTH_BYTE_STRING_LIT
 178                     && !args[0].span.from_expansion()
 179                 {
 180                     span_lint_and_sugg(
 181                         cx,
 182                         STRING_LIT_AS_BYTES,
 183                         e.span,
 184                         "calling `as_bytes()` on a string literal",
 185                         "consider using a byte string literal instead",
 186                         format!(
 187                             "b{}",
 188                             snippet_with_applicability(cx, args[0].span, r#""foo""#, &mut applicability)
 189                         ),
 190                         applicability,
 191                     );
 192                 }
 193             }
 194         }
 195     }
 196 }