clippy_lints/src/matches.rs

   1 use rustc::hir::*;
   2 use rustc::lint::*;
   3 use rustc::middle::const_val::ConstVal;
   4 use rustc::ty;
   5 use rustc_const_eval::EvalHint::ExprTypeChecked;
   6 use rustc_const_eval::ConstContext;
   7 use rustc_const_math::ConstInt;
   8 use std::cmp::Ordering;
   9 use std::collections::Bound;
  10 use syntax::ast::LitKind;
  11 use syntax::codemap::Span;
  12 use utils::paths;
  13 use utils::{match_type, snippet, span_note_and_lint, span_lint_and_then, in_external_macro, expr_block, walk_ptrs_ty, is_expn_of};
  14 use utils::sugg::Sugg;
  15
  16 /// **What it does:** Checks for matches with a single arm where an `if let`
  17 /// will usually suffice.
  18 ///
  19 /// **Why is this bad?** Just readability – `if let` nests less than a `match`.
  20 ///
  21 /// **Known problems:** None.
  22 ///
  23 /// **Example:**
  24 /// ```rust
  25 /// match x {
  26 ///     Some(ref foo) => bar(foo),
  27 ///     _ => ()
  28 /// }
  29 /// ```
  30 declare_lint! {
  31     pub SINGLE_MATCH,
  32     Warn,
  33     "a match statement with a single nontrivial arm (i.e, where the other arm \
  34      is `_ => {}`) instead of `if let`"
  35 }
  36
  37 /// **What it does:** Checks for matches with a two arms where an `if let` will
  38 /// usually suffice.
  39 ///
  40 /// **Why is this bad?** Just readability – `if let` nests less than a `match`.
  41 ///
  42 /// **Known problems:** Personal style preferences may differ.
  43 ///
  44 /// **Example:**
  45 /// ```rust
  46 /// match x {
  47 ///     Some(ref foo) => bar(foo),
  48 ///     _ => bar(other_ref),
  49 /// }
  50 /// ```
  51 declare_lint! {
  52     pub SINGLE_MATCH_ELSE,
  53     Allow,
  54     "a match statement with a two arms where the second arm's pattern is a wildcard \
  55      instead of `if let`"
  56 }
  57
  58 /// **What it does:** Checks for matches where all arms match a reference,
  59 /// suggesting to remove the reference and deref the matched expression
  60 /// instead. It also checks for `if let &foo = bar` blocks.
  61 ///
  62 /// **Why is this bad?** It just makes the code less readable. That reference
  63 /// destructuring adds nothing to the code.
  64 ///
  65 /// **Known problems:** None.
  66 ///
  67 /// **Example:**
  68 /// ```rust
  69 /// match x {
  70 ///     &A(ref y) => foo(y),
  71 ///     &B => bar(),
  72 ///     _ => frob(&x),
  73 /// }
  74 /// ```
  75 declare_lint! {
  76     pub MATCH_REF_PATS,
  77     Warn,
  78     "a match or `if let` with all arms prefixed with `&` instead of deref-ing the match expression"
  79 }
  80
  81 /// **What it does:** Checks for matches where match expression is a `bool`. It
  82 /// suggests to replace the expression with an `if...else` block.
  83 ///
  84 /// **Why is this bad?** It makes the code less readable.
  85 ///
  86 /// **Known problems:** None.
  87 ///
  88 /// **Example:**
  89 /// ```rust
  90 /// let condition: bool = true;
  91 /// match condition {
  92 ///     true => foo(),
  93 ///     false => bar(),
  94 /// }
  95 /// ```
  96 declare_lint! {
  97     pub MATCH_BOOL,
  98     Warn,
  99     "a match on a boolean expression instead of an `if..else` block"
 100 }
 101
 102 /// **What it does:** Checks for overlapping match arms.
 103 ///
 104 /// **Why is this bad?** It is likely to be an error and if not, makes the code
 105 /// less obvious.
 106 ///
 107 /// **Known problems:** None.
 108 ///
 109 /// **Example:**
 110 /// ```rust
 111 /// let x = 5;
 112 /// match x {
 113 ///     1 ... 10 => println!("1 ... 10"),
 114 ///     5 ... 15 => println!("5 ... 15"),
 115 ///     _ => (),
 116 /// }
 117 /// ```
 118 declare_lint! {
 119     pub MATCH_OVERLAPPING_ARM,
 120     Warn,
 121     "a match with overlapping arms"
 122 }
 123
 124 /// **What it does:** Checks for arm which matches all errors with `Err(_)`
 125 /// and take drastic actions like `panic!`.
 126 ///
 127 /// **Why is this bad?** It is generally a bad practice, just like
 128 /// catching all exceptions in java with `catch(Exception)`
 129 ///
 130 /// **Known problems:** None.
 131 ///
 132 /// **Example:**
 133 /// ```rust
 134 /// let x : Result(i32, &str) = Ok(3);
 135 /// match x {
 136 ///     Ok(_) => println!("ok"),
 137 ///     Err(_) => panic!("err"),
 138 /// }
 139 /// ```
 140 declare_lint! {
 141     pub MATCH_WILD_ERR_ARM,
 142     Warn,
 143     "a match with `Err(_)` arm and take drastic actions"
 144 }
 145
 146 #[allow(missing_copy_implementations)]
 147 pub struct MatchPass;
 148
 149 impl LintPass for MatchPass {
 150     fn get_lints(&self) -> LintArray {
 151         lint_array!(SINGLE_MATCH,
 152                     MATCH_REF_PATS,
 153                     MATCH_BOOL,
 154                     SINGLE_MATCH_ELSE,
 155                     MATCH_OVERLAPPING_ARM,
 156                     MATCH_WILD_ERR_ARM)
 157     }
 158 }
 159
 160 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for MatchPass {
 161     fn check_expr(&mut self, cx: &LateContext<'a, 'tcx>, expr: &'tcx Expr) {
 162         if in_external_macro(cx, expr.span) {
 163             return;
 164         }
 165         if let ExprMatch(ref ex, ref arms, MatchSource::Normal) = expr.node {
 166             check_single_match(cx, ex, arms, expr);
 167             check_match_bool(cx, ex, arms, expr);
 168             check_overlapping_arms(cx, ex, arms);
 169             check_wild_err_arm(cx, ex, arms);
 170         }
 171         if let ExprMatch(ref ex, ref arms, source) = expr.node {
 172             check_match_ref_pats(cx, ex, arms, source, expr);
 173         }
 174     }
 175 }
 176
 177 #[cfg_attr(rustfmt, rustfmt_skip)]
 178 fn check_single_match(cx: &LateContext, ex: &Expr, arms: &[Arm], expr: &Expr) {
 179     if arms.len() == 2 &&
 180       arms[0].pats.len() == 1 && arms[0].guard.is_none() &&
 181       arms[1].pats.len() == 1 && arms[1].guard.is_none() {
 182         let els = if is_unit_expr(&arms[1].body) {
 183             None
 184         } else if let ExprBlock(_) = arms[1].body.node {
 185             // matches with blocks that contain statements are prettier as `if let + else`
 186             Some(&*arms[1].body)
 187         } else {
 188             // allow match arms with just expressions
 189             return;
 190         };
 191         let ty = cx.tables.expr_ty(ex);
 192         if ty.sty != ty::TyBool || cx.current_level(MATCH_BOOL) == Allow {
 193             check_single_match_single_pattern(cx, ex, arms, expr, els);
 194             check_single_match_opt_like(cx, ex, arms, expr, ty, els);
 195         }
 196     }
 197 }
 198
 199 fn check_single_match_single_pattern(cx: &LateContext, ex: &Expr, arms: &[Arm], expr: &Expr, els: Option<&Expr>) {
 200     if arms[1].pats[0].node == PatKind::Wild {
 201         let lint = if els.is_some() {
 202             SINGLE_MATCH_ELSE
 203         } else {
 204             SINGLE_MATCH
 205         };
 206         let els_str = els.map_or(String::new(), |els| format!(" else {}", expr_block(cx, els, None, "..")));
 207         span_lint_and_then(cx,
 208                            lint,
 209                            expr.span,
 210                            "you seem to be trying to use match for destructuring a single pattern. \
 211                            Consider using `if let`",
 212                            |db| {
 213             db.span_suggestion(expr.span,
 214                                "try this",
 215                                format!("if let {} = {} {}{}",
 216                                        snippet(cx, arms[0].pats[0].span, ".."),
 217                                        snippet(cx, ex.span, ".."),
 218                                        expr_block(cx, &arms[0].body, None, ".."),
 219                                        els_str));
 220         });
 221     }
 222 }
 223
 224 fn check_single_match_opt_like(
 225     cx: &LateContext,
 226     ex: &Expr,
 227     arms: &[Arm],
 228     expr: &Expr,
 229     ty: ty::Ty,
 230     els: Option<&Expr>
 231 ) {
 232     // list of candidate Enums we know will never get any more members
 233     let candidates = &[(&paths::COW, "Borrowed"),
 234                        (&paths::COW, "Cow::Borrowed"),
 235                        (&paths::COW, "Cow::Owned"),
 236                        (&paths::COW, "Owned"),
 237                        (&paths::OPTION, "None"),
 238                        (&paths::RESULT, "Err"),
 239                        (&paths::RESULT, "Ok")];
 240
 241     let path = match arms[1].pats[0].node {
 242         PatKind::TupleStruct(ref path, ref inner, _) => {
 243             // contains any non wildcard patterns? e.g. Err(err)
 244             if inner.iter().any(|pat| pat.node != PatKind::Wild) {
 245                 return;
 246             }
 247             print::to_string(print::NO_ANN, |s| s.print_qpath(path, false))
 248         },
 249         PatKind::Binding(BindByValue(MutImmutable), _, ident, None) => ident.node.to_string(),
 250         PatKind::Path(ref path) => print::to_string(print::NO_ANN, |s| s.print_qpath(path, false)),
 251         _ => return,
 252     };
 253
 254     for &(ty_path, pat_path) in candidates {
 255         if &path == pat_path && match_type(cx, ty, ty_path) {
 256             let lint = if els.is_some() {
 257                 SINGLE_MATCH_ELSE
 258             } else {
 259                 SINGLE_MATCH
 260             };
 261             let els_str = els.map_or(String::new(), |els| format!(" else {}", expr_block(cx, els, None, "..")));
 262             span_lint_and_then(cx,
 263                                lint,
 264                                expr.span,
 265                                "you seem to be trying to use match for destructuring a single pattern. Consider \
 266                                 using `if let`",
 267                                |db| {
 268                 db.span_suggestion(expr.span,
 269                                    "try this",
 270                                    format!("if let {} = {} {}{}",
 271                                            snippet(cx, arms[0].pats[0].span, ".."),
 272                                            snippet(cx, ex.span, ".."),
 273                                            expr_block(cx, &arms[0].body, None, ".."),
 274                                            els_str));
 275             });
 276         }
 277     }
 278 }
 279
 280 fn check_match_bool(cx: &LateContext, ex: &Expr, arms: &[Arm], expr: &Expr) {
 281     // type of expression == bool
 282     if cx.tables.expr_ty(ex).sty == ty::TyBool {
 283         span_lint_and_then(cx,
 284                            MATCH_BOOL,
 285                            expr.span,
 286                            "you seem to be trying to match on a boolean expression",
 287                            move |db| {
 288             if arms.len() == 2 && arms[0].pats.len() == 1 {
 289                 // no guards
 290                 let exprs = if let PatKind::Lit(ref arm_bool) = arms[0].pats[0].node {
 291                     if let ExprLit(ref lit) = arm_bool.node {
 292                         match lit.node {
 293                             LitKind::Bool(true) => Some((&*arms[0].body, &*arms[1].body)),
 294                             LitKind::Bool(false) => Some((&*arms[1].body, &*arms[0].body)),
 295                             _ => None,
 296                         }
 297                     } else {
 298                         None
 299                     }
 300                 } else {
 301                     None
 302                 };
 303
 304                 if let Some((true_expr, false_expr)) = exprs {
 305                     let sugg = match (is_unit_expr(true_expr), is_unit_expr(false_expr)) {
 306                         (false, false) => {
 307                             Some(format!("if {} {} else {}",
 308                                          snippet(cx, ex.span, "b"),
 309                                          expr_block(cx, true_expr, None, ".."),
 310                                          expr_block(cx, false_expr, None, "..")))
 311                         },
 312                         (false, true) => {
 313                             Some(format!("if {} {}", snippet(cx, ex.span, "b"), expr_block(cx, true_expr, None, "..")))
 314                         },
 315                         (true, false) => {
 316                             let test = Sugg::hir(cx, ex, "..");
 317                             Some(format!("if {} {}", !test, expr_block(cx, false_expr, None, "..")))
 318                         },
 319                         (true, true) => None,
 320                     };
 321
 322                     if let Some(sugg) = sugg {
 323                         db.span_suggestion(expr.span, "consider using an if/else expression", sugg);
 324                     }
 325                 }
 326             }
 327
 328         });
 329     }
 330 }
 331
 332 fn check_overlapping_arms(cx: &LateContext, ex: &Expr, arms: &[Arm]) {
 333     if arms.len() >= 2 && cx.tables.expr_ty(ex).is_integral() {
 334         let ranges = all_ranges(cx, arms);
 335         let type_ranges = type_ranges(&ranges);
 336         if !type_ranges.is_empty() {
 337             if let Some((start, end)) = overlapping(&type_ranges) {
 338                 span_note_and_lint(cx,
 339                                    MATCH_OVERLAPPING_ARM,
 340                                    start.span,
 341                                    "some ranges overlap",
 342                                    end.span,
 343                                    "overlaps with this");
 344             }
 345         }
 346     }
 347 }
 348
 349 fn check_wild_err_arm(cx: &LateContext, ex: &Expr, arms: &[Arm]) {
 350     let ex_ty = walk_ptrs_ty(cx.tables.expr_ty(ex));
 351     if match_type(cx, ex_ty, &paths::RESULT) {
 352         for arm in arms {
 353             if let PatKind::TupleStruct(ref path, ref inner, _) = arm.pats[0].node {
 354                 let path_str = print::to_string(print::NO_ANN, |s| s.print_qpath(path, false));
 355                 if inner.iter().any(|pat| pat.node == PatKind::Wild) &&
 356                     path_str == "Err" {
 357                         // `Err(_)` arm found
 358                         if_let_chain! {[
 359                             let ExprBlock(ref block) = arm.body.node,
 360                             is_panic_block(cx, block)
 361                         ], {
 362                             span_note_and_lint(cx,
 363                                                MATCH_WILD_ERR_ARM,
 364                                                arm.pats[0].span,
 365                                                "Err(_) will match all errors, maybe not a good idea",
 366                                                arm.pats[0].span,
 367                                                "to remove this warning, match each error seperately or use unreachable macro");
 368                         }}
 369                 }
 370             }
 371         }
 372     }
 373 }
 374
 375 // If the block contains only a `panic!` macro (as expression or statement)
 376 fn is_panic_block(cx: &LateContext, block: &Block) -> bool {
 377     match (&block.expr, block.stmts.len(), block.stmts.first()) {
 378         (&Some(ref exp), 0, _) => is_expn_of(cx, exp.span, "panic").is_some() && is_expn_of(cx, exp.span, "unreachable").is_none(),
 379         (&None, 1, Some(ref stmt)) => is_expn_of(cx, stmt.span, "panic").is_some() && is_expn_of(cx, stmt.span, "unreachable").is_none(),
 380         _ => false
 381     }
 382 }
 383
 384 fn check_match_ref_pats(cx: &LateContext, ex: &Expr, arms: &[Arm], source: MatchSource, expr: &Expr) {
 385     if has_only_ref_pats(arms) {
 386         if let ExprAddrOf(Mutability::MutImmutable, ref inner) = ex.node {
 387             span_lint_and_then(cx,
 388                                MATCH_REF_PATS,
 389                                expr.span,
 390                                "you don't need to add `&` to both the expression and the patterns",
 391                                |db| {
 392                 let inner = Sugg::hir(cx, inner, "..");
 393                 let template = match_template(expr.span, source, inner);
 394                 db.span_suggestion(expr.span, "try", template);
 395             });
 396         } else {
 397             span_lint_and_then(cx,
 398                                MATCH_REF_PATS,
 399                                expr.span,
 400                                "you don't need to add `&` to all patterns",
 401                                |db| {
 402                 let ex = Sugg::hir(cx, ex, "..");
 403                 let template = match_template(expr.span, source, ex.deref());
 404                 db.span_suggestion(expr.span,
 405                                    "instead of prefixing all patterns with `&`, you can dereference the expression",
 406                                    template);
 407             });
 408         }
 409     }
 410 }
 411
 412 /// Get all arms that are unbounded `PatRange`s.
 413 fn all_ranges(cx: &LateContext, arms: &[Arm]) -> Vec<SpannedRange<ConstVal>> {
 414     let constcx = ConstContext::with_tables(cx.tcx, cx.tables);
 415     arms.iter()
 416         .flat_map(|arm| {
 417             if let Arm { ref pats, guard: None, .. } = *arm {
 418                     pats.iter()
 419                 } else {
 420                     [].iter()
 421                 }
 422                 .filter_map(|pat| {
 423                     if_let_chain! {[
 424                     let PatKind::Range(ref lhs, ref rhs, ref range_end) = pat.node,
 425                     let Ok(lhs) = constcx.eval(lhs, ExprTypeChecked),
 426                     let Ok(rhs) = constcx.eval(rhs, ExprTypeChecked)
 427                 ], {
 428                     let rhs = match *range_end {
 429                         RangeEnd::Included => Bound::Included(rhs),
 430                         RangeEnd::Excluded => Bound::Excluded(rhs),
 431                     };
 432                     return Some(SpannedRange { span: pat.span, node: (lhs, rhs) });
 433                 }}
 434
 435                     if_let_chain! {[
 436                     let PatKind::Lit(ref value) = pat.node,
 437                     let Ok(value) = constcx.eval(value, ExprTypeChecked)
 438                 ], {
 439                     return Some(SpannedRange { span: pat.span, node: (value.clone(), Bound::Included(value)) });
 440                 }}
 441
 442                     None
 443                 })
 444         })
 445         .collect()
 446 }
 447
 448 #[derive(Debug, Eq, PartialEq)]
 449 pub struct SpannedRange<T> {
 450     pub span: Span,
 451     pub node: (T, Bound<T>),
 452 }
 453
 454 type TypedRanges = Vec<SpannedRange<ConstInt>>;
 455
 456 /// Get all `Int` ranges or all `Uint` ranges. Mixed types are an error anyway and other types than
 457 /// `Uint` and `Int` probably don't make sense.
 458 fn type_ranges(ranges: &[SpannedRange<ConstVal>]) -> TypedRanges {
 459     ranges.iter()
 460         .filter_map(|range| match range.node {
 461             (ConstVal::Integral(start), Bound::Included(ConstVal::Integral(end))) => {
 462                 Some(SpannedRange {
 463                     span: range.span,
 464                     node: (start, Bound::Included(end)),
 465                 })
 466             },
 467             (ConstVal::Integral(start), Bound::Excluded(ConstVal::Integral(end))) => {
 468                 Some(SpannedRange {
 469                     span: range.span,
 470                     node: (start, Bound::Excluded(end)),
 471                 })
 472             },
 473             (ConstVal::Integral(start), Bound::Unbounded) => {
 474                 Some(SpannedRange {
 475                     span: range.span,
 476                     node: (start, Bound::Unbounded),
 477                 })
 478             },
 479             _ => None,
 480         })
 481         .collect()
 482 }
 483
 484 fn is_unit_expr(expr: &Expr) -> bool {
 485     match expr.node {
 486         ExprTup(ref v) if v.is_empty() => true,
 487         ExprBlock(ref b) if b.stmts.is_empty() && b.expr.is_none() => true,
 488         _ => false,
 489     }
 490 }
 491
 492 fn has_only_ref_pats(arms: &[Arm]) -> bool {
 493     let mapped = arms.iter()
 494         .flat_map(|a| &a.pats)
 495         .map(|p| {
 496             match p.node {
 497                 PatKind::Ref(..) => Some(true),  // &-patterns
 498                 PatKind::Wild => Some(false),   // an "anything" wildcard is also fine
 499                 _ => None,                    // any other pattern is not fine
 500             }
 501         })
 502         .collect::<Option<Vec<bool>>>();
 503     // look for Some(v) where there's at least one true element
 504     mapped.map_or(false, |v| v.iter().any(|el| *el))
 505 }
 506
 507 fn match_template(span: Span, source: MatchSource, expr: Sugg) -> String {
 508     match source {
 509         MatchSource::Normal => format!("match {} {{ .. }}", expr),
 510         MatchSource::IfLetDesugar { .. } => format!("if let .. = {} {{ .. }}", expr),
 511         MatchSource::WhileLetDesugar => format!("while let .. = {} {{ .. }}", expr),
 512         MatchSource::ForLoopDesugar => span_bug!(span, "for loop desugared to match with &-patterns!"),
 513         MatchSource::TryDesugar => span_bug!(span, "`?` operator desugared to match with &-patterns!"),
 514     }
 515 }
 516
 517 pub fn overlapping<T>(ranges: &[SpannedRange<T>]) -> Option<(&SpannedRange<T>, &SpannedRange<T>)>
 518     where T: Copy + Ord
 519 {
 520     #[derive(Copy, Clone, Debug, Eq, PartialEq)]
 521     enum Kind<'a, T: 'a> {
 522         Start(T, &'a SpannedRange<T>),
 523         End(Bound<T>, &'a SpannedRange<T>),
 524     }
 525
 526     impl<'a, T: Copy> Kind<'a, T> {
 527         fn range(&self) -> &'a SpannedRange<T> {
 528             match *self {
 529                 Kind::Start(_, r) |
 530                 Kind::End(_, r) => r,
 531             }
 532         }
 533
 534         fn value(self) -> Bound<T> {
 535             match self {
 536                 Kind::Start(t, _) => Bound::Included(t),
 537                 Kind::End(t, _) => t,
 538             }
 539         }
 540     }
 541
 542     impl<'a, T: Copy + Ord> PartialOrd for Kind<'a, T> {
 543         fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
 544             Some(self.cmp(other))
 545         }
 546     }
 547
 548     impl<'a, T: Copy + Ord> Ord for Kind<'a, T> {
 549         fn cmp(&self, other: &Self) -> Ordering {
 550             match (self.value(), other.value()) {
 551                 (Bound::Included(a), Bound::Included(b)) |
 552                 (Bound::Excluded(a), Bound::Excluded(b)) => a.cmp(&b),
 553                 // Range patterns cannot be unbounded (yet)
 554                 (Bound::Unbounded, _) |
 555                 (_, Bound::Unbounded) => unimplemented!(),
 556                 (Bound::Included(a), Bound::Excluded(b)) => {
 557                     match a.cmp(&b) {
 558                         Ordering::Equal => Ordering::Greater,
 559                         other => other,
 560                     }
 561                 },
 562                 (Bound::Excluded(a), Bound::Included(b)) => {
 563                     match a.cmp(&b) {
 564                         Ordering::Equal => Ordering::Less,
 565                         other => other,
 566                     }
 567                 },
 568             }
 569         }
 570     }
 571
 572     let mut values = Vec::with_capacity(2 * ranges.len());
 573
 574     for r in ranges {
 575         values.push(Kind::Start(r.node.0, r));
 576         values.push(Kind::End(r.node.1, r));
 577     }
 578
 579     values.sort();
 580
 581     for (a, b) in values.iter().zip(values.iter().skip(1)) {
 582         match (a, b) {
 583             (&Kind::Start(_, ra), &Kind::End(_, rb)) => {
 584                 if ra.node != rb.node {
 585                     return Some((ra, rb));
 586                 }
 587             },
 588             (&Kind::End(a, _), &Kind::Start(b, _)) if a != Bound::Included(b) => (),
 589             _ => return Some((a.range(), b.range())),
 590         }
 591     }
 592
 593     None
 594 }