mirror of
https://github.com/rust-lang/rust.git
synced 2024-11-23 23:34:48 +00:00
Make if_chain more readable
This commit is contained in:
parent
72a5d7b612
commit
d66acc23a9
@ -1,5 +1,6 @@
|
||||
use crate::consts::{constant, Constant};
|
||||
use crate::utils::{is_direct_expn_of, is_expn_of, match_qpath, span_help_and_lint};
|
||||
use crate::utils::paths;
|
||||
use crate::utils::{is_direct_expn_of, is_expn_of, match_def_path, resolve_node, span_help_and_lint};
|
||||
use if_chain::if_chain;
|
||||
use rustc::hir::*;
|
||||
use rustc::lint::{LateContext, LateLintPass, LintArray, LintPass};
|
||||
@ -99,59 +100,75 @@ impl<'a, 'tcx> LateLintPass<'a, 'tcx> for AssertionsOnConstants {
|
||||
}
|
||||
}
|
||||
|
||||
// fn get_assert_args(snip: String) -> Option<Vec<String>> {
|
||||
//
|
||||
// }
|
||||
|
||||
/// Check if the expression matches
|
||||
///
|
||||
/// ```rust,ignore
|
||||
/// match { let _t = !c; _t } {
|
||||
/// true => {
|
||||
/// {
|
||||
/// ::std::rt::begin_panic(message, _)
|
||||
/// }
|
||||
/// }
|
||||
/// _ => { }
|
||||
/// };
|
||||
/// ```
|
||||
///
|
||||
/// where `message` is a string literal and `c` is a constant bool.
|
||||
///
|
||||
/// TODO extend this to match anything as message not just string literals
|
||||
///
|
||||
/// Returns the `message` argument of `begin_panic` and the value of `c` which is the
|
||||
/// first argument of `assert!`.
|
||||
fn assert_with_message<'a, 'tcx>(cx: &LateContext<'a, 'tcx>, expr: &'tcx Expr) -> Option<(LocalInternedString, bool)> {
|
||||
if_chain! {
|
||||
if let ExprKind::Match(ref expr, ref arms, MatchSource::IfDesugar { contains_else_clause: false }) = expr.kind;
|
||||
// match expr
|
||||
if let ExprKind::Match(ref expr, ref arms, _) = expr.kind;
|
||||
// matches { let _t = expr; _t }
|
||||
if let ExprKind::DropTemps(ref expr) = expr.kind;
|
||||
if let ExprKind::Unary(UnOp::UnNot, ref expr) = expr.kind;
|
||||
//if let ExprKind::Lit(ref lit) = expr.kind;
|
||||
// bind the first argument of the `assert!` macro
|
||||
if let Some((Constant::Bool(is_true), _)) = constant(cx, cx.tables, expr);
|
||||
//if is_true;
|
||||
// match arms
|
||||
// arm 1 pattern
|
||||
if let PatKind::Lit(ref lit_expr) = arms[0].pat.kind;
|
||||
if let ExprKind::Lit(ref lit) = lit_expr.kind;
|
||||
if let LitKind::Bool(true) = lit.node;
|
||||
//if let LitKind::Bool(true) = lit1.node;
|
||||
// arm 1 block
|
||||
if let ExprKind::Block(ref block1, _) = arms[0].body.kind;
|
||||
if let Some(trailing_expr1) = &block1.expr;
|
||||
if block1.stmts.len() == 0;
|
||||
//
|
||||
if let ExprKind::Block(ref actual_block1, _) = trailing_expr1.kind;
|
||||
if let Some(block1_expr) = &actual_block1.expr;
|
||||
if let ExprKind::Block(ref block, _) = arms[0].body.kind;
|
||||
if block.stmts.len() == 0;
|
||||
if let Some(block_expr) = &block.expr;
|
||||
if let ExprKind::Block(ref inner_block, _) = block_expr.kind;
|
||||
if let Some(begin_panic_call) = &inner_block.expr;
|
||||
// function call
|
||||
if let ExprKind::Call(ref func, ref args) = block1_expr.kind;
|
||||
if let ExprKind::Path(ref path) = func.kind;
|
||||
// ["{{root}}", "std", "rt", "begin_panic"] does not work
|
||||
if match_qpath(path, &["$crate", "rt", "begin_panic"]);
|
||||
// arguments
|
||||
if let Some(args) = match_function_call(cx, begin_panic_call, &paths::BEGIN_PANIC);
|
||||
if args.len() == 2;
|
||||
if let ExprKind::Lit(ref lit) = args[0].kind;
|
||||
if let LitKind::Str(ref s, _) = lit.node;
|
||||
let panic_message = s.as_str(); // bind the panic message
|
||||
if let ExprKind::AddrOf(MutImmutable, ref inner) = args[1].kind;
|
||||
if let ExprKind::Tup(ref elements) = inner.kind;
|
||||
if elements.len() == 3;
|
||||
if let ExprKind::Lit(ref lit1) = elements[0].kind;
|
||||
if let LitKind::Str(ref s1, _) = lit1.node;
|
||||
if let ExprKind::Lit(ref lit2) = elements[1].kind;
|
||||
if let LitKind::Int(_, _) = lit2.node;
|
||||
if let ExprKind::Lit(ref lit3) = elements[2].kind;
|
||||
if let LitKind::Int(_, _) = lit3.node;
|
||||
// arm 2 block
|
||||
if let PatKind::Wild = arms[1].pat.kind;
|
||||
if let ExprKind::Block(ref block2, _) = arms[1].body.kind;
|
||||
if let None = &block2.expr;
|
||||
if block2.stmts.len() == 0;
|
||||
// bind the second argument of the `assert!` macro
|
||||
let panic_message = s.as_str();
|
||||
// second argument of begin_panic is irrelevant
|
||||
// as is the second match arm
|
||||
then {
|
||||
return Some((panic_message, is_true));
|
||||
}
|
||||
}
|
||||
return None;
|
||||
None
|
||||
}
|
||||
|
||||
/// Matches a function call with the given path and returns the arguments.
|
||||
///
|
||||
/// Usage:
|
||||
///
|
||||
/// ```rust,ignore
|
||||
/// if let Some(args) = match_function_call(cx, begin_panic_call, &paths::BEGIN_PANIC);
|
||||
/// ```
|
||||
fn match_function_call<'a, 'tcx>(cx: &LateContext<'a, 'tcx>, expr: &'tcx Expr, path: &[&str]) -> Option<&'a [Expr]> {
|
||||
if_chain! {
|
||||
if let ExprKind::Call(ref fun, ref args) = expr.kind;
|
||||
if let ExprKind::Path(ref qpath) = fun.kind;
|
||||
if let Some(fun_def_id) = resolve_node(cx, qpath, fun.hir_id).opt_def_id();
|
||||
if match_def_path(cx, fun_def_id, path);
|
||||
then {
|
||||
return Some(&args)
|
||||
}
|
||||
};
|
||||
None
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user