]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_hir/src/hir.rs
Rollup merge of #79554 - b-naber:generic-associated-types-in-trait-paths, r=jackh726
[rust.git] / compiler / rustc_hir / src / hir.rs
1 use crate::def::{DefKind, Namespace, Res};
2 use crate::def_id::DefId;
3 crate use crate::hir_id::HirId;
4 use crate::{itemlikevisit, LangItem};
5
6 use rustc_ast::util::parser::ExprPrecedence;
7 use rustc_ast::{self as ast, CrateSugar, LlvmAsmDialect};
8 use rustc_ast::{AttrVec, Attribute, FloatTy, IntTy, Label, LitKind, StrStyle, UintTy};
9 pub use rustc_ast::{BorrowKind, ImplPolarity, IsAuto};
10 pub use rustc_ast::{CaptureBy, Movability, Mutability};
11 use rustc_ast::{InlineAsmOptions, InlineAsmTemplatePiece};
12 use rustc_data_structures::sync::{par_for_each_in, Send, Sync};
13 use rustc_macros::HashStable_Generic;
14 use rustc_span::source_map::{SourceMap, Spanned};
15 use rustc_span::symbol::{kw, sym, Ident, Symbol};
16 use rustc_span::{def_id::LocalDefId, BytePos};
17 use rustc_span::{MultiSpan, Span, DUMMY_SP};
18 use rustc_target::asm::InlineAsmRegOrRegClass;
19 use rustc_target::spec::abi::Abi;
20
21 use smallvec::SmallVec;
22 use std::collections::{BTreeMap, BTreeSet};
23 use std::fmt;
24
25 #[derive(Copy, Clone, Encodable, HashStable_Generic)]
26 pub struct Lifetime {
27     pub hir_id: HirId,
28     pub span: Span,
29
30     /// Either "`'a`", referring to a named lifetime definition,
31     /// or "``" (i.e., `kw::Empty`), for elision placeholders.
32     ///
33     /// HIR lowering inserts these placeholders in type paths that
34     /// refer to type definitions needing lifetime parameters,
35     /// `&T` and `&mut T`, and trait objects without `... + 'a`.
36     pub name: LifetimeName,
37 }
38
39 #[derive(Debug, Clone, PartialEq, Eq, Encodable, Hash, Copy)]
40 #[derive(HashStable_Generic)]
41 pub enum ParamName {
42     /// Some user-given name like `T` or `'x`.
43     Plain(Ident),
44
45     /// Synthetic name generated when user elided a lifetime in an impl header.
46     ///
47     /// E.g., the lifetimes in cases like these:
48     ///
49     ///     impl Foo for &u32
50     ///     impl Foo<'_> for u32
51     ///
52     /// in that case, we rewrite to
53     ///
54     ///     impl<'f> Foo for &'f u32
55     ///     impl<'f> Foo<'f> for u32
56     ///
57     /// where `'f` is something like `Fresh(0)`. The indices are
58     /// unique per impl, but not necessarily continuous.
59     Fresh(usize),
60
61     /// Indicates an illegal name was given and an error has been
62     /// reported (so we should squelch other derived errors). Occurs
63     /// when, e.g., `'_` is used in the wrong place.
64     Error,
65 }
66
67 impl ParamName {
68     pub fn ident(&self) -> Ident {
69         match *self {
70             ParamName::Plain(ident) => ident,
71             ParamName::Fresh(_) | ParamName::Error => {
72                 Ident::with_dummy_span(kw::UnderscoreLifetime)
73             }
74         }
75     }
76
77     pub fn normalize_to_macros_2_0(&self) -> ParamName {
78         match *self {
79             ParamName::Plain(ident) => ParamName::Plain(ident.normalize_to_macros_2_0()),
80             param_name => param_name,
81         }
82     }
83 }
84
85 #[derive(Debug, Clone, PartialEq, Eq, Encodable, Hash, Copy)]
86 #[derive(HashStable_Generic)]
87 pub enum LifetimeName {
88     /// User-given names or fresh (synthetic) names.
89     Param(ParamName),
90
91     /// User wrote nothing (e.g., the lifetime in `&u32`).
92     Implicit,
93
94     /// Implicit lifetime in a context like `dyn Foo`. This is
95     /// distinguished from implicit lifetimes elsewhere because the
96     /// lifetime that they default to must appear elsewhere within the
97     /// enclosing type.  This means that, in an `impl Trait` context, we
98     /// don't have to create a parameter for them. That is, `impl
99     /// Trait<Item = &u32>` expands to an opaque type like `type
100     /// Foo<'a> = impl Trait<Item = &'a u32>`, but `impl Trait<item =
101     /// dyn Bar>` expands to `type Foo = impl Trait<Item = dyn Bar +
102     /// 'static>`. The latter uses `ImplicitObjectLifetimeDefault` so
103     /// that surrounding code knows not to create a lifetime
104     /// parameter.
105     ImplicitObjectLifetimeDefault,
106
107     /// Indicates an error during lowering (usually `'_` in wrong place)
108     /// that was already reported.
109     Error,
110
111     /// User wrote specifies `'_`.
112     Underscore,
113
114     /// User wrote `'static`.
115     Static,
116 }
117
118 impl LifetimeName {
119     pub fn ident(&self) -> Ident {
120         match *self {
121             LifetimeName::ImplicitObjectLifetimeDefault
122             | LifetimeName::Implicit
123             | LifetimeName::Error => Ident::invalid(),
124             LifetimeName::Underscore => Ident::with_dummy_span(kw::UnderscoreLifetime),
125             LifetimeName::Static => Ident::with_dummy_span(kw::StaticLifetime),
126             LifetimeName::Param(param_name) => param_name.ident(),
127         }
128     }
129
130     pub fn is_elided(&self) -> bool {
131         match self {
132             LifetimeName::ImplicitObjectLifetimeDefault
133             | LifetimeName::Implicit
134             | LifetimeName::Underscore => true,
135
136             // It might seem surprising that `Fresh(_)` counts as
137             // *not* elided -- but this is because, as far as the code
138             // in the compiler is concerned -- `Fresh(_)` variants act
139             // equivalently to "some fresh name". They correspond to
140             // early-bound regions on an impl, in other words.
141             LifetimeName::Error | LifetimeName::Param(_) | LifetimeName::Static => false,
142         }
143     }
144
145     fn is_static(&self) -> bool {
146         self == &LifetimeName::Static
147     }
148
149     pub fn normalize_to_macros_2_0(&self) -> LifetimeName {
150         match *self {
151             LifetimeName::Param(param_name) => {
152                 LifetimeName::Param(param_name.normalize_to_macros_2_0())
153             }
154             lifetime_name => lifetime_name,
155         }
156     }
157 }
158
159 impl fmt::Display for Lifetime {
160     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
161         self.name.ident().fmt(f)
162     }
163 }
164
165 impl fmt::Debug for Lifetime {
166     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
167         write!(f, "lifetime({}: {})", self.hir_id, self.name.ident())
168     }
169 }
170
171 impl Lifetime {
172     pub fn is_elided(&self) -> bool {
173         self.name.is_elided()
174     }
175
176     pub fn is_static(&self) -> bool {
177         self.name.is_static()
178     }
179 }
180
181 /// A `Path` is essentially Rust's notion of a name; for instance,
182 /// `std::cmp::PartialEq`. It's represented as a sequence of identifiers,
183 /// along with a bunch of supporting information.
184 #[derive(Debug, HashStable_Generic)]
185 pub struct Path<'hir> {
186     pub span: Span,
187     /// The resolution for the path.
188     pub res: Res,
189     /// The segments in the path: the things separated by `::`.
190     pub segments: &'hir [PathSegment<'hir>],
191 }
192
193 impl Path<'_> {
194     pub fn is_global(&self) -> bool {
195         !self.segments.is_empty() && self.segments[0].ident.name == kw::PathRoot
196     }
197 }
198
199 /// A segment of a path: an identifier, an optional lifetime, and a set of
200 /// types.
201 #[derive(Debug, HashStable_Generic)]
202 pub struct PathSegment<'hir> {
203     /// The identifier portion of this path segment.
204     #[stable_hasher(project(name))]
205     pub ident: Ident,
206     // `id` and `res` are optional. We currently only use these in save-analysis,
207     // any path segments without these will not have save-analysis info and
208     // therefore will not have 'jump to def' in IDEs, but otherwise will not be
209     // affected. (In general, we don't bother to get the defs for synthesized
210     // segments, only for segments which have come from the AST).
211     pub hir_id: Option<HirId>,
212     pub res: Option<Res>,
213
214     /// Type/lifetime parameters attached to this path. They come in
215     /// two flavors: `Path<A,B,C>` and `Path(A,B) -> C`. Note that
216     /// this is more than just simple syntactic sugar; the use of
217     /// parens affects the region binding rules, so we preserve the
218     /// distinction.
219     pub args: Option<&'hir GenericArgs<'hir>>,
220
221     /// Whether to infer remaining type parameters, if any.
222     /// This only applies to expression and pattern paths, and
223     /// out of those only the segments with no type parameters
224     /// to begin with, e.g., `Vec::new` is `<Vec<..>>::new::<..>`.
225     pub infer_args: bool,
226 }
227
228 impl<'hir> PathSegment<'hir> {
229     /// Converts an identifier to the corresponding segment.
230     pub fn from_ident(ident: Ident) -> PathSegment<'hir> {
231         PathSegment { ident, hir_id: None, res: None, infer_args: true, args: None }
232     }
233
234     pub fn invalid() -> Self {
235         Self::from_ident(Ident::invalid())
236     }
237
238     pub fn args(&self) -> &GenericArgs<'hir> {
239         if let Some(ref args) = self.args {
240             args
241         } else {
242             const DUMMY: &GenericArgs<'_> = &GenericArgs::none();
243             DUMMY
244         }
245     }
246 }
247
248 #[derive(Encodable, Debug, HashStable_Generic)]
249 pub struct ConstArg {
250     pub value: AnonConst,
251     pub span: Span,
252 }
253
254 #[derive(Debug, HashStable_Generic)]
255 pub enum GenericArg<'hir> {
256     Lifetime(Lifetime),
257     Type(Ty<'hir>),
258     Const(ConstArg),
259 }
260
261 impl GenericArg<'_> {
262     pub fn span(&self) -> Span {
263         match self {
264             GenericArg::Lifetime(l) => l.span,
265             GenericArg::Type(t) => t.span,
266             GenericArg::Const(c) => c.span,
267         }
268     }
269
270     pub fn id(&self) -> HirId {
271         match self {
272             GenericArg::Lifetime(l) => l.hir_id,
273             GenericArg::Type(t) => t.hir_id,
274             GenericArg::Const(c) => c.value.hir_id,
275         }
276     }
277
278     pub fn is_const(&self) -> bool {
279         matches!(self, GenericArg::Const(_))
280     }
281
282     pub fn is_synthetic(&self) -> bool {
283         matches!(self, GenericArg::Lifetime(lifetime) if lifetime.name.ident() == Ident::invalid())
284     }
285
286     pub fn descr(&self) -> &'static str {
287         match self {
288             GenericArg::Lifetime(_) => "lifetime",
289             GenericArg::Type(_) => "type",
290             GenericArg::Const(_) => "constant",
291         }
292     }
293
294     pub fn to_ord(&self, feats: &rustc_feature::Features) -> ast::ParamKindOrd {
295         match self {
296             GenericArg::Lifetime(_) => ast::ParamKindOrd::Lifetime,
297             GenericArg::Type(_) => ast::ParamKindOrd::Type,
298             GenericArg::Const(_) => ast::ParamKindOrd::Const { unordered: feats.const_generics },
299         }
300     }
301 }
302
303 #[derive(Debug, HashStable_Generic)]
304 pub struct GenericArgs<'hir> {
305     /// The generic arguments for this path segment.
306     pub args: &'hir [GenericArg<'hir>],
307     /// Bindings (equality constraints) on associated types, if present.
308     /// E.g., `Foo<A = Bar>`.
309     pub bindings: &'hir [TypeBinding<'hir>],
310     /// Were arguments written in parenthesized form `Fn(T) -> U`?
311     /// This is required mostly for pretty-printing and diagnostics,
312     /// but also for changing lifetime elision rules to be "function-like".
313     pub parenthesized: bool,
314 }
315
316 impl GenericArgs<'_> {
317     pub const fn none() -> Self {
318         Self { args: &[], bindings: &[], parenthesized: false }
319     }
320
321     pub fn inputs(&self) -> &[Ty<'_>] {
322         if self.parenthesized {
323             for arg in self.args {
324                 match arg {
325                     GenericArg::Lifetime(_) => {}
326                     GenericArg::Type(ref ty) => {
327                         if let TyKind::Tup(ref tys) = ty.kind {
328                             return tys;
329                         }
330                         break;
331                     }
332                     GenericArg::Const(_) => {}
333                 }
334             }
335         }
336         panic!("GenericArgs::inputs: not a `Fn(T) -> U`");
337     }
338
339     pub fn own_counts(&self) -> GenericParamCount {
340         // We could cache this as a property of `GenericParamCount`, but
341         // the aim is to refactor this away entirely eventually and the
342         // presence of this method will be a constant reminder.
343         let mut own_counts: GenericParamCount = Default::default();
344
345         for arg in self.args {
346             match arg {
347                 GenericArg::Lifetime(_) => own_counts.lifetimes += 1,
348                 GenericArg::Type(_) => own_counts.types += 1,
349                 GenericArg::Const(_) => own_counts.consts += 1,
350             };
351         }
352
353         own_counts
354     }
355
356     pub fn span(&self) -> Option<Span> {
357         self.args
358             .iter()
359             .filter(|arg| !arg.is_synthetic())
360             .map(|arg| arg.span())
361             .reduce(|span1, span2| span1.to(span2))
362     }
363
364     /// Returns span encompassing arguments and their surrounding `<>` or `()`
365     pub fn span_ext(&self, sm: &SourceMap) -> Option<Span> {
366         let mut span = self.span()?;
367
368         let (o, c) = if self.parenthesized { ('(', ')') } else { ('<', '>') };
369
370         if let Ok(snippet) = sm.span_to_snippet(span) {
371             let snippet = snippet.as_bytes();
372
373             if snippet[0] != (o as u8) || snippet[snippet.len() - 1] != (c as u8) {
374                 span = sm.span_extend_to_prev_char(span, o, true);
375                 span = span.with_lo(span.lo() - BytePos(1));
376
377                 span = sm.span_extend_to_next_char(span, c, true);
378                 span = span.with_hi(span.hi() + BytePos(1));
379             }
380         }
381
382         Some(span)
383     }
384
385     pub fn is_empty(&self) -> bool {
386         self.args.is_empty()
387     }
388 }
389
390 /// A modifier on a bound, currently this is only used for `?Sized`, where the
391 /// modifier is `Maybe`. Negative bounds should also be handled here.
392 #[derive(Copy, Clone, PartialEq, Eq, Encodable, Hash, Debug)]
393 #[derive(HashStable_Generic)]
394 pub enum TraitBoundModifier {
395     None,
396     Maybe,
397     MaybeConst,
398 }
399
400 /// The AST represents all type param bounds as types.
401 /// `typeck::collect::compute_bounds` matches these against
402 /// the "special" built-in traits (see `middle::lang_items`) and
403 /// detects `Copy`, `Send` and `Sync`.
404 #[derive(Debug, HashStable_Generic)]
405 pub enum GenericBound<'hir> {
406     Trait(PolyTraitRef<'hir>, TraitBoundModifier),
407     // FIXME(davidtwco): Introduce `PolyTraitRef::LangItem`
408     LangItemTrait(LangItem, Span, HirId, &'hir GenericArgs<'hir>),
409     Outlives(Lifetime),
410 }
411
412 impl GenericBound<'_> {
413     pub fn trait_ref(&self) -> Option<&TraitRef<'_>> {
414         match self {
415             GenericBound::Trait(data, _) => Some(&data.trait_ref),
416             _ => None,
417         }
418     }
419
420     pub fn span(&self) -> Span {
421         match self {
422             GenericBound::Trait(t, ..) => t.span,
423             GenericBound::LangItemTrait(_, span, ..) => *span,
424             GenericBound::Outlives(l) => l.span,
425         }
426     }
427 }
428
429 pub type GenericBounds<'hir> = &'hir [GenericBound<'hir>];
430
431 #[derive(Copy, Clone, PartialEq, Eq, Encodable, Debug, HashStable_Generic)]
432 pub enum LifetimeParamKind {
433     // Indicates that the lifetime definition was explicitly declared (e.g., in
434     // `fn foo<'a>(x: &'a u8) -> &'a u8 { x }`).
435     Explicit,
436
437     // Indicates that the lifetime definition was synthetically added
438     // as a result of an in-band lifetime usage (e.g., in
439     // `fn foo(x: &'a u8) -> &'a u8 { x }`).
440     InBand,
441
442     // Indication that the lifetime was elided (e.g., in both cases in
443     // `fn foo(x: &u8) -> &'_ u8 { x }`).
444     Elided,
445
446     // Indication that the lifetime name was somehow in error.
447     Error,
448 }
449
450 #[derive(Debug, HashStable_Generic)]
451 pub enum GenericParamKind<'hir> {
452     /// A lifetime definition (e.g., `'a: 'b + 'c + 'd`).
453     Lifetime {
454         kind: LifetimeParamKind,
455     },
456     Type {
457         default: Option<&'hir Ty<'hir>>,
458         synthetic: Option<SyntheticTyParamKind>,
459     },
460     Const {
461         ty: &'hir Ty<'hir>,
462         /// Optional default value for the const generic param
463         default: Option<AnonConst>,
464     },
465 }
466
467 #[derive(Debug, HashStable_Generic)]
468 pub struct GenericParam<'hir> {
469     pub hir_id: HirId,
470     pub name: ParamName,
471     pub attrs: &'hir [Attribute],
472     pub bounds: GenericBounds<'hir>,
473     pub span: Span,
474     pub pure_wrt_drop: bool,
475     pub kind: GenericParamKind<'hir>,
476 }
477
478 impl GenericParam<'hir> {
479     pub fn bounds_span(&self) -> Option<Span> {
480         self.bounds.iter().fold(None, |span, bound| {
481             let span = span.map(|s| s.to(bound.span())).unwrap_or_else(|| bound.span());
482
483             Some(span)
484         })
485     }
486 }
487
488 #[derive(Default)]
489 pub struct GenericParamCount {
490     pub lifetimes: usize,
491     pub types: usize,
492     pub consts: usize,
493 }
494
495 /// Represents lifetimes and type parameters attached to a declaration
496 /// of a function, enum, trait, etc.
497 #[derive(Debug, HashStable_Generic)]
498 pub struct Generics<'hir> {
499     pub params: &'hir [GenericParam<'hir>],
500     pub where_clause: WhereClause<'hir>,
501     pub span: Span,
502 }
503
504 impl Generics<'hir> {
505     pub const fn empty() -> Generics<'hir> {
506         Generics {
507             params: &[],
508             where_clause: WhereClause { predicates: &[], span: DUMMY_SP },
509             span: DUMMY_SP,
510         }
511     }
512
513     pub fn get_named(&self, name: Symbol) -> Option<&GenericParam<'_>> {
514         for param in self.params {
515             if name == param.name.ident().name {
516                 return Some(param);
517             }
518         }
519         None
520     }
521
522     pub fn spans(&self) -> MultiSpan {
523         if self.params.is_empty() {
524             self.span.into()
525         } else {
526             self.params.iter().map(|p| p.span).collect::<Vec<Span>>().into()
527         }
528     }
529 }
530
531 /// Synthetic type parameters are converted to another form during lowering; this allows
532 /// us to track the original form they had, and is useful for error messages.
533 #[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
534 #[derive(HashStable_Generic)]
535 pub enum SyntheticTyParamKind {
536     ImplTrait,
537     // Created by the `#[rustc_synthetic]` attribute.
538     FromAttr,
539 }
540
541 /// A where-clause in a definition.
542 #[derive(Debug, HashStable_Generic)]
543 pub struct WhereClause<'hir> {
544     pub predicates: &'hir [WherePredicate<'hir>],
545     // Only valid if predicates aren't empty.
546     pub span: Span,
547 }
548
549 impl WhereClause<'_> {
550     pub fn span(&self) -> Option<Span> {
551         if self.predicates.is_empty() { None } else { Some(self.span) }
552     }
553
554     /// The `WhereClause` under normal circumstances points at either the predicates or the empty
555     /// space where the `where` clause should be. Only of use for diagnostic suggestions.
556     pub fn span_for_predicates_or_empty_place(&self) -> Span {
557         self.span
558     }
559
560     /// `Span` where further predicates would be suggested, accounting for trailing commas, like
561     ///  in `fn foo<T>(t: T) where T: Foo,` so we don't suggest two trailing commas.
562     pub fn tail_span_for_suggestion(&self) -> Span {
563         let end = self.span_for_predicates_or_empty_place().shrink_to_hi();
564         self.predicates.last().map_or(end, |p| p.span()).shrink_to_hi().to(end)
565     }
566 }
567
568 /// A single predicate in a where-clause.
569 #[derive(Debug, HashStable_Generic)]
570 pub enum WherePredicate<'hir> {
571     /// A type binding (e.g., `for<'c> Foo: Send + Clone + 'c`).
572     BoundPredicate(WhereBoundPredicate<'hir>),
573     /// A lifetime predicate (e.g., `'a: 'b + 'c`).
574     RegionPredicate(WhereRegionPredicate<'hir>),
575     /// An equality predicate (unsupported).
576     EqPredicate(WhereEqPredicate<'hir>),
577 }
578
579 impl WherePredicate<'_> {
580     pub fn span(&self) -> Span {
581         match self {
582             WherePredicate::BoundPredicate(p) => p.span,
583             WherePredicate::RegionPredicate(p) => p.span,
584             WherePredicate::EqPredicate(p) => p.span,
585         }
586     }
587 }
588
589 /// A type bound (e.g., `for<'c> Foo: Send + Clone + 'c`).
590 #[derive(Debug, HashStable_Generic)]
591 pub struct WhereBoundPredicate<'hir> {
592     pub span: Span,
593     /// Any generics from a `for` binding.
594     pub bound_generic_params: &'hir [GenericParam<'hir>],
595     /// The type being bounded.
596     pub bounded_ty: &'hir Ty<'hir>,
597     /// Trait and lifetime bounds (e.g., `Clone + Send + 'static`).
598     pub bounds: GenericBounds<'hir>,
599 }
600
601 /// A lifetime predicate (e.g., `'a: 'b + 'c`).
602 #[derive(Debug, HashStable_Generic)]
603 pub struct WhereRegionPredicate<'hir> {
604     pub span: Span,
605     pub lifetime: Lifetime,
606     pub bounds: GenericBounds<'hir>,
607 }
608
609 /// An equality predicate (e.g., `T = int`); currently unsupported.
610 #[derive(Debug, HashStable_Generic)]
611 pub struct WhereEqPredicate<'hir> {
612     pub hir_id: HirId,
613     pub span: Span,
614     pub lhs_ty: &'hir Ty<'hir>,
615     pub rhs_ty: &'hir Ty<'hir>,
616 }
617
618 #[derive(Encodable, Debug, HashStable_Generic)]
619 pub struct ModuleItems {
620     // Use BTreeSets here so items are in the same order as in the
621     // list of all items in Crate
622     pub items: BTreeSet<HirId>,
623     pub trait_items: BTreeSet<TraitItemId>,
624     pub impl_items: BTreeSet<ImplItemId>,
625     pub foreign_items: BTreeSet<ForeignItemId>,
626 }
627
628 /// A type representing only the top-level module.
629 #[derive(Encodable, Debug, HashStable_Generic)]
630 pub struct CrateItem<'hir> {
631     pub module: Mod<'hir>,
632     pub attrs: &'hir [Attribute],
633     pub span: Span,
634 }
635
636 /// The top-level data structure that stores the entire contents of
637 /// the crate currently being compiled.
638 ///
639 /// For more details, see the [rustc dev guide].
640 ///
641 /// [rustc dev guide]: https://rustc-dev-guide.rust-lang.org/hir.html
642 #[derive(Debug)]
643 pub struct Crate<'hir> {
644     pub item: CrateItem<'hir>,
645     pub exported_macros: &'hir [MacroDef<'hir>],
646     // Attributes from non-exported macros, kept only for collecting the library feature list.
647     pub non_exported_macro_attrs: &'hir [Attribute],
648
649     // N.B., we use a `BTreeMap` here so that `visit_all_items` iterates
650     // over the ids in increasing order. In principle it should not
651     // matter what order we visit things in, but in *practice* it
652     // does, because it can affect the order in which errors are
653     // detected, which in turn can make UI tests yield
654     // slightly different results.
655     pub items: BTreeMap<HirId, Item<'hir>>,
656
657     pub trait_items: BTreeMap<TraitItemId, TraitItem<'hir>>,
658     pub impl_items: BTreeMap<ImplItemId, ImplItem<'hir>>,
659     pub foreign_items: BTreeMap<ForeignItemId, ForeignItem<'hir>>,
660     pub bodies: BTreeMap<BodyId, Body<'hir>>,
661     pub trait_impls: BTreeMap<DefId, Vec<HirId>>,
662
663     /// A list of the body ids written out in the order in which they
664     /// appear in the crate. If you're going to process all the bodies
665     /// in the crate, you should iterate over this list rather than the keys
666     /// of bodies.
667     pub body_ids: Vec<BodyId>,
668
669     /// A list of modules written out in the order in which they
670     /// appear in the crate. This includes the main crate module.
671     pub modules: BTreeMap<HirId, ModuleItems>,
672     /// A list of proc macro HirIds, written out in the order in which
673     /// they are declared in the static array generated by proc_macro_harness.
674     pub proc_macros: Vec<HirId>,
675
676     pub trait_map: BTreeMap<HirId, Vec<TraitCandidate>>,
677 }
678
679 impl Crate<'hir> {
680     pub fn item(&self, id: HirId) -> &Item<'hir> {
681         &self.items[&id]
682     }
683
684     pub fn trait_item(&self, id: TraitItemId) -> &TraitItem<'hir> {
685         &self.trait_items[&id]
686     }
687
688     pub fn impl_item(&self, id: ImplItemId) -> &ImplItem<'hir> {
689         &self.impl_items[&id]
690     }
691
692     pub fn foreign_item(&self, id: ForeignItemId) -> &ForeignItem<'hir> {
693         &self.foreign_items[&id]
694     }
695
696     pub fn body(&self, id: BodyId) -> &Body<'hir> {
697         &self.bodies[&id]
698     }
699 }
700
701 impl Crate<'_> {
702     /// Visits all items in the crate in some deterministic (but
703     /// unspecified) order. If you just need to process every item,
704     /// but don't care about nesting, this method is the best choice.
705     ///
706     /// If you do care about nesting -- usually because your algorithm
707     /// follows lexical scoping rules -- then you want a different
708     /// approach. You should override `visit_nested_item` in your
709     /// visitor and then call `intravisit::walk_crate` instead.
710     pub fn visit_all_item_likes<'hir, V>(&'hir self, visitor: &mut V)
711     where
712         V: itemlikevisit::ItemLikeVisitor<'hir>,
713     {
714         for item in self.items.values() {
715             visitor.visit_item(item);
716         }
717
718         for trait_item in self.trait_items.values() {
719             visitor.visit_trait_item(trait_item);
720         }
721
722         for impl_item in self.impl_items.values() {
723             visitor.visit_impl_item(impl_item);
724         }
725
726         for foreign_item in self.foreign_items.values() {
727             visitor.visit_foreign_item(foreign_item);
728         }
729     }
730
731     /// A parallel version of `visit_all_item_likes`.
732     pub fn par_visit_all_item_likes<'hir, V>(&'hir self, visitor: &V)
733     where
734         V: itemlikevisit::ParItemLikeVisitor<'hir> + Sync + Send,
735     {
736         parallel!(
737             {
738                 par_for_each_in(&self.items, |(_, item)| {
739                     visitor.visit_item(item);
740                 });
741             },
742             {
743                 par_for_each_in(&self.trait_items, |(_, trait_item)| {
744                     visitor.visit_trait_item(trait_item);
745                 });
746             },
747             {
748                 par_for_each_in(&self.impl_items, |(_, impl_item)| {
749                     visitor.visit_impl_item(impl_item);
750                 });
751             },
752             {
753                 par_for_each_in(&self.foreign_items, |(_, foreign_item)| {
754                     visitor.visit_foreign_item(foreign_item);
755                 });
756             }
757         );
758     }
759 }
760
761 /// A macro definition, in this crate or imported from another.
762 ///
763 /// Not parsed directly, but created on macro import or `macro_rules!` expansion.
764 #[derive(Debug, HashStable_Generic)]
765 pub struct MacroDef<'hir> {
766     pub ident: Ident,
767     pub vis: Visibility<'hir>,
768     pub attrs: &'hir [Attribute],
769     pub hir_id: HirId,
770     pub span: Span,
771     pub ast: ast::MacroDef,
772 }
773
774 /// A block of statements `{ .. }`, which may have a label (in this case the
775 /// `targeted_by_break` field will be `true`) and may be `unsafe` by means of
776 /// the `rules` being anything but `DefaultBlock`.
777 #[derive(Debug, HashStable_Generic)]
778 pub struct Block<'hir> {
779     /// Statements in a block.
780     pub stmts: &'hir [Stmt<'hir>],
781     /// An expression at the end of the block
782     /// without a semicolon, if any.
783     pub expr: Option<&'hir Expr<'hir>>,
784     #[stable_hasher(ignore)]
785     pub hir_id: HirId,
786     /// Distinguishes between `unsafe { ... }` and `{ ... }`.
787     pub rules: BlockCheckMode,
788     pub span: Span,
789     /// If true, then there may exist `break 'a` values that aim to
790     /// break out of this block early.
791     /// Used by `'label: {}` blocks and by `try {}` blocks.
792     pub targeted_by_break: bool,
793 }
794
795 #[derive(Debug, HashStable_Generic)]
796 pub struct Pat<'hir> {
797     #[stable_hasher(ignore)]
798     pub hir_id: HirId,
799     pub kind: PatKind<'hir>,
800     pub span: Span,
801     // Whether to use default binding modes.
802     // At present, this is false only for destructuring assignment.
803     pub default_binding_modes: bool,
804 }
805
806 impl<'hir> Pat<'hir> {
807     // FIXME(#19596) this is a workaround, but there should be a better way
808     fn walk_short_(&self, it: &mut impl FnMut(&Pat<'hir>) -> bool) -> bool {
809         if !it(self) {
810             return false;
811         }
812
813         use PatKind::*;
814         match &self.kind {
815             Wild | Lit(_) | Range(..) | Binding(.., None) | Path(_) => true,
816             Box(s) | Ref(s, _) | Binding(.., Some(s)) => s.walk_short_(it),
817             Struct(_, fields, _) => fields.iter().all(|field| field.pat.walk_short_(it)),
818             TupleStruct(_, s, _) | Tuple(s, _) | Or(s) => s.iter().all(|p| p.walk_short_(it)),
819             Slice(before, slice, after) => {
820                 before.iter().chain(slice.iter()).chain(after.iter()).all(|p| p.walk_short_(it))
821             }
822         }
823     }
824
825     /// Walk the pattern in left-to-right order,
826     /// short circuiting (with `.all(..)`) if `false` is returned.
827     ///
828     /// Note that when visiting e.g. `Tuple(ps)`,
829     /// if visiting `ps[0]` returns `false`,
830     /// then `ps[1]` will not be visited.
831     pub fn walk_short(&self, mut it: impl FnMut(&Pat<'hir>) -> bool) -> bool {
832         self.walk_short_(&mut it)
833     }
834
835     // FIXME(#19596) this is a workaround, but there should be a better way
836     fn walk_(&self, it: &mut impl FnMut(&Pat<'hir>) -> bool) {
837         if !it(self) {
838             return;
839         }
840
841         use PatKind::*;
842         match &self.kind {
843             Wild | Lit(_) | Range(..) | Binding(.., None) | Path(_) => {}
844             Box(s) | Ref(s, _) | Binding(.., Some(s)) => s.walk_(it),
845             Struct(_, fields, _) => fields.iter().for_each(|field| field.pat.walk_(it)),
846             TupleStruct(_, s, _) | Tuple(s, _) | Or(s) => s.iter().for_each(|p| p.walk_(it)),
847             Slice(before, slice, after) => {
848                 before.iter().chain(slice.iter()).chain(after.iter()).for_each(|p| p.walk_(it))
849             }
850         }
851     }
852
853     /// Walk the pattern in left-to-right order.
854     ///
855     /// If `it(pat)` returns `false`, the children are not visited.
856     pub fn walk(&self, mut it: impl FnMut(&Pat<'hir>) -> bool) {
857         self.walk_(&mut it)
858     }
859
860     /// Walk the pattern in left-to-right order.
861     ///
862     /// If you always want to recurse, prefer this method over `walk`.
863     pub fn walk_always(&self, mut it: impl FnMut(&Pat<'_>)) {
864         self.walk(|p| {
865             it(p);
866             true
867         })
868     }
869 }
870
871 /// A single field in a struct pattern.
872 ///
873 /// Patterns like the fields of Foo `{ x, ref y, ref mut z }`
874 /// are treated the same as` x: x, y: ref y, z: ref mut z`,
875 /// except `is_shorthand` is true.
876 #[derive(Debug, HashStable_Generic)]
877 pub struct FieldPat<'hir> {
878     #[stable_hasher(ignore)]
879     pub hir_id: HirId,
880     /// The identifier for the field.
881     #[stable_hasher(project(name))]
882     pub ident: Ident,
883     /// The pattern the field is destructured to.
884     pub pat: &'hir Pat<'hir>,
885     pub is_shorthand: bool,
886     pub span: Span,
887 }
888
889 /// Explicit binding annotations given in the HIR for a binding. Note
890 /// that this is not the final binding *mode* that we infer after type
891 /// inference.
892 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
893 pub enum BindingAnnotation {
894     /// No binding annotation given: this means that the final binding mode
895     /// will depend on whether we have skipped through a `&` reference
896     /// when matching. For example, the `x` in `Some(x)` will have binding
897     /// mode `None`; if you do `let Some(x) = &Some(22)`, it will
898     /// ultimately be inferred to be by-reference.
899     ///
900     /// Note that implicit reference skipping is not implemented yet (#42640).
901     Unannotated,
902
903     /// Annotated with `mut x` -- could be either ref or not, similar to `None`.
904     Mutable,
905
906     /// Annotated as `ref`, like `ref x`
907     Ref,
908
909     /// Annotated as `ref mut x`.
910     RefMut,
911 }
912
913 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
914 pub enum RangeEnd {
915     Included,
916     Excluded,
917 }
918
919 impl fmt::Display for RangeEnd {
920     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
921         f.write_str(match self {
922             RangeEnd::Included => "..=",
923             RangeEnd::Excluded => "..",
924         })
925     }
926 }
927
928 #[derive(Debug, HashStable_Generic)]
929 pub enum PatKind<'hir> {
930     /// Represents a wildcard pattern (i.e., `_`).
931     Wild,
932
933     /// A fresh binding `ref mut binding @ OPT_SUBPATTERN`.
934     /// The `HirId` is the canonical ID for the variable being bound,
935     /// (e.g., in `Ok(x) | Err(x)`, both `x` use the same canonical ID),
936     /// which is the pattern ID of the first `x`.
937     Binding(BindingAnnotation, HirId, Ident, Option<&'hir Pat<'hir>>),
938
939     /// A struct or struct variant pattern (e.g., `Variant {x, y, ..}`).
940     /// The `bool` is `true` in the presence of a `..`.
941     Struct(QPath<'hir>, &'hir [FieldPat<'hir>], bool),
942
943     /// A tuple struct/variant pattern `Variant(x, y, .., z)`.
944     /// If the `..` pattern fragment is present, then `Option<usize>` denotes its position.
945     /// `0 <= position <= subpats.len()`
946     TupleStruct(QPath<'hir>, &'hir [&'hir Pat<'hir>], Option<usize>),
947
948     /// An or-pattern `A | B | C`.
949     /// Invariant: `pats.len() >= 2`.
950     Or(&'hir [&'hir Pat<'hir>]),
951
952     /// A path pattern for an unit struct/variant or a (maybe-associated) constant.
953     Path(QPath<'hir>),
954
955     /// A tuple pattern (e.g., `(a, b)`).
956     /// If the `..` pattern fragment is present, then `Option<usize>` denotes its position.
957     /// `0 <= position <= subpats.len()`
958     Tuple(&'hir [&'hir Pat<'hir>], Option<usize>),
959
960     /// A `box` pattern.
961     Box(&'hir Pat<'hir>),
962
963     /// A reference pattern (e.g., `&mut (a, b)`).
964     Ref(&'hir Pat<'hir>, Mutability),
965
966     /// A literal.
967     Lit(&'hir Expr<'hir>),
968
969     /// A range pattern (e.g., `1..=2` or `1..2`).
970     Range(Option<&'hir Expr<'hir>>, Option<&'hir Expr<'hir>>, RangeEnd),
971
972     /// A slice pattern, `[before_0, ..., before_n, (slice, after_0, ..., after_n)?]`.
973     ///
974     /// Here, `slice` is lowered from the syntax `($binding_mode $ident @)? ..`.
975     /// If `slice` exists, then `after` can be non-empty.
976     ///
977     /// The representation for e.g., `[a, b, .., c, d]` is:
978     /// ```
979     /// PatKind::Slice([Binding(a), Binding(b)], Some(Wild), [Binding(c), Binding(d)])
980     /// ```
981     Slice(&'hir [&'hir Pat<'hir>], Option<&'hir Pat<'hir>>, &'hir [&'hir Pat<'hir>]),
982 }
983
984 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
985 pub enum BinOpKind {
986     /// The `+` operator (addition).
987     Add,
988     /// The `-` operator (subtraction).
989     Sub,
990     /// The `*` operator (multiplication).
991     Mul,
992     /// The `/` operator (division).
993     Div,
994     /// The `%` operator (modulus).
995     Rem,
996     /// The `&&` operator (logical and).
997     And,
998     /// The `||` operator (logical or).
999     Or,
1000     /// The `^` operator (bitwise xor).
1001     BitXor,
1002     /// The `&` operator (bitwise and).
1003     BitAnd,
1004     /// The `|` operator (bitwise or).
1005     BitOr,
1006     /// The `<<` operator (shift left).
1007     Shl,
1008     /// The `>>` operator (shift right).
1009     Shr,
1010     /// The `==` operator (equality).
1011     Eq,
1012     /// The `<` operator (less than).
1013     Lt,
1014     /// The `<=` operator (less than or equal to).
1015     Le,
1016     /// The `!=` operator (not equal to).
1017     Ne,
1018     /// The `>=` operator (greater than or equal to).
1019     Ge,
1020     /// The `>` operator (greater than).
1021     Gt,
1022 }
1023
1024 impl BinOpKind {
1025     pub fn as_str(self) -> &'static str {
1026         match self {
1027             BinOpKind::Add => "+",
1028             BinOpKind::Sub => "-",
1029             BinOpKind::Mul => "*",
1030             BinOpKind::Div => "/",
1031             BinOpKind::Rem => "%",
1032             BinOpKind::And => "&&",
1033             BinOpKind::Or => "||",
1034             BinOpKind::BitXor => "^",
1035             BinOpKind::BitAnd => "&",
1036             BinOpKind::BitOr => "|",
1037             BinOpKind::Shl => "<<",
1038             BinOpKind::Shr => ">>",
1039             BinOpKind::Eq => "==",
1040             BinOpKind::Lt => "<",
1041             BinOpKind::Le => "<=",
1042             BinOpKind::Ne => "!=",
1043             BinOpKind::Ge => ">=",
1044             BinOpKind::Gt => ">",
1045         }
1046     }
1047
1048     pub fn is_lazy(self) -> bool {
1049         matches!(self, BinOpKind::And | BinOpKind::Or)
1050     }
1051
1052     pub fn is_shift(self) -> bool {
1053         matches!(self, BinOpKind::Shl | BinOpKind::Shr)
1054     }
1055
1056     pub fn is_comparison(self) -> bool {
1057         match self {
1058             BinOpKind::Eq
1059             | BinOpKind::Lt
1060             | BinOpKind::Le
1061             | BinOpKind::Ne
1062             | BinOpKind::Gt
1063             | BinOpKind::Ge => true,
1064             BinOpKind::And
1065             | BinOpKind::Or
1066             | BinOpKind::Add
1067             | BinOpKind::Sub
1068             | BinOpKind::Mul
1069             | BinOpKind::Div
1070             | BinOpKind::Rem
1071             | BinOpKind::BitXor
1072             | BinOpKind::BitAnd
1073             | BinOpKind::BitOr
1074             | BinOpKind::Shl
1075             | BinOpKind::Shr => false,
1076         }
1077     }
1078
1079     /// Returns `true` if the binary operator takes its arguments by value.
1080     pub fn is_by_value(self) -> bool {
1081         !self.is_comparison()
1082     }
1083 }
1084
1085 impl Into<ast::BinOpKind> for BinOpKind {
1086     fn into(self) -> ast::BinOpKind {
1087         match self {
1088             BinOpKind::Add => ast::BinOpKind::Add,
1089             BinOpKind::Sub => ast::BinOpKind::Sub,
1090             BinOpKind::Mul => ast::BinOpKind::Mul,
1091             BinOpKind::Div => ast::BinOpKind::Div,
1092             BinOpKind::Rem => ast::BinOpKind::Rem,
1093             BinOpKind::And => ast::BinOpKind::And,
1094             BinOpKind::Or => ast::BinOpKind::Or,
1095             BinOpKind::BitXor => ast::BinOpKind::BitXor,
1096             BinOpKind::BitAnd => ast::BinOpKind::BitAnd,
1097             BinOpKind::BitOr => ast::BinOpKind::BitOr,
1098             BinOpKind::Shl => ast::BinOpKind::Shl,
1099             BinOpKind::Shr => ast::BinOpKind::Shr,
1100             BinOpKind::Eq => ast::BinOpKind::Eq,
1101             BinOpKind::Lt => ast::BinOpKind::Lt,
1102             BinOpKind::Le => ast::BinOpKind::Le,
1103             BinOpKind::Ne => ast::BinOpKind::Ne,
1104             BinOpKind::Ge => ast::BinOpKind::Ge,
1105             BinOpKind::Gt => ast::BinOpKind::Gt,
1106         }
1107     }
1108 }
1109
1110 pub type BinOp = Spanned<BinOpKind>;
1111
1112 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
1113 pub enum UnOp {
1114     /// The `*` operator (deferencing).
1115     UnDeref,
1116     /// The `!` operator (logical negation).
1117     UnNot,
1118     /// The `-` operator (negation).
1119     UnNeg,
1120 }
1121
1122 impl UnOp {
1123     pub fn as_str(self) -> &'static str {
1124         match self {
1125             Self::UnDeref => "*",
1126             Self::UnNot => "!",
1127             Self::UnNeg => "-",
1128         }
1129     }
1130
1131     /// Returns `true` if the unary operator takes its argument by value.
1132     pub fn is_by_value(self) -> bool {
1133         matches!(self, Self::UnNeg | Self::UnNot)
1134     }
1135 }
1136
1137 /// A statement.
1138 #[derive(Debug, HashStable_Generic)]
1139 pub struct Stmt<'hir> {
1140     pub hir_id: HirId,
1141     pub kind: StmtKind<'hir>,
1142     pub span: Span,
1143 }
1144
1145 /// The contents of a statement.
1146 #[derive(Debug, HashStable_Generic)]
1147 pub enum StmtKind<'hir> {
1148     /// A local (`let`) binding.
1149     Local(&'hir Local<'hir>),
1150
1151     /// An item binding.
1152     Item(ItemId),
1153
1154     /// An expression without a trailing semi-colon (must have unit type).
1155     Expr(&'hir Expr<'hir>),
1156
1157     /// An expression with a trailing semi-colon (may have any type).
1158     Semi(&'hir Expr<'hir>),
1159 }
1160
1161 impl<'hir> StmtKind<'hir> {
1162     pub fn attrs(&self, get_item: impl FnOnce(ItemId) -> &'hir Item<'hir>) -> &'hir [Attribute] {
1163         match *self {
1164             StmtKind::Local(ref l) => &l.attrs,
1165             StmtKind::Item(ref item_id) => &get_item(*item_id).attrs,
1166             StmtKind::Expr(ref e) | StmtKind::Semi(ref e) => &e.attrs,
1167         }
1168     }
1169 }
1170
1171 /// Represents a `let` statement (i.e., `let <pat>:<ty> = <expr>;`).
1172 #[derive(Debug, HashStable_Generic)]
1173 pub struct Local<'hir> {
1174     pub pat: &'hir Pat<'hir>,
1175     /// Type annotation, if any (otherwise the type will be inferred).
1176     pub ty: Option<&'hir Ty<'hir>>,
1177     /// Initializer expression to set the value, if any.
1178     pub init: Option<&'hir Expr<'hir>>,
1179     pub hir_id: HirId,
1180     pub span: Span,
1181     pub attrs: AttrVec,
1182     /// Can be `ForLoopDesugar` if the `let` statement is part of a `for` loop
1183     /// desugaring. Otherwise will be `Normal`.
1184     pub source: LocalSource,
1185 }
1186
1187 /// Represents a single arm of a `match` expression, e.g.
1188 /// `<pat> (if <guard>) => <body>`.
1189 #[derive(Debug, HashStable_Generic)]
1190 pub struct Arm<'hir> {
1191     #[stable_hasher(ignore)]
1192     pub hir_id: HirId,
1193     pub span: Span,
1194     pub attrs: &'hir [Attribute],
1195     /// If this pattern and the optional guard matches, then `body` is evaluated.
1196     pub pat: &'hir Pat<'hir>,
1197     /// Optional guard clause.
1198     pub guard: Option<Guard<'hir>>,
1199     /// The expression the arm evaluates to if this arm matches.
1200     pub body: &'hir Expr<'hir>,
1201 }
1202
1203 #[derive(Debug, HashStable_Generic)]
1204 pub enum Guard<'hir> {
1205     If(&'hir Expr<'hir>),
1206     IfLet(&'hir Pat<'hir>, &'hir Expr<'hir>),
1207 }
1208
1209 #[derive(Debug, HashStable_Generic)]
1210 pub struct Field<'hir> {
1211     #[stable_hasher(ignore)]
1212     pub hir_id: HirId,
1213     pub ident: Ident,
1214     pub expr: &'hir Expr<'hir>,
1215     pub span: Span,
1216     pub is_shorthand: bool,
1217 }
1218
1219 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
1220 pub enum BlockCheckMode {
1221     DefaultBlock,
1222     UnsafeBlock(UnsafeSource),
1223     PushUnsafeBlock(UnsafeSource),
1224     PopUnsafeBlock(UnsafeSource),
1225 }
1226
1227 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
1228 pub enum UnsafeSource {
1229     CompilerGenerated,
1230     UserProvided,
1231 }
1232
1233 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Encodable, Hash, Debug)]
1234 pub struct BodyId {
1235     pub hir_id: HirId,
1236 }
1237
1238 /// The body of a function, closure, or constant value. In the case of
1239 /// a function, the body contains not only the function body itself
1240 /// (which is an expression), but also the argument patterns, since
1241 /// those are something that the caller doesn't really care about.
1242 ///
1243 /// # Examples
1244 ///
1245 /// ```
1246 /// fn foo((x, y): (u32, u32)) -> u32 {
1247 ///     x + y
1248 /// }
1249 /// ```
1250 ///
1251 /// Here, the `Body` associated with `foo()` would contain:
1252 ///
1253 /// - an `params` array containing the `(x, y)` pattern
1254 /// - a `value` containing the `x + y` expression (maybe wrapped in a block)
1255 /// - `generator_kind` would be `None`
1256 ///
1257 /// All bodies have an **owner**, which can be accessed via the HIR
1258 /// map using `body_owner_def_id()`.
1259 #[derive(Debug)]
1260 pub struct Body<'hir> {
1261     pub params: &'hir [Param<'hir>],
1262     pub value: Expr<'hir>,
1263     pub generator_kind: Option<GeneratorKind>,
1264 }
1265
1266 impl Body<'hir> {
1267     pub fn id(&self) -> BodyId {
1268         BodyId { hir_id: self.value.hir_id }
1269     }
1270
1271     pub fn generator_kind(&self) -> Option<GeneratorKind> {
1272         self.generator_kind
1273     }
1274 }
1275
1276 /// The type of source expression that caused this generator to be created.
1277 #[derive(Clone, PartialEq, Eq, HashStable_Generic, Encodable, Decodable, Debug, Copy)]
1278 pub enum GeneratorKind {
1279     /// An explicit `async` block or the body of an async function.
1280     Async(AsyncGeneratorKind),
1281
1282     /// A generator literal created via a `yield` inside a closure.
1283     Gen,
1284 }
1285
1286 impl fmt::Display for GeneratorKind {
1287     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1288         match self {
1289             GeneratorKind::Async(k) => fmt::Display::fmt(k, f),
1290             GeneratorKind::Gen => f.write_str("generator"),
1291         }
1292     }
1293 }
1294
1295 /// In the case of a generator created as part of an async construct,
1296 /// which kind of async construct caused it to be created?
1297 ///
1298 /// This helps error messages but is also used to drive coercions in
1299 /// type-checking (see #60424).
1300 #[derive(Clone, PartialEq, Eq, HashStable_Generic, Encodable, Decodable, Debug, Copy)]
1301 pub enum AsyncGeneratorKind {
1302     /// An explicit `async` block written by the user.
1303     Block,
1304
1305     /// An explicit `async` block written by the user.
1306     Closure,
1307
1308     /// The `async` block generated as the body of an async function.
1309     Fn,
1310 }
1311
1312 impl fmt::Display for AsyncGeneratorKind {
1313     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1314         f.write_str(match self {
1315             AsyncGeneratorKind::Block => "`async` block",
1316             AsyncGeneratorKind::Closure => "`async` closure body",
1317             AsyncGeneratorKind::Fn => "`async fn` body",
1318         })
1319     }
1320 }
1321
1322 #[derive(Copy, Clone, Debug)]
1323 pub enum BodyOwnerKind {
1324     /// Functions and methods.
1325     Fn,
1326
1327     /// Closures
1328     Closure,
1329
1330     /// Constants and associated constants.
1331     Const,
1332
1333     /// Initializer of a `static` item.
1334     Static(Mutability),
1335 }
1336
1337 impl BodyOwnerKind {
1338     pub fn is_fn_or_closure(self) -> bool {
1339         match self {
1340             BodyOwnerKind::Fn | BodyOwnerKind::Closure => true,
1341             BodyOwnerKind::Const | BodyOwnerKind::Static(_) => false,
1342         }
1343     }
1344 }
1345
1346 /// The kind of an item that requires const-checking.
1347 #[derive(Clone, Copy, Debug, PartialEq, Eq)]
1348 pub enum ConstContext {
1349     /// A `const fn`.
1350     ConstFn,
1351
1352     /// A `static` or `static mut`.
1353     Static(Mutability),
1354
1355     /// A `const`, associated `const`, or other const context.
1356     ///
1357     /// Other contexts include:
1358     /// - Array length expressions
1359     /// - Enum discriminants
1360     /// - Const generics
1361     ///
1362     /// For the most part, other contexts are treated just like a regular `const`, so they are
1363     /// lumped into the same category.
1364     Const,
1365 }
1366
1367 impl ConstContext {
1368     /// A description of this const context that can appear between backticks in an error message.
1369     ///
1370     /// E.g. `const` or `static mut`.
1371     pub fn keyword_name(self) -> &'static str {
1372         match self {
1373             Self::Const => "const",
1374             Self::Static(Mutability::Not) => "static",
1375             Self::Static(Mutability::Mut) => "static mut",
1376             Self::ConstFn => "const fn",
1377         }
1378     }
1379 }
1380
1381 /// A colloquial, trivially pluralizable description of this const context for use in error
1382 /// messages.
1383 impl fmt::Display for ConstContext {
1384     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1385         match *self {
1386             Self::Const => write!(f, "constant"),
1387             Self::Static(_) => write!(f, "static"),
1388             Self::ConstFn => write!(f, "constant function"),
1389         }
1390     }
1391 }
1392
1393 /// A literal.
1394 pub type Lit = Spanned<LitKind>;
1395
1396 /// A constant (expression) that's not an item or associated item,
1397 /// but needs its own `DefId` for type-checking, const-eval, etc.
1398 /// These are usually found nested inside types (e.g., array lengths)
1399 /// or expressions (e.g., repeat counts), and also used to define
1400 /// explicit discriminant values for enum variants.
1401 #[derive(Copy, Clone, PartialEq, Eq, Encodable, Debug, HashStable_Generic)]
1402 pub struct AnonConst {
1403     pub hir_id: HirId,
1404     pub body: BodyId,
1405 }
1406
1407 /// An expression.
1408 #[derive(Debug)]
1409 pub struct Expr<'hir> {
1410     pub hir_id: HirId,
1411     pub kind: ExprKind<'hir>,
1412     pub attrs: AttrVec,
1413     pub span: Span,
1414 }
1415
1416 // `Expr` is used a lot. Make sure it doesn't unintentionally get bigger.
1417 #[cfg(target_arch = "x86_64")]
1418 rustc_data_structures::static_assert_size!(Expr<'static>, 72);
1419
1420 impl Expr<'_> {
1421     pub fn precedence(&self) -> ExprPrecedence {
1422         match self.kind {
1423             ExprKind::Box(_) => ExprPrecedence::Box,
1424             ExprKind::ConstBlock(_) => ExprPrecedence::ConstBlock,
1425             ExprKind::Array(_) => ExprPrecedence::Array,
1426             ExprKind::Call(..) => ExprPrecedence::Call,
1427             ExprKind::MethodCall(..) => ExprPrecedence::MethodCall,
1428             ExprKind::Tup(_) => ExprPrecedence::Tup,
1429             ExprKind::Binary(op, ..) => ExprPrecedence::Binary(op.node.into()),
1430             ExprKind::Unary(..) => ExprPrecedence::Unary,
1431             ExprKind::Lit(_) => ExprPrecedence::Lit,
1432             ExprKind::Type(..) | ExprKind::Cast(..) => ExprPrecedence::Cast,
1433             ExprKind::DropTemps(ref expr, ..) => expr.precedence(),
1434             ExprKind::If(..) => ExprPrecedence::If,
1435             ExprKind::Loop(..) => ExprPrecedence::Loop,
1436             ExprKind::Match(..) => ExprPrecedence::Match,
1437             ExprKind::Closure(..) => ExprPrecedence::Closure,
1438             ExprKind::Block(..) => ExprPrecedence::Block,
1439             ExprKind::Assign(..) => ExprPrecedence::Assign,
1440             ExprKind::AssignOp(..) => ExprPrecedence::AssignOp,
1441             ExprKind::Field(..) => ExprPrecedence::Field,
1442             ExprKind::Index(..) => ExprPrecedence::Index,
1443             ExprKind::Path(..) => ExprPrecedence::Path,
1444             ExprKind::AddrOf(..) => ExprPrecedence::AddrOf,
1445             ExprKind::Break(..) => ExprPrecedence::Break,
1446             ExprKind::Continue(..) => ExprPrecedence::Continue,
1447             ExprKind::Ret(..) => ExprPrecedence::Ret,
1448             ExprKind::InlineAsm(..) => ExprPrecedence::InlineAsm,
1449             ExprKind::LlvmInlineAsm(..) => ExprPrecedence::InlineAsm,
1450             ExprKind::Struct(..) => ExprPrecedence::Struct,
1451             ExprKind::Repeat(..) => ExprPrecedence::Repeat,
1452             ExprKind::Yield(..) => ExprPrecedence::Yield,
1453             ExprKind::Err => ExprPrecedence::Err,
1454         }
1455     }
1456
1457     // Whether this looks like a place expr, without checking for deref
1458     // adjustments.
1459     // This will return `true` in some potentially surprising cases such as
1460     // `CONSTANT.field`.
1461     pub fn is_syntactic_place_expr(&self) -> bool {
1462         self.is_place_expr(|_| true)
1463     }
1464
1465     /// Whether this is a place expression.
1466     ///
1467     /// `allow_projections_from` should return `true` if indexing a field or index expression based
1468     /// on the given expression should be considered a place expression.
1469     pub fn is_place_expr(&self, mut allow_projections_from: impl FnMut(&Self) -> bool) -> bool {
1470         match self.kind {
1471             ExprKind::Path(QPath::Resolved(_, ref path)) => {
1472                 matches!(path.res, Res::Local(..) | Res::Def(DefKind::Static, _) | Res::Err)
1473             }
1474
1475             // Type ascription inherits its place expression kind from its
1476             // operand. See:
1477             // https://github.com/rust-lang/rfcs/blob/master/text/0803-type-ascription.md#type-ascription-and-temporaries
1478             ExprKind::Type(ref e, _) => e.is_place_expr(allow_projections_from),
1479
1480             ExprKind::Unary(UnOp::UnDeref, _) => true,
1481
1482             ExprKind::Field(ref base, _) | ExprKind::Index(ref base, _) => {
1483                 allow_projections_from(base) || base.is_place_expr(allow_projections_from)
1484             }
1485
1486             // Lang item paths cannot currently be local variables or statics.
1487             ExprKind::Path(QPath::LangItem(..)) => false,
1488
1489             // Partially qualified paths in expressions can only legally
1490             // refer to associated items which are always rvalues.
1491             ExprKind::Path(QPath::TypeRelative(..))
1492             | ExprKind::Call(..)
1493             | ExprKind::MethodCall(..)
1494             | ExprKind::Struct(..)
1495             | ExprKind::Tup(..)
1496             | ExprKind::If(..)
1497             | ExprKind::Match(..)
1498             | ExprKind::Closure(..)
1499             | ExprKind::Block(..)
1500             | ExprKind::Repeat(..)
1501             | ExprKind::Array(..)
1502             | ExprKind::Break(..)
1503             | ExprKind::Continue(..)
1504             | ExprKind::Ret(..)
1505             | ExprKind::Loop(..)
1506             | ExprKind::Assign(..)
1507             | ExprKind::InlineAsm(..)
1508             | ExprKind::LlvmInlineAsm(..)
1509             | ExprKind::AssignOp(..)
1510             | ExprKind::Lit(_)
1511             | ExprKind::ConstBlock(..)
1512             | ExprKind::Unary(..)
1513             | ExprKind::Box(..)
1514             | ExprKind::AddrOf(..)
1515             | ExprKind::Binary(..)
1516             | ExprKind::Yield(..)
1517             | ExprKind::Cast(..)
1518             | ExprKind::DropTemps(..)
1519             | ExprKind::Err => false,
1520         }
1521     }
1522
1523     /// If `Self.kind` is `ExprKind::DropTemps(expr)`, drill down until we get a non-`DropTemps`
1524     /// `Expr`. This is used in suggestions to ignore this `ExprKind` as it is semantically
1525     /// silent, only signaling the ownership system. By doing this, suggestions that check the
1526     /// `ExprKind` of any given `Expr` for presentation don't have to care about `DropTemps`
1527     /// beyond remembering to call this function before doing analysis on it.
1528     pub fn peel_drop_temps(&self) -> &Self {
1529         let mut expr = self;
1530         while let ExprKind::DropTemps(inner) = &expr.kind {
1531             expr = inner;
1532         }
1533         expr
1534     }
1535 }
1536
1537 /// Checks if the specified expression is a built-in range literal.
1538 /// (See: `LoweringContext::lower_expr()`).
1539 pub fn is_range_literal(expr: &Expr<'_>) -> bool {
1540     match expr.kind {
1541         // All built-in range literals but `..=` and `..` desugar to `Struct`s.
1542         ExprKind::Struct(ref qpath, _, _) => matches!(
1543             **qpath,
1544             QPath::LangItem(
1545                 LangItem::Range
1546                     | LangItem::RangeTo
1547                     | LangItem::RangeFrom
1548                     | LangItem::RangeFull
1549                     | LangItem::RangeToInclusive,
1550                 _,
1551             )
1552         ),
1553
1554         // `..=` desugars into `::std::ops::RangeInclusive::new(...)`.
1555         ExprKind::Call(ref func, _) => {
1556             matches!(func.kind, ExprKind::Path(QPath::LangItem(LangItem::RangeInclusiveNew, _)))
1557         }
1558
1559         _ => false,
1560     }
1561 }
1562
1563 #[derive(Debug, HashStable_Generic)]
1564 pub enum ExprKind<'hir> {
1565     /// A `box x` expression.
1566     Box(&'hir Expr<'hir>),
1567     /// Allow anonymous constants from an inline `const` block
1568     ConstBlock(AnonConst),
1569     /// An array (e.g., `[a, b, c, d]`).
1570     Array(&'hir [Expr<'hir>]),
1571     /// A function call.
1572     ///
1573     /// The first field resolves to the function itself (usually an `ExprKind::Path`),
1574     /// and the second field is the list of arguments.
1575     /// This also represents calling the constructor of
1576     /// tuple-like ADTs such as tuple structs and enum variants.
1577     Call(&'hir Expr<'hir>, &'hir [Expr<'hir>]),
1578     /// A method call (e.g., `x.foo::<'static, Bar, Baz>(a, b, c, d)`).
1579     ///
1580     /// The `PathSegment`/`Span` represent the method name and its generic arguments
1581     /// (within the angle brackets).
1582     /// The first element of the vector of `Expr`s is the expression that evaluates
1583     /// to the object on which the method is being called on (the receiver),
1584     /// and the remaining elements are the rest of the arguments.
1585     /// Thus, `x.foo::<Bar, Baz>(a, b, c, d)` is represented as
1586     /// `ExprKind::MethodCall(PathSegment { foo, [Bar, Baz] }, [x, a, b, c, d])`.
1587     /// The final `Span` represents the span of the function and arguments
1588     /// (e.g. `foo::<Bar, Baz>(a, b, c, d)` in `x.foo::<Bar, Baz>(a, b, c, d)`
1589     ///
1590     /// To resolve the called method to a `DefId`, call [`type_dependent_def_id`] with
1591     /// the `hir_id` of the `MethodCall` node itself.
1592     ///
1593     /// [`type_dependent_def_id`]: ../ty/struct.TypeckResults.html#method.type_dependent_def_id
1594     MethodCall(&'hir PathSegment<'hir>, Span, &'hir [Expr<'hir>], Span),
1595     /// A tuple (e.g., `(a, b, c, d)`).
1596     Tup(&'hir [Expr<'hir>]),
1597     /// A binary operation (e.g., `a + b`, `a * b`).
1598     Binary(BinOp, &'hir Expr<'hir>, &'hir Expr<'hir>),
1599     /// A unary operation (e.g., `!x`, `*x`).
1600     Unary(UnOp, &'hir Expr<'hir>),
1601     /// A literal (e.g., `1`, `"foo"`).
1602     Lit(Lit),
1603     /// A cast (e.g., `foo as f64`).
1604     Cast(&'hir Expr<'hir>, &'hir Ty<'hir>),
1605     /// A type reference (e.g., `Foo`).
1606     Type(&'hir Expr<'hir>, &'hir Ty<'hir>),
1607     /// Wraps the expression in a terminating scope.
1608     /// This makes it semantically equivalent to `{ let _t = expr; _t }`.
1609     ///
1610     /// This construct only exists to tweak the drop order in HIR lowering.
1611     /// An example of that is the desugaring of `for` loops.
1612     DropTemps(&'hir Expr<'hir>),
1613     /// An `if` block, with an optional else block.
1614     ///
1615     /// I.e., `if <expr> { <expr> } else { <expr> }`.
1616     If(&'hir Expr<'hir>, &'hir Expr<'hir>, Option<&'hir Expr<'hir>>),
1617     /// A conditionless loop (can be exited with `break`, `continue`, or `return`).
1618     ///
1619     /// I.e., `'label: loop { <block> }`.
1620     ///
1621     /// The `Span` is the loop header (`for x in y`/`while let pat = expr`).
1622     Loop(&'hir Block<'hir>, Option<Label>, LoopSource, Span),
1623     /// A `match` block, with a source that indicates whether or not it is
1624     /// the result of a desugaring, and if so, which kind.
1625     Match(&'hir Expr<'hir>, &'hir [Arm<'hir>], MatchSource),
1626     /// A closure (e.g., `move |a, b, c| {a + b + c}`).
1627     ///
1628     /// The `Span` is the argument block `|...|`.
1629     ///
1630     /// This may also be a generator literal or an `async block` as indicated by the
1631     /// `Option<Movability>`.
1632     Closure(CaptureBy, &'hir FnDecl<'hir>, BodyId, Span, Option<Movability>),
1633     /// A block (e.g., `'label: { ... }`).
1634     Block(&'hir Block<'hir>, Option<Label>),
1635
1636     /// An assignment (e.g., `a = foo()`).
1637     Assign(&'hir Expr<'hir>, &'hir Expr<'hir>, Span),
1638     /// An assignment with an operator.
1639     ///
1640     /// E.g., `a += 1`.
1641     AssignOp(BinOp, &'hir Expr<'hir>, &'hir Expr<'hir>),
1642     /// Access of a named (e.g., `obj.foo`) or unnamed (e.g., `obj.0`) struct or tuple field.
1643     Field(&'hir Expr<'hir>, Ident),
1644     /// An indexing operation (`foo[2]`).
1645     Index(&'hir Expr<'hir>, &'hir Expr<'hir>),
1646
1647     /// Path to a definition, possibly containing lifetime or type parameters.
1648     Path(QPath<'hir>),
1649
1650     /// A referencing operation (i.e., `&a` or `&mut a`).
1651     AddrOf(BorrowKind, Mutability, &'hir Expr<'hir>),
1652     /// A `break`, with an optional label to break.
1653     Break(Destination, Option<&'hir Expr<'hir>>),
1654     /// A `continue`, with an optional label.
1655     Continue(Destination),
1656     /// A `return`, with an optional value to be returned.
1657     Ret(Option<&'hir Expr<'hir>>),
1658
1659     /// Inline assembly (from `asm!`), with its outputs and inputs.
1660     InlineAsm(&'hir InlineAsm<'hir>),
1661     /// Inline assembly (from `llvm_asm!`), with its outputs and inputs.
1662     LlvmInlineAsm(&'hir LlvmInlineAsm<'hir>),
1663
1664     /// A struct or struct-like variant literal expression.
1665     ///
1666     /// E.g., `Foo {x: 1, y: 2}`, or `Foo {x: 1, .. base}`,
1667     /// where `base` is the `Option<Expr>`.
1668     Struct(&'hir QPath<'hir>, &'hir [Field<'hir>], Option<&'hir Expr<'hir>>),
1669
1670     /// An array literal constructed from one repeated element.
1671     ///
1672     /// E.g., `[1; 5]`. The first expression is the element
1673     /// to be repeated; the second is the number of times to repeat it.
1674     Repeat(&'hir Expr<'hir>, AnonConst),
1675
1676     /// A suspension point for generators (i.e., `yield <expr>`).
1677     Yield(&'hir Expr<'hir>, YieldSource),
1678
1679     /// A placeholder for an expression that wasn't syntactically well formed in some way.
1680     Err,
1681 }
1682
1683 /// Represents an optionally `Self`-qualified value/type path or associated extension.
1684 ///
1685 /// To resolve the path to a `DefId`, call [`qpath_res`].
1686 ///
1687 /// [`qpath_res`]: ../rustc_middle/ty/struct.TypeckResults.html#method.qpath_res
1688 #[derive(Debug, HashStable_Generic)]
1689 pub enum QPath<'hir> {
1690     /// Path to a definition, optionally "fully-qualified" with a `Self`
1691     /// type, if the path points to an associated item in a trait.
1692     ///
1693     /// E.g., an unqualified path like `Clone::clone` has `None` for `Self`,
1694     /// while `<Vec<T> as Clone>::clone` has `Some(Vec<T>)` for `Self`,
1695     /// even though they both have the same two-segment `Clone::clone` `Path`.
1696     Resolved(Option<&'hir Ty<'hir>>, &'hir Path<'hir>),
1697
1698     /// Type-related paths (e.g., `<T>::default` or `<T>::Output`).
1699     /// Will be resolved by type-checking to an associated item.
1700     ///
1701     /// UFCS source paths can desugar into this, with `Vec::new` turning into
1702     /// `<Vec>::new`, and `T::X::Y::method` into `<<<T>::X>::Y>::method`,
1703     /// the `X` and `Y` nodes each being a `TyKind::Path(QPath::TypeRelative(..))`.
1704     TypeRelative(&'hir Ty<'hir>, &'hir PathSegment<'hir>),
1705
1706     /// Reference to a `#[lang = "foo"]` item.
1707     LangItem(LangItem, Span),
1708 }
1709
1710 impl<'hir> QPath<'hir> {
1711     /// Returns the span of this `QPath`.
1712     pub fn span(&self) -> Span {
1713         match *self {
1714             QPath::Resolved(_, path) => path.span,
1715             QPath::TypeRelative(_, ps) => ps.ident.span,
1716             QPath::LangItem(_, span) => span,
1717         }
1718     }
1719
1720     /// Returns the span of the qself of this `QPath`. For example, `()` in
1721     /// `<() as Trait>::method`.
1722     pub fn qself_span(&self) -> Span {
1723         match *self {
1724             QPath::Resolved(_, path) => path.span,
1725             QPath::TypeRelative(qself, _) => qself.span,
1726             QPath::LangItem(_, span) => span,
1727         }
1728     }
1729
1730     /// Returns the span of the last segment of this `QPath`. For example, `method` in
1731     /// `<() as Trait>::method`.
1732     pub fn last_segment_span(&self) -> Span {
1733         match *self {
1734             QPath::Resolved(_, path) => path.segments.last().unwrap().ident.span,
1735             QPath::TypeRelative(_, segment) => segment.ident.span,
1736             QPath::LangItem(_, span) => span,
1737         }
1738     }
1739 }
1740
1741 /// Hints at the original code for a let statement.
1742 #[derive(Copy, Clone, Encodable, Debug, HashStable_Generic)]
1743 pub enum LocalSource {
1744     /// A `match _ { .. }`.
1745     Normal,
1746     /// A desugared `for _ in _ { .. }` loop.
1747     ForLoopDesugar,
1748     /// When lowering async functions, we create locals within the `async move` so that
1749     /// all parameters are dropped after the future is polled.
1750     ///
1751     /// ```ignore (pseudo-Rust)
1752     /// async fn foo(<pattern> @ x: Type) {
1753     ///     async move {
1754     ///         let <pattern> = x;
1755     ///     }
1756     /// }
1757     /// ```
1758     AsyncFn,
1759     /// A desugared `<expr>.await`.
1760     AwaitDesugar,
1761     /// A desugared `expr = expr`, where the LHS is a tuple, struct or array.
1762     /// The span is that of the `=` sign.
1763     AssignDesugar(Span),
1764 }
1765
1766 /// Hints at the original code for a `match _ { .. }`.
1767 #[derive(Copy, Clone, PartialEq, Eq, Encodable, Hash, Debug)]
1768 #[derive(HashStable_Generic)]
1769 pub enum MatchSource {
1770     /// A `match _ { .. }`.
1771     Normal,
1772     /// An `if let _ = _ { .. }` (optionally with `else { .. }`).
1773     IfLetDesugar { contains_else_clause: bool },
1774     /// An `if let _ = _ => { .. }` match guard.
1775     IfLetGuardDesugar,
1776     /// A `while _ { .. }` (which was desugared to a `loop { match _ { .. } }`).
1777     WhileDesugar,
1778     /// A `while let _ = _ { .. }` (which was desugared to a
1779     /// `loop { match _ { .. } }`).
1780     WhileLetDesugar,
1781     /// A desugared `for _ in _ { .. }` loop.
1782     ForLoopDesugar,
1783     /// A desugared `?` operator.
1784     TryDesugar,
1785     /// A desugared `<expr>.await`.
1786     AwaitDesugar,
1787 }
1788
1789 impl MatchSource {
1790     pub fn name(self) -> &'static str {
1791         use MatchSource::*;
1792         match self {
1793             Normal => "match",
1794             IfLetDesugar { .. } | IfLetGuardDesugar => "if",
1795             WhileDesugar | WhileLetDesugar => "while",
1796             ForLoopDesugar => "for",
1797             TryDesugar => "?",
1798             AwaitDesugar => ".await",
1799         }
1800     }
1801 }
1802
1803 /// The loop type that yielded an `ExprKind::Loop`.
1804 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
1805 pub enum LoopSource {
1806     /// A `loop { .. }` loop.
1807     Loop,
1808     /// A `while _ { .. }` loop.
1809     While,
1810     /// A `while let _ = _ { .. }` loop.
1811     WhileLet,
1812     /// A `for _ in _ { .. }` loop.
1813     ForLoop,
1814 }
1815
1816 impl LoopSource {
1817     pub fn name(self) -> &'static str {
1818         match self {
1819             LoopSource::Loop => "loop",
1820             LoopSource::While | LoopSource::WhileLet => "while",
1821             LoopSource::ForLoop => "for",
1822         }
1823     }
1824 }
1825
1826 #[derive(Copy, Clone, Encodable, Debug, HashStable_Generic)]
1827 pub enum LoopIdError {
1828     OutsideLoopScope,
1829     UnlabeledCfInWhileCondition,
1830     UnresolvedLabel,
1831 }
1832
1833 impl fmt::Display for LoopIdError {
1834     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1835         f.write_str(match self {
1836             LoopIdError::OutsideLoopScope => "not inside loop scope",
1837             LoopIdError::UnlabeledCfInWhileCondition => {
1838                 "unlabeled control flow (break or continue) in while condition"
1839             }
1840             LoopIdError::UnresolvedLabel => "label not found",
1841         })
1842     }
1843 }
1844
1845 #[derive(Copy, Clone, Encodable, Debug, HashStable_Generic)]
1846 pub struct Destination {
1847     // This is `Some(_)` iff there is an explicit user-specified `label
1848     pub label: Option<Label>,
1849
1850     // These errors are caught and then reported during the diagnostics pass in
1851     // librustc_passes/loops.rs
1852     pub target_id: Result<HirId, LoopIdError>,
1853 }
1854
1855 /// The yield kind that caused an `ExprKind::Yield`.
1856 #[derive(Copy, Clone, PartialEq, Eq, Debug, Encodable, Decodable, HashStable_Generic)]
1857 pub enum YieldSource {
1858     /// An `<expr>.await`.
1859     Await { expr: Option<HirId> },
1860     /// A plain `yield`.
1861     Yield,
1862 }
1863
1864 impl YieldSource {
1865     pub fn is_await(&self) -> bool {
1866         match self {
1867             YieldSource::Await { .. } => true,
1868             YieldSource::Yield => false,
1869         }
1870     }
1871 }
1872
1873 impl fmt::Display for YieldSource {
1874     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1875         f.write_str(match self {
1876             YieldSource::Await { .. } => "`await`",
1877             YieldSource::Yield => "`yield`",
1878         })
1879     }
1880 }
1881
1882 impl From<GeneratorKind> for YieldSource {
1883     fn from(kind: GeneratorKind) -> Self {
1884         match kind {
1885             // Guess based on the kind of the current generator.
1886             GeneratorKind::Gen => Self::Yield,
1887             GeneratorKind::Async(_) => Self::Await { expr: None },
1888         }
1889     }
1890 }
1891
1892 // N.B., if you change this, you'll probably want to change the corresponding
1893 // type structure in middle/ty.rs as well.
1894 #[derive(Debug, HashStable_Generic)]
1895 pub struct MutTy<'hir> {
1896     pub ty: &'hir Ty<'hir>,
1897     pub mutbl: Mutability,
1898 }
1899
1900 /// Represents a function's signature in a trait declaration,
1901 /// trait implementation, or a free function.
1902 #[derive(Debug, HashStable_Generic)]
1903 pub struct FnSig<'hir> {
1904     pub header: FnHeader,
1905     pub decl: &'hir FnDecl<'hir>,
1906     pub span: Span,
1907 }
1908
1909 // The bodies for items are stored "out of line", in a separate
1910 // hashmap in the `Crate`. Here we just record the hir-id of the item
1911 // so it can fetched later.
1912 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Encodable, Debug)]
1913 pub struct TraitItemId {
1914     pub hir_id: HirId,
1915 }
1916
1917 /// Represents an item declaration within a trait declaration,
1918 /// possibly including a default implementation. A trait item is
1919 /// either required (meaning it doesn't have an implementation, just a
1920 /// signature) or provided (meaning it has a default implementation).
1921 #[derive(Debug)]
1922 pub struct TraitItem<'hir> {
1923     pub ident: Ident,
1924     pub hir_id: HirId,
1925     pub attrs: &'hir [Attribute],
1926     pub generics: Generics<'hir>,
1927     pub kind: TraitItemKind<'hir>,
1928     pub span: Span,
1929 }
1930
1931 /// Represents a trait method's body (or just argument names).
1932 #[derive(Encodable, Debug, HashStable_Generic)]
1933 pub enum TraitFn<'hir> {
1934     /// No default body in the trait, just a signature.
1935     Required(&'hir [Ident]),
1936
1937     /// Both signature and body are provided in the trait.
1938     Provided(BodyId),
1939 }
1940
1941 /// Represents a trait method or associated constant or type
1942 #[derive(Debug, HashStable_Generic)]
1943 pub enum TraitItemKind<'hir> {
1944     /// An associated constant with an optional value (otherwise `impl`s must contain a value).
1945     Const(&'hir Ty<'hir>, Option<BodyId>),
1946     /// An associated function with an optional body.
1947     Fn(FnSig<'hir>, TraitFn<'hir>),
1948     /// An associated type with (possibly empty) bounds and optional concrete
1949     /// type.
1950     Type(GenericBounds<'hir>, Option<&'hir Ty<'hir>>),
1951 }
1952
1953 // The bodies for items are stored "out of line", in a separate
1954 // hashmap in the `Crate`. Here we just record the hir-id of the item
1955 // so it can fetched later.
1956 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Encodable, Debug)]
1957 pub struct ImplItemId {
1958     pub hir_id: HirId,
1959 }
1960
1961 /// Represents anything within an `impl` block.
1962 #[derive(Debug)]
1963 pub struct ImplItem<'hir> {
1964     pub ident: Ident,
1965     pub hir_id: HirId,
1966     pub vis: Visibility<'hir>,
1967     pub defaultness: Defaultness,
1968     pub attrs: &'hir [Attribute],
1969     pub generics: Generics<'hir>,
1970     pub kind: ImplItemKind<'hir>,
1971     pub span: Span,
1972 }
1973
1974 /// Represents various kinds of content within an `impl`.
1975 #[derive(Debug, HashStable_Generic)]
1976 pub enum ImplItemKind<'hir> {
1977     /// An associated constant of the given type, set to the constant result
1978     /// of the expression.
1979     Const(&'hir Ty<'hir>, BodyId),
1980     /// An associated function implementation with the given signature and body.
1981     Fn(FnSig<'hir>, BodyId),
1982     /// An associated type.
1983     TyAlias(&'hir Ty<'hir>),
1984 }
1985
1986 impl ImplItemKind<'_> {
1987     pub fn namespace(&self) -> Namespace {
1988         match self {
1989             ImplItemKind::TyAlias(..) => Namespace::TypeNS,
1990             ImplItemKind::Const(..) | ImplItemKind::Fn(..) => Namespace::ValueNS,
1991         }
1992     }
1993 }
1994
1995 // The name of the associated type for `Fn` return types.
1996 pub const FN_OUTPUT_NAME: Symbol = sym::Output;
1997
1998 /// Bind a type to an associated type (i.e., `A = Foo`).
1999 ///
2000 /// Bindings like `A: Debug` are represented as a special type `A =
2001 /// $::Debug` that is understood by the astconv code.
2002 ///
2003 /// FIXME(alexreg): why have a separate type for the binding case,
2004 /// wouldn't it be better to make the `ty` field an enum like the
2005 /// following?
2006 ///
2007 /// ```
2008 /// enum TypeBindingKind {
2009 ///    Equals(...),
2010 ///    Binding(...),
2011 /// }
2012 /// ```
2013 #[derive(Debug, HashStable_Generic)]
2014 pub struct TypeBinding<'hir> {
2015     pub hir_id: HirId,
2016     #[stable_hasher(project(name))]
2017     pub ident: Ident,
2018     pub gen_args: &'hir GenericArgs<'hir>,
2019     pub kind: TypeBindingKind<'hir>,
2020     pub span: Span,
2021 }
2022
2023 // Represents the two kinds of type bindings.
2024 #[derive(Debug, HashStable_Generic)]
2025 pub enum TypeBindingKind<'hir> {
2026     /// E.g., `Foo<Bar: Send>`.
2027     Constraint { bounds: &'hir [GenericBound<'hir>] },
2028     /// E.g., `Foo<Bar = ()>`.
2029     Equality { ty: &'hir Ty<'hir> },
2030 }
2031
2032 impl TypeBinding<'_> {
2033     pub fn ty(&self) -> &Ty<'_> {
2034         match self.kind {
2035             TypeBindingKind::Equality { ref ty } => ty,
2036             _ => panic!("expected equality type binding for parenthesized generic args"),
2037         }
2038     }
2039 }
2040
2041 #[derive(Debug)]
2042 pub struct Ty<'hir> {
2043     pub hir_id: HirId,
2044     pub kind: TyKind<'hir>,
2045     pub span: Span,
2046 }
2047
2048 /// Not represented directly in the AST; referred to by name through a `ty_path`.
2049 #[derive(Copy, Clone, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
2050 #[derive(HashStable_Generic)]
2051 pub enum PrimTy {
2052     Int(IntTy),
2053     Uint(UintTy),
2054     Float(FloatTy),
2055     Str,
2056     Bool,
2057     Char,
2058 }
2059
2060 impl PrimTy {
2061     pub fn name_str(self) -> &'static str {
2062         match self {
2063             PrimTy::Int(i) => i.name_str(),
2064             PrimTy::Uint(u) => u.name_str(),
2065             PrimTy::Float(f) => f.name_str(),
2066             PrimTy::Str => "str",
2067             PrimTy::Bool => "bool",
2068             PrimTy::Char => "char",
2069         }
2070     }
2071
2072     pub fn name(self) -> Symbol {
2073         match self {
2074             PrimTy::Int(i) => i.name(),
2075             PrimTy::Uint(u) => u.name(),
2076             PrimTy::Float(f) => f.name(),
2077             PrimTy::Str => sym::str,
2078             PrimTy::Bool => sym::bool,
2079             PrimTy::Char => sym::char,
2080         }
2081     }
2082 }
2083
2084 #[derive(Debug, HashStable_Generic)]
2085 pub struct BareFnTy<'hir> {
2086     pub unsafety: Unsafety,
2087     pub abi: Abi,
2088     pub generic_params: &'hir [GenericParam<'hir>],
2089     pub decl: &'hir FnDecl<'hir>,
2090     pub param_names: &'hir [Ident],
2091 }
2092
2093 #[derive(Debug, HashStable_Generic)]
2094 pub struct OpaqueTy<'hir> {
2095     pub generics: Generics<'hir>,
2096     pub bounds: GenericBounds<'hir>,
2097     pub impl_trait_fn: Option<DefId>,
2098     pub origin: OpaqueTyOrigin,
2099 }
2100
2101 /// From whence the opaque type came.
2102 #[derive(Copy, Clone, Encodable, Decodable, Debug, HashStable_Generic)]
2103 pub enum OpaqueTyOrigin {
2104     /// `-> impl Trait`
2105     FnReturn,
2106     /// `async fn`
2107     AsyncFn,
2108     /// `let _: impl Trait = ...`
2109     Binding,
2110     /// Impl trait in type aliases, consts, statics, bounds.
2111     Misc,
2112 }
2113
2114 /// The various kinds of types recognized by the compiler.
2115 #[derive(Debug, HashStable_Generic)]
2116 pub enum TyKind<'hir> {
2117     /// A variable length slice (i.e., `[T]`).
2118     Slice(&'hir Ty<'hir>),
2119     /// A fixed length array (i.e., `[T; n]`).
2120     Array(&'hir Ty<'hir>, AnonConst),
2121     /// A raw pointer (i.e., `*const T` or `*mut T`).
2122     Ptr(MutTy<'hir>),
2123     /// A reference (i.e., `&'a T` or `&'a mut T`).
2124     Rptr(Lifetime, MutTy<'hir>),
2125     /// A bare function (e.g., `fn(usize) -> bool`).
2126     BareFn(&'hir BareFnTy<'hir>),
2127     /// The never type (`!`).
2128     Never,
2129     /// A tuple (`(A, B, C, D, ...)`).
2130     Tup(&'hir [Ty<'hir>]),
2131     /// A path to a type definition (`module::module::...::Type`), or an
2132     /// associated type (e.g., `<Vec<T> as Trait>::Type` or `<T>::Target`).
2133     ///
2134     /// Type parameters may be stored in each `PathSegment`.
2135     Path(QPath<'hir>),
2136     /// A opaque type definition itself. This is currently only used for the
2137     /// `opaque type Foo: Trait` item that `impl Trait` in desugars to.
2138     ///
2139     /// The generic argument list contains the lifetimes (and in the future
2140     /// possibly parameters) that are actually bound on the `impl Trait`.
2141     OpaqueDef(ItemId, &'hir [GenericArg<'hir>]),
2142     /// A trait object type `Bound1 + Bound2 + Bound3`
2143     /// where `Bound` is a trait or a lifetime.
2144     TraitObject(&'hir [PolyTraitRef<'hir>], Lifetime),
2145     /// Unused for now.
2146     Typeof(AnonConst),
2147     /// `TyKind::Infer` means the type should be inferred instead of it having been
2148     /// specified. This can appear anywhere in a type.
2149     Infer,
2150     /// Placeholder for a type that has failed to be defined.
2151     Err,
2152 }
2153
2154 #[derive(Debug, HashStable_Generic)]
2155 pub enum InlineAsmOperand<'hir> {
2156     In {
2157         reg: InlineAsmRegOrRegClass,
2158         expr: Expr<'hir>,
2159     },
2160     Out {
2161         reg: InlineAsmRegOrRegClass,
2162         late: bool,
2163         expr: Option<Expr<'hir>>,
2164     },
2165     InOut {
2166         reg: InlineAsmRegOrRegClass,
2167         late: bool,
2168         expr: Expr<'hir>,
2169     },
2170     SplitInOut {
2171         reg: InlineAsmRegOrRegClass,
2172         late: bool,
2173         in_expr: Expr<'hir>,
2174         out_expr: Option<Expr<'hir>>,
2175     },
2176     Const {
2177         expr: Expr<'hir>,
2178     },
2179     Sym {
2180         expr: Expr<'hir>,
2181     },
2182 }
2183
2184 impl<'hir> InlineAsmOperand<'hir> {
2185     pub fn reg(&self) -> Option<InlineAsmRegOrRegClass> {
2186         match *self {
2187             Self::In { reg, .. }
2188             | Self::Out { reg, .. }
2189             | Self::InOut { reg, .. }
2190             | Self::SplitInOut { reg, .. } => Some(reg),
2191             Self::Const { .. } | Self::Sym { .. } => None,
2192         }
2193     }
2194 }
2195
2196 #[derive(Debug, HashStable_Generic)]
2197 pub struct InlineAsm<'hir> {
2198     pub template: &'hir [InlineAsmTemplatePiece],
2199     pub operands: &'hir [(InlineAsmOperand<'hir>, Span)],
2200     pub options: InlineAsmOptions,
2201     pub line_spans: &'hir [Span],
2202 }
2203
2204 #[derive(Copy, Clone, Encodable, Decodable, Debug, HashStable_Generic, PartialEq)]
2205 pub struct LlvmInlineAsmOutput {
2206     pub constraint: Symbol,
2207     pub is_rw: bool,
2208     pub is_indirect: bool,
2209     pub span: Span,
2210 }
2211
2212 // NOTE(eddyb) This is used within MIR as well, so unlike the rest of the HIR,
2213 // it needs to be `Clone` and `Decodable` and use plain `Vec<T>` instead of
2214 // arena-allocated slice.
2215 #[derive(Clone, Encodable, Decodable, Debug, HashStable_Generic, PartialEq)]
2216 pub struct LlvmInlineAsmInner {
2217     pub asm: Symbol,
2218     pub asm_str_style: StrStyle,
2219     pub outputs: Vec<LlvmInlineAsmOutput>,
2220     pub inputs: Vec<Symbol>,
2221     pub clobbers: Vec<Symbol>,
2222     pub volatile: bool,
2223     pub alignstack: bool,
2224     pub dialect: LlvmAsmDialect,
2225 }
2226
2227 #[derive(Debug, HashStable_Generic)]
2228 pub struct LlvmInlineAsm<'hir> {
2229     pub inner: LlvmInlineAsmInner,
2230     pub outputs_exprs: &'hir [Expr<'hir>],
2231     pub inputs_exprs: &'hir [Expr<'hir>],
2232 }
2233
2234 /// Represents a parameter in a function header.
2235 #[derive(Debug, HashStable_Generic)]
2236 pub struct Param<'hir> {
2237     pub attrs: &'hir [Attribute],
2238     pub hir_id: HirId,
2239     pub pat: &'hir Pat<'hir>,
2240     pub ty_span: Span,
2241     pub span: Span,
2242 }
2243
2244 /// Represents the header (not the body) of a function declaration.
2245 #[derive(Debug, HashStable_Generic)]
2246 pub struct FnDecl<'hir> {
2247     /// The types of the function's parameters.
2248     ///
2249     /// Additional argument data is stored in the function's [body](Body::params).
2250     pub inputs: &'hir [Ty<'hir>],
2251     pub output: FnRetTy<'hir>,
2252     pub c_variadic: bool,
2253     /// Does the function have an implicit self?
2254     pub implicit_self: ImplicitSelfKind,
2255 }
2256
2257 /// Represents what type of implicit self a function has, if any.
2258 #[derive(Copy, Clone, Encodable, Decodable, Debug, HashStable_Generic)]
2259 pub enum ImplicitSelfKind {
2260     /// Represents a `fn x(self);`.
2261     Imm,
2262     /// Represents a `fn x(mut self);`.
2263     Mut,
2264     /// Represents a `fn x(&self);`.
2265     ImmRef,
2266     /// Represents a `fn x(&mut self);`.
2267     MutRef,
2268     /// Represents when a function does not have a self argument or
2269     /// when a function has a `self: X` argument.
2270     None,
2271 }
2272
2273 impl ImplicitSelfKind {
2274     /// Does this represent an implicit self?
2275     pub fn has_implicit_self(&self) -> bool {
2276         !matches!(*self, ImplicitSelfKind::None)
2277     }
2278 }
2279
2280 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Encodable, Decodable, Debug)]
2281 #[derive(HashStable_Generic)]
2282 pub enum IsAsync {
2283     Async,
2284     NotAsync,
2285 }
2286
2287 #[derive(Copy, Clone, PartialEq, Eq, Hash, Debug, Encodable, Decodable, HashStable_Generic)]
2288 pub enum Defaultness {
2289     Default { has_value: bool },
2290     Final,
2291 }
2292
2293 impl Defaultness {
2294     pub fn has_value(&self) -> bool {
2295         match *self {
2296             Defaultness::Default { has_value } => has_value,
2297             Defaultness::Final => true,
2298         }
2299     }
2300
2301     pub fn is_final(&self) -> bool {
2302         *self == Defaultness::Final
2303     }
2304
2305     pub fn is_default(&self) -> bool {
2306         matches!(*self, Defaultness::Default { .. })
2307     }
2308 }
2309
2310 #[derive(Debug, HashStable_Generic)]
2311 pub enum FnRetTy<'hir> {
2312     /// Return type is not specified.
2313     ///
2314     /// Functions default to `()` and
2315     /// closures default to inference. Span points to where return
2316     /// type would be inserted.
2317     DefaultReturn(Span),
2318     /// Everything else.
2319     Return(&'hir Ty<'hir>),
2320 }
2321
2322 impl FnRetTy<'_> {
2323     pub fn span(&self) -> Span {
2324         match *self {
2325             Self::DefaultReturn(span) => span,
2326             Self::Return(ref ty) => ty.span,
2327         }
2328     }
2329 }
2330
2331 #[derive(Encodable, Debug)]
2332 pub struct Mod<'hir> {
2333     /// A span from the first token past `{` to the last token until `}`.
2334     /// For `mod foo;`, the inner span ranges from the first token
2335     /// to the last token in the external file.
2336     pub inner: Span,
2337     pub item_ids: &'hir [ItemId],
2338 }
2339
2340 #[derive(Encodable, Debug, HashStable_Generic)]
2341 pub struct GlobalAsm {
2342     pub asm: Symbol,
2343 }
2344
2345 #[derive(Debug, HashStable_Generic)]
2346 pub struct EnumDef<'hir> {
2347     pub variants: &'hir [Variant<'hir>],
2348 }
2349
2350 #[derive(Debug, HashStable_Generic)]
2351 pub struct Variant<'hir> {
2352     /// Name of the variant.
2353     #[stable_hasher(project(name))]
2354     pub ident: Ident,
2355     /// Attributes of the variant.
2356     pub attrs: &'hir [Attribute],
2357     /// Id of the variant (not the constructor, see `VariantData::ctor_hir_id()`).
2358     pub id: HirId,
2359     /// Fields and constructor id of the variant.
2360     pub data: VariantData<'hir>,
2361     /// Explicit discriminant (e.g., `Foo = 1`).
2362     pub disr_expr: Option<AnonConst>,
2363     /// Span
2364     pub span: Span,
2365 }
2366
2367 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
2368 pub enum UseKind {
2369     /// One import, e.g., `use foo::bar` or `use foo::bar as baz`.
2370     /// Also produced for each element of a list `use`, e.g.
2371     /// `use foo::{a, b}` lowers to `use foo::a; use foo::b;`.
2372     Single,
2373
2374     /// Glob import, e.g., `use foo::*`.
2375     Glob,
2376
2377     /// Degenerate list import, e.g., `use foo::{a, b}` produces
2378     /// an additional `use foo::{}` for performing checks such as
2379     /// unstable feature gating. May be removed in the future.
2380     ListStem,
2381 }
2382
2383 /// References to traits in impls.
2384 ///
2385 /// `resolve` maps each `TraitRef`'s `ref_id` to its defining trait; that's all
2386 /// that the `ref_id` is for. Note that `ref_id`'s value is not the `HirId` of the
2387 /// trait being referred to but just a unique `HirId` that serves as a key
2388 /// within the resolution map.
2389 #[derive(Debug, HashStable_Generic)]
2390 pub struct TraitRef<'hir> {
2391     pub path: &'hir Path<'hir>,
2392     // Don't hash the `ref_id`. It is tracked via the thing it is used to access.
2393     #[stable_hasher(ignore)]
2394     pub hir_ref_id: HirId,
2395 }
2396
2397 impl TraitRef<'_> {
2398     /// Gets the `DefId` of the referenced trait. It _must_ actually be a trait or trait alias.
2399     pub fn trait_def_id(&self) -> Option<DefId> {
2400         match self.path.res {
2401             Res::Def(DefKind::Trait | DefKind::TraitAlias, did) => Some(did),
2402             Res::Err => None,
2403             _ => unreachable!(),
2404         }
2405     }
2406 }
2407
2408 #[derive(Debug, HashStable_Generic)]
2409 pub struct PolyTraitRef<'hir> {
2410     /// The `'a` in `for<'a> Foo<&'a T>`.
2411     pub bound_generic_params: &'hir [GenericParam<'hir>],
2412
2413     /// The `Foo<&'a T>` in `for<'a> Foo<&'a T>`.
2414     pub trait_ref: TraitRef<'hir>,
2415
2416     pub span: Span,
2417 }
2418
2419 pub type Visibility<'hir> = Spanned<VisibilityKind<'hir>>;
2420
2421 #[derive(Debug)]
2422 pub enum VisibilityKind<'hir> {
2423     Public,
2424     Crate(CrateSugar),
2425     Restricted { path: &'hir Path<'hir>, hir_id: HirId },
2426     Inherited,
2427 }
2428
2429 impl VisibilityKind<'_> {
2430     pub fn is_pub(&self) -> bool {
2431         matches!(*self, VisibilityKind::Public)
2432     }
2433
2434     pub fn is_pub_restricted(&self) -> bool {
2435         match *self {
2436             VisibilityKind::Public | VisibilityKind::Inherited => false,
2437             VisibilityKind::Crate(..) | VisibilityKind::Restricted { .. } => true,
2438         }
2439     }
2440 }
2441
2442 #[derive(Debug, HashStable_Generic)]
2443 pub struct StructField<'hir> {
2444     pub span: Span,
2445     #[stable_hasher(project(name))]
2446     pub ident: Ident,
2447     pub vis: Visibility<'hir>,
2448     pub hir_id: HirId,
2449     pub ty: &'hir Ty<'hir>,
2450     pub attrs: &'hir [Attribute],
2451 }
2452
2453 impl StructField<'_> {
2454     // Still necessary in couple of places
2455     pub fn is_positional(&self) -> bool {
2456         let first = self.ident.as_str().as_bytes()[0];
2457         (b'0'..=b'9').contains(&first)
2458     }
2459 }
2460
2461 /// Fields and constructor IDs of enum variants and structs.
2462 #[derive(Debug, HashStable_Generic)]
2463 pub enum VariantData<'hir> {
2464     /// A struct variant.
2465     ///
2466     /// E.g., `Bar { .. }` as in `enum Foo { Bar { .. } }`.
2467     Struct(&'hir [StructField<'hir>], /* recovered */ bool),
2468     /// A tuple variant.
2469     ///
2470     /// E.g., `Bar(..)` as in `enum Foo { Bar(..) }`.
2471     Tuple(&'hir [StructField<'hir>], HirId),
2472     /// A unit variant.
2473     ///
2474     /// E.g., `Bar = ..` as in `enum Foo { Bar = .. }`.
2475     Unit(HirId),
2476 }
2477
2478 impl VariantData<'hir> {
2479     /// Return the fields of this variant.
2480     pub fn fields(&self) -> &'hir [StructField<'hir>] {
2481         match *self {
2482             VariantData::Struct(ref fields, ..) | VariantData::Tuple(ref fields, ..) => fields,
2483             _ => &[],
2484         }
2485     }
2486
2487     /// Return the `HirId` of this variant's constructor, if it has one.
2488     pub fn ctor_hir_id(&self) -> Option<HirId> {
2489         match *self {
2490             VariantData::Struct(_, _) => None,
2491             VariantData::Tuple(_, hir_id) | VariantData::Unit(hir_id) => Some(hir_id),
2492         }
2493     }
2494 }
2495
2496 // The bodies for items are stored "out of line", in a separate
2497 // hashmap in the `Crate`. Here we just record the hir-id of the item
2498 // so it can fetched later.
2499 #[derive(Copy, Clone, Encodable, Debug)]
2500 pub struct ItemId {
2501     pub id: HirId,
2502 }
2503
2504 /// An item
2505 ///
2506 /// The name might be a dummy name in case of anonymous items
2507 #[derive(Debug)]
2508 pub struct Item<'hir> {
2509     pub ident: Ident,
2510     pub hir_id: HirId,
2511     pub attrs: &'hir [Attribute],
2512     pub kind: ItemKind<'hir>,
2513     pub vis: Visibility<'hir>,
2514     pub span: Span,
2515 }
2516
2517 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
2518 #[derive(Encodable, Decodable, HashStable_Generic)]
2519 pub enum Unsafety {
2520     Unsafe,
2521     Normal,
2522 }
2523
2524 impl Unsafety {
2525     pub fn prefix_str(&self) -> &'static str {
2526         match self {
2527             Self::Unsafe => "unsafe ",
2528             Self::Normal => "",
2529         }
2530     }
2531 }
2532
2533 impl fmt::Display for Unsafety {
2534     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2535         f.write_str(match *self {
2536             Self::Unsafe => "unsafe",
2537             Self::Normal => "normal",
2538         })
2539     }
2540 }
2541
2542 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
2543 #[derive(Encodable, Decodable, HashStable_Generic)]
2544 pub enum Constness {
2545     Const,
2546     NotConst,
2547 }
2548
2549 #[derive(Copy, Clone, Encodable, Debug, HashStable_Generic)]
2550 pub struct FnHeader {
2551     pub unsafety: Unsafety,
2552     pub constness: Constness,
2553     pub asyncness: IsAsync,
2554     pub abi: Abi,
2555 }
2556
2557 impl FnHeader {
2558     pub fn is_const(&self) -> bool {
2559         matches!(&self.constness, Constness::Const)
2560     }
2561 }
2562
2563 #[derive(Debug, HashStable_Generic)]
2564 pub enum ItemKind<'hir> {
2565     /// An `extern crate` item, with optional *original* crate name if the crate was renamed.
2566     ///
2567     /// E.g., `extern crate foo` or `extern crate foo_bar as foo`.
2568     ExternCrate(Option<Symbol>),
2569
2570     /// `use foo::bar::*;` or `use foo::bar::baz as quux;`
2571     ///
2572     /// or just
2573     ///
2574     /// `use foo::bar::baz;` (with `as baz` implicitly on the right).
2575     Use(&'hir Path<'hir>, UseKind),
2576
2577     /// A `static` item.
2578     Static(&'hir Ty<'hir>, Mutability, BodyId),
2579     /// A `const` item.
2580     Const(&'hir Ty<'hir>, BodyId),
2581     /// A function declaration.
2582     Fn(FnSig<'hir>, Generics<'hir>, BodyId),
2583     /// A module.
2584     Mod(Mod<'hir>),
2585     /// An external module, e.g. `extern { .. }`.
2586     ForeignMod { abi: Abi, items: &'hir [ForeignItemRef<'hir>] },
2587     /// Module-level inline assembly (from `global_asm!`).
2588     GlobalAsm(&'hir GlobalAsm),
2589     /// A type alias, e.g., `type Foo = Bar<u8>`.
2590     TyAlias(&'hir Ty<'hir>, Generics<'hir>),
2591     /// An opaque `impl Trait` type alias, e.g., `type Foo = impl Bar;`.
2592     OpaqueTy(OpaqueTy<'hir>),
2593     /// An enum definition, e.g., `enum Foo<A, B> {C<A>, D<B>}`.
2594     Enum(EnumDef<'hir>, Generics<'hir>),
2595     /// A struct definition, e.g., `struct Foo<A> {x: A}`.
2596     Struct(VariantData<'hir>, Generics<'hir>),
2597     /// A union definition, e.g., `union Foo<A, B> {x: A, y: B}`.
2598     Union(VariantData<'hir>, Generics<'hir>),
2599     /// A trait definition.
2600     Trait(IsAuto, Unsafety, Generics<'hir>, GenericBounds<'hir>, &'hir [TraitItemRef]),
2601     /// A trait alias.
2602     TraitAlias(Generics<'hir>, GenericBounds<'hir>),
2603
2604     /// An implementation, e.g., `impl<A> Trait for Foo { .. }`.
2605     Impl(Impl<'hir>),
2606 }
2607
2608 #[derive(Debug, HashStable_Generic)]
2609 pub struct Impl<'hir> {
2610     pub unsafety: Unsafety,
2611     pub polarity: ImplPolarity,
2612     pub defaultness: Defaultness,
2613     // We do not put a `Span` in `Defaultness` because it breaks foreign crate metadata
2614     // decoding as `Span`s cannot be decoded when a `Session` is not available.
2615     pub defaultness_span: Option<Span>,
2616     pub constness: Constness,
2617     pub generics: Generics<'hir>,
2618
2619     /// The trait being implemented, if any.
2620     pub of_trait: Option<TraitRef<'hir>>,
2621
2622     pub self_ty: &'hir Ty<'hir>,
2623     pub items: &'hir [ImplItemRef<'hir>],
2624 }
2625
2626 impl ItemKind<'_> {
2627     pub fn generics(&self) -> Option<&Generics<'_>> {
2628         Some(match *self {
2629             ItemKind::Fn(_, ref generics, _)
2630             | ItemKind::TyAlias(_, ref generics)
2631             | ItemKind::OpaqueTy(OpaqueTy { ref generics, impl_trait_fn: None, .. })
2632             | ItemKind::Enum(_, ref generics)
2633             | ItemKind::Struct(_, ref generics)
2634             | ItemKind::Union(_, ref generics)
2635             | ItemKind::Trait(_, _, ref generics, _, _)
2636             | ItemKind::Impl(Impl { ref generics, .. }) => generics,
2637             _ => return None,
2638         })
2639     }
2640 }
2641
2642 /// A reference from an trait to one of its associated items. This
2643 /// contains the item's id, naturally, but also the item's name and
2644 /// some other high-level details (like whether it is an associated
2645 /// type or method, and whether it is public). This allows other
2646 /// passes to find the impl they want without loading the ID (which
2647 /// means fewer edges in the incremental compilation graph).
2648 #[derive(Encodable, Debug, HashStable_Generic)]
2649 pub struct TraitItemRef {
2650     pub id: TraitItemId,
2651     #[stable_hasher(project(name))]
2652     pub ident: Ident,
2653     pub kind: AssocItemKind,
2654     pub span: Span,
2655     pub defaultness: Defaultness,
2656 }
2657
2658 /// A reference from an impl to one of its associated items. This
2659 /// contains the item's ID, naturally, but also the item's name and
2660 /// some other high-level details (like whether it is an associated
2661 /// type or method, and whether it is public). This allows other
2662 /// passes to find the impl they want without loading the ID (which
2663 /// means fewer edges in the incremental compilation graph).
2664 #[derive(Debug, HashStable_Generic)]
2665 pub struct ImplItemRef<'hir> {
2666     pub id: ImplItemId,
2667     #[stable_hasher(project(name))]
2668     pub ident: Ident,
2669     pub kind: AssocItemKind,
2670     pub span: Span,
2671     pub vis: Visibility<'hir>,
2672     pub defaultness: Defaultness,
2673 }
2674
2675 #[derive(Copy, Clone, PartialEq, Encodable, Debug, HashStable_Generic)]
2676 pub enum AssocItemKind {
2677     Const,
2678     Fn { has_self: bool },
2679     Type,
2680 }
2681
2682 // The bodies for items are stored "out of line", in a separate
2683 // hashmap in the `Crate`. Here we just record the hir-id of the item
2684 // so it can fetched later.
2685 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Encodable, Debug)]
2686 pub struct ForeignItemId {
2687     pub hir_id: HirId,
2688 }
2689
2690 /// A reference from a foreign block to one of its items. This
2691 /// contains the item's ID, naturally, but also the item's name and
2692 /// some other high-level details (like whether it is an associated
2693 /// type or method, and whether it is public). This allows other
2694 /// passes to find the impl they want without loading the ID (which
2695 /// means fewer edges in the incremental compilation graph).
2696 #[derive(Debug, HashStable_Generic)]
2697 pub struct ForeignItemRef<'hir> {
2698     pub id: ForeignItemId,
2699     #[stable_hasher(project(name))]
2700     pub ident: Ident,
2701     pub span: Span,
2702     pub vis: Visibility<'hir>,
2703 }
2704
2705 #[derive(Debug, HashStable_Generic)]
2706 pub struct ForeignItem<'hir> {
2707     #[stable_hasher(project(name))]
2708     pub ident: Ident,
2709     pub attrs: &'hir [Attribute],
2710     pub kind: ForeignItemKind<'hir>,
2711     pub hir_id: HirId,
2712     pub span: Span,
2713     pub vis: Visibility<'hir>,
2714 }
2715
2716 /// An item within an `extern` block.
2717 #[derive(Debug, HashStable_Generic)]
2718 pub enum ForeignItemKind<'hir> {
2719     /// A foreign function.
2720     Fn(&'hir FnDecl<'hir>, &'hir [Ident], Generics<'hir>),
2721     /// A foreign static item (`static ext: u8`).
2722     Static(&'hir Ty<'hir>, Mutability),
2723     /// A foreign type.
2724     Type,
2725 }
2726
2727 /// A variable captured by a closure.
2728 #[derive(Debug, Copy, Clone, Encodable, HashStable_Generic)]
2729 pub struct Upvar {
2730     // First span where it is accessed (there can be multiple).
2731     pub span: Span,
2732 }
2733
2734 // The TraitCandidate's import_ids is empty if the trait is defined in the same module, and
2735 // has length > 0 if the trait is found through an chain of imports, starting with the
2736 // import/use statement in the scope where the trait is used.
2737 #[derive(Encodable, Decodable, Clone, Debug)]
2738 pub struct TraitCandidate {
2739     pub def_id: DefId,
2740     pub import_ids: SmallVec<[LocalDefId; 1]>,
2741 }
2742
2743 #[derive(Copy, Clone, Debug, HashStable_Generic)]
2744 pub enum Node<'hir> {
2745     Param(&'hir Param<'hir>),
2746     Item(&'hir Item<'hir>),
2747     ForeignItem(&'hir ForeignItem<'hir>),
2748     TraitItem(&'hir TraitItem<'hir>),
2749     ImplItem(&'hir ImplItem<'hir>),
2750     Variant(&'hir Variant<'hir>),
2751     Field(&'hir StructField<'hir>),
2752     AnonConst(&'hir AnonConst),
2753     Expr(&'hir Expr<'hir>),
2754     Stmt(&'hir Stmt<'hir>),
2755     PathSegment(&'hir PathSegment<'hir>),
2756     Ty(&'hir Ty<'hir>),
2757     TraitRef(&'hir TraitRef<'hir>),
2758     Binding(&'hir Pat<'hir>),
2759     Pat(&'hir Pat<'hir>),
2760     Arm(&'hir Arm<'hir>),
2761     Block(&'hir Block<'hir>),
2762     Local(&'hir Local<'hir>),
2763     MacroDef(&'hir MacroDef<'hir>),
2764
2765     /// `Ctor` refers to the constructor of an enum variant or struct. Only tuple or unit variants
2766     /// with synthesized constructors.
2767     Ctor(&'hir VariantData<'hir>),
2768
2769     Lifetime(&'hir Lifetime),
2770     GenericParam(&'hir GenericParam<'hir>),
2771     Visibility(&'hir Visibility<'hir>),
2772
2773     Crate(&'hir CrateItem<'hir>),
2774 }
2775
2776 impl<'hir> Node<'hir> {
2777     pub fn ident(&self) -> Option<Ident> {
2778         match self {
2779             Node::TraitItem(TraitItem { ident, .. })
2780             | Node::ImplItem(ImplItem { ident, .. })
2781             | Node::ForeignItem(ForeignItem { ident, .. })
2782             | Node::Field(StructField { ident, .. })
2783             | Node::Variant(Variant { ident, .. })
2784             | Node::MacroDef(MacroDef { ident, .. })
2785             | Node::Item(Item { ident, .. }) => Some(*ident),
2786             _ => None,
2787         }
2788     }
2789
2790     pub fn fn_decl(&self) -> Option<&FnDecl<'hir>> {
2791         match self {
2792             Node::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
2793             | Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
2794             | Node::Item(Item { kind: ItemKind::Fn(fn_sig, _, _), .. }) => Some(fn_sig.decl),
2795             Node::ForeignItem(ForeignItem { kind: ForeignItemKind::Fn(fn_decl, _, _), .. }) => {
2796                 Some(fn_decl)
2797             }
2798             _ => None,
2799         }
2800     }
2801
2802     pub fn body_id(&self) -> Option<BodyId> {
2803         match self {
2804             Node::TraitItem(TraitItem {
2805                 kind: TraitItemKind::Fn(_, TraitFn::Provided(body_id)),
2806                 ..
2807             })
2808             | Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(_, body_id), .. })
2809             | Node::Item(Item { kind: ItemKind::Fn(.., body_id), .. }) => Some(*body_id),
2810             _ => None,
2811         }
2812     }
2813
2814     pub fn generics(&self) -> Option<&'hir Generics<'hir>> {
2815         match self {
2816             Node::TraitItem(TraitItem { generics, .. })
2817             | Node::ImplItem(ImplItem { generics, .. }) => Some(generics),
2818             Node::Item(item) => item.kind.generics(),
2819             _ => None,
2820         }
2821     }
2822
2823     pub fn hir_id(&self) -> Option<HirId> {
2824         match self {
2825             Node::Item(Item { hir_id, .. })
2826             | Node::ForeignItem(ForeignItem { hir_id, .. })
2827             | Node::TraitItem(TraitItem { hir_id, .. })
2828             | Node::ImplItem(ImplItem { hir_id, .. })
2829             | Node::Field(StructField { hir_id, .. })
2830             | Node::AnonConst(AnonConst { hir_id, .. })
2831             | Node::Expr(Expr { hir_id, .. })
2832             | Node::Stmt(Stmt { hir_id, .. })
2833             | Node::Ty(Ty { hir_id, .. })
2834             | Node::Binding(Pat { hir_id, .. })
2835             | Node::Pat(Pat { hir_id, .. })
2836             | Node::Arm(Arm { hir_id, .. })
2837             | Node::Block(Block { hir_id, .. })
2838             | Node::Local(Local { hir_id, .. })
2839             | Node::MacroDef(MacroDef { hir_id, .. })
2840             | Node::Lifetime(Lifetime { hir_id, .. })
2841             | Node::Param(Param { hir_id, .. })
2842             | Node::GenericParam(GenericParam { hir_id, .. }) => Some(*hir_id),
2843             Node::TraitRef(TraitRef { hir_ref_id, .. }) => Some(*hir_ref_id),
2844             Node::PathSegment(PathSegment { hir_id, .. }) => *hir_id,
2845             Node::Variant(Variant { id, .. }) => Some(*id),
2846             Node::Ctor(variant) => variant.ctor_hir_id(),
2847             Node::Crate(_) | Node::Visibility(_) => None,
2848         }
2849     }
2850 }