refactor: use pattern matching for MethodCall arguments

Co-authored-by: xFrednet <xFrednet@gmail.com>
This commit is contained in:
Max Baumann 2022-03-18 00:23:55 +01:00
parent c22ff6cd6e
commit 4493fc7d27
No known key found for this signature in database
GPG Key ID: 20FA1609B03B1D6D

View File

@ -4,7 +4,7 @@ use if_chain::if_chain;
use rustc_hir::{Expr, ExprKind, QPath}; use rustc_hir::{Expr, ExprKind, QPath};
use rustc_lint::{LateContext, LateLintPass}; use rustc_lint::{LateContext, LateLintPass};
use rustc_session::{declare_lint_pass, declare_tool_lint}; use rustc_session::{declare_lint_pass, declare_tool_lint};
use rustc_span::{sym, Span}; use rustc_span::sym;
declare_clippy_lint! { declare_clippy_lint! {
/// ### What it does /// ### What it does
@ -47,25 +47,22 @@ impl<'tcx> LateLintPass<'tcx> for UseUnwrapOr {
fn check_expr(&mut self, cx: &LateContext<'tcx>, expr: &'tcx Expr<'_>) { fn check_expr(&mut self, cx: &LateContext<'tcx>, expr: &'tcx Expr<'_>) {
// look for x.or().unwrap() // look for x.or().unwrap()
if_chain! { if_chain! {
if let ExprKind::MethodCall(path, args, unwrap_span) = expr.kind; if let ExprKind::MethodCall(path, [unwrap_self], unwrap_span) = expr.kind;
if path.ident.name == sym::unwrap; if path.ident.name == sym::unwrap;
if let Some(caller) = args.first(); if let ExprKind::MethodCall(caller_path, [or_self, or_arg], or_span) = unwrap_self.kind;
if let ExprKind::MethodCall(caller_path, caller_args, or_span) = caller.kind;
if caller_path.ident.name == sym::or; if caller_path.ident.name == sym::or;
then { then {
let ty = cx.typeck_results().expr_ty(&caller_args[0]); // get type of x (we later check if it's Option or Result) let ty = cx.typeck_results().expr_ty(&or_self); // get type of x (we later check if it's Option or Result)
let title; let title;
let arg = &caller_args[1]; // the argument or(xyz) is called with
if is_type_diagnostic_item(cx, ty, sym::Option) { if is_type_diagnostic_item(cx, ty, sym::Option) {
title = ".or(Some(…)).unwrap() found"; title = ".or(Some(…)).unwrap() found";
if !is(arg, "Some") { if !is(or_arg, "Some") {
return; return;
} }
} else if is_type_diagnostic_item(cx, ty, sym::Result) { } else if is_type_diagnostic_item(cx, ty, sym::Result) {
title = ".or(Ok(…)).unwrap() found"; title = ".or(Ok(…)).unwrap() found";
if !is(arg, "Ok") { if !is(or_arg, "Ok") {
return; return;
} }
} else { } else {
@ -74,13 +71,10 @@ impl<'tcx> LateLintPass<'tcx> for UseUnwrapOr {
return; return;
} }
// span = or_span + unwrap_span
let span = Span::new(or_span.lo(), unwrap_span.hi(), or_span.ctxt(), or_span.parent());
span_lint_and_help( span_lint_and_help(
cx, cx,
USE_UNWRAP_OR, USE_UNWRAP_OR,
span, or_span.to(unwrap_span),
title, title,
None, None,
"use `unwrap_or()` instead" "use `unwrap_or()` instead"