mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

needless_collect: use snippet_with_applicability

Browse Source 
+ small code refactor - using early returns.
This commit is contained in:
Mateusz Gacek 2021-05-13 17:09:59 +02:00
parent 171789eb45
commit b249290448
1 changed files with 18 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
clippy_lints/src/loops/needless_collect.rs
View File

@ -1,6 +1,6 @@
use super::NEEDLESS_COLLECT; use super::NEEDLESS_COLLECT;
use clippy_utils::diagnostics::{span_lint_and_sugg, span_lint_and_then}; use clippy_utils::diagnostics::{span_lint_and_sugg, span_lint_and_then};
use clippy_utils::source::snippet; use clippy_utils::source::{snippet, snippet_with_applicability};
use clippy_utils::sugg::Sugg; use clippy_utils::sugg::Sugg;
use clippy_utils::ty::is_type_diagnostic_item; use clippy_utils::ty::is_type_diagnostic_item;
use clippy_utils::{is_trait_method, path_to_local_id}; use clippy_utils::{is_trait_method, path_to_local_id};
@ -28,46 +28,45 @@ fn check_needless_collect_direct_usage<'tcx>(expr: &'tcx Expr<'_>, cx: &LateCont
if let Some(GenericArg::Type(ref ty)) = generic_args.args.get(0); if let Some(GenericArg::Type(ref ty)) = generic_args.args.get(0);
if let Some(ty) = cx.typeck_results().node_type_opt(ty.hir_id); if let Some(ty) = cx.typeck_results().node_type_opt(ty.hir_id);
then { then {
let is_empty_sugg = Some("next().is_none()".to_string()); let mut applicability = Applicability::MachineApplicable;
let is_empty_sugg = "next().is_none()".to_string();
let method_name = &*method.ident.name.as_str(); let method_name = &*method.ident.name.as_str();
let sugg = if is_type_diagnostic_item(cx, ty, sym::vec_type) || let sugg = if is_type_diagnostic_item(cx, ty, sym::vec_type) ||
is_type_diagnostic_item(cx, ty, sym::vecdeque_type) || is_type_diagnostic_item(cx, ty, sym::vecdeque_type) ||
is_type_diagnostic_item(cx, ty, sym::LinkedList) || is_type_diagnostic_item(cx, ty, sym::LinkedList) ||
is_type_diagnostic_item(cx, ty, sym::BinaryHeap) { is_type_diagnostic_item(cx, ty, sym::BinaryHeap) {
match method_name { match method_name {
"len" => Some("count()".to_string()), "len" => "count()".to_string(),
"is_empty" => is_empty_sugg, "is_empty" => is_empty_sugg,
"contains" => { "contains" => {
let contains_arg = snippet(cx, args[1].span, "??"); let contains_arg = snippet_with_applicability(cx, args[1].span, "??", &mut applicability);
let (arg, pred) = contains_arg let (arg, pred) = contains_arg
.strip_prefix('&') .strip_prefix('&')
.map_or(("&x", &*contains_arg), |s| ("x", s)); .map_or(("&x", &*contains_arg), |s| ("x", s));
Some(format!("any(|{}| x == {})", arg, pred)) format!("any(|{}| x == {})", arg, pred)
} }
_ => None, _ => return,
} }
} }
else if is_type_diagnostic_item(cx, ty, sym::BTreeMap) || else if is_type_diagnostic_item(cx, ty, sym::BTreeMap) ||
is_type_diagnostic_item(cx, ty, sym::hashmap_type) { is_type_diagnostic_item(cx, ty, sym::hashmap_type) {
match method_name { match method_name {
"is_empty" => is_empty_sugg, "is_empty" => is_empty_sugg,
_ => None, _ => return,
} }
} }
else { else {
None return;
}; };
if let Some(sugg) = sugg { span_lint_and_sugg(
span_lint_and_sugg( cx,
cx, NEEDLESS_COLLECT,
NEEDLESS_COLLECT, method0_span.with_hi(expr.span.hi()),
method0_span.with_hi(expr.span.hi()), NEEDLESS_COLLECT_MSG,
NEEDLESS_COLLECT_MSG, "replace with",
"replace with", sugg,
sugg, applicability,
Applicability::MachineApplicable, );
);
}
} }
} }
} }