1 use crate::def_id::{DefId, CRATE_DEF_INDEX, LOCAL_CRATE};
6 use rustc_macros::HashStable_Generic;
7 use rustc_span::hygiene::MacroKind;
8 use rustc_span::Symbol;
10 use std::array::IntoIter;
13 /// Encodes if a `DefKind::Ctor` is the constructor of an enum variant or a struct.
14 #[derive(Clone, Copy, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
15 #[derive(HashStable_Generic)]
17 /// This `DefKind::Ctor` is a synthesized constructor of a tuple or unit struct.
19 /// This `DefKind::Ctor` is a synthesized constructor of a tuple or unit variant.
23 /// What kind of constructor something is.
24 #[derive(Clone, Copy, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
25 #[derive(HashStable_Generic)]
27 /// Constructor function automatically created by a tuple struct/variant.
29 /// Constructor constant automatically created by a unit struct/variant.
31 /// Unusable name in value namespace created by a struct variant.
35 /// An attribute that is not a macro; e.g., `#[inline]` or `#[rustfmt::skip]`.
36 #[derive(Clone, Copy, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
37 #[derive(HashStable_Generic)]
38 pub enum NonMacroAttrKind {
39 /// Single-segment attribute defined by the language (`#[inline]`)
41 /// Multi-segment custom attribute living in a "tool module" (`#[rustfmt::skip]`).
43 /// Single-segment custom attribute registered by a derive macro (`#[serde(default)]`).
45 /// Single-segment custom attribute registered by a derive macro
46 /// but used before that derive macro was expanded (deprecated).
48 /// Single-segment custom attribute registered with `#[register_attr]`.
52 /// What kind of definition something is; e.g., `mod` vs `struct`.
53 #[derive(Clone, Copy, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
54 #[derive(HashStable_Generic)]
58 /// Refers to the struct itself, [`DefKind::Ctor`] refers to its constructor if it exists.
62 /// Refers to the variant itself, [`DefKind::Ctor`] refers to its constructor if it exists.
65 /// Type alias: `type Foo = Bar;`
67 /// Type from an `extern` block.
69 /// Trait alias: `trait IntIterator = Iterator<Item = i32>;`
71 /// Associated type: `trait MyTrait { type Assoc; }`
73 /// Type parameter: the `T` in `struct Vec<T> { ... }`
79 /// Constant generic parameter: `struct Foo<const N: usize> { ... }`
81 Static(ast::Mutability),
82 /// Refers to the struct or enum variant's constructor.
84 /// The reason `Ctor` exists in addition to [`DefKind::Struct`] and
85 /// [`DefKind::Variant`] is because structs and enum variants exist
86 /// in the *type* namespace, whereas struct and enum variant *constructors*
87 /// exist in the *value* namespace.
89 /// You may wonder why enum variants exist in the type namespace as opposed
90 /// to the value namespace. Check out [RFC 2593] for intuition on why that is.
92 /// [RFC 2593]: https://github.com/rust-lang/rfcs/pull/2593
93 Ctor(CtorOf, CtorKind),
94 /// Associated function: `impl MyStruct { fn associated() {} }`
96 /// Associated constant: `trait MyTrait { const ASSOC: usize; }`
102 // Not namespaced (or they are, but we don't treat them so)
105 /// An `extern` block.
107 /// Anonymous constant, e.g. the `1 + 2` in `[u8; 1 + 2]`
109 /// An inline constant, e.g. `const { 1 + 2 }`
111 /// Opaque type, aka `impl Trait`.
114 /// Lifetime parameter: the `'a` in `struct Foo<'a> { ... }`
116 /// A use of `global_asm!`.
124 pub fn descr(self, def_id: DefId) -> &'static str {
126 DefKind::Fn => "function",
127 DefKind::Mod if def_id.index == CRATE_DEF_INDEX && def_id.krate != LOCAL_CRATE => {
130 DefKind::Mod => "module",
131 DefKind::Static(..) => "static",
132 DefKind::Enum => "enum",
133 DefKind::Variant => "variant",
134 DefKind::Ctor(CtorOf::Variant, CtorKind::Fn) => "tuple variant",
135 DefKind::Ctor(CtorOf::Variant, CtorKind::Const) => "unit variant",
136 DefKind::Ctor(CtorOf::Variant, CtorKind::Fictive) => "struct variant",
137 DefKind::Struct => "struct",
138 DefKind::Ctor(CtorOf::Struct, CtorKind::Fn) => "tuple struct",
139 DefKind::Ctor(CtorOf::Struct, CtorKind::Const) => "unit struct",
140 DefKind::Ctor(CtorOf::Struct, CtorKind::Fictive) => {
141 panic!("impossible struct constructor")
143 DefKind::OpaqueTy => "opaque type",
144 DefKind::TyAlias => "type alias",
145 DefKind::TraitAlias => "trait alias",
146 DefKind::AssocTy => "associated type",
147 DefKind::Union => "union",
148 DefKind::Trait => "trait",
149 DefKind::ForeignTy => "foreign type",
150 DefKind::AssocFn => "associated function",
151 DefKind::Const => "constant",
152 DefKind::AssocConst => "associated constant",
153 DefKind::TyParam => "type parameter",
154 DefKind::ConstParam => "const parameter",
155 DefKind::Macro(macro_kind) => macro_kind.descr(),
156 DefKind::LifetimeParam => "lifetime parameter",
157 DefKind::Use => "import",
158 DefKind::ForeignMod => "foreign module",
159 DefKind::AnonConst => "constant expression",
160 DefKind::InlineConst => "inline constant",
161 DefKind::Field => "field",
162 DefKind::Impl => "implementation",
163 DefKind::Closure => "closure",
164 DefKind::Generator => "generator",
165 DefKind::ExternCrate => "extern crate",
166 DefKind::GlobalAsm => "global assembly block",
170 /// Gets an English article for the definition.
171 pub fn article(&self) -> &'static str {
174 | DefKind::AssocConst
180 | DefKind::InlineConst
181 | DefKind::ExternCrate => "an",
182 DefKind::Macro(macro_kind) => macro_kind.article(),
187 pub fn ns(&self) -> Option<Namespace> {
198 | DefKind::TraitAlias
200 | DefKind::TyParam => Some(Namespace::TypeNS),
204 | DefKind::ConstParam
205 | DefKind::Static(..)
208 | DefKind::AssocConst => Some(Namespace::ValueNS),
210 DefKind::Macro(..) => Some(Namespace::MacroNS),
214 | DefKind::InlineConst
216 | DefKind::LifetimeParam
217 | DefKind::ExternCrate
221 | DefKind::ForeignMod
223 | DefKind::Impl => None,
228 pub fn is_fn_like(self) -> bool {
230 DefKind::Fn | DefKind::AssocFn | DefKind::Closure | DefKind::Generator => true,
236 /// The resolution of a path or export.
238 /// For every path or identifier in Rust, the compiler must determine
239 /// what the path refers to. This process is called name resolution,
240 /// and `Res` is the primary result of name resolution.
242 /// For example, everything prefixed with `/* Res */` in this example has
243 /// an associated `Res`:
246 /// fn str_to_string(s: & /* Res */ str) -> /* Res */ String {
247 /// /* Res */ String::from(/* Res */ s)
250 /// /* Res */ str_to_string("hello");
253 /// The associated `Res`s will be:
255 /// - `str` will resolve to [`Res::PrimTy`];
256 /// - `String` will resolve to [`Res::Def`], and the `Res` will include the [`DefId`]
257 /// for `String` as defined in the standard library;
258 /// - `String::from` will also resolve to [`Res::Def`], with the [`DefId`]
259 /// pointing to `String::from`;
260 /// - `s` will resolve to [`Res::Local`];
261 /// - the call to `str_to_string` will resolve to [`Res::Def`], with the [`DefId`]
262 /// pointing to the definition of `str_to_string` in the current crate.
264 #[derive(Clone, Copy, PartialEq, Eq, Encodable, Decodable, Hash, Debug)]
265 #[derive(HashStable_Generic)]
266 pub enum Res<Id = hir::HirId> {
267 /// Definition having a unique ID (`DefId`), corresponds to something defined in user code.
269 /// **Not bound to a specific namespace.**
273 /// A primitive type such as `i32` or `str`.
275 /// **Belongs to the type namespace.**
277 /// The `Self` type, optionally with the [`DefId`] of the trait it belongs to and
278 /// optionally with the [`DefId`] of the item introducing the `Self` type alias.
280 /// **Belongs to the type namespace.**
284 /// struct Bar(Box<Self>);
285 /// // `Res::SelfTy { trait_: None, alias_of: Some(Bar) }`
288 /// fn foo() -> Box<Self>;
289 /// // `Res::SelfTy { trait_: Some(Foo), alias_of: None }`
295 /// // `Res::SelfTy { trait_: None, alias_of: Some(::{impl#0}) }`
299 /// impl Foo for Bar {
300 /// fn foo() -> Box<Self> {
301 /// // `Res::SelfTy { trait_: Some(Foo), alias_of: Some(::{impl#1}) }`
303 /// // `Res::SelfTy { trait_: Some(Foo), alias_of: Some(::{impl#1}) }`
310 /// *See also [`Res::SelfCtor`].*
314 /// HACK(min_const_generics): self types also have an optional requirement to **not** mention
315 /// any generic parameters to allow the following with `min_const_generics`:
317 /// impl Foo { fn test() -> [u8; std::mem::size_of::<Self>()] { todo!() } }
319 /// struct Bar([u8; baz::<Self>()]);
320 /// const fn baz<T>() -> usize { 10 }
322 /// We do however allow `Self` in repeat expression even if it is generic to not break code
323 /// which already works on stable while causing the `const_evaluatable_unchecked` future compat lint:
326 /// let _bar = [1_u8; std::mem::size_of::<*mut T>()];
329 // FIXME(generic_const_exprs): Remove this bodge once that feature is stable.
331 /// The trait this `Self` is a generic arg for.
332 trait_: Option<DefId>,
333 /// The item introducing the `Self` type alias. Can be used in the `type_of` query
334 /// to get the underlying type. Additionally whether the `Self` type is disallowed
335 /// from mentioning generics (i.e. when used in an anonymous constant).
336 alias_to: Option<(DefId, bool)>,
338 /// A tool attribute module; e.g., the `rustfmt` in `#[rustfmt::skip]`.
340 /// **Belongs to the type namespace.**
344 /// The `Self` constructor, along with the [`DefId`]
345 /// of the impl it is associated with.
347 /// **Belongs to the value namespace.**
349 /// *See also [`Res::SelfTy`].*
351 /// A local variable or function parameter.
353 /// **Belongs to the value namespace.**
357 /// An attribute that is *not* implemented via macro.
358 /// E.g., `#[inline]` and `#[rustfmt::skip]`, which are essentially directives,
359 /// as opposed to `#[test]`, which is a builtin macro.
361 /// **Belongs to the macro namespace.**
362 NonMacroAttr(NonMacroAttrKind), // e.g., `#[inline]` or `#[rustfmt::skip]`
365 /// Name resolution failed. We use a dummy `Res` variant so later phases
366 /// of the compiler won't crash and can instead report more errors.
368 /// **Not bound to a specific namespace.**
372 /// The result of resolving a path before lowering to HIR,
373 /// with "module" segments resolved and associated item
374 /// segments deferred to type checking.
375 /// `base_res` is the resolution of the resolved part of the
376 /// path, `unresolved_segments` is the number of unresolved
380 /// module::Type::AssocX::AssocY::MethodOrAssocType
381 /// ^~~~~~~~~~~~ ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
382 /// base_res unresolved_segments = 3
384 /// <T as Trait>::AssocX::AssocY::MethodOrAssocType
385 /// ^~~~~~~~~~~~~~ ^~~~~~~~~~~~~~~~~~~~~~~~~
386 /// base_res unresolved_segments = 2
388 #[derive(Copy, Clone, Debug)]
389 pub struct PartialRes {
390 base_res: Res<NodeId>,
391 unresolved_segments: usize,
396 pub fn new(base_res: Res<NodeId>) -> Self {
397 PartialRes { base_res, unresolved_segments: 0 }
401 pub fn with_unresolved_segments(base_res: Res<NodeId>, mut unresolved_segments: usize) -> Self {
402 if base_res == Res::Err {
403 unresolved_segments = 0
405 PartialRes { base_res, unresolved_segments }
409 pub fn base_res(&self) -> Res<NodeId> {
414 pub fn unresolved_segments(&self) -> usize {
415 self.unresolved_segments
419 /// Different kinds of symbols can coexist even if they share the same textual name.
420 /// Therefore, they each have a separate universe (known as a "namespace").
421 #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, Debug)]
423 /// The type namespace includes `struct`s, `enum`s, `union`s, `trait`s, and `mod`s
424 /// (and, by extension, crates).
426 /// Note that the type namespace includes other items; this is not an
429 /// The value namespace includes `fn`s, `const`s, `static`s, and local variables (including function arguments).
431 /// The macro namespace includes `macro_rules!` macros, declarative `macro`s,
432 /// procedural macros, attribute macros, `derive` macros, and non-macro attributes
433 /// like `#[inline]` and `#[rustfmt::skip]`.
438 /// The English description of the namespace.
439 pub fn descr(self) -> &'static str {
441 Self::TypeNS => "type",
442 Self::ValueNS => "value",
443 Self::MacroNS => "macro",
448 /// Just a helper ‒ separate structure for each namespace.
449 #[derive(Copy, Clone, Default, Debug)]
450 pub struct PerNS<T> {
457 pub fn map<U, F: FnMut(T) -> U>(self, mut f: F) -> PerNS<U> {
458 PerNS { value_ns: f(self.value_ns), type_ns: f(self.type_ns), macro_ns: f(self.macro_ns) }
461 pub fn into_iter(self) -> IntoIter<T, 3> {
462 [self.value_ns, self.type_ns, self.macro_ns].into_iter()
465 pub fn iter(&self) -> IntoIter<&T, 3> {
466 [&self.value_ns, &self.type_ns, &self.macro_ns].into_iter()
470 impl<T> ::std::ops::Index<Namespace> for PerNS<T> {
473 fn index(&self, ns: Namespace) -> &T {
475 Namespace::ValueNS => &self.value_ns,
476 Namespace::TypeNS => &self.type_ns,
477 Namespace::MacroNS => &self.macro_ns,
482 impl<T> ::std::ops::IndexMut<Namespace> for PerNS<T> {
483 fn index_mut(&mut self, ns: Namespace) -> &mut T {
485 Namespace::ValueNS => &mut self.value_ns,
486 Namespace::TypeNS => &mut self.type_ns,
487 Namespace::MacroNS => &mut self.macro_ns,
492 impl<T> PerNS<Option<T>> {
493 /// Returns `true` if all the items in this collection are `None`.
494 pub fn is_empty(&self) -> bool {
495 self.type_ns.is_none() && self.value_ns.is_none() && self.macro_ns.is_none()
498 /// Returns an iterator over the items which are `Some`.
499 pub fn present_items(self) -> impl Iterator<Item = T> {
500 [self.type_ns, self.value_ns, self.macro_ns].into_iter().flatten()
505 pub fn from_ast(vdata: &ast::VariantData) -> CtorKind {
507 ast::VariantData::Tuple(..) => CtorKind::Fn,
508 ast::VariantData::Unit(..) => CtorKind::Const,
509 ast::VariantData::Struct(..) => CtorKind::Fictive,
513 pub fn from_hir(vdata: &hir::VariantData<'_>) -> CtorKind {
515 hir::VariantData::Tuple(..) => CtorKind::Fn,
516 hir::VariantData::Unit(..) => CtorKind::Const,
517 hir::VariantData::Struct(..) => CtorKind::Fictive,
522 impl NonMacroAttrKind {
523 pub fn descr(self) -> &'static str {
525 NonMacroAttrKind::Builtin(..) => "built-in attribute",
526 NonMacroAttrKind::Tool => "tool attribute",
527 NonMacroAttrKind::DeriveHelper | NonMacroAttrKind::DeriveHelperCompat => {
528 "derive helper attribute"
530 NonMacroAttrKind::Registered => "explicitly registered attribute",
534 pub fn article(self) -> &'static str {
536 NonMacroAttrKind::Registered => "an",
541 /// Users of some attributes cannot mark them as used, so they are considered always used.
542 pub fn is_used(self) -> bool {
544 NonMacroAttrKind::Tool
545 | NonMacroAttrKind::DeriveHelper
546 | NonMacroAttrKind::DeriveHelperCompat => true,
547 NonMacroAttrKind::Builtin(..) | NonMacroAttrKind::Registered => false,
553 /// Return the `DefId` of this `Def` if it has an ID, else panic.
554 pub fn def_id(&self) -> DefId
559 .unwrap_or_else(|| panic!("attempted .def_id() on invalid res: {:?}", self))
562 /// Return `Some(..)` with the `DefId` of this `Res` if it has a ID, else `None`.
563 pub fn opt_def_id(&self) -> Option<DefId> {
565 Res::Def(_, id) => Some(id),
572 | Res::NonMacroAttr(..)
577 /// Return the `DefId` of this `Res` if it represents a module.
578 pub fn mod_def_id(&self) -> Option<DefId> {
580 Res::Def(DefKind::Mod, id) => Some(id),
585 /// A human readable name for the res kind ("function", "module", etc.).
586 pub fn descr(&self) -> &'static str {
588 Res::Def(kind, def_id) => kind.descr(def_id),
589 Res::SelfCtor(..) => "self constructor",
590 Res::PrimTy(..) => "builtin type",
591 Res::Local(..) => "local variable",
592 Res::SelfTy { .. } => "self type",
593 Res::ToolMod => "tool module",
594 Res::NonMacroAttr(attr_kind) => attr_kind.descr(),
595 Res::Err => "unresolved item",
599 /// Gets an English article for the `Res`.
600 pub fn article(&self) -> &'static str {
602 Res::Def(kind, _) => kind.article(),
603 Res::NonMacroAttr(kind) => kind.article(),
609 pub fn map_id<R>(self, mut map: impl FnMut(Id) -> R) -> Res<R> {
611 Res::Def(kind, id) => Res::Def(kind, id),
612 Res::SelfCtor(id) => Res::SelfCtor(id),
613 Res::PrimTy(id) => Res::PrimTy(id),
614 Res::Local(id) => Res::Local(map(id)),
615 Res::SelfTy { trait_, alias_to } => Res::SelfTy { trait_, alias_to },
616 Res::ToolMod => Res::ToolMod,
617 Res::NonMacroAttr(attr_kind) => Res::NonMacroAttr(attr_kind),
618 Res::Err => Res::Err,
622 pub fn apply_id<R, E>(self, mut map: impl FnMut(Id) -> Result<R, E>) -> Result<Res<R>, E> {
624 Res::Def(kind, id) => Res::Def(kind, id),
625 Res::SelfCtor(id) => Res::SelfCtor(id),
626 Res::PrimTy(id) => Res::PrimTy(id),
627 Res::Local(id) => Res::Local(map(id)?),
628 Res::SelfTy { trait_, alias_to } => Res::SelfTy { trait_, alias_to },
629 Res::ToolMod => Res::ToolMod,
630 Res::NonMacroAttr(attr_kind) => Res::NonMacroAttr(attr_kind),
631 Res::Err => Res::Err,
636 pub fn expect_non_local<OtherId>(self) -> Res<OtherId> {
637 self.map_id(|_| panic!("unexpected `Res::Local`"))
640 pub fn macro_kind(self) -> Option<MacroKind> {
642 Res::Def(DefKind::Macro(kind), _) => Some(kind),
643 Res::NonMacroAttr(..) => Some(MacroKind::Attr),
648 /// Returns `None` if this is `Res::Err`
649 pub fn ns(&self) -> Option<Namespace> {
651 Res::Def(kind, ..) => kind.ns(),
652 Res::PrimTy(..) | Res::SelfTy { .. } | Res::ToolMod => Some(Namespace::TypeNS),
653 Res::SelfCtor(..) | Res::Local(..) => Some(Namespace::ValueNS),
654 Res::NonMacroAttr(..) => Some(Namespace::MacroNS),
659 /// Always returns `true` if `self` is `Res::Err`
660 pub fn matches_ns(&self, ns: Namespace) -> bool {
661 self.ns().map_or(true, |actual_ns| actual_ns == ns)
664 /// Returns whether such a resolved path can occur in a tuple struct/variant pattern
665 pub fn expected_in_tuple_struct_pat(&self) -> bool {
666 matches!(self, Res::Def(DefKind::Ctor(_, CtorKind::Fn), _) | Res::SelfCtor(..))