Add higher docs and remove some unneeded fields

This commit is contained in:
Cameron Steffen 2021-08-27 08:38:07 -05:00 committed by flip1995
parent c7c2036cb9
commit c2bb313e7a
No known key found for this signature in database
GPG key ID: 1CA0DF2AF59D68A5
3 changed files with 65 additions and 49 deletions

View file

@ -580,8 +580,8 @@ impl<'tcx> LateLintPass<'tcx> for Loops {
while_let_on_iterator::check(cx, expr); while_let_on_iterator::check(cx, expr);
if let Some(higher::While { if_cond, if_then, .. }) = higher::While::hir(expr) { if let Some(higher::While { condition, body }) = higher::While::hir(expr) {
while_immutable_condition::check(cx, if_cond, if_then); while_immutable_condition::check(cx, condition, body);
} }
needless_collect::check(expr, cx); needless_collect::check(expr, cx);

View file

@ -14,12 +14,7 @@ use rustc_span::{symbol::sym, Span, Symbol};
pub(super) fn check(cx: &LateContext<'tcx>, expr: &'tcx Expr<'_>) { pub(super) fn check(cx: &LateContext<'tcx>, expr: &'tcx Expr<'_>) {
let (scrutinee_expr, iter_expr, some_pat, loop_expr) = if_chain! { let (scrutinee_expr, iter_expr, some_pat, loop_expr) = if_chain! {
if let Some(higher::WhileLet { if let Some(higher::WhileLet { if_then, let_pat, let_expr }) = higher::WhileLet::hir(expr);
if_then,
let_pat,
let_expr,
..
}) = higher::WhileLet::hir(expr);
// check for `Some(..)` pattern // check for `Some(..)` pattern
if let PatKind::TupleStruct(QPath::Resolved(None, pat_path), some_pat, _) = let_pat.kind; if let PatKind::TupleStruct(QPath::Resolved(None, pat_path), some_pat, _) = let_pat.kind;
if let Res::Def(_, pat_did) = pat_path.res; if let Res::Def(_, pat_did) = pat_path.res;

View file

@ -13,14 +13,19 @@ use rustc_span::{sym, ExpnKind, Span, Symbol};
/// The essential nodes of a desugared for loop as well as the entire span: /// The essential nodes of a desugared for loop as well as the entire span:
/// `for pat in arg { body }` becomes `(pat, arg, body)`. Return `(pat, arg, body, span)`. /// `for pat in arg { body }` becomes `(pat, arg, body)`. Return `(pat, arg, body, span)`.
pub struct ForLoop<'tcx> { pub struct ForLoop<'tcx> {
/// `for` loop item
pub pat: &'tcx hir::Pat<'tcx>, pub pat: &'tcx hir::Pat<'tcx>,
/// `IntoIterator` argument
pub arg: &'tcx hir::Expr<'tcx>, pub arg: &'tcx hir::Expr<'tcx>,
/// `for` loop body
pub body: &'tcx hir::Expr<'tcx>, pub body: &'tcx hir::Expr<'tcx>,
/// entire `for` loop span
pub span: Span, pub span: Span,
} }
impl<'tcx> ForLoop<'tcx> { impl<'tcx> ForLoop<'tcx> {
#[inline] #[inline]
/// Parses a desugared `for` loop
pub fn hir(expr: &Expr<'tcx>) -> Option<Self> { pub fn hir(expr: &Expr<'tcx>) -> Option<Self> {
if_chain! { if_chain! {
if let hir::ExprKind::Match(iterexpr, arms, hir::MatchSource::ForLoopDesugar) = expr.kind; if let hir::ExprKind::Match(iterexpr, arms, hir::MatchSource::ForLoopDesugar) = expr.kind;
@ -46,14 +51,19 @@ impl<'tcx> ForLoop<'tcx> {
} }
} }
/// An `if` expression without `DropTemps`
pub struct If<'hir> { pub struct If<'hir> {
/// `if` condition
pub cond: &'hir Expr<'hir>, pub cond: &'hir Expr<'hir>,
pub r#else: Option<&'hir Expr<'hir>>, /// `if` then expression
pub then: &'hir Expr<'hir>, pub then: &'hir Expr<'hir>,
/// `else` expression
pub r#else: Option<&'hir Expr<'hir>>,
} }
impl<'hir> If<'hir> { impl<'hir> If<'hir> {
#[inline] #[inline]
/// Parses an `if` expression
pub const fn hir(expr: &Expr<'hir>) -> Option<Self> { pub const fn hir(expr: &Expr<'hir>) -> Option<Self> {
if let ExprKind::If( if let ExprKind::If(
Expr { Expr {
@ -64,21 +74,27 @@ impl<'hir> If<'hir> {
r#else, r#else,
) = expr.kind ) = expr.kind
{ {
Some(Self { cond, r#else, then }) Some(Self { cond, then, r#else })
} else { } else {
None None
} }
} }
} }
/// An `if let` expression
pub struct IfLet<'hir> { pub struct IfLet<'hir> {
/// `if let` pattern
pub let_pat: &'hir Pat<'hir>, pub let_pat: &'hir Pat<'hir>,
/// `if let` scrutinee
pub let_expr: &'hir Expr<'hir>, pub let_expr: &'hir Expr<'hir>,
/// `if let` then expression
pub if_then: &'hir Expr<'hir>, pub if_then: &'hir Expr<'hir>,
/// `if let` else expression
pub if_else: Option<&'hir Expr<'hir>>, pub if_else: Option<&'hir Expr<'hir>>,
} }
impl<'hir> IfLet<'hir> { impl<'hir> IfLet<'hir> {
/// Parses an `if let` expression
pub fn hir(cx: &LateContext<'_>, expr: &Expr<'hir>) -> Option<Self> { pub fn hir(cx: &LateContext<'_>, expr: &Expr<'hir>) -> Option<Self> {
if let ExprKind::If( if let ExprKind::If(
Expr { Expr {
@ -115,7 +131,9 @@ impl<'hir> IfLet<'hir> {
} }
} }
/// An `if let` or `match` expression. Useful for lints that trigger on one or the other.
pub enum IfLetOrMatch<'hir> { pub enum IfLetOrMatch<'hir> {
/// Any `match` expression
Match(&'hir Expr<'hir>, &'hir [Arm<'hir>], MatchSource), Match(&'hir Expr<'hir>, &'hir [Arm<'hir>], MatchSource),
/// scrutinee, pattern, then block, else block /// scrutinee, pattern, then block, else block
IfLet( IfLet(
@ -127,6 +145,7 @@ pub enum IfLetOrMatch<'hir> {
} }
impl<'hir> IfLetOrMatch<'hir> { impl<'hir> IfLetOrMatch<'hir> {
/// Parses an `if let` or `match` expression
pub fn parse(cx: &LateContext<'_>, expr: &Expr<'hir>) -> Option<Self> { pub fn parse(cx: &LateContext<'_>, expr: &Expr<'hir>) -> Option<Self> {
match expr.kind { match expr.kind {
ExprKind::Match(expr, arms, source) => Some(Self::Match(expr, arms, source)), ExprKind::Match(expr, arms, source) => Some(Self::Match(expr, arms, source)),
@ -142,14 +161,19 @@ impl<'hir> IfLetOrMatch<'hir> {
} }
} }
/// An `if` or `if let` expression
pub struct IfOrIfLet<'hir> { pub struct IfOrIfLet<'hir> {
/// `if` condition that is maybe a `let` expression
pub cond: &'hir Expr<'hir>, pub cond: &'hir Expr<'hir>,
pub r#else: Option<&'hir Expr<'hir>>, /// `if` then expression
pub then: &'hir Expr<'hir>, pub then: &'hir Expr<'hir>,
/// `else` expression
pub r#else: Option<&'hir Expr<'hir>>,
} }
impl<'hir> IfOrIfLet<'hir> { impl<'hir> IfOrIfLet<'hir> {
#[inline] #[inline]
/// Parses an `if` or `if let` expression
pub const fn hir(expr: &Expr<'hir>) -> Option<Self> { pub const fn hir(expr: &Expr<'hir>) -> Option<Self> {
if let ExprKind::If(cond, then, r#else) = expr.kind { if let ExprKind::If(cond, then, r#else) = expr.kind {
if let ExprKind::DropTemps(new_cond) = cond.kind { if let ExprKind::DropTemps(new_cond) = cond.kind {
@ -160,7 +184,7 @@ impl<'hir> IfOrIfLet<'hir> {
}); });
} }
if let ExprKind::Let(..) = cond.kind { if let ExprKind::Let(..) = cond.kind {
return Some(Self { cond, r#else, then }); return Some(Self { cond, then, r#else });
} }
} }
None None
@ -281,14 +305,17 @@ impl<'a> VecArgs<'a> {
} }
} }
/// A desugared `while` loop
pub struct While<'hir> { pub struct While<'hir> {
pub if_cond: &'hir Expr<'hir>, /// `while` loop condition
pub if_then: &'hir Expr<'hir>, pub condition: &'hir Expr<'hir>,
pub if_else: Option<&'hir Expr<'hir>>, /// `while` loop body
pub body: &'hir Expr<'hir>,
} }
impl<'hir> While<'hir> { impl<'hir> While<'hir> {
#[inline] #[inline]
/// Parses a desugared `while` loop
pub const fn hir(expr: &Expr<'hir>) -> Option<Self> { pub const fn hir(expr: &Expr<'hir>) -> Option<Self> {
if let ExprKind::Loop( if let ExprKind::Loop(
Block { Block {
@ -297,11 +324,11 @@ impl<'hir> While<'hir> {
kind: kind:
ExprKind::If( ExprKind::If(
Expr { Expr {
kind: ExprKind::DropTemps(if_cond), kind: ExprKind::DropTemps(condition),
.. ..
}, },
if_then, body,
if_else_ref, _,
), ),
.. ..
}), }),
@ -312,38 +339,30 @@ impl<'hir> While<'hir> {
_, _,
) = expr.kind ) = expr.kind
{ {
let if_else = *if_else_ref; return Some(Self { condition, body });
return Some(Self {
if_cond,
if_then,
if_else,
});
} }
None None
} }
} }
/// A desugared `while let` loop
pub struct WhileLet<'hir> { pub struct WhileLet<'hir> {
pub if_expr: &'hir Expr<'hir>, /// `while let` loop item pattern
pub let_pat: &'hir Pat<'hir>, pub let_pat: &'hir Pat<'hir>,
/// `while let` loop scrutinee
pub let_expr: &'hir Expr<'hir>, pub let_expr: &'hir Expr<'hir>,
/// `while let` loop body
pub if_then: &'hir Expr<'hir>, pub if_then: &'hir Expr<'hir>,
pub if_else: Option<&'hir Expr<'hir>>,
} }
impl<'hir> WhileLet<'hir> { impl<'hir> WhileLet<'hir> {
#[inline] #[inline]
/// Parses a desugared `while let` loop
pub const fn hir(expr: &Expr<'hir>) -> Option<Self> { pub const fn hir(expr: &Expr<'hir>) -> Option<Self> {
if let ExprKind::Loop( if let ExprKind::Loop(
Block { Block {
expr: Some(if_expr), .. expr:
}, Some(Expr {
_,
LoopSource::While,
_,
) = expr.kind
{
if let Expr {
kind: kind:
ExprKind::If( ExprKind::If(
Expr { Expr {
@ -351,21 +370,23 @@ impl<'hir> WhileLet<'hir> {
.. ..
}, },
if_then, if_then,
if_else_ref, _,
), ),
.. ..
} = if_expr }),
..
},
_,
LoopSource::While,
_,
) = expr.kind
{ {
let if_else = *if_else_ref;
return Some(Self { return Some(Self {
if_expr,
let_pat, let_pat,
let_expr, let_expr,
if_then, if_then,
if_else,
}); });
} }
}
None None
} }
} }