161 lines
4.0 KiB
Rust
161 lines
4.0 KiB
Rust
//! Checks for if expressions that contain only an if expression.
|
|
//!
|
|
//! For example, the lint would catch:
|
|
//!
|
|
//! ```rust
|
|
//! if x {
|
|
//! if y {
|
|
//! println!("Hello world");
|
|
//! }
|
|
//! }
|
|
//! ```
|
|
//!
|
|
//! This lint is **warn** by default
|
|
|
|
use rustc::lint::*;
|
|
use syntax::ast;
|
|
|
|
use utils::{in_macro, snippet_block, span_lint_and_then};
|
|
use utils::sugg::Sugg;
|
|
|
|
/// **What it does:** Checks for nested `if` statements which can be collapsed
|
|
/// by `&&`-combining their conditions and for `else { if ... }` expressions that
|
|
/// can be collapsed to `else if ...`.
|
|
///
|
|
/// **Why is this bad?** Each `if`-statement adds one level of nesting, which
|
|
/// makes code look more complex than it really is.
|
|
///
|
|
/// **Known problems:** None.
|
|
///
|
|
/// **Example:**
|
|
/// ```rust
|
|
/// if x {
|
|
/// if y {
|
|
/// …
|
|
/// }
|
|
/// }
|
|
///
|
|
/// // or
|
|
///
|
|
/// if x {
|
|
/// …
|
|
/// } else {
|
|
/// if y {
|
|
/// …
|
|
/// }
|
|
/// }
|
|
/// ```
|
|
///
|
|
/// Should be written:
|
|
///
|
|
/// ```rust
|
|
/// if x && y {
|
|
/// …
|
|
/// }
|
|
///
|
|
/// // or
|
|
///
|
|
/// if x {
|
|
/// …
|
|
/// } else if y {
|
|
/// …
|
|
/// }
|
|
/// ```
|
|
declare_lint! {
|
|
pub COLLAPSIBLE_IF,
|
|
Warn,
|
|
"`if`s that can be collapsed (e.g. `if x { if y { ... } }` and `else { if x { ... } }`)"
|
|
}
|
|
|
|
#[derive(Copy,Clone)]
|
|
pub struct CollapsibleIf;
|
|
|
|
impl LintPass for CollapsibleIf {
|
|
fn get_lints(&self) -> LintArray {
|
|
lint_array!(COLLAPSIBLE_IF)
|
|
}
|
|
}
|
|
|
|
impl EarlyLintPass for CollapsibleIf {
|
|
fn check_expr(&mut self, cx: &EarlyContext, expr: &ast::Expr) {
|
|
if !in_macro(cx, expr.span) {
|
|
check_if(cx, expr)
|
|
}
|
|
}
|
|
}
|
|
|
|
fn check_if(cx: &EarlyContext, expr: &ast::Expr) {
|
|
match expr.node {
|
|
ast::ExprKind::If(ref check, ref then, ref else_) => {
|
|
if let Some(ref else_) = *else_ {
|
|
check_collapsible_maybe_if_let(cx, else_);
|
|
} else {
|
|
check_collapsible_no_if_let(cx, expr, check, then);
|
|
}
|
|
}
|
|
ast::ExprKind::IfLet(_, _, _, Some(ref else_)) => {
|
|
check_collapsible_maybe_if_let(cx, else_);
|
|
}
|
|
_ => (),
|
|
}
|
|
}
|
|
|
|
fn check_collapsible_maybe_if_let(cx: &EarlyContext, else_: &ast::Expr) {
|
|
if_let_chain! {[
|
|
let ast::ExprKind::Block(ref block) = else_.node,
|
|
let Some(ref else_) = expr_block(block),
|
|
!in_macro(cx, else_.span),
|
|
], {
|
|
match else_.node {
|
|
ast::ExprKind::If(..) | ast::ExprKind::IfLet(..) => {
|
|
span_lint_and_then(cx,
|
|
COLLAPSIBLE_IF,
|
|
block.span,
|
|
"this `else { if .. }` block can be collapsed", |db| {
|
|
db.span_suggestion(block.span, "try", snippet_block(cx, else_.span, "..").into_owned());
|
|
});
|
|
}
|
|
_ => (),
|
|
}
|
|
}}
|
|
}
|
|
|
|
fn check_collapsible_no_if_let(
|
|
cx: &EarlyContext,
|
|
expr: &ast::Expr,
|
|
check: &ast::Expr,
|
|
then: &ast::Block,
|
|
) {
|
|
if_let_chain! {[
|
|
let Some(inner) = expr_block(then),
|
|
let ast::ExprKind::If(ref check_inner, ref content, None) = inner.node,
|
|
], {
|
|
if expr.span.expn_id != inner.span.expn_id {
|
|
return;
|
|
}
|
|
span_lint_and_then(cx, COLLAPSIBLE_IF, expr.span, "this if statement can be collapsed", |db| {
|
|
let lhs = Sugg::ast(cx, check, "..");
|
|
let rhs = Sugg::ast(cx, check_inner, "..");
|
|
db.span_suggestion(expr.span,
|
|
"try",
|
|
format!("if {} {}",
|
|
lhs.and(rhs),
|
|
snippet_block(cx, content.span, "..")));
|
|
});
|
|
}}
|
|
}
|
|
|
|
/// If the block contains only one expression, returns it.
|
|
fn expr_block(block: &ast::Block) -> Option<&ast::Expr> {
|
|
let mut it = block.stmts.iter();
|
|
|
|
if let (Some(stmt), None) = (it.next(), it.next()) {
|
|
match stmt.node {
|
|
ast::StmtKind::Expr(ref expr) | ast::StmtKind::Semi(ref expr) => Some(expr),
|
|
_ => None,
|
|
}
|
|
} else {
|
|
None
|
|
}
|
|
}
|