#[allow(unused_imports)]
use alloc::string::ToString;
#[allow(unused_imports)]
use {alloc::boxed::Box, alloc::string::String, alloc::vec::Vec};
use alloc::borrow::Cow;
use core::fmt;
use core::ops::Not;
use crate::rustc_abi::ExternAbi;
use crate::rustc_ast::util::parser::ExprPrecedence;
use crate::rustc_ast::{
self as ast, FloatTy, InlineAsmOptions, InlineAsmTemplatePiece, IntTy, Label, LitIntType,
LitKind, TraitObjectSyntax, UintTy, UnsafeBinderCastKind,
};
pub use crate::rustc_ast::{
AssignOp, AssignOpKind, AttrId, AttrStyle, BinOp, BinOpKind, BindingMode, BorrowKind,
BoundConstness, BoundPolarity, ByRef, CaptureBy, DelimArgs, ImplPolarity, IsAuto,
MetaItemInner, MetaItemLit, Movability, Mutability, Pinnedness, UnOp,
};
use crate::rustc_attr_ir::Attribute;
use crate::rustc_data_structures::fingerprint::Fingerprint;
use crate::rustc_data_structures::fx::FxIndexSet;
use crate::rustc_data_structures::sorted_map::SortedMap;
use crate::rustc_data_structures::steal::Steal;
use crate::rustc_data_structures::tagged_ptr::TaggedRef;
use crate::rustc_data_structures::unord::UnordMap;
use crate::rustc_error_messages::{DiagArgValue, IntoDiagArg};
use crate::rustc_hir_id::{HirId, ItemLocalId, ItemLocalMap, OwnerId};
use crate::rustc_index::IndexVec;
use rustc_macros::{Decodable, Encodable, StableHash};
use crate::rustc_span::def_id::LocalDefId;
use crate::rustc_span::{
BytePos, DUMMY_SP, DesugaringKind, ErrorGuaranteed, Ident, LocalExpnId, Span, Spanned, Symbol,
kw, sym,
};
use crate::rustc_target::asm::InlineAsmRegOrRegClass;
use tracing::debug;
use crate::rustc_hir::def::{CtorKind, DefKind, MacroKinds, PerNS, Res};
use crate::rustc_hir::def_id::{DefId, LocalDefIdMap};
use crate::rustc_hir::intravisit::{FnKind, VisitorExt};
use crate::rustc_hir::lints::DelayedLints;
#[derive(Debug, Copy, Clone, PartialEq, Eq, StableHash)]
pub enum AngleBrackets {
Missing,
Empty,
Full,
}
#[derive(Debug, Copy, Clone, PartialEq, Eq, StableHash)]
pub enum LifetimeSource {
Reference,
Path { angle_brackets: AngleBrackets },
OutlivesBound,
PreciseCapturing,
Other,
}
#[derive(Debug, Copy, Clone, PartialEq, Eq, StableHash)]
pub enum LifetimeSyntax {
Implicit,
ExplicitAnonymous,
ExplicitBound,
}
impl From<Ident> for LifetimeSyntax {
fn from(ident: Ident) -> Self {
let name = ident.name;
if name == sym::empty {
unreachable!("A lifetime name should never be empty");
} else if name == kw::UnderscoreLifetime {
LifetimeSyntax::ExplicitAnonymous
} else {
debug_assert!(name.as_str().starts_with('\''));
LifetimeSyntax::ExplicitBound
}
}
}
#[derive(Debug, Copy, Clone, StableHash)]
#[repr(align(4))]
pub struct Lifetime {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub ident: Ident,
pub kind: LifetimeKind,
pub source: LifetimeSource,
pub syntax: LifetimeSyntax,
}
#[derive(Debug, Copy, Clone, StableHash)]
pub enum ParamName {
Plain(Ident),
Error(Ident),
Fresh,
}
impl ParamName {
pub fn ident(&self) -> Ident {
match *self {
ParamName::Plain(ident) | ParamName::Error(ident) => ident,
ParamName::Fresh => Ident::with_dummy_span(kw::UnderscoreLifetime),
}
}
}
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, StableHash)]
pub enum LifetimeKind {
Param(LocalDefId),
ImplicitObjectLifetimeDefault,
Error(ErrorGuaranteed),
Infer,
Static,
}
impl LifetimeKind {
fn is_elided(&self) -> bool {
match self {
LifetimeKind::ImplicitObjectLifetimeDefault | LifetimeKind::Infer => true,
LifetimeKind::Error(..) | LifetimeKind::Param(..) | LifetimeKind::Static => false,
}
}
}
impl fmt::Display for Lifetime {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.ident.name.fmt(f)
}
}
impl Lifetime {
pub fn new(
hir_id: HirId,
ident: Ident,
kind: LifetimeKind,
source: LifetimeSource,
syntax: LifetimeSyntax,
) -> Lifetime {
let lifetime = Lifetime { hir_id, ident, kind, source, syntax };
#[cfg(debug_assertions)]
match (lifetime.is_elided(), lifetime.is_anonymous()) {
(false, false) => {} (false, true) => {} (true, true) => {} (true, false) => panic!("bad Lifetime"),
}
lifetime
}
pub fn is_elided(&self) -> bool {
self.kind.is_elided()
}
pub fn is_anonymous(&self) -> bool {
self.ident.name == kw::UnderscoreLifetime
}
pub fn is_implicit(&self) -> bool {
matches!(self.syntax, LifetimeSyntax::Implicit)
}
pub fn is_static(&self) -> bool {
self.kind == LifetimeKind::Static
}
pub fn suggestion(&self, new_lifetime: &str) -> (Span, String) {
use LifetimeSource::*;
use LifetimeSyntax::*;
debug_assert!(new_lifetime.starts_with('\''));
match (self.syntax, self.source) {
(ExplicitBound | ExplicitAnonymous, _) => (self.ident.span, format!("{new_lifetime}")),
(Implicit, Path { angle_brackets: AngleBrackets::Full }) => {
(self.ident.span, format!("{new_lifetime}, "))
}
(Implicit, Path { angle_brackets: AngleBrackets::Empty }) => {
(self.ident.span, format!("{new_lifetime}"))
}
(Implicit, Path { angle_brackets: AngleBrackets::Missing }) => {
(self.ident.span.shrink_to_hi(), format!("<{new_lifetime}>"))
}
(Implicit, Reference) => (self.ident.span, format!("{new_lifetime} ")),
(Implicit, source) => {
unreachable!("can't suggest for a implicit lifetime of {source:?}")
}
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Path<'hir, R = Res> {
pub span: Span,
pub res: R,
pub segments: &'hir [PathSegment<'hir>],
}
pub type UsePath<'hir> = Path<'hir, PerNS<Option<Res>>>;
impl Path<'_> {
pub fn is_global(&self) -> bool {
self.segments.first().is_some_and(|segment| segment.ident.name == kw::PathRoot)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct PathSegment<'hir> {
pub ident: Ident,
#[stable_hash(ignore)]
pub hir_id: HirId,
pub res: Res,
pub args: Option<&'hir GenericArgs<'hir>>,
pub infer_args: bool,
pub delegation_child_segment: bool,
}
impl<'hir> PathSegment<'hir> {
pub fn new(ident: Ident, hir_id: HirId, res: Res) -> PathSegment<'hir> {
PathSegment {
ident,
hir_id,
res,
infer_args: true,
args: None,
delegation_child_segment: false,
}
}
pub fn invalid() -> Self {
Self::new(Ident::dummy(), HirId::INVALID, Res::Err)
}
pub fn args(&self) -> &GenericArgs<'hir> {
if let Some(ref args) = self.args { args } else { GenericArgs::NONE }
}
}
#[derive(Clone, Copy, Debug, StableHash)]
pub enum ConstItemRhs<'hir> {
Body(BodyId),
TypeConst(&'hir ConstArg<'hir>),
}
impl<'hir> ConstItemRhs<'hir> {
pub fn hir_id(&self) -> HirId {
match self {
ConstItemRhs::Body(body_id) => body_id.hir_id,
ConstItemRhs::TypeConst(ct_arg) => ct_arg.hir_id,
}
}
pub fn span<'tcx>(&self, tcx: impl crate::rustc_hir::intravisit::HirTyCtxt<'tcx>) -> Span {
match self {
ConstItemRhs::Body(body_id) => tcx.hir_body(*body_id).value.span,
ConstItemRhs::TypeConst(ct_arg) => ct_arg.span,
}
}
}
#[derive(Clone, Copy, Debug, StableHash)]
#[repr(C)]
pub struct ConstArg<'hir, Unambig = ()> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub kind: ConstArgKind<'hir, Unambig>,
pub span: Span,
}
impl<'hir> ConstArg<'hir, AmbigArg> {
pub fn as_unambig_ct(&self) -> &ConstArg<'hir> {
let ptr = self as *const ConstArg<'hir, AmbigArg> as *const ConstArg<'hir, ()>;
unsafe { &*ptr }
}
}
impl<'hir> ConstArg<'hir> {
pub fn try_as_ambig_ct(&self) -> Option<&ConstArg<'hir, AmbigArg>> {
if let ConstArgKind::Infer(()) = self.kind {
return None;
}
let ptr = self as *const ConstArg<'hir> as *const ConstArg<'hir, AmbigArg>;
Some(unsafe { &*ptr })
}
}
impl<'hir, Unambig> ConstArg<'hir, Unambig> {
pub fn anon_const_hir_id(&self) -> Option<HirId> {
match self.kind {
ConstArgKind::Anon(ac) => Some(ac.hir_id),
_ => None,
}
}
}
#[derive(Clone, Copy, Debug, StableHash)]
#[repr(u8, C)]
pub enum ConstArgKind<'hir, Unambig = ()> {
Tup(&'hir [&'hir ConstArg<'hir>]),
Path(QPath<'hir>),
Anon(&'hir AnonConst),
Struct(QPath<'hir>, &'hir [&'hir ConstArgExprField<'hir>]),
TupleCall(QPath<'hir>, &'hir [&'hir ConstArg<'hir>]),
Array(&'hir ConstArgArrayExpr<'hir>),
Error(ErrorGuaranteed),
Infer(Unambig),
Literal {
lit: LitKind,
negated: bool,
},
}
#[derive(Clone, Copy, Debug, StableHash)]
pub struct ConstArgExprField<'hir> {
pub hir_id: HirId,
pub span: Span,
pub field: Ident,
pub expr: &'hir ConstArg<'hir>,
}
#[derive(Clone, Copy, Debug, StableHash)]
pub struct ConstArgArrayExpr<'hir> {
pub span: Span,
pub elems: &'hir [&'hir ConstArg<'hir>],
}
#[derive(Clone, Copy, Debug, PartialEq, Eq, StableHash)]
pub enum InferArgKind {
TypeOrConst,
Const,
}
#[derive(Clone, Copy, Debug, StableHash)]
pub struct InferArg {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub span: Span,
pub kind: InferArgKind,
}
impl InferArg {
pub fn to_ty(&self) -> Ty<'static> {
Ty { kind: TyKind::Infer(()), span: self.span, hir_id: self.hir_id }
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum GenericArg<'hir> {
Lifetime(&'hir Lifetime),
Type(&'hir Ty<'hir, AmbigArg>),
Const(&'hir ConstArg<'hir, AmbigArg>),
Infer(&'hir InferArg),
}
impl GenericArg<'_> {
pub fn span(&self) -> Span {
match self {
GenericArg::Lifetime(l) => l.ident.span,
GenericArg::Type(t) => t.span,
GenericArg::Const(c) => c.span,
GenericArg::Infer(i) => i.span,
}
}
pub fn hir_id(&self) -> HirId {
match self {
GenericArg::Lifetime(l) => l.hir_id,
GenericArg::Type(t) => t.hir_id,
GenericArg::Const(c) => c.hir_id,
GenericArg::Infer(i) => i.hir_id,
}
}
pub fn descr(&self) -> &'static str {
match self {
GenericArg::Lifetime(_) => "lifetime",
GenericArg::Type(_) => "type",
GenericArg::Const(_) => "constant",
GenericArg::Infer(InferArg { kind: InferArgKind::TypeOrConst, .. }) => "placeholder",
GenericArg::Infer(InferArg { kind: InferArgKind::Const, .. }) => "constant",
}
}
pub fn to_ord(&self) -> ast::ParamKindOrd {
match self {
GenericArg::Lifetime(_) => ast::ParamKindOrd::Lifetime,
GenericArg::Type(_) | GenericArg::Const(_) | GenericArg::Infer(_) => {
ast::ParamKindOrd::TypeOrConst
}
}
}
pub fn is_ty_or_const(&self) -> bool {
match self {
GenericArg::Lifetime(_) => false,
GenericArg::Type(_) | GenericArg::Const(_) | GenericArg::Infer(_) => true,
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct GenericArgs<'hir> {
pub args: &'hir [GenericArg<'hir>],
pub constraints: &'hir [AssocItemConstraint<'hir>],
pub parenthesized: GenericArgsParentheses,
pub span_ext: Span,
}
impl<'hir> GenericArgs<'hir> {
pub const NONE: &'hir GenericArgs<'hir> = &GenericArgs {
args: &[],
constraints: &[],
parenthesized: GenericArgsParentheses::No,
span_ext: DUMMY_SP,
};
pub fn paren_sugar_inputs_output(&self) -> Option<(&[Ty<'hir>], &Ty<'hir>)> {
if self.parenthesized != GenericArgsParentheses::ParenSugar {
return None;
}
let inputs = self
.args
.iter()
.find_map(|arg| {
let GenericArg::Type(ty) = arg else { return None };
let TyKind::Tup(tys) = &ty.kind else { return None };
Some(tys)
})
.unwrap();
Some((inputs, self.paren_sugar_output_inner()))
}
pub fn paren_sugar_output(&self) -> Option<&Ty<'hir>> {
(self.parenthesized == GenericArgsParentheses::ParenSugar)
.then(|| self.paren_sugar_output_inner())
}
fn paren_sugar_output_inner(&self) -> &Ty<'hir> {
let [constraint] = self.constraints.try_into().unwrap();
debug_assert_eq!(constraint.ident.name, sym::Output);
constraint.ty().unwrap()
}
pub fn has_err(&self) -> Option<ErrorGuaranteed> {
self.args
.iter()
.find_map(|arg| {
let GenericArg::Type(ty) = arg else { return None };
let TyKind::Err(guar) = ty.kind else { return None };
Some(guar)
})
.or_else(|| {
self.constraints.iter().find_map(|constraint| {
let TyKind::Err(guar) = constraint.ty()?.kind else { return None };
Some(guar)
})
})
}
#[inline]
pub fn num_lifetime_args(&self) -> usize {
self.args.iter().filter(|arg| matches!(arg, GenericArg::Lifetime(_))).count()
}
#[inline]
pub fn has_lifetime_args(&self) -> bool {
self.args.iter().any(|arg| matches!(arg, GenericArg::Lifetime(_)))
}
#[inline]
pub fn num_generic_params(&self) -> usize {
self.args.iter().filter(|arg| !matches!(arg, GenericArg::Lifetime(_))).count()
}
pub fn span(&self) -> Option<Span> {
let span_ext = self.span_ext()?;
Some(span_ext.with_lo(span_ext.lo() + BytePos(1)).with_hi(span_ext.hi() - BytePos(1)))
}
pub fn span_ext(&self) -> Option<Span> {
Some(self.span_ext).filter(|span| !span.is_empty())
}
pub fn is_empty(&self) -> bool {
self.args.is_empty()
}
}
#[derive(Copy, Clone, PartialEq, Eq, Debug, StableHash)]
pub enum GenericArgsParentheses {
No,
ReturnTypeNotation,
ParenSugar,
}
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug, StableHash)]
pub struct TraitBoundModifiers {
pub constness: BoundConstness,
pub polarity: BoundPolarity,
}
impl TraitBoundModifiers {
pub const NONE: Self =
TraitBoundModifiers { constness: BoundConstness::Never, polarity: BoundPolarity::Positive };
}
#[derive(Clone, Copy, Debug, StableHash)]
pub enum GenericBound<'hir> {
Trait(PolyTraitRef<'hir>),
Outlives(&'hir Lifetime),
Use(&'hir [PreciseCapturingArg<'hir>], Span),
}
impl GenericBound<'_> {
pub fn trait_ref(&self) -> Option<&TraitRef<'_>> {
match self {
GenericBound::Trait(data) => Some(&data.trait_ref),
_ => None,
}
}
pub fn span(&self) -> Span {
match self {
GenericBound::Trait(t, ..) => t.span,
GenericBound::Outlives(l) => l.ident.span,
GenericBound::Use(_, span) => *span,
}
}
}
pub type GenericBounds<'hir> = &'hir [GenericBound<'hir>];
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, StableHash, Debug)]
pub enum MissingLifetimeKind {
Underscore,
Ampersand,
Comma,
Brackets,
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum LifetimeParamKind {
Explicit,
Elided(MissingLifetimeKind),
Error,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum GenericParamKind<'hir> {
Lifetime {
kind: LifetimeParamKind,
},
Type {
default: Option<&'hir Ty<'hir>>,
synthetic: bool,
},
Const {
ty: &'hir Ty<'hir>,
default: Option<&'hir ConstArg<'hir>>,
},
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct GenericParam<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub def_id: LocalDefId,
pub name: ParamName,
pub span: Span,
pub pure_wrt_drop: bool,
pub kind: GenericParamKind<'hir>,
pub colon_span: Option<Span>,
pub source: GenericParamSource,
}
impl<'hir> GenericParam<'hir> {
pub fn is_impl_trait(&self) -> bool {
matches!(self.kind, GenericParamKind::Type { synthetic: true, .. })
}
pub fn is_elided_lifetime(&self) -> bool {
matches!(self.kind, GenericParamKind::Lifetime { kind: LifetimeParamKind::Elided(_) })
}
pub fn is_lifetime(&self) -> bool {
matches!(self.kind, GenericParamKind::Lifetime { .. })
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum GenericParamSource {
Generics,
Binder,
}
#[derive(Default)]
pub struct GenericParamCount {
pub lifetimes: usize,
pub types: usize,
pub consts: usize,
pub infer: usize,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Generics<'hir> {
pub params: &'hir [GenericParam<'hir>],
pub predicates: &'hir [WherePredicate<'hir>],
pub has_where_clause_predicates: bool,
pub where_clause_span: Span,
pub span: Span,
}
impl<'hir> Generics<'hir> {
pub const fn empty() -> &'hir Generics<'hir> {
const NOPE: Generics<'_> = Generics {
params: &[],
predicates: &[],
has_where_clause_predicates: false,
where_clause_span: DUMMY_SP,
span: DUMMY_SP,
};
&NOPE
}
pub fn get_named(&self, name: Symbol) -> Option<&GenericParam<'hir>> {
self.params.iter().find(|¶m| name == param.name.ident().name)
}
pub fn span_for_lifetime_suggestion(&self) -> Option<Span> {
if let Some(first) = self.params.first()
&& self.span.contains(first.span)
{
Some(first.span.shrink_to_lo())
} else {
None
}
}
pub fn span_for_param_suggestion(&self) -> Option<Span> {
self.params.iter().any(|p| self.span.contains(p.span)).then(|| {
self.span.with_lo(self.span.hi() - BytePos(1)).shrink_to_lo()
})
}
pub fn tail_span_for_predicate_suggestion(&self) -> Span {
let end = self.where_clause_span.shrink_to_hi();
if self.has_where_clause_predicates {
self.predicates
.iter()
.rfind(|&p| p.kind.in_where_clause())
.map_or(end, |p| p.span)
.shrink_to_hi()
.to(end)
} else {
end
}
}
pub fn add_where_or_trailing_comma(&self) -> &'static str {
if self.has_where_clause_predicates {
","
} else if self.where_clause_span.is_empty() {
" where"
} else {
""
}
}
pub fn bounds_for_param(
&self,
param_def_id: LocalDefId,
) -> impl Iterator<Item = &WhereBoundPredicate<'hir>> {
self.predicates.iter().filter_map(move |pred| match pred.kind {
WherePredicateKind::BoundPredicate(bp)
if bp.is_param_bound(param_def_id.to_def_id()) =>
{
Some(bp)
}
_ => None,
})
}
pub fn outlives_for_param(
&self,
param_def_id: LocalDefId,
) -> impl Iterator<Item = &WhereRegionPredicate<'_>> {
self.predicates.iter().filter_map(move |pred| match pred.kind {
WherePredicateKind::RegionPredicate(rp) if rp.is_param_bound(param_def_id) => Some(rp),
_ => None,
})
}
pub fn bounds_span_for_suggestions(
&self,
param_def_id: LocalDefId,
) -> Option<(Span, Option<Span>)> {
self.bounds_for_param(param_def_id).flat_map(|bp| bp.bounds.iter().rev()).find_map(
|bound| {
let span_for_parentheses = if let Some(trait_ref) = bound.trait_ref()
&& let [.., segment] = trait_ref.path.segments
&& let Some(ret_ty) = segment.args().paren_sugar_output()
&& let ret_ty = ret_ty.peel_refs()
&& let TyKind::TraitObject(_, tagged_ptr) = ret_ty.kind
&& let TraitObjectSyntax::Dyn = tagged_ptr.tag()
&& ret_ty.span.can_be_used_for_suggestions()
{
Some(ret_ty.span)
} else {
None
};
span_for_parentheses.map_or_else(
|| {
let bs = bound.span();
bs.from_expansion().not().then(|| (bs.shrink_to_hi(), None))
},
|span| Some((span.shrink_to_hi(), Some(span.shrink_to_lo()))),
)
},
)
}
pub fn span_for_predicate_removal(&self, pos: usize) -> Span {
let predicate = &self.predicates[pos];
let span = predicate.span;
if !predicate.kind.in_where_clause() {
return span;
}
if pos < self.predicates.len() - 1 {
let next_pred = &self.predicates[pos + 1];
if next_pred.kind.in_where_clause() {
return span.until(next_pred.span);
}
}
if pos > 0 {
let prev_pred = &self.predicates[pos - 1];
if prev_pred.kind.in_where_clause() {
return prev_pred.span.shrink_to_hi().to(span);
}
}
self.where_clause_span
}
pub fn span_for_bound_removal(&self, predicate_pos: usize, bound_pos: usize) -> Span {
let predicate = &self.predicates[predicate_pos];
let bounds = predicate.kind.bounds();
if bounds.len() == 1 {
return self.span_for_predicate_removal(predicate_pos);
}
let bound_span = bounds[bound_pos].span();
if bound_pos < bounds.len() - 1 {
bound_span.to(bounds[bound_pos + 1].span().shrink_to_lo())
} else {
bound_span.with_lo(bounds[bound_pos - 1].span().hi())
}
}
pub fn span_for_param_removal(&self, param_index: usize) -> Span {
if param_index >= self.params.len() {
return self.span.shrink_to_hi();
}
let is_param_explicit = |par: &&GenericParam<'_>| match par.kind {
GenericParamKind::Type { .. }
| GenericParamKind::Const { .. }
| GenericParamKind::Lifetime { kind: LifetimeParamKind::Explicit } => true,
_ => false,
};
if let Some(next) = self.params[param_index + 1..].iter().find(is_param_explicit) {
self.params[param_index].span.until(next.span)
} else if let Some(prev) = self.params[..param_index].iter().rfind(is_param_explicit) {
let mut prev_span = prev.span;
if let Some(prev_bounds_span) = self.span_for_param_bounds(prev) {
prev_span = prev_span.to(prev_bounds_span);
}
prev_span.shrink_to_hi().to(
if let Some(cur_bounds_span) = self.span_for_param_bounds(&self.params[param_index])
{
cur_bounds_span
} else {
self.params[param_index].span
},
)
} else {
self.span
}
}
fn span_for_param_bounds(&self, param: &GenericParam<'hir>) -> Option<Span> {
self.predicates
.iter()
.find(|pred| {
if let WherePredicateKind::BoundPredicate(WhereBoundPredicate {
origin: PredicateOrigin::GenericParam,
bounded_ty,
..
}) = pred.kind
{
bounded_ty.span == param.span
} else {
false
}
})
.map(|pred| pred.span)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct WherePredicate<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub span: Span,
pub kind: &'hir WherePredicateKind<'hir>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum WherePredicateKind<'hir> {
BoundPredicate(WhereBoundPredicate<'hir>),
RegionPredicate(WhereRegionPredicate<'hir>),
}
impl<'hir> WherePredicateKind<'hir> {
pub fn in_where_clause(&self) -> bool {
match self {
WherePredicateKind::BoundPredicate(p) => p.origin == PredicateOrigin::WhereClause,
WherePredicateKind::RegionPredicate(p) => p.in_where_clause,
}
}
pub fn bounds(&self) -> GenericBounds<'hir> {
match self {
WherePredicateKind::BoundPredicate(p) => p.bounds,
WherePredicateKind::RegionPredicate(p) => p.bounds,
}
}
}
#[derive(Copy, Clone, Debug, StableHash, PartialEq, Eq)]
pub enum PredicateOrigin {
WhereClause,
GenericParam,
ImplTrait,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct WhereBoundPredicate<'hir> {
pub origin: PredicateOrigin,
pub bound_generic_params: &'hir [GenericParam<'hir>],
pub bounded_ty: &'hir Ty<'hir>,
pub bounds: GenericBounds<'hir>,
}
impl<'hir> WhereBoundPredicate<'hir> {
pub fn is_param_bound(&self, param_def_id: DefId) -> bool {
self.bounded_ty.as_generic_param().is_some_and(|(def_id, _)| def_id == param_def_id)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct WhereRegionPredicate<'hir> {
pub in_where_clause: bool,
pub lifetime: &'hir Lifetime,
pub bounds: GenericBounds<'hir>,
}
impl<'hir> WhereRegionPredicate<'hir> {
fn is_param_bound(&self, param_def_id: LocalDefId) -> bool {
self.lifetime.kind == LifetimeKind::Param(param_def_id)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct WhereEqPredicate<'hir> {
pub lhs_ty: &'hir Ty<'hir>,
pub rhs_ty: &'hir Ty<'hir>,
}
#[derive(Clone, Copy, Debug)]
pub struct ParentedNode<'tcx> {
pub parent: ItemLocalId,
pub node: Node<'tcx>,
}
#[derive(Debug)]
pub struct AttributeMap<'tcx> {
pub map: SortedMap<ItemLocalId, &'tcx [Attribute]>,
pub define_opaque: Option<&'tcx [(Span, LocalDefId)]>,
pub opt_hash: Option<Fingerprint>,
}
impl<'tcx> AttributeMap<'tcx> {
pub const EMPTY: &'static AttributeMap<'static> = &AttributeMap {
map: SortedMap::new(),
opt_hash: Some(Fingerprint::ZERO),
define_opaque: None,
};
#[inline]
pub fn get(&self, id: ItemLocalId) -> &'tcx [Attribute] {
self.map.get(&id).copied().unwrap_or(&[])
}
}
pub struct OwnerNodes<'tcx> {
pub opt_hash: Option<Fingerprint>,
pub nodes: IndexVec<ItemLocalId, ParentedNode<'tcx>>,
pub bodies: SortedMap<ItemLocalId, &'tcx Body<'tcx>>,
}
impl<'tcx> OwnerNodes<'tcx> {
pub fn node(&self) -> OwnerNode<'tcx> {
self.nodes[ItemLocalId::ZERO].node.as_owner().unwrap()
}
pub fn synthetic() -> OwnerNodes<'tcx> {
OwnerNodes {
opt_hash: Some(Fingerprint::ZERO),
nodes: IndexVec::from_elem_n(
ParentedNode { parent: ItemLocalId::INVALID, node: OwnerNode::Synthetic.into() },
1,
),
bodies: SortedMap::new(),
}
}
}
impl fmt::Debug for OwnerNodes<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("OwnerNodes")
.field("node", &self.nodes[ItemLocalId::ZERO])
.field(
"parents",
&fmt::from_fn(|f| {
f.debug_list()
.entries(self.nodes.iter_enumerated().map(|(id, parented_node)| {
fmt::from_fn(move |f| write!(f, "({id:?}, {:?})", parented_node.parent))
}))
.finish()
}),
)
.field("bodies", &self.bodies)
.field("opt_hash", &self.opt_hash)
.finish()
}
}
#[derive(Debug)]
pub struct OwnerInfo<'hir> {
pub nodes: OwnerNodes<'hir>,
pub parenting: LocalDefIdMap<ItemLocalId>,
pub attrs: AttributeMap<'hir>,
pub trait_map: ItemLocalMap<&'hir [TraitCandidate<'hir>]>,
pub children: UnordMap<LocalDefId, MaybeOwner<'hir>>,
pub delayed_lints: Steal<DelayedLints>,
pub opt_hash: Option<Fingerprint>,
}
impl<'tcx> OwnerInfo<'tcx> {
#[inline]
pub fn node(&self) -> OwnerNode<'tcx> {
self.nodes.node()
}
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum MaybeOwner<'tcx> {
Owner(&'tcx OwnerInfo<'tcx>),
NonOwner(HirId),
}
impl<'tcx> MaybeOwner<'tcx> {
#[inline]
pub fn as_owner(self) -> Option<&'tcx OwnerInfo<'tcx>> {
match self {
MaybeOwner::Owner(i) => Some(i),
MaybeOwner::NonOwner(_) => None,
}
}
#[inline]
pub fn unwrap(self) -> &'tcx OwnerInfo<'tcx> {
self.as_owner().unwrap_or_else(|| panic!("Not a HIR owner"))
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Closure<'hir> {
pub def_id: LocalDefId,
pub binder: ClosureBinder,
pub constness: Constness,
pub capture_clause: CaptureBy,
pub bound_generic_params: &'hir [GenericParam<'hir>],
pub fn_decl: &'hir FnDecl<'hir>,
pub body: BodyId,
pub fn_decl_span: Span,
pub fn_arg_span: Option<Span>,
pub kind: ClosureKind,
pub explicit_captures: &'hir [ExplicitCapture],
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct ExplicitCapture {
pub var_hir_id: HirId,
}
#[derive(Clone, PartialEq, Eq, Debug, Copy, Hash, StableHash, Encodable, Decodable)]
pub enum ClosureKind {
Closure,
Coroutine(CoroutineKind),
CoroutineClosure(CoroutineDesugaring),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Block<'hir> {
pub stmts: &'hir [Stmt<'hir>],
pub expr: Option<&'hir Expr<'hir>>,
#[stable_hash(ignore)]
pub hir_id: HirId,
pub rules: BlockCheckMode,
pub span: Span,
pub targeted_by_break: bool,
}
impl<'hir> Block<'hir> {
pub fn innermost_block(&self) -> &Block<'hir> {
let mut block = self;
while let Some(Expr { kind: ExprKind::Block(inner_block, _), .. }) = block.expr {
block = inner_block;
}
block
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TyFieldPath {
pub variant: Option<Ident>,
pub field: Ident,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TyPat<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub kind: TyPatKind<'hir>,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Pat<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub kind: PatKind<'hir>,
pub span: Span,
pub default_binding_modes: bool,
}
impl<'hir> Pat<'hir> {
fn walk_short_(&self, it: &mut impl FnMut(&Pat<'hir>) -> bool) -> bool {
if !it(self) {
return false;
}
use PatKind::*;
match self.kind {
Missing => unreachable!(),
Wild | Never | Expr(_) | Range(..) | Binding(.., None) | Err(_) => true,
Deref(s) | Ref(s, _, _) | Binding(.., Some(s)) | Guard(s, _) => s.walk_short_(it),
Struct(_, fields, _) => fields.iter().all(|field| field.pat.walk_short_(it)),
TupleStruct(_, s, _) | Tuple(s, _) | Or(s) => s.iter().all(|p| p.walk_short_(it)),
Slice(before, slice, after) => {
before.iter().chain(slice).chain(after.iter()).all(|p| p.walk_short_(it))
}
}
}
pub fn walk_short(&self, mut it: impl FnMut(&Pat<'hir>) -> bool) -> bool {
self.walk_short_(&mut it)
}
fn walk_(&self, it: &mut impl FnMut(&Pat<'hir>) -> bool) {
if !it(self) {
return;
}
use PatKind::*;
match self.kind {
Missing | Wild | Never | Expr(_) | Range(..) | Binding(.., None) | Err(_) => {}
Deref(s) | Ref(s, _, _) | Binding(.., Some(s)) | Guard(s, _) => s.walk_(it),
Struct(_, fields, _) => fields.iter().for_each(|field| field.pat.walk_(it)),
TupleStruct(_, s, _) | Tuple(s, _) | Or(s) => s.iter().for_each(|p| p.walk_(it)),
Slice(before, slice, after) => {
before.iter().chain(slice).chain(after.iter()).for_each(|p| p.walk_(it))
}
}
}
pub fn walk(&self, mut it: impl FnMut(&Pat<'hir>) -> bool) {
self.walk_(&mut it)
}
pub fn walk_always(&self, mut it: impl FnMut(&Pat<'_>)) {
self.walk(|p| {
it(p);
true
})
}
pub fn is_never_pattern(&self) -> bool {
let mut is_never_pattern = false;
self.walk(|pat| match &pat.kind {
PatKind::Never => {
is_never_pattern = true;
false
}
PatKind::Or(s) => {
is_never_pattern = s.iter().all(|p| p.is_never_pattern());
false
}
_ => true,
});
is_never_pattern
}
pub fn is_guaranteed_to_constitute_read_for_never(&self) -> bool {
match self.kind {
PatKind::Wild => false,
PatKind::Guard(pat, _) => pat.is_guaranteed_to_constitute_read_for_never(),
PatKind::Or(subpats) => {
subpats.iter().all(|pat| pat.is_guaranteed_to_constitute_read_for_never())
}
PatKind::Never => true,
PatKind::Missing
| PatKind::Binding(_, _, _, _)
| PatKind::Struct(_, _, _)
| PatKind::TupleStruct(_, _, _)
| PatKind::Tuple(_, _)
| PatKind::Ref(_, _, _)
| PatKind::Deref(_)
| PatKind::Expr(_)
| PatKind::Range(_, _, _)
| PatKind::Slice(_, _, _)
| PatKind::Err(_) => true,
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct PatField<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub ident: Ident,
pub pat: &'hir Pat<'hir>,
pub is_shorthand: bool,
pub span: Span,
}
#[derive(Copy, Clone, PartialEq, Debug, StableHash, Hash, Eq, Encodable, Decodable)]
pub enum RangeEnd {
Included,
Excluded,
}
impl fmt::Display for RangeEnd {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(match self {
RangeEnd::Included => "..=",
RangeEnd::Excluded => "..",
})
}
}
#[derive(Clone, Copy, PartialEq, Eq, Hash, StableHash)]
pub struct DotDotPos(u32);
impl DotDotPos {
pub fn new(n: Option<usize>) -> Self {
match n {
Some(n) => {
assert!(n < u32::MAX as usize);
Self(n as u32)
}
None => Self(u32::MAX),
}
}
pub fn as_opt_usize(&self) -> Option<usize> {
if self.0 == u32::MAX { None } else { Some(self.0 as usize) }
}
}
impl fmt::Debug for DotDotPos {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.as_opt_usize().fmt(f)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct PatExpr<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub span: Span,
pub kind: PatExprKind<'hir>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum PatExprKind<'hir> {
Lit {
lit: Lit,
negated: bool,
},
Path(QPath<'hir>),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum TyPatKind<'hir> {
Range(&'hir ConstArg<'hir>, &'hir ConstArg<'hir>),
NotNull,
Or(&'hir [TyPat<'hir>]),
Err(ErrorGuaranteed),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum PatKind<'hir> {
Missing,
Wild,
Binding(BindingMode, HirId, Ident, Option<&'hir Pat<'hir>>),
Struct(QPath<'hir>, &'hir [PatField<'hir>], Option<Span>),
TupleStruct(QPath<'hir>, &'hir [Pat<'hir>], DotDotPos),
Or(&'hir [Pat<'hir>]),
Never,
Tuple(&'hir [Pat<'hir>], DotDotPos),
Deref(&'hir Pat<'hir>),
Ref(&'hir Pat<'hir>, Pinnedness, Mutability),
Expr(&'hir PatExpr<'hir>),
Guard(&'hir Pat<'hir>, &'hir Expr<'hir>),
Range(Option<&'hir PatExpr<'hir>>, Option<&'hir PatExpr<'hir>>, RangeEnd),
Slice(&'hir [Pat<'hir>], Option<&'hir Pat<'hir>>, &'hir [Pat<'hir>]),
Err(ErrorGuaranteed),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Stmt<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub kind: StmtKind<'hir>,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum StmtKind<'hir> {
Let(&'hir LetStmt<'hir>),
Item(ItemId),
Expr(&'hir Expr<'hir>),
Semi(&'hir Expr<'hir>),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct LetStmt<'hir> {
pub super_: Option<Span>,
pub pat: &'hir Pat<'hir>,
pub ty: Option<&'hir Ty<'hir>>,
pub init: Option<&'hir Expr<'hir>>,
pub els: Option<&'hir Block<'hir>>,
#[stable_hash(ignore)]
pub hir_id: HirId,
pub span: Span,
pub source: LocalSource,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Arm<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub span: Span,
pub pat: &'hir Pat<'hir>,
pub guard: Option<&'hir Expr<'hir>>,
pub body: &'hir Expr<'hir>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct LetExpr<'hir> {
pub span: Span,
pub pat: &'hir Pat<'hir>,
pub ty: Option<&'hir Ty<'hir>>,
pub init: &'hir Expr<'hir>,
pub recovered: ast::Recovered,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct ExprField<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub ident: Ident,
pub expr: &'hir Expr<'hir>,
pub span: Span,
pub is_shorthand: bool,
}
#[derive(Copy, Clone, PartialEq, Debug, StableHash)]
pub enum BlockCheckMode {
DefaultBlock,
UnsafeBlock(UnsafeSource),
}
#[derive(Copy, Clone, PartialEq, Debug, StableHash)]
pub enum UnsafeSource {
CompilerGenerated,
UserProvided,
}
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug, StableHash)]
pub struct BodyId {
pub hir_id: HirId,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Body<'hir> {
pub params: &'hir [Param<'hir>],
pub value: &'hir Expr<'hir>,
}
impl<'hir> Body<'hir> {
pub fn id(&self) -> BodyId {
BodyId { hir_id: self.value.hir_id }
}
}
#[derive(Clone, PartialEq, Eq, Debug, Copy, Hash, StableHash, Encodable, Decodable)]
pub enum CoroutineKind {
Desugared(CoroutineDesugaring, CoroutineSource),
Coroutine(Movability),
}
impl CoroutineKind {
pub fn movability(self) -> Movability {
match self {
CoroutineKind::Desugared(CoroutineDesugaring::Async, _)
| CoroutineKind::Desugared(CoroutineDesugaring::AsyncGen, _) => Movability::Static,
CoroutineKind::Desugared(CoroutineDesugaring::Gen, _) => Movability::Movable,
CoroutineKind::Coroutine(mov) => mov,
}
}
pub fn is_fn_like(self) -> bool {
matches!(self, CoroutineKind::Desugared(_, CoroutineSource::Fn))
}
pub fn is_async_desugaring(self) -> bool {
matches!(
self,
CoroutineKind::Desugared(CoroutineDesugaring::Async | CoroutineDesugaring::AsyncGen, _)
)
}
pub fn to_plural_string(&self) -> String {
match self {
CoroutineKind::Desugared(d, CoroutineSource::Fn) => format!("{d:#}fn bodies"),
CoroutineKind::Desugared(d, CoroutineSource::Block) => format!("{d:#}blocks"),
CoroutineKind::Desugared(d, CoroutineSource::Closure) => format!("{d:#}closure bodies"),
CoroutineKind::Coroutine(_) => "coroutines".to_string(),
}
}
}
impl fmt::Display for CoroutineKind {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
CoroutineKind::Desugared(d, k) => {
d.fmt(f)?;
k.fmt(f)
}
CoroutineKind::Coroutine(_) => f.write_str("coroutine"),
}
}
}
#[derive(Clone, PartialEq, Eq, Hash, Debug, Copy, StableHash, Encodable, Decodable)]
pub enum CoroutineSource {
Block,
Closure,
Fn,
}
impl fmt::Display for CoroutineSource {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
CoroutineSource::Block => "block",
CoroutineSource::Closure => "closure body",
CoroutineSource::Fn => "fn body",
}
.fmt(f)
}
}
#[derive(Clone, PartialEq, Eq, Debug, Copy, Hash, StableHash, Encodable, Decodable)]
pub enum CoroutineDesugaring {
Async,
Gen,
AsyncGen,
}
impl fmt::Display for CoroutineDesugaring {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
CoroutineDesugaring::Async => {
if f.alternate() {
f.write_str("`async` ")?;
} else {
f.write_str("async ")?
}
}
CoroutineDesugaring::Gen => {
if f.alternate() {
f.write_str("`gen` ")?;
} else {
f.write_str("gen ")?
}
}
CoroutineDesugaring::AsyncGen => {
if f.alternate() {
f.write_str("`async gen` ")?;
} else {
f.write_str("async gen ")?
}
}
}
Ok(())
}
}
#[derive(Copy, Clone, Debug)]
pub enum BodyOwnerKind {
Fn,
Closure,
Const { inline: bool },
Static(Mutability),
GlobalAsm,
}
impl BodyOwnerKind {
pub fn is_fn_or_closure(self) -> bool {
match self {
BodyOwnerKind::Fn | BodyOwnerKind::Closure => true,
BodyOwnerKind::Const { .. } | BodyOwnerKind::Static(_) | BodyOwnerKind::GlobalAsm => {
false
}
}
}
}
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum ConstContext {
ConstFn,
Static(Mutability),
Const {
allow_const_fn_promotion: bool,
},
}
impl ConstContext {
pub fn keyword_name(self) -> &'static str {
match self {
Self::Const { .. } => "const",
Self::Static(Mutability::Not) => "static",
Self::Static(Mutability::Mut) => "static mut",
Self::ConstFn => "const fn",
}
}
}
impl fmt::Display for ConstContext {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match *self {
Self::Const { .. } => write!(f, "constant"),
Self::Static(_) => write!(f, "static"),
Self::ConstFn => write!(f, "constant function"),
}
}
}
impl IntoDiagArg for ConstContext {
fn into_diag_arg(self, _: &mut crate::rustc_error_messages::LongTyPath) -> DiagArgValue {
DiagArgValue::Str(Cow::Borrowed(match self {
ConstContext::ConstFn => "constant function",
ConstContext::Static(_) => "static",
ConstContext::Const { .. } => "constant",
}))
}
}
pub type Lit = Spanned<LitKind>;
#[derive(Copy, Clone, Debug, StableHash)]
pub struct AnonConst {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub def_id: LocalDefId,
pub body: BodyId,
pub span: Span,
}
#[derive(Copy, Clone, Debug, StableHash)]
pub struct ConstBlock {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub def_id: LocalDefId,
pub body: BodyId,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Expr<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub kind: ExprKind<'hir>,
pub span: Span,
}
impl Expr<'_> {
pub fn precedence(&self, has_attr: &dyn Fn(HirId) -> bool) -> ExprPrecedence {
let prefix_attrs_precedence = || -> ExprPrecedence {
if has_attr(self.hir_id) { ExprPrecedence::Prefix } else { ExprPrecedence::Unambiguous }
};
match &self.kind {
ExprKind::Closure(closure) => match closure.fn_decl.output {
FnRetTy::DefaultReturn(_) => ExprPrecedence::Jump,
FnRetTy::Return(_) => prefix_attrs_precedence(),
},
ExprKind::Break(..)
| ExprKind::Ret(..)
| ExprKind::Yield(..)
| ExprKind::Become(..) => ExprPrecedence::Jump,
ExprKind::Binary(op, ..) => op.node.precedence(),
ExprKind::Cast(..) => ExprPrecedence::Cast,
ExprKind::Assign(..) | ExprKind::AssignOp(..) => ExprPrecedence::Assign,
ExprKind::AddrOf(..) => ExprPrecedence::Prefix,
ExprKind::Let(..) | ExprKind::Unary(..) => ExprPrecedence::Prefix,
ExprKind::Array(_)
| ExprKind::Block(..)
| ExprKind::Call(..)
| ExprKind::ConstBlock(_)
| ExprKind::Continue(..)
| ExprKind::Field(..)
| ExprKind::If(..)
| ExprKind::Index(..)
| ExprKind::InlineAsm(..)
| ExprKind::Lit(_)
| ExprKind::Loop(..)
| ExprKind::Match(..)
| ExprKind::MethodCall(..)
| ExprKind::OffsetOf(..)
| ExprKind::Path(..)
| ExprKind::Repeat(..)
| ExprKind::Struct(..)
| ExprKind::Tup(_)
| ExprKind::Type(..)
| ExprKind::UnsafeBinderCast(..)
| ExprKind::Use(..)
| ExprKind::Err(_) => prefix_attrs_precedence(),
ExprKind::DropTemps(expr, ..) => expr.precedence(has_attr),
}
}
pub fn is_syntactic_place_expr(&self) -> bool {
self.is_place_expr(|_| true)
}
pub fn is_place_expr(&self, mut allow_projections_from: impl FnMut(&Self) -> bool) -> bool {
match self.kind {
ExprKind::Path(QPath::Resolved(_, ref path)) => {
matches!(path.res, Res::Local(..) | Res::Def(DefKind::Static { .. }, _) | Res::Err)
}
ExprKind::Type(ref e, _) => e.is_place_expr(allow_projections_from),
ExprKind::UnsafeBinderCast(_, e, _) => e.is_place_expr(allow_projections_from),
ExprKind::Unary(UnOp::Deref, _) => true,
ExprKind::Field(ref base, _) | ExprKind::Index(ref base, _, _) => {
allow_projections_from(base) || base.is_place_expr(allow_projections_from)
}
ExprKind::Err(_guar)
| ExprKind::Let(&LetExpr { recovered: ast::Recovered::Yes(_guar), .. }) => true,
ExprKind::Path(QPath::TypeRelative(..))
| ExprKind::Call(..)
| ExprKind::MethodCall(..)
| ExprKind::Use(..)
| ExprKind::Struct(..)
| ExprKind::Tup(..)
| ExprKind::If(..)
| ExprKind::Match(..)
| ExprKind::Closure { .. }
| ExprKind::Block(..)
| ExprKind::Repeat(..)
| ExprKind::Array(..)
| ExprKind::Break(..)
| ExprKind::Continue(..)
| ExprKind::Ret(..)
| ExprKind::Become(..)
| ExprKind::Let(..)
| ExprKind::Loop(..)
| ExprKind::Assign(..)
| ExprKind::InlineAsm(..)
| ExprKind::OffsetOf(..)
| ExprKind::AssignOp(..)
| ExprKind::Lit(_)
| ExprKind::ConstBlock(..)
| ExprKind::Unary(..)
| ExprKind::AddrOf(..)
| ExprKind::Binary(..)
| ExprKind::Yield(..)
| ExprKind::Cast(..)
| ExprKind::DropTemps(..) => false,
}
}
pub fn range_span(&self) -> Option<Span> {
is_range_literal(self).then(|| self.span.parent_callsite().unwrap())
}
pub fn is_size_lit(&self) -> bool {
matches!(
self.kind,
ExprKind::Lit(Lit {
node: LitKind::Int(_, LitIntType::Unsuffixed | LitIntType::Unsigned(UintTy::Usize)),
..
})
)
}
pub fn peel_drop_temps(&self) -> &Self {
let mut expr = self;
while let ExprKind::DropTemps(inner) = &expr.kind {
expr = inner;
}
expr
}
pub fn peel_blocks(&self) -> &Self {
let mut expr = self;
while let ExprKind::Block(Block { expr: Some(inner), .. }, _) = &expr.kind {
expr = inner;
}
expr
}
pub fn peel_borrows(&self) -> &Self {
let mut expr = self;
while let ExprKind::AddrOf(.., inner) = &expr.kind {
expr = inner;
}
expr
}
pub fn can_have_side_effects(&self) -> bool {
match self.peel_drop_temps().kind {
ExprKind::Path(_) | ExprKind::Lit(_) | ExprKind::OffsetOf(..) | ExprKind::Use(..) => {
false
}
ExprKind::Type(base, _)
| ExprKind::Unary(_, base)
| ExprKind::Field(base, _)
| ExprKind::Index(base, _, _)
| ExprKind::AddrOf(.., base)
| ExprKind::Cast(base, _)
| ExprKind::UnsafeBinderCast(_, base, _) => {
base.can_have_side_effects()
}
ExprKind::Binary(_, lhs, rhs) => {
lhs.can_have_side_effects() || rhs.can_have_side_effects()
}
ExprKind::Struct(_, fields, init) => {
let init_side_effects = match init {
StructTailExpr::Base(init) => init.can_have_side_effects(),
StructTailExpr::DefaultFields(_)
| StructTailExpr::None
| StructTailExpr::NoneWithError(_) => false,
};
fields.iter().map(|field| field.expr).any(|e| e.can_have_side_effects())
|| init_side_effects
}
ExprKind::Array(args)
| ExprKind::Tup(args)
| ExprKind::Call(
Expr {
kind:
ExprKind::Path(QPath::Resolved(
None,
Path { res: Res::Def(DefKind::Ctor(_, CtorKind::Fn), _), .. },
)),
..
},
args,
) => args.iter().any(|arg| arg.can_have_side_effects()),
ExprKind::Repeat(arg, _) => arg.can_have_side_effects(),
ExprKind::If(..)
| ExprKind::Match(..)
| ExprKind::MethodCall(..)
| ExprKind::Call(..)
| ExprKind::Closure { .. }
| ExprKind::Block(..)
| ExprKind::Break(..)
| ExprKind::Continue(..)
| ExprKind::Ret(..)
| ExprKind::Become(..)
| ExprKind::Let(..)
| ExprKind::Loop(..)
| ExprKind::Assign(..)
| ExprKind::InlineAsm(..)
| ExprKind::AssignOp(..)
| ExprKind::ConstBlock(..)
| ExprKind::Yield(..)
| ExprKind::DropTemps(..)
| ExprKind::Err(_) => true,
}
}
pub fn is_approximately_pattern(&self) -> bool {
match &self.kind {
ExprKind::Array(_)
| ExprKind::Call(..)
| ExprKind::Tup(_)
| ExprKind::Lit(_)
| ExprKind::Path(_)
| ExprKind::Struct(..) => true,
_ => false,
}
}
pub fn equivalent_for_indexing(&self, other: &Expr<'_>) -> bool {
match (self.kind, other.kind) {
(ExprKind::Lit(lit1), ExprKind::Lit(lit2)) => lit1.node == lit2.node,
(
ExprKind::Path(QPath::Resolved(None, path1)),
ExprKind::Path(QPath::Resolved(None, path2)),
) => path1.res == path2.res,
(
ExprKind::Struct(
&QPath::Resolved(None, &Path { res: Res::Def(_, path1_def_id), .. }),
args1,
StructTailExpr::None,
),
ExprKind::Struct(
&QPath::Resolved(None, &Path { res: Res::Def(_, path2_def_id), .. }),
args2,
StructTailExpr::None,
),
) => {
path2_def_id == path1_def_id
&& is_range_literal(self)
&& is_range_literal(other)
&& core::iter::zip(args1, args2)
.all(|(a, b)| a.expr.equivalent_for_indexing(b.expr))
}
_ => false,
}
}
pub fn method_ident(&self) -> Option<Ident> {
match self.kind {
ExprKind::MethodCall(receiver_method, ..) => Some(receiver_method.ident),
ExprKind::Unary(_, expr) | ExprKind::AddrOf(.., expr) => expr.method_ident(),
_ => None,
}
}
}
pub fn is_range_literal(expr: &Expr<'_>) -> bool {
if let ExprKind::Struct(QPath::Resolved(None, path), _, StructTailExpr::None) = expr.kind
&& let [.., segment] = path.segments
&& let sym::RangeFrom
| sym::RangeFull
| sym::Range
| sym::RangeToInclusive
| sym::RangeTo
| sym::RangeFromCopy
| sym::RangeCopy
| sym::RangeInclusiveCopy
| sym::RangeToInclusiveCopy = segment.ident.name
&& expr.span.is_desugaring(DesugaringKind::RangeExpr)
{
true
} else if let ExprKind::Call(func, _) = &expr.kind
&& let ExprKind::Path(QPath::Resolved(None, path)) = func.kind
&& let [.., segment] = path.segments
&& let sym::range_inclusive_new = segment.ident.name
&& expr.span.is_desugaring(DesugaringKind::RangeExpr)
{
true
} else {
false
}
}
pub fn expr_needs_parens(expr: &Expr<'_>) -> bool {
match expr.kind {
ExprKind::Cast(_, _) | ExprKind::Binary(_, _, _) => true,
_ if is_range_literal(expr) => true,
_ => false,
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum ExprKind<'hir> {
ConstBlock(ConstBlock),
Array(&'hir [Expr<'hir>]),
Call(&'hir Expr<'hir>, &'hir [Expr<'hir>]),
MethodCall(&'hir PathSegment<'hir>, &'hir Expr<'hir>, &'hir [Expr<'hir>], Span),
Use(&'hir Expr<'hir>, Span),
Tup(&'hir [Expr<'hir>]),
Binary(BinOp, &'hir Expr<'hir>, &'hir Expr<'hir>),
Unary(UnOp, &'hir Expr<'hir>),
Lit(Lit),
Cast(&'hir Expr<'hir>, &'hir Ty<'hir>),
Type(&'hir Expr<'hir>, &'hir Ty<'hir>),
DropTemps(&'hir Expr<'hir>),
Let(&'hir LetExpr<'hir>),
If(&'hir Expr<'hir>, &'hir Expr<'hir>, Option<&'hir Expr<'hir>>),
Loop(&'hir Block<'hir>, Option<Label>, LoopSource, Span),
Match(&'hir Expr<'hir>, &'hir [Arm<'hir>], MatchSource),
Closure(&'hir Closure<'hir>),
Block(&'hir Block<'hir>, Option<Label>),
Assign(&'hir Expr<'hir>, &'hir Expr<'hir>, Span),
AssignOp(AssignOp, &'hir Expr<'hir>, &'hir Expr<'hir>),
Field(&'hir Expr<'hir>, Ident),
Index(&'hir Expr<'hir>, &'hir Expr<'hir>, Span),
Path(QPath<'hir>),
AddrOf(BorrowKind, Mutability, &'hir Expr<'hir>),
Break(Destination, Option<&'hir Expr<'hir>>),
Continue(Destination),
Ret(Option<&'hir Expr<'hir>>),
Become(&'hir Expr<'hir>),
InlineAsm(&'hir InlineAsm<'hir>),
OffsetOf(&'hir Ty<'hir>, &'hir [Ident]),
Struct(&'hir QPath<'hir>, &'hir [ExprField<'hir>], StructTailExpr<'hir>),
Repeat(&'hir Expr<'hir>, &'hir ConstArg<'hir>),
Yield(&'hir Expr<'hir>, YieldSource),
UnsafeBinderCast(UnsafeBinderCastKind, &'hir Expr<'hir>, Option<&'hir Ty<'hir>>),
Err(crate::rustc_span::ErrorGuaranteed),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum StructTailExpr<'hir> {
None,
Base(&'hir Expr<'hir>),
DefaultFields(Span),
NoneWithError(ErrorGuaranteed),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum QPath<'hir> {
Resolved(Option<&'hir Ty<'hir>>, &'hir Path<'hir>),
TypeRelative(&'hir Ty<'hir>, &'hir PathSegment<'hir>),
}
impl<'hir> QPath<'hir> {
pub fn span(&self) -> Span {
match *self {
QPath::Resolved(_, path) => path.span,
QPath::TypeRelative(qself, ps) => qself.span.to(ps.ident.span),
}
}
pub fn qself_span(&self) -> Span {
match *self {
QPath::Resolved(_, path) => path.span,
QPath::TypeRelative(qself, _) => qself.span,
}
}
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum LocalSource {
Normal,
AsyncFn,
AwaitDesugar,
AssignDesugar,
Contract,
}
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug, StableHash, Encodable, Decodable)]
pub enum MatchSource {
Normal,
Postfix,
ForLoopDesugar,
TryDesugar(HirId),
AwaitDesugar,
FormatArgs,
}
impl MatchSource {
#[inline]
pub const fn name(self) -> &'static str {
use MatchSource::*;
match self {
Normal => "match",
Postfix => ".match",
ForLoopDesugar => "for",
TryDesugar(_) => "?",
AwaitDesugar => ".await",
FormatArgs => "format_args!()",
}
}
}
#[derive(Copy, Clone, PartialEq, Debug, StableHash)]
pub enum LoopSource {
Loop,
While,
ForLoop,
}
impl LoopSource {
pub fn name(self) -> &'static str {
match self {
LoopSource::Loop => "loop",
LoopSource::While => "while",
LoopSource::ForLoop => "for",
}
}
}
#[derive(Copy, Clone, Debug, PartialEq, StableHash)]
pub enum LoopIdError {
OutsideLoopScope,
UnlabeledCfInWhileCondition,
UnresolvedLabel,
}
impl fmt::Display for LoopIdError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(match self {
LoopIdError::OutsideLoopScope => "not inside loop scope",
LoopIdError::UnlabeledCfInWhileCondition => {
"unlabeled control flow (break or continue) in while condition"
}
LoopIdError::UnresolvedLabel => "label not found",
})
}
}
#[derive(Copy, Clone, Debug, PartialEq, StableHash)]
pub struct Destination {
pub label: Option<Label>,
pub target_id: Result<HirId, LoopIdError>,
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum YieldSource {
Await { expr: Option<HirId> },
Yield,
}
impl fmt::Display for YieldSource {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(match self {
YieldSource::Await { .. } => "`await`",
YieldSource::Yield => "`yield`",
})
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct MutTy<'hir> {
pub ty: &'hir Ty<'hir>,
pub mutbl: Mutability,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct FnSig<'hir> {
pub header: FnHeader,
pub decl: &'hir FnDecl<'hir>,
pub span: Span,
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Debug, StableHash)]
pub struct TraitItemId {
pub owner_id: OwnerId,
}
impl TraitItemId {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TraitItem<'hir> {
pub ident: Ident,
pub owner_id: OwnerId,
pub generics: &'hir Generics<'hir>,
pub kind: TraitItemKind<'hir>,
pub span: Span,
pub defaultness: Defaultness,
}
macro_rules! expect_methods_self_kind {
( $( $name:ident, $ret_ty:ty, $pat:pat, $ret_val:expr; )* ) => {
$(
#[track_caller]
pub fn $name(&self) -> $ret_ty {
let $pat = &self.kind else { expect_failed(stringify!($name), self) };
$ret_val
}
)*
}
}
macro_rules! expect_methods_self {
( $( $name:ident, $ret_ty:ty, $pat:pat, $ret_val:expr; )* ) => {
$(
#[track_caller]
pub fn $name(&self) -> $ret_ty {
let $pat = self else { expect_failed(stringify!($name), self) };
$ret_val
}
)*
}
}
#[track_caller]
fn expect_failed<T: fmt::Debug>(ident: &'static str, found: T) -> ! {
panic!("{ident}: found {found:?}")
}
impl<'hir> TraitItem<'hir> {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
pub fn trait_item_id(&self) -> TraitItemId {
TraitItemId { owner_id: self.owner_id }
}
expect_methods_self_kind! {
expect_const, (&'hir Ty<'hir>, Option<ConstItemRhs<'hir>>),
TraitItemKind::Const(ty, rhs), (ty, *rhs);
expect_fn, (&FnSig<'hir>, &TraitFn<'hir>),
TraitItemKind::Fn(ty, trfn), (ty, trfn);
expect_type, (GenericBounds<'hir>, Option<&'hir Ty<'hir>>),
TraitItemKind::Type(bounds, ty), (bounds, *ty);
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum TraitFn<'hir> {
Required(&'hir [Option<Ident>]),
Provided(BodyId),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum TraitItemKind<'hir> {
Const(&'hir Ty<'hir>, Option<ConstItemRhs<'hir>>),
Fn(FnSig<'hir>, TraitFn<'hir>),
Type(GenericBounds<'hir>, Option<&'hir Ty<'hir>>),
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Debug, StableHash)]
pub struct ImplItemId {
pub owner_id: OwnerId,
}
impl ImplItemId {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct ImplItem<'hir> {
pub ident: Ident,
pub owner_id: OwnerId,
pub generics: &'hir Generics<'hir>,
pub kind: ImplItemKind<'hir>,
pub impl_kind: ImplItemImplKind,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum ImplItemImplKind {
Inherent {
vis_span: Span,
},
Trait {
defaultness: Defaultness,
trait_item_def_id: Result<DefId, ErrorGuaranteed>,
},
}
impl<'hir> ImplItem<'hir> {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
pub fn impl_item_id(&self) -> ImplItemId {
ImplItemId { owner_id: self.owner_id }
}
pub fn vis_span(&self) -> Option<Span> {
match self.impl_kind {
ImplItemImplKind::Trait { .. } => None,
ImplItemImplKind::Inherent { vis_span, .. } => Some(vis_span),
}
}
expect_methods_self_kind! {
expect_const, (&'hir Ty<'hir>, ConstItemRhs<'hir>), ImplItemKind::Const(ty, rhs), (ty, *rhs);
expect_fn, (&FnSig<'hir>, BodyId), ImplItemKind::Fn(ty, body), (ty, *body);
expect_type, &'hir Ty<'hir>, ImplItemKind::Type(ty), ty;
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum ImplItemKind<'hir> {
Const(&'hir Ty<'hir>, ConstItemRhs<'hir>),
Fn(FnSig<'hir>, BodyId),
Type(&'hir Ty<'hir>),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct AssocItemConstraint<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub ident: Ident,
pub gen_args: &'hir GenericArgs<'hir>,
pub kind: AssocItemConstraintKind<'hir>,
pub span: Span,
}
impl<'hir> AssocItemConstraint<'hir> {
pub fn ty(self) -> Option<&'hir Ty<'hir>> {
match self.kind {
AssocItemConstraintKind::Equality { term: Term::Ty(ty) } => Some(ty),
_ => None,
}
}
pub fn ct(self) -> Option<&'hir ConstArg<'hir>> {
match self.kind {
AssocItemConstraintKind::Equality { term: Term::Const(ct) } => Some(ct),
_ => None,
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum Term<'hir> {
Ty(&'hir Ty<'hir>),
Const(&'hir ConstArg<'hir>),
}
impl<'hir> From<&'hir Ty<'hir>> for Term<'hir> {
fn from(ty: &'hir Ty<'hir>) -> Self {
Term::Ty(ty)
}
}
impl<'hir> From<&'hir ConstArg<'hir>> for Term<'hir> {
fn from(c: &'hir ConstArg<'hir>) -> Self {
Term::Const(c)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum AssocItemConstraintKind<'hir> {
Equality { term: Term<'hir> },
Bound { bounds: &'hir [GenericBound<'hir>] },
}
impl<'hir> AssocItemConstraintKind<'hir> {
pub fn descr(&self) -> &'static str {
match self {
AssocItemConstraintKind::Equality { .. } => "binding",
AssocItemConstraintKind::Bound { .. } => "constraint",
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum AmbigArg {}
#[derive(Debug, Clone, Copy, StableHash)]
#[repr(C)]
pub struct Ty<'hir, Unambig = ()> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub span: Span,
pub kind: TyKind<'hir, Unambig>,
}
impl<'hir> Ty<'hir, AmbigArg> {
pub fn as_unambig_ty(&self) -> &Ty<'hir> {
let ptr = self as *const Ty<'hir, AmbigArg> as *const Ty<'hir, ()>;
unsafe { &*ptr }
}
}
impl<'hir> Ty<'hir> {
pub fn try_as_ambig_ty(&self) -> Option<&Ty<'hir, AmbigArg>> {
if let TyKind::Infer(()) = self.kind {
return None;
}
let ptr = self as *const Ty<'hir> as *const Ty<'hir, AmbigArg>;
Some(unsafe { &*ptr })
}
}
impl<'hir> Ty<'hir, AmbigArg> {
pub fn peel_refs(&self) -> &Ty<'hir> {
let mut final_ty = self.as_unambig_ty();
while let TyKind::Ref(_, MutTy { ty, .. }) = &final_ty.kind {
final_ty = ty;
}
final_ty
}
}
impl<'hir> Ty<'hir> {
pub fn peel_refs(&self) -> &Self {
let mut final_ty = self;
while let TyKind::Ref(_, MutTy { ty, .. }) = &final_ty.kind {
final_ty = ty;
}
final_ty
}
pub fn as_generic_param(&self) -> Option<(DefId, Ident)> {
let TyKind::Path(QPath::Resolved(None, path)) = self.kind else {
return None;
};
let [segment] = &path.segments else {
return None;
};
match path.res {
Res::Def(DefKind::TyParam, def_id) | Res::SelfTyParam { trait_: def_id } => {
Some((def_id, segment.ident))
}
_ => None,
}
}
pub fn find_self_aliases(&self) -> Vec<Span> {
use crate::rustc_hir::intravisit::Visitor;
struct MyVisitor(Vec<Span>);
impl<'v> Visitor<'v> for MyVisitor {
type NestedFilter = crate::rustc_hir::intravisit::IgnoreNested;
type Result = ();
fn visit_ty(&mut self, t: &'v Ty<'v, AmbigArg>) {
if matches!(
&t.kind,
TyKind::Path(QPath::Resolved(
_,
Path { res: crate::rustc_hir::def::Res::SelfTyAlias { .. }, .. },
))
) {
self.0.push(t.span);
return;
}
crate::rustc_hir::intravisit::walk_ty(self, t);
}
}
let mut my_visitor = MyVisitor(vec![]);
my_visitor.visit_ty_unambig(self);
my_visitor.0
}
pub fn is_suggestable_infer_ty(&self) -> bool {
fn are_suggestable_generic_args(generic_args: &[GenericArg<'_>]) -> bool {
generic_args.iter().any(|arg| match arg {
GenericArg::Type(ty) => ty.as_unambig_ty().is_suggestable_infer_ty(),
GenericArg::Infer(_) => true,
_ => false,
})
}
debug!(?self);
match &self.kind {
TyKind::Infer(()) => true,
TyKind::Slice(ty) => ty.is_suggestable_infer_ty(),
TyKind::Array(ty, length) => {
ty.is_suggestable_infer_ty() || matches!(length.kind, ConstArgKind::Infer(..))
}
TyKind::Tup(tys) => tys.iter().any(Self::is_suggestable_infer_ty),
TyKind::Ptr(mut_ty) | TyKind::Ref(_, mut_ty) => mut_ty.ty.is_suggestable_infer_ty(),
TyKind::Path(QPath::TypeRelative(ty, segment)) => {
ty.is_suggestable_infer_ty() || are_suggestable_generic_args(segment.args().args)
}
TyKind::Path(QPath::Resolved(ty_opt, Path { segments, .. })) => {
ty_opt.is_some_and(Self::is_suggestable_infer_ty)
|| segments
.iter()
.any(|segment| are_suggestable_generic_args(segment.args().args))
}
_ => false,
}
}
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Hash, Debug, StableHash)]
pub enum PrimTy {
Int(IntTy),
Uint(UintTy),
Float(FloatTy),
Str,
Bool,
Char,
}
impl PrimTy {
pub const ALL: [Self; 19] = [
Self::Int(IntTy::I8),
Self::Int(IntTy::I16),
Self::Int(IntTy::I32),
Self::Int(IntTy::I64),
Self::Int(IntTy::I128),
Self::Int(IntTy::Isize),
Self::Uint(UintTy::U8),
Self::Uint(UintTy::U16),
Self::Uint(UintTy::U32),
Self::Uint(UintTy::U64),
Self::Uint(UintTy::U128),
Self::Uint(UintTy::Usize),
Self::Float(FloatTy::F16),
Self::Float(FloatTy::F32),
Self::Float(FloatTy::F64),
Self::Float(FloatTy::F128),
Self::Bool,
Self::Char,
Self::Str,
];
pub fn name_str(self) -> &'static str {
match self {
PrimTy::Int(i) => i.name_str(),
PrimTy::Uint(u) => u.name_str(),
PrimTy::Float(f) => f.name_str(),
PrimTy::Str => "str",
PrimTy::Bool => "bool",
PrimTy::Char => "char",
}
}
pub fn name(self) -> Symbol {
match self {
PrimTy::Int(i) => i.name(),
PrimTy::Uint(u) => u.name(),
PrimTy::Float(f) => f.name(),
PrimTy::Str => sym::str,
PrimTy::Bool => sym::bool,
PrimTy::Char => sym::char,
}
}
pub fn from_name(name: Symbol) -> Option<Self> {
let ty = match name {
sym::i8 => Self::Int(IntTy::I8),
sym::i16 => Self::Int(IntTy::I16),
sym::i32 => Self::Int(IntTy::I32),
sym::i64 => Self::Int(IntTy::I64),
sym::i128 => Self::Int(IntTy::I128),
sym::isize => Self::Int(IntTy::Isize),
sym::u8 => Self::Uint(UintTy::U8),
sym::u16 => Self::Uint(UintTy::U16),
sym::u32 => Self::Uint(UintTy::U32),
sym::u64 => Self::Uint(UintTy::U64),
sym::u128 => Self::Uint(UintTy::U128),
sym::usize => Self::Uint(UintTy::Usize),
sym::f16 => Self::Float(FloatTy::F16),
sym::f32 => Self::Float(FloatTy::F32),
sym::f64 => Self::Float(FloatTy::F64),
sym::f128 => Self::Float(FloatTy::F128),
sym::bool => Self::Bool,
sym::char => Self::Char,
sym::str => Self::Str,
_ => return None,
};
Some(ty)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct FnPtrTy<'hir> {
pub safety: Safety,
pub abi: ExternAbi,
pub generic_params: &'hir [GenericParam<'hir>],
pub decl: &'hir FnDecl<'hir>,
pub param_idents: &'hir [Option<Ident>],
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct UnsafeBinderTy<'hir> {
pub generic_params: &'hir [GenericParam<'hir>],
pub inner_ty: &'hir Ty<'hir>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct OpaqueTy<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub def_id: LocalDefId,
pub bounds: GenericBounds<'hir>,
pub origin: OpaqueTyOrigin<LocalDefId>,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash, Encodable, Decodable)]
pub enum PreciseCapturingArgKind<T, U> {
Lifetime(T),
Param(U),
}
pub type PreciseCapturingArg<'hir> =
PreciseCapturingArgKind<&'hir Lifetime, PreciseCapturingNonLifetimeArg>;
impl PreciseCapturingArg<'_> {
pub fn hir_id(self) -> HirId {
match self {
PreciseCapturingArg::Lifetime(lt) => lt.hir_id,
PreciseCapturingArg::Param(param) => param.hir_id,
}
}
pub fn name(self) -> Symbol {
match self {
PreciseCapturingArg::Lifetime(lt) => lt.ident.name,
PreciseCapturingArg::Param(param) => param.ident.name,
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct PreciseCapturingNonLifetimeArg {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub ident: Ident,
pub res: Res,
}
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
#[derive(StableHash, Encodable, Decodable)]
pub enum RpitContext {
Trait,
TraitImpl,
}
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
#[derive(StableHash, Encodable, Decodable)]
pub enum OpaqueTyOrigin<D> {
FnReturn {
parent: D,
in_trait_or_impl: Option<RpitContext>,
},
AsyncFn {
parent: D,
in_trait_or_impl: Option<RpitContext>,
},
TyAlias {
parent: D,
in_assoc_ty: bool,
},
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, StableHash)]
pub enum DelegationSelfTyPropagationKind {
SelfTy(HirId ),
SelfParam,
}
#[derive(Debug, StableHash)]
pub struct DelegationInfo {
pub call_expr_id: HirId,
pub call_path_res: DefId,
pub child_seg_id: HirId,
pub parent_seg_id_for_sig: Option<HirId>,
pub child_seg_id_for_sig: Option<HirId>,
pub self_ty_propagation_kind: Option<DelegationSelfTyPropagationKind>,
pub group_id: Option<(LocalExpnId, bool /* unused_target_expr */)>,
pub arguments_to_map: FxIndexSet<usize>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum InferDelegationSig<'hir> {
Input(usize),
Output(&'hir DelegationInfo),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum InferDelegation<'hir> {
DefId(DefId),
Sig(DefId, InferDelegationSig<'hir>),
}
#[repr(u8, C)]
#[derive(Debug, Clone, Copy, StableHash)]
pub enum TyKind<'hir, Unambig = ()> {
InferDelegation(InferDelegation<'hir>),
Slice(&'hir Ty<'hir>),
Array(&'hir Ty<'hir>, &'hir ConstArg<'hir>),
Ptr(MutTy<'hir>),
Ref(&'hir Lifetime, MutTy<'hir>),
FnPtr(&'hir FnPtrTy<'hir>),
UnsafeBinder(&'hir UnsafeBinderTy<'hir>),
Never,
Tup(&'hir [Ty<'hir>]),
Path(QPath<'hir>),
OpaqueDef(&'hir OpaqueTy<'hir>),
TraitAscription(GenericBounds<'hir>),
TraitObject(&'hir [PolyTraitRef<'hir>], TaggedRef<'hir, Lifetime, TraitObjectSyntax>),
Err(crate::rustc_span::ErrorGuaranteed),
Pat(&'hir Ty<'hir>, &'hir TyPat<'hir>),
FieldOf(&'hir Ty<'hir>, &'hir TyFieldPath),
View(&'hir Ty<'hir>, &'hir [Ident]),
Infer(Unambig),
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum InlineAsmOperand<'hir> {
In {
reg: InlineAsmRegOrRegClass,
expr: &'hir Expr<'hir>,
},
Out {
reg: InlineAsmRegOrRegClass,
late: bool,
expr: Option<&'hir Expr<'hir>>,
},
InOut {
reg: InlineAsmRegOrRegClass,
late: bool,
expr: &'hir Expr<'hir>,
},
SplitInOut {
reg: InlineAsmRegOrRegClass,
late: bool,
in_expr: &'hir Expr<'hir>,
out_expr: Option<&'hir Expr<'hir>>,
},
Const {
anon_const: ConstBlock,
},
SymFn {
expr: &'hir Expr<'hir>,
},
SymStatic {
path: QPath<'hir>,
def_id: DefId,
},
Label {
block: &'hir Block<'hir>,
},
}
impl<'hir> InlineAsmOperand<'hir> {
pub fn reg(&self) -> Option<InlineAsmRegOrRegClass> {
match *self {
Self::In { reg, .. }
| Self::Out { reg, .. }
| Self::InOut { reg, .. }
| Self::SplitInOut { reg, .. } => Some(reg),
Self::Const { .. }
| Self::SymFn { .. }
| Self::SymStatic { .. }
| Self::Label { .. } => None,
}
}
pub fn is_clobber(&self) -> bool {
matches!(
self,
InlineAsmOperand::Out { reg: InlineAsmRegOrRegClass::Reg(_), late: _, expr: None }
)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct InlineAsm<'hir> {
pub asm_macro: ast::AsmMacro,
pub template: &'hir [InlineAsmTemplatePiece],
pub template_strs: &'hir [(Symbol, Option<Symbol>, Span)],
pub operands: &'hir [(InlineAsmOperand<'hir>, Span)],
pub options: InlineAsmOptions,
pub line_spans: &'hir [Span],
}
impl InlineAsm<'_> {
pub fn contains_label(&self) -> bool {
self.operands.iter().any(|x| matches!(x.0, InlineAsmOperand::Label { .. }))
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Param<'hir> {
#[stable_hash(ignore)]
pub hir_id: HirId,
pub pat: &'hir Pat<'hir>,
pub ty_span: Span,
pub span: Span,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum SplattedArgIndexError {
InvalidIndex { splatted_arg_index: u8 },
OutOfBounds { splatted_arg_index: u8, args_len: u16 },
}
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[derive(Encodable, Decodable, StableHash)]
pub struct FnDeclFlags {
flags: u8,
splatted: u8,
}
impl fmt::Debug for FnDeclFlags {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let mut f = f.debug_tuple("FnDeclFlags");
f.field(&format!("ImplicitSelfKind({:?})", self.implicit_self()));
if self.lifetime_elision_allowed() {
f.field(&"LifetimeElisionAllowed");
} else {
f.field(&"NoLifetimeElision");
}
if self.c_variadic() {
f.field(&"CVariadic");
}
if let Some(index) = self.splatted() {
f.field(&format!("Splatted({})", index));
}
f.finish()
}
}
impl FnDeclFlags {
const IMPLICIT_SELF_MASK: u8 = 0b111;
const C_VARIADIC_FLAG: u8 = 1 << 3;
const LIFETIME_ELISION_ALLOWED_FLAG: u8 = 1 << 4;
const NO_SPLATTED_ARG_INDEX: u8 = u8::MAX;
pub fn default() -> Self {
Self { flags: 0, splatted: 0 }
.set_implicit_self(ImplicitSelfKind::None)
.set_lifetime_elision_allowed(false)
.set_c_variadic(false)
.set_no_splatted_args()
}
#[must_use = "this method does not modify the receiver"]
pub fn set_implicit_self(mut self, implicit_self: ImplicitSelfKind) -> Self {
self.flags &= !Self::IMPLICIT_SELF_MASK;
match implicit_self {
ImplicitSelfKind::None => self.flags |= 0,
ImplicitSelfKind::Imm => self.flags |= 1,
ImplicitSelfKind::Mut => self.flags |= 2,
ImplicitSelfKind::RefImm => self.flags |= 3,
ImplicitSelfKind::RefMut => self.flags |= 4,
}
self
}
#[must_use = "this method does not modify the receiver"]
pub fn set_c_variadic(mut self, c_variadic: bool) -> Self {
if c_variadic {
self.flags |= Self::C_VARIADIC_FLAG;
} else {
self.flags &= !Self::C_VARIADIC_FLAG;
}
self
}
#[must_use = "this method does not modify the receiver"]
pub fn set_lifetime_elision_allowed(mut self, allowed: bool) -> Self {
if allowed {
self.flags |= Self::LIFETIME_ELISION_ALLOWED_FLAG;
} else {
self.flags &= !Self::LIFETIME_ELISION_ALLOWED_FLAG;
}
self
}
#[must_use = "this method does not modify the receiver"]
pub fn set_splatted(
mut self,
splatted: Option<u8>,
args_len: usize,
) -> Result<Self, SplattedArgIndexError> {
if let Some(splatted_arg_index) = splatted {
if splatted_arg_index == Self::NO_SPLATTED_ARG_INDEX {
return Err(SplattedArgIndexError::InvalidIndex { splatted_arg_index });
} else if usize::from(splatted_arg_index) >= args_len {
return Err(SplattedArgIndexError::OutOfBounds {
splatted_arg_index,
args_len: args_len as u16,
});
}
self.splatted = splatted_arg_index;
} else {
self.splatted = Self::NO_SPLATTED_ARG_INDEX;
}
Ok(self)
}
#[must_use = "this method does not modify the receiver"]
pub fn set_no_splatted_args(mut self) -> Self {
self.splatted = Self::NO_SPLATTED_ARG_INDEX;
self
}
pub fn implicit_self(self) -> ImplicitSelfKind {
match self.flags & Self::IMPLICIT_SELF_MASK {
0 => ImplicitSelfKind::None,
1 => ImplicitSelfKind::Imm,
2 => ImplicitSelfKind::Mut,
3 => ImplicitSelfKind::RefImm,
4 => ImplicitSelfKind::RefMut,
_ => unreachable!(),
}
}
pub fn c_variadic(self) -> bool {
self.flags & Self::C_VARIADIC_FLAG != 0
}
pub fn lifetime_elision_allowed(self) -> bool {
self.flags & Self::LIFETIME_ELISION_ALLOWED_FLAG != 0
}
pub fn splatted(self) -> Option<u8> {
if self.splatted == Self::NO_SPLATTED_ARG_INDEX { None } else { Some(self.splatted) }
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct FnDecl<'hir> {
pub inputs: &'hir [Ty<'hir>],
pub output: FnRetTy<'hir>,
pub fn_decl_kind: FnDeclFlags,
}
impl<'hir> FnDecl<'hir> {
pub fn opt_delegation_sig_id(&self) -> Option<DefId> {
if let FnRetTy::Return(ty) = self.output
&& let TyKind::InferDelegation(InferDelegation::Sig(sig_id, _)) = ty.kind
{
return Some(sig_id);
}
None
}
pub fn opt_delegation_info(&self) -> Option<&'hir DelegationInfo> {
if let FnRetTy::Return(ty) = self.output
&& let TyKind::InferDelegation(InferDelegation::Sig(_, kind)) = ty.kind
&& let InferDelegationSig::Output(generics) = kind
{
return Some(generics);
}
None
}
pub fn implicit_self(&self) -> ImplicitSelfKind {
self.fn_decl_kind.implicit_self()
}
pub fn c_variadic(&self) -> bool {
self.fn_decl_kind.c_variadic()
}
pub fn lifetime_elision_allowed(&self) -> bool {
self.fn_decl_kind.lifetime_elision_allowed()
}
pub fn splatted(&self) -> Option<u8> {
self.fn_decl_kind.splatted()
}
pub fn dummy(span: Span) -> Self {
Self {
inputs: &[],
output: FnRetTy::DefaultReturn(span),
fn_decl_kind: FnDeclFlags::default().set_lifetime_elision_allowed(true),
}
}
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Debug, StableHash)]
pub enum ImplicitSelfKind {
Imm,
Mut,
RefImm,
RefMut,
None,
}
impl ImplicitSelfKind {
pub fn has_implicit_self(&self) -> bool {
!matches!(*self, ImplicitSelfKind::None)
}
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Debug, StableHash)]
pub enum IsAsync {
Async(Span),
NotAsync,
}
impl IsAsync {
pub fn is_async(self) -> bool {
matches!(self, IsAsync::Async(_))
}
}
#[derive(Copy, Clone, PartialEq, Eq, Debug, Encodable, Decodable, StableHash)]
#[derive(Default)]
pub enum Defaultness {
Default {
has_value: bool,
},
#[default]
Final,
}
impl Defaultness {
pub fn has_value(&self) -> bool {
match *self {
Defaultness::Default { has_value } => has_value,
Defaultness::Final => true,
}
}
pub fn is_final(&self) -> bool {
*self == Defaultness::Final
}
pub fn is_default(&self) -> bool {
matches!(*self, Defaultness::Default { .. })
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum FnRetTy<'hir> {
DefaultReturn(Span),
Return(&'hir Ty<'hir>),
}
impl<'hir> FnRetTy<'hir> {
#[inline]
pub fn span(&self) -> Span {
match *self {
Self::DefaultReturn(span) => span,
Self::Return(ref ty) => ty.span,
}
}
pub fn is_suggestable_infer_ty(&self) -> Option<&'hir Ty<'hir>> {
if let Self::Return(ty) = self
&& ty.is_suggestable_infer_ty()
{
return Some(*ty);
}
None
}
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum ClosureBinder {
Default,
For { span: Span },
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Mod<'hir> {
pub spans: ModSpans,
pub item_ids: &'hir [ItemId],
}
#[derive(Copy, Clone, Debug, StableHash)]
pub struct ModSpans {
pub inner_span: Span,
pub inject_use_span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct EnumDef<'hir> {
pub variants: &'hir [Variant<'hir>],
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Variant<'hir> {
pub ident: Ident,
#[stable_hash(ignore)]
pub hir_id: HirId,
pub def_id: LocalDefId,
pub data: VariantData<'hir>,
pub disr_expr: Option<&'hir AnonConst>,
pub span: Span,
}
#[derive(Copy, Clone, PartialEq, Debug, StableHash)]
pub enum UseKind {
Single(Ident),
Glob,
ListStem,
}
#[derive(Clone, Debug, Copy, StableHash)]
pub struct TraitRef<'hir> {
pub path: &'hir Path<'hir>,
#[stable_hash(ignore)]
pub hir_ref_id: HirId,
}
impl TraitRef<'_> {
pub fn trait_def_id(&self) -> Option<DefId> {
match self.path.res {
Res::Def(DefKind::Trait | DefKind::TraitAlias, did) => Some(did),
Res::Err => None,
res => panic!("{res:?} did not resolve to a trait or trait alias"),
}
}
}
#[derive(Clone, Debug, Copy, StableHash)]
pub struct PolyTraitRef<'hir> {
pub bound_generic_params: &'hir [GenericParam<'hir>],
pub modifiers: TraitBoundModifiers,
pub trait_ref: TraitRef<'hir>,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct FieldDef<'hir> {
pub span: Span,
pub vis_span: Span,
pub mut_restriction: &'hir MutRestriction<'hir>,
pub ident: Ident,
#[stable_hash(ignore)]
pub hir_id: HirId,
pub def_id: LocalDefId,
pub ty: &'hir Ty<'hir>,
pub safety: Safety,
pub default: Option<&'hir AnonConst>,
}
impl FieldDef<'_> {
pub fn is_positional(&self) -> bool {
self.ident.as_str().as_bytes()[0].is_ascii_digit()
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum VariantData<'hir> {
Struct { fields: &'hir [FieldDef<'hir>], recovered: ast::Recovered },
Tuple(&'hir [FieldDef<'hir>], #[stable_hash(ignore)] HirId, LocalDefId),
Unit(#[stable_hash(ignore)] HirId, LocalDefId),
}
impl<'hir> VariantData<'hir> {
pub fn fields(&self) -> &'hir [FieldDef<'hir>] {
match *self {
VariantData::Struct { fields, .. } | VariantData::Tuple(fields, ..) => fields,
_ => &[],
}
}
pub fn ctor(&self) -> Option<(CtorKind, HirId, LocalDefId)> {
match *self {
VariantData::Tuple(_, hir_id, def_id) => Some((CtorKind::Fn, hir_id, def_id)),
VariantData::Unit(hir_id, def_id) => Some((CtorKind::Const, hir_id, def_id)),
VariantData::Struct { .. } => None,
}
}
#[inline]
pub fn ctor_kind(&self) -> Option<CtorKind> {
self.ctor().map(|(kind, ..)| kind)
}
#[inline]
pub fn ctor_hir_id(&self) -> Option<HirId> {
self.ctor().map(|(_, hir_id, _)| hir_id)
}
#[inline]
pub fn ctor_def_id(&self) -> Option<LocalDefId> {
self.ctor().map(|(.., def_id)| def_id)
}
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Debug, Hash, StableHash)]
pub struct ItemId {
pub owner_id: OwnerId,
}
impl ItemId {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Item<'hir> {
pub owner_id: OwnerId,
pub kind: ItemKind<'hir>,
pub span: Span,
pub vis_span: Span,
pub eii: bool,
}
impl<'hir> Item<'hir> {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
#[inline]
pub fn item_id(&self) -> ItemId {
ItemId { owner_id: self.owner_id }
}
pub fn is_adt(&self) -> bool {
matches!(self.kind, ItemKind::Enum(..) | ItemKind::Struct(..) | ItemKind::Union(..))
}
pub fn is_struct_or_union(&self) -> bool {
matches!(self.kind, ItemKind::Struct(..) | ItemKind::Union(..))
}
expect_methods_self_kind! {
expect_extern_crate, (Option<Symbol>, Ident),
ItemKind::ExternCrate(s, ident), (*s, *ident);
expect_use, (&'hir UsePath<'hir>, UseKind), ItemKind::Use(p, uk), (p, *uk);
expect_static, (Mutability, Ident, &'hir Ty<'hir>, BodyId),
ItemKind::Static(mutbl, ident, ty, body), (*mutbl, *ident, ty, *body);
expect_const, (Ident, &'hir Generics<'hir>, &'hir Ty<'hir>, ConstItemRhs<'hir>),
ItemKind::Const(ident, generics, ty, rhs), (*ident, generics, ty, *rhs);
expect_fn, (Ident, &FnSig<'hir>, &'hir Generics<'hir>, BodyId),
ItemKind::Fn { ident, sig, generics, body, .. }, (*ident, sig, generics, *body);
expect_macro, (Ident, &ast::MacroDef, MacroKinds),
ItemKind::Macro(ident, def, mk), (*ident, def, *mk);
expect_mod, (Ident, &'hir Mod<'hir>), ItemKind::Mod(ident, m), (*ident, m);
expect_foreign_mod, (ExternAbi, &'hir [ForeignItemId]),
ItemKind::ForeignMod { abi, items }, (*abi, items);
expect_global_asm, &'hir InlineAsm<'hir>, ItemKind::GlobalAsm { asm, .. }, asm;
expect_ty_alias, (Ident, &'hir Generics<'hir>, &'hir Ty<'hir>),
ItemKind::TyAlias(ident, generics, ty), (*ident, generics, ty);
expect_enum, (Ident, &'hir Generics<'hir>, &EnumDef<'hir>),
ItemKind::Enum(ident, generics, def), (*ident, generics, def);
expect_struct, (Ident, &'hir Generics<'hir>, &VariantData<'hir>),
ItemKind::Struct(ident, generics, data), (*ident, generics, data);
expect_union, (Ident, &'hir Generics<'hir>, &VariantData<'hir>),
ItemKind::Union(ident, generics, data), (*ident, generics, data);
expect_trait,
(
&'hir ImplRestriction<'hir>,
Constness,
IsAuto,
Safety,
Ident,
&'hir Generics<'hir>,
GenericBounds<'hir>,
&'hir [TraitItemId]
),
ItemKind::Trait { impl_restriction, constness, is_auto, safety, ident, generics, bounds, items },
(impl_restriction, *constness, *is_auto, *safety, *ident, generics, bounds, items);
expect_trait_alias, (Constness, Ident, &'hir Generics<'hir>, GenericBounds<'hir>),
ItemKind::TraitAlias(constness, ident, generics, bounds), (*constness, *ident, generics, bounds);
expect_impl, &Impl<'hir>, ItemKind::Impl(imp), imp;
expect_test_binder_constraints, (&'hir Generics<'hir>, &'hir TestBinderBody<'hir>),
ItemKind::TestBinderConstraints { generics, body }, (generics, body);
}
}
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
#[derive(Encodable, Decodable, StableHash, Default)]
pub enum Safety {
#[default]
Unsafe,
Safe,
}
impl Safety {
pub fn prefix_str(self) -> &'static str {
match self {
Self::Unsafe => "unsafe ",
Self::Safe => "",
}
}
#[inline]
pub fn is_unsafe(self) -> bool {
!self.is_safe()
}
#[inline]
pub fn is_safe(self) -> bool {
match self {
Self::Unsafe => false,
Self::Safe => true,
}
}
}
impl fmt::Display for Safety {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(match *self {
Self::Unsafe => "unsafe",
Self::Safe => "safe",
})
}
}
#[derive(Copy, Clone, PartialEq, Eq, Debug, Encodable, Decodable, StableHash)]
pub enum Constness {
Const { always: bool },
NotConst,
}
impl Default for Constness {
fn default() -> Self {
Self::Const { always: false }
}
}
impl fmt::Display for Constness {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(match *self {
Self::Const { always: true } => "comptime",
Self::Const { always: false } => "const",
Self::NotConst => "non-const",
})
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct ImplRestriction<'hir> {
pub kind: RestrictionKind<'hir>,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct MutRestriction<'hir> {
pub kind: RestrictionKind<'hir>,
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum RestrictionKind<'hir> {
Unrestricted,
Restricted(&'hir Path<'hir, DefId>),
}
#[derive(Copy, Clone, Debug, StableHash, PartialEq, Eq)]
pub enum HeaderSafety {
SafeTargetFeatures,
Normal(Safety),
}
impl From<Safety> for HeaderSafety {
fn from(v: Safety) -> Self {
Self::Normal(v)
}
}
#[derive(Copy, Clone, Debug, StableHash)]
pub struct FnHeader {
pub safety: HeaderSafety,
pub constness: Constness,
pub asyncness: IsAsync,
pub abi: ExternAbi,
}
impl FnHeader {
pub fn is_async(&self) -> bool {
matches!(self.asyncness, IsAsync::Async(_))
}
pub fn is_unsafe(&self) -> bool {
self.safety().is_unsafe()
}
pub fn is_safe(&self) -> bool {
self.safety().is_safe()
}
pub fn safety(&self) -> Safety {
match self.safety {
HeaderSafety::SafeTargetFeatures => Safety::Unsafe,
HeaderSafety::Normal(safety) => safety,
}
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TestBinderBody<'hir> {
pub foralls: &'hir [TestBinderForall<'hir>],
pub exists: &'hir [TestBinderExists<'hir>],
pub constraints: TestBinderConstraint<'hir>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TestBinderForall<'hir> {
pub span: Span,
pub hir_id: HirId,
pub generics: &'hir Generics<'hir>,
pub body: &'hir TestBinderBody<'hir>,
pub assert_on_exit: Option<&'hir TestBinderConstraint<'hir>>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TestBinderExists<'hir> {
pub span: Span,
pub hir_id: HirId,
pub params: &'hir [GenericParam<'hir>],
pub body: &'hir TestBinderBody<'hir>,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum TestBinderConstraint<'hir> {
And { items: &'hir [TestBinderConstraint<'hir>] },
Or { items: &'hir [TestBinderConstraint<'hir>] },
Lifetime { lhs: &'hir Lifetime, rhs: &'hir Lifetime },
Type { lhs: &'hir Ty<'hir>, rhs: &'hir Lifetime },
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum ItemKind<'hir> {
ExternCrate(Option<Symbol>, Ident),
Use(&'hir UsePath<'hir>, UseKind),
Static(Mutability, Ident, &'hir Ty<'hir>, BodyId),
Const(Ident, &'hir Generics<'hir>, &'hir Ty<'hir>, ConstItemRhs<'hir>),
Fn {
sig: FnSig<'hir>,
ident: Ident,
generics: &'hir Generics<'hir>,
body: BodyId,
has_body: bool,
},
Macro(Ident, &'hir ast::MacroDef, MacroKinds),
Mod(Ident, &'hir Mod<'hir>),
ForeignMod {
abi: ExternAbi,
items: &'hir [ForeignItemId],
},
GlobalAsm {
asm: &'hir InlineAsm<'hir>,
fake_body: BodyId,
},
TyAlias(Ident, &'hir Generics<'hir>, &'hir Ty<'hir>),
Enum(Ident, &'hir Generics<'hir>, EnumDef<'hir>),
Struct(Ident, &'hir Generics<'hir>, VariantData<'hir>),
Union(Ident, &'hir Generics<'hir>, VariantData<'hir>),
Trait {
impl_restriction: &'hir ImplRestriction<'hir>,
constness: Constness,
is_auto: IsAuto,
safety: Safety,
ident: Ident,
generics: &'hir Generics<'hir>,
bounds: GenericBounds<'hir>,
items: &'hir [TraitItemId],
},
TraitAlias(Constness, Ident, &'hir Generics<'hir>, GenericBounds<'hir>),
Impl(Impl<'hir>),
TestBinderConstraints {
generics: &'hir Generics<'hir>,
body: &'hir TestBinderBody<'hir>,
},
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct Impl<'hir> {
pub generics: &'hir Generics<'hir>,
pub of_trait: Option<&'hir TraitImplHeader<'hir>>,
pub self_ty: &'hir Ty<'hir>,
pub items: &'hir [ImplItemId],
pub constness: Constness,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TraitImplHeader<'hir> {
pub safety: Safety,
pub polarity: ImplPolarity,
pub defaultness: Defaultness,
pub defaultness_span: Option<Span>,
pub trait_ref: TraitRef<'hir>,
}
impl ItemKind<'_> {
pub fn ident(&self) -> Option<Ident> {
match *self {
ItemKind::ExternCrate(_, ident)
| ItemKind::Use(_, UseKind::Single(ident))
| ItemKind::Static(_, ident, ..)
| ItemKind::Const(ident, ..)
| ItemKind::Fn { ident, .. }
| ItemKind::Macro(ident, ..)
| ItemKind::Mod(ident, ..)
| ItemKind::TyAlias(ident, ..)
| ItemKind::Enum(ident, ..)
| ItemKind::Struct(ident, ..)
| ItemKind::Union(ident, ..)
| ItemKind::Trait { ident, .. }
| ItemKind::TraitAlias(_, ident, ..) => Some(ident),
ItemKind::Use(_, UseKind::Glob | UseKind::ListStem)
| ItemKind::ForeignMod { .. }
| ItemKind::GlobalAsm { .. }
| ItemKind::Impl(_)
| ItemKind::TestBinderConstraints { .. } => None,
}
}
pub fn generics(&self) -> Option<&Generics<'_>> {
Some(match self {
ItemKind::Fn { generics, .. }
| ItemKind::TyAlias(_, generics, _)
| ItemKind::Const(_, generics, _, _)
| ItemKind::Enum(_, generics, _)
| ItemKind::Struct(_, generics, _)
| ItemKind::Union(_, generics, _)
| ItemKind::Trait { generics, .. }
| ItemKind::TraitAlias(_, _, generics, _)
| ItemKind::Impl(Impl { generics, .. })
| ItemKind::TestBinderConstraints { generics, .. } => generics,
_ => return None,
})
}
pub fn recovered(&self) -> bool {
match self {
ItemKind::Struct(
_,
_,
VariantData::Struct { recovered: ast::Recovered::Yes(_), .. },
) => true,
ItemKind::Union(
_,
_,
VariantData::Struct { recovered: ast::Recovered::Yes(_), .. },
) => true,
ItemKind::Enum(_, _, def) => def.variants.iter().any(|v| match v.data {
VariantData::Struct { recovered: ast::Recovered::Yes(_), .. } => true,
_ => false,
}),
_ => false,
}
}
}
#[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Debug, StableHash)]
pub struct ForeignItemId {
pub owner_id: OwnerId,
}
impl ForeignItemId {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct ForeignItem<'hir> {
pub ident: Ident,
pub kind: ForeignItemKind<'hir>,
pub owner_id: OwnerId,
pub span: Span,
pub vis_span: Span,
}
impl ForeignItem<'_> {
#[inline]
pub fn hir_id(&self) -> HirId {
HirId::make_owner(self.owner_id.def_id)
}
pub fn foreign_item_id(&self) -> ForeignItemId {
ForeignItemId { owner_id: self.owner_id }
}
}
#[derive(Debug, Clone, Copy, StableHash)]
pub enum ForeignItemKind<'hir> {
Fn(FnSig<'hir>, &'hir [Option<Ident>], &'hir Generics<'hir>),
Static(&'hir Ty<'hir>, Mutability, Safety),
Type,
}
#[derive(Debug, Copy, Clone, StableHash)]
pub struct Upvar {
pub span: Span,
}
#[derive(Debug, Clone, Copy, StableHash)]
pub struct TraitCandidate<'hir> {
pub def_id: DefId,
pub import_ids: &'hir [LocalDefId],
pub lint_ambiguous: bool,
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum OwnerNode<'hir> {
Item(&'hir Item<'hir>),
ForeignItem(&'hir ForeignItem<'hir>),
TraitItem(&'hir TraitItem<'hir>),
ImplItem(&'hir ImplItem<'hir>),
Crate(&'hir Mod<'hir>),
Synthetic,
}
impl<'hir> OwnerNode<'hir> {
pub fn span(&self) -> Span {
match self {
OwnerNode::Item(Item { span, .. })
| OwnerNode::ForeignItem(ForeignItem { span, .. })
| OwnerNode::ImplItem(ImplItem { span, .. })
| OwnerNode::TraitItem(TraitItem { span, .. }) => *span,
OwnerNode::Crate(Mod { spans: ModSpans { inner_span, .. }, .. }) => *inner_span,
OwnerNode::Synthetic => unreachable!(),
}
}
pub fn fn_sig(self) -> Option<&'hir FnSig<'hir>> {
match self {
OwnerNode::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
| OwnerNode::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
| OwnerNode::Item(Item { kind: ItemKind::Fn { sig: fn_sig, .. }, .. })
| OwnerNode::ForeignItem(ForeignItem {
kind: ForeignItemKind::Fn(fn_sig, _, _), ..
}) => Some(fn_sig),
_ => None,
}
}
pub fn fn_decl(self) -> Option<&'hir FnDecl<'hir>> {
match self {
OwnerNode::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
| OwnerNode::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
| OwnerNode::Item(Item { kind: ItemKind::Fn { sig: fn_sig, .. }, .. })
| OwnerNode::ForeignItem(ForeignItem {
kind: ForeignItemKind::Fn(fn_sig, _, _), ..
}) => Some(fn_sig.decl),
_ => None,
}
}
pub fn body_id(&self) -> Option<BodyId> {
match self {
OwnerNode::Item(Item {
kind:
ItemKind::Static(_, _, _, body)
| ItemKind::Const(.., ConstItemRhs::Body(body))
| ItemKind::Fn { body, .. },
..
})
| OwnerNode::TraitItem(TraitItem {
kind:
TraitItemKind::Fn(_, TraitFn::Provided(body))
| TraitItemKind::Const(_, Some(ConstItemRhs::Body(body))),
..
})
| OwnerNode::ImplItem(ImplItem {
kind: ImplItemKind::Fn(_, body) | ImplItemKind::Const(_, ConstItemRhs::Body(body)),
..
}) => Some(*body),
_ => None,
}
}
pub fn generics(self) -> Option<&'hir Generics<'hir>> {
Node::generics(self.into())
}
pub fn def_id(self) -> OwnerId {
match self {
OwnerNode::Item(Item { owner_id, .. })
| OwnerNode::TraitItem(TraitItem { owner_id, .. })
| OwnerNode::ImplItem(ImplItem { owner_id, .. })
| OwnerNode::ForeignItem(ForeignItem { owner_id, .. }) => *owner_id,
OwnerNode::Crate(..) => crate::rustc_hir_id::CRATE_HIR_ID.owner,
OwnerNode::Synthetic => unreachable!(),
}
}
pub fn is_impl_block(&self) -> bool {
matches!(self, OwnerNode::Item(Item { kind: ItemKind::Impl(_), .. }))
}
expect_methods_self! {
expect_item, &'hir Item<'hir>, OwnerNode::Item(n), n;
expect_foreign_item, &'hir ForeignItem<'hir>, OwnerNode::ForeignItem(n), n;
expect_impl_item, &'hir ImplItem<'hir>, OwnerNode::ImplItem(n), n;
expect_trait_item, &'hir TraitItem<'hir>, OwnerNode::TraitItem(n), n;
}
}
impl<'hir> From<&'hir Item<'hir>> for OwnerNode<'hir> {
fn from(val: &'hir Item<'hir>) -> Self {
OwnerNode::Item(val)
}
}
impl<'hir> From<&'hir ForeignItem<'hir>> for OwnerNode<'hir> {
fn from(val: &'hir ForeignItem<'hir>) -> Self {
OwnerNode::ForeignItem(val)
}
}
impl<'hir> From<&'hir ImplItem<'hir>> for OwnerNode<'hir> {
fn from(val: &'hir ImplItem<'hir>) -> Self {
OwnerNode::ImplItem(val)
}
}
impl<'hir> From<&'hir TraitItem<'hir>> for OwnerNode<'hir> {
fn from(val: &'hir TraitItem<'hir>) -> Self {
OwnerNode::TraitItem(val)
}
}
impl<'hir> From<OwnerNode<'hir>> for Node<'hir> {
fn from(val: OwnerNode<'hir>) -> Self {
match val {
OwnerNode::Item(n) => Node::Item(n),
OwnerNode::ForeignItem(n) => Node::ForeignItem(n),
OwnerNode::ImplItem(n) => Node::ImplItem(n),
OwnerNode::TraitItem(n) => Node::TraitItem(n),
OwnerNode::Crate(n) => Node::Crate(n),
OwnerNode::Synthetic => Node::Synthetic,
}
}
}
#[derive(Copy, Clone, Debug, StableHash)]
pub enum Node<'hir> {
Param(&'hir Param<'hir>),
Item(&'hir Item<'hir>),
ForeignItem(&'hir ForeignItem<'hir>),
TraitItem(&'hir TraitItem<'hir>),
ImplItem(&'hir ImplItem<'hir>),
Variant(&'hir Variant<'hir>),
Field(&'hir FieldDef<'hir>),
AnonConst(&'hir AnonConst),
ConstBlock(&'hir ConstBlock),
ConstArg(&'hir ConstArg<'hir>),
Expr(&'hir Expr<'hir>),
ExprField(&'hir ExprField<'hir>),
ConstArgExprField(&'hir ConstArgExprField<'hir>),
Stmt(&'hir Stmt<'hir>),
PathSegment(&'hir PathSegment<'hir>),
Ty(&'hir Ty<'hir>),
AssocItemConstraint(&'hir AssocItemConstraint<'hir>),
TraitRef(&'hir TraitRef<'hir>),
OpaqueTy(&'hir OpaqueTy<'hir>),
TyPat(&'hir TyPat<'hir>),
Pat(&'hir Pat<'hir>),
PatField(&'hir PatField<'hir>),
PatExpr(&'hir PatExpr<'hir>),
Arm(&'hir Arm<'hir>),
Block(&'hir Block<'hir>),
LetStmt(&'hir LetStmt<'hir>),
Ctor(&'hir VariantData<'hir>),
Lifetime(&'hir Lifetime),
GenericParam(&'hir GenericParam<'hir>),
Crate(&'hir Mod<'hir>),
Infer(&'hir InferArg),
WherePredicate(&'hir WherePredicate<'hir>),
PreciseCapturingNonLifetimeArg(&'hir PreciseCapturingNonLifetimeArg),
TestBinderForall(&'hir TestBinderForall<'hir>),
TestBinderExists(&'hir TestBinderExists<'hir>),
Synthetic,
Err(Span),
}
impl<'hir> Node<'hir> {
pub fn ident(&self) -> Option<Ident> {
match self {
Node::Item(item) => item.kind.ident(),
Node::TraitItem(TraitItem { ident, .. })
| Node::ImplItem(ImplItem { ident, .. })
| Node::ForeignItem(ForeignItem { ident, .. })
| Node::Field(FieldDef { ident, .. })
| Node::Variant(Variant { ident, .. })
| Node::PathSegment(PathSegment { ident, .. }) => Some(*ident),
Node::Lifetime(lt) => Some(lt.ident),
Node::GenericParam(p) => Some(p.name.ident()),
Node::AssocItemConstraint(c) => Some(c.ident),
Node::PatField(f) => Some(f.ident),
Node::ExprField(f) => Some(f.ident),
Node::ConstArgExprField(f) => Some(f.field),
Node::PreciseCapturingNonLifetimeArg(a) => Some(a.ident),
Node::Param(..)
| Node::AnonConst(..)
| Node::ConstBlock(..)
| Node::ConstArg(..)
| Node::Expr(..)
| Node::Stmt(..)
| Node::Block(..)
| Node::Ctor(..)
| Node::Pat(..)
| Node::TyPat(..)
| Node::PatExpr(..)
| Node::Arm(..)
| Node::LetStmt(..)
| Node::Crate(..)
| Node::Ty(..)
| Node::TraitRef(..)
| Node::OpaqueTy(..)
| Node::Infer(..)
| Node::WherePredicate(..)
| Node::TestBinderForall(..)
| Node::TestBinderExists(..)
| Node::Synthetic
| Node::Err(..) => None,
}
}
pub fn fn_decl(self) -> Option<&'hir FnDecl<'hir>> {
match self {
Node::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
| Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
| Node::Item(Item { kind: ItemKind::Fn { sig: fn_sig, .. }, .. })
| Node::ForeignItem(ForeignItem { kind: ForeignItemKind::Fn(fn_sig, _, _), .. }) => {
Some(fn_sig.decl)
}
Node::Expr(Expr { kind: ExprKind::Closure(Closure { fn_decl, .. }), .. }) => {
Some(fn_decl)
}
_ => None,
}
}
pub fn impl_block_of_trait(self, trait_def_id: DefId) -> Option<&'hir Impl<'hir>> {
if let Node::Item(Item { kind: ItemKind::Impl(impl_block), .. }) = self
&& let Some(of_trait) = impl_block.of_trait
&& let Some(trait_id) = of_trait.trait_ref.trait_def_id()
&& trait_id == trait_def_id
{
Some(impl_block)
} else {
None
}
}
pub fn fn_sig(self) -> Option<&'hir FnSig<'hir>> {
match self {
Node::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
| Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
| Node::Item(Item { kind: ItemKind::Fn { sig: fn_sig, .. }, .. })
| Node::ForeignItem(ForeignItem { kind: ForeignItemKind::Fn(fn_sig, _, _), .. }) => {
Some(fn_sig)
}
_ => None,
}
}
pub fn ty(self) -> Option<&'hir Ty<'hir>> {
match self {
Node::Item(it) => match it.kind {
ItemKind::TyAlias(_, _, ty)
| ItemKind::Static(_, _, ty, _)
| ItemKind::Const(_, _, ty, _) => Some(ty),
ItemKind::Impl(impl_item) => Some(&impl_item.self_ty),
_ => None,
},
Node::TraitItem(it) => match it.kind {
TraitItemKind::Const(ty, _) => Some(ty),
TraitItemKind::Type(_, ty) => ty,
_ => None,
},
Node::ImplItem(it) => match it.kind {
ImplItemKind::Const(ty, _) => Some(ty),
ImplItemKind::Type(ty) => Some(ty),
_ => None,
},
Node::ForeignItem(it) => match it.kind {
ForeignItemKind::Static(ty, ..) => Some(ty),
_ => None,
},
Node::GenericParam(param) => match param.kind {
GenericParamKind::Lifetime { .. } => None,
GenericParamKind::Type { default, .. } => default,
GenericParamKind::Const { ty, .. } => Some(ty),
},
Node::Field(f) => Some(f.ty),
_ => None,
}
}
pub fn alias_ty(self) -> Option<&'hir Ty<'hir>> {
match self {
Node::Item(Item { kind: ItemKind::TyAlias(_, _, ty), .. }) => Some(ty),
_ => None,
}
}
#[inline]
pub fn associated_body(&self) -> Option<(LocalDefId, BodyId)> {
match self {
Node::Item(Item {
owner_id,
kind:
ItemKind::Const(.., ConstItemRhs::Body(body))
| ItemKind::Static(.., body)
| ItemKind::Fn { body, .. },
..
})
| Node::TraitItem(TraitItem {
owner_id,
kind:
TraitItemKind::Const(_, Some(ConstItemRhs::Body(body)))
| TraitItemKind::Fn(_, TraitFn::Provided(body)),
..
})
| Node::ImplItem(ImplItem {
owner_id,
kind: ImplItemKind::Const(.., ConstItemRhs::Body(body)) | ImplItemKind::Fn(_, body),
..
}) => Some((owner_id.def_id, *body)),
Node::Item(Item {
owner_id, kind: ItemKind::GlobalAsm { asm: _, fake_body }, ..
}) => Some((owner_id.def_id, *fake_body)),
Node::Expr(Expr { kind: ExprKind::Closure(Closure { def_id, body, .. }), .. }) => {
Some((*def_id, *body))
}
Node::AnonConst(constant) => Some((constant.def_id, constant.body)),
Node::ConstBlock(constant) => Some((constant.def_id, constant.body)),
_ => None,
}
}
pub fn body_id(&self) -> Option<BodyId> {
Some(self.associated_body()?.1)
}
pub fn generics(self) -> Option<&'hir Generics<'hir>> {
match self {
Node::ForeignItem(ForeignItem {
kind: ForeignItemKind::Fn(_, _, generics), ..
})
| Node::TraitItem(TraitItem { generics, .. })
| Node::ImplItem(ImplItem { generics, .. }) => Some(generics),
Node::Item(item) => item.kind.generics(),
_ => None,
}
}
pub fn as_owner(self) -> Option<OwnerNode<'hir>> {
match self {
Node::Item(i) => Some(OwnerNode::Item(i)),
Node::ForeignItem(i) => Some(OwnerNode::ForeignItem(i)),
Node::TraitItem(i) => Some(OwnerNode::TraitItem(i)),
Node::ImplItem(i) => Some(OwnerNode::ImplItem(i)),
Node::Crate(i) => Some(OwnerNode::Crate(i)),
Node::Synthetic => Some(OwnerNode::Synthetic),
_ => None,
}
}
pub fn fn_kind(self) -> Option<FnKind<'hir>> {
match self {
Node::Item(i) => match i.kind {
ItemKind::Fn { ident, sig, generics, .. } => {
Some(FnKind::ItemFn(ident, generics, sig.header))
}
_ => None,
},
Node::TraitItem(ti) => match ti.kind {
TraitItemKind::Fn(ref sig, _) => Some(FnKind::Method(ti.ident, sig)),
_ => None,
},
Node::ImplItem(ii) => match ii.kind {
ImplItemKind::Fn(ref sig, _) => Some(FnKind::Method(ii.ident, sig)),
_ => None,
},
Node::Expr(e) => match e.kind {
ExprKind::Closure { .. } => Some(FnKind::Closure),
_ => None,
},
_ => None,
}
}
pub fn path(self) -> Option<&'hir QPath<'hir>> {
match self {
Node::Ty(Ty { kind: TyKind::Path(path), .. })
| Node::Expr(Expr { kind: ExprKind::Path(path), .. })
| Node::PatExpr(PatExpr { kind: PatExprKind::Path(path), .. }) => Some(path),
_ => None,
}
}
expect_methods_self! {
expect_param, &'hir Param<'hir>, Node::Param(n), n;
expect_item, &'hir Item<'hir>, Node::Item(n), n;
expect_foreign_item, &'hir ForeignItem<'hir>, Node::ForeignItem(n), n;
expect_trait_item, &'hir TraitItem<'hir>, Node::TraitItem(n), n;
expect_impl_item, &'hir ImplItem<'hir>, Node::ImplItem(n), n;
expect_variant, &'hir Variant<'hir>, Node::Variant(n), n;
expect_field, &'hir FieldDef<'hir>, Node::Field(n), n;
expect_anon_const, &'hir AnonConst, Node::AnonConst(n), n;
expect_inline_const, &'hir ConstBlock, Node::ConstBlock(n), n;
expect_expr, &'hir Expr<'hir>, Node::Expr(n), n;
expect_expr_field, &'hir ExprField<'hir>, Node::ExprField(n), n;
expect_stmt, &'hir Stmt<'hir>, Node::Stmt(n), n;
expect_path_segment, &'hir PathSegment<'hir>, Node::PathSegment(n), n;
expect_ty, &'hir Ty<'hir>, Node::Ty(n), n;
expect_assoc_item_constraint, &'hir AssocItemConstraint<'hir>, Node::AssocItemConstraint(n), n;
expect_trait_ref, &'hir TraitRef<'hir>, Node::TraitRef(n), n;
expect_opaque_ty, &'hir OpaqueTy<'hir>, Node::OpaqueTy(n), n;
expect_pat, &'hir Pat<'hir>, Node::Pat(n), n;
expect_pat_field, &'hir PatField<'hir>, Node::PatField(n), n;
expect_arm, &'hir Arm<'hir>, Node::Arm(n), n;
expect_block, &'hir Block<'hir>, Node::Block(n), n;
expect_let_stmt, &'hir LetStmt<'hir>, Node::LetStmt(n), n;
expect_ctor, &'hir VariantData<'hir>, Node::Ctor(n), n;
expect_lifetime, &'hir Lifetime, Node::Lifetime(n), n;
expect_generic_param, &'hir GenericParam<'hir>, Node::GenericParam(n), n;
expect_crate, &'hir Mod<'hir>, Node::Crate(n), n;
expect_infer, &'hir InferArg, Node::Infer(n), n;
expect_closure, &'hir Closure<'hir>, Node::Expr(Expr { kind: ExprKind::Closure(n), .. }), n;
}
}
#[cfg(target_pointer_width = "64")]
mod size_asserts {
use crate::static_assert_size;
use super::*;
static_assert_size!(Block<'_>, 48);
static_assert_size!(Body<'_>, 24);
static_assert_size!(Expr<'_>, 64);
static_assert_size!(ExprKind<'_>, 48);
static_assert_size!(FnDecl<'_>, 40);
static_assert_size!(ForeignItem<'_>, 88);
static_assert_size!(ForeignItemKind<'_>, 56);
static_assert_size!(GenericArg<'_>, 16);
static_assert_size!(GenericBound<'_>, 64);
static_assert_size!(Generics<'_>, 56);
static_assert_size!(Impl<'_>, 48);
static_assert_size!(ImplItem<'_>, 88);
static_assert_size!(ImplItemKind<'_>, 40);
static_assert_size!(Item<'_>, 88);
static_assert_size!(ItemKind<'_>, 64);
static_assert_size!(LetStmt<'_>, 64);
static_assert_size!(Param<'_>, 32);
static_assert_size!(Pat<'_>, 80);
static_assert_size!(PatKind<'_>, 56);
static_assert_size!(Path<'_>, 40);
static_assert_size!(PathSegment<'_>, 48);
static_assert_size!(QPath<'_>, 24);
static_assert_size!(Res, 12);
static_assert_size!(Stmt<'_>, 32);
static_assert_size!(StmtKind<'_>, 16);
static_assert_size!(TraitImplHeader<'_>, 48);
static_assert_size!(TraitItem<'_>, 88);
static_assert_size!(TraitItemKind<'_>, 48);
static_assert_size!(Ty<'_>, 48);
static_assert_size!(TyKind<'_>, 32);
}
#[cfg(test)]
mod tests;