[PoC] Track type-dependent defs in ItemCtxts (minimally)

Author: fmease

compiler/rustc_hir_analysis/src/collect.rs

@@ -15,7 +15,7 @@
 //! crate as a kind of pass. This should eventually be factored away.
 
 use std::assert_matches::assert_matches;
-use std::cell::Cell;
+use std::cell::{Cell, RefCell};
 use std::iter;
 use std::ops::Bound;
 
@@ -33,6 +33,10 @@ use rustc_hir::{self as hir, GenericParamKind, HirId, Node, PreciseCapturingArgK
 use rustc_infer::infer::{InferCtxt, TyCtxtInferExt};
 use rustc_infer::traits::{DynCompatibilityViolation, ObligationCause};
 use rustc_middle::query::Providers;
+use rustc_middle::ty::typeck_results::{
+    HasTypeDependentDefs, LocalTableInContext, LocalTableInContextMut, TypeDependentDef,
+    TypeDependentDefs,
+};
 use rustc_middle::ty::util::{Discr, IntTypeExt};
 use rustc_middle::ty::{
     self, AdtKind, Const, IsSuggestable, Ty, TyCtxt, TypeVisitableExt, TypingMode, fold_regions,
@@ -132,6 +136,7 @@ pub(crate) fn provide(providers: &mut Providers) {
 pub(crate) struct ItemCtxt<'tcx> {
     tcx: TyCtxt<'tcx>,
     item_def_id: LocalDefId,
+    type_dependent_defs: RefCell<TypeDependentDefs>,
     tainted_by_errors: Cell<Option<ErrorGuaranteed>>,
 }
 
@@ -243,7 +248,12 @@ fn bad_placeholder<'cx, 'tcx>(
 
 impl<'tcx> ItemCtxt<'tcx> {
     pub(crate) fn new(tcx: TyCtxt<'tcx>, item_def_id: LocalDefId) -> ItemCtxt<'tcx> {
-        ItemCtxt { tcx, item_def_id, tainted_by_errors: Cell::new(None) }
+        ItemCtxt {
+            tcx,
+            item_def_id,
+            type_dependent_defs: Default::default(),
+            tainted_by_errors: Cell::new(None),
+        }
     }
 
     pub(crate) fn lower_ty(&self, hir_ty: &hir::Ty<'tcx>) -> Ty<'tcx> {
@@ -510,6 +520,14 @@ impl<'tcx> HirTyLowerer<'tcx> for ItemCtxt<'tcx> {
         // There's no place to record types from signatures?
     }
 
+    fn record_res(&self, hir_id: hir::HirId, result: TypeDependentDef) {
+        LocalTableInContextMut::new(
+            self.hir_id().owner,
+            &mut self.type_dependent_defs.borrow_mut(),
+        )
+        .insert(hir_id, result);
+    }
+
     fn infcx(&self) -> Option<&InferCtxt<'tcx>> {
         None
     }
@@ -568,6 +586,15 @@ impl<'tcx> HirTyLowerer<'tcx> for ItemCtxt<'tcx> {
     }
 }
 
+impl HasTypeDependentDefs for ItemCtxt<'_> {
+    fn type_dependent_def(&self, id: hir::HirId) -> Option<(DefKind, DefId)> {
+        LocalTableInContext::new(self.hir_id().owner, &self.type_dependent_defs.borrow())
+            .get(id)
+            .copied()
+            .and_then(|result| result.ok())
+    }
+}
+
 /// Synthesize a new lifetime name that doesn't clash with any of the lifetimes already present.
 fn get_new_lifetime_name<'tcx>(
     tcx: TyCtxt<'tcx>,

compiler/rustc_hir_analysis/src/hir_ty_lowering/mod.rs

@@ -37,6 +37,7 @@ use rustc_macros::{TypeFoldable, TypeVisitable};
 use rustc_middle::middle::stability::AllowUnstable;
 use rustc_middle::mir::interpret::LitToConstInput;
 use rustc_middle::ty::print::PrintPolyTraitRefExt as _;
+use rustc_middle::ty::typeck_results::{HasTypeDependentDefs, TypeDependentDef};
 use rustc_middle::ty::{
     self, Const, GenericArgKind, GenericArgsRef, GenericParamDefKind, Ty, TyCtxt, TypeVisitableExt,
     TypingMode, Upcast, fold_regions,
@@ -123,7 +124,7 @@ pub struct InherentAssocCandidate {
 /// the [`rustc_middle::ty`] representation.
 ///
 /// This trait used to be called `AstConv`.
-pub trait HirTyLowerer<'tcx> {
+pub trait HirTyLowerer<'tcx>: HasTypeDependentDefs {
     fn tcx(&self) -> TyCtxt<'tcx>;
 
     fn dcx(&self) -> DiagCtxtHandle<'_>;
@@ -214,6 +215,9 @@ pub trait HirTyLowerer<'tcx> {
     /// Record the lowered type of a HIR node in this context.
     fn record_ty(&self, hir_id: HirId, ty: Ty<'tcx>, span: Span);
 
+    /// Record the resolution of a HIR node corresponding to a type-dependent definition in this context.
+    fn record_res(&self, hir_id: hir::HirId, result: TypeDependentDef);
+
     /// The inference context of the lowering context if applicable.
     fn infcx(&self) -> Option<&InferCtxt<'tcx>>;
 
@@ -1233,6 +1237,7 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
     ///
     /// [#22519]: https://github.com/rust-lang/rust/issues/22519
     /// [iat]: https://github.com/rust-lang/rust/issues/8995#issuecomment-1569208403
+    // FIXME(fmease): Update docs
     //
     // NOTE: When this function starts resolving `Trait::AssocTy` successfully
     // it should also start reporting the `BARE_TRAIT_OBJECTS` lint.
@@ -1256,9 +1261,11 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
             LowerTypeRelativePathMode::Type(permit_variants),
         )? {
             TypeRelativePath::AssocItem(def_id, args) => {
+                let kind = tcx.def_kind(def_id);
+                self.record_res(qpath_hir_id, Ok((kind, def_id)));
                 let alias_ty = ty::AliasTy::new_from_args(tcx, def_id, args);
                 let ty = Ty::new_alias(tcx, alias_ty.kind(tcx), alias_ty);
-                Ok((ty, tcx.def_kind(def_id), def_id))
+                Ok((ty, kind, def_id))
             }
             TypeRelativePath::Variant { adt, variant_did } => {
                 Ok((adt, DefKind::Variant, variant_did))
@@ -1420,7 +1427,7 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
         let tcx = self.tcx();
 
         let self_ty_res = match hir_self_ty.kind {
-            hir::TyKind::Path(hir::QPath::Resolved(_, path)) => path.res,
+            hir::TyKind::Path(qpath) => self.qpath_res(&qpath, hir_self_ty.hir_id),
             _ => Res::Err,
         };
 
@@ -1454,14 +1461,7 @@ impl<'tcx> dyn HirTyLowerer<'tcx> + '_ {
                 segment.ident,
                 span,
             )?,
-            // FIXME(fmease):
-            // Require the pre-lowered projectee (the HIR QSelf) to have `DefKind::AssocTy`. Rephrased,
-            // `T::Assoc::Assoc` typeck'ing shouldn't imply `Identity<T::Assoc>::Assoc` typeck'ing where
-            // `Identity` is an eager (i.e., non-lazy) type alias. We should do this
-            // * for consistency with lazy type aliases (`ty::Weak`)
-            // * for consistency with the fact that `T::Assoc` typeck'ing doesn't imply `Identity<T>::Assoc`
-            //   typeck'ing
-            (ty::Alias(ty::Projection, alias_ty), _ /* Res::Def(DefKind::AssocTy, _) */) => {
+            (ty::Alias(ty::Projection, alias_ty), Res::Def(DefKind::AssocTy, _)) => {
                 // FIXME: Utilizing `item_bounds` for this is cycle-prone.
                 let predicates = tcx.item_bounds(alias_ty.def_id).instantiate(tcx, alias_ty.args);
 

compiler/rustc_hir_typeck/src/fn_ctxt/_impl.rs

@@ -171,9 +171,9 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
     pub(crate) fn write_resolution(
         &self,
         hir_id: HirId,
-        r: Result<(DefKind, DefId), ErrorGuaranteed>,
+        result: Result<(DefKind, DefId), ErrorGuaranteed>,
     ) {
-        self.typeck_results.borrow_mut().type_dependent_defs_mut().insert(hir_id, r);
+        self.typeck_results.borrow_mut().type_dependent_defs_mut().insert(hir_id, result);
     }
 
     #[instrument(level = "debug", skip(self))]

compiler/rustc_hir_typeck/src/fn_ctxt/checks.rs

@@ -1311,12 +1311,8 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                     .map(|(ty, _, _)| ty)
                     .unwrap_or_else(|guar| Ty::new_error(self.tcx(), guar));
                 let ty = LoweredTy::from_raw(self, path_span, ty);
-                let result = result.map(|(_, kind, def_id)| (kind, def_id));
 
-                // Write back the new resolution.
-                self.write_resolution(hir_id, result);
-
-                (result.map_or(Res::Err, |(kind, def_id)| Res::Def(kind, def_id)), ty)
+                (result.map_or(Res::Err, |(_, kind, def_id)| Res::Def(kind, def_id)), ty)
             }
         }
     }

compiler/rustc_hir_typeck/src/fn_ctxt/mod.rs

@@ -8,15 +8,16 @@ mod suggestions;
 use std::cell::{Cell, RefCell};
 use std::ops::Deref;
 
-use hir::def_id::CRATE_DEF_ID;
 use rustc_errors::DiagCtxtHandle;
-use rustc_hir::def_id::{DefId, LocalDefId};
+use rustc_hir::def::DefKind;
+use rustc_hir::def_id::{CRATE_DEF_ID, DefId, LocalDefId};
 use rustc_hir::{self as hir, HirId, ItemLocalMap};
 use rustc_hir_analysis::hir_ty_lowering::{
     HirTyLowerer, InherentAssocCandidate, RegionInferReason,
 };
 use rustc_infer::infer::{self, RegionVariableOrigin};
 use rustc_infer::traits::{DynCompatibilityViolation, Obligation};
+use rustc_middle::ty::typeck_results::{HasTypeDependentDefs, TypeDependentDef};
 use rustc_middle::ty::{self, Const, Ty, TyCtxt, TypeVisitableExt};
 use rustc_session::Session;
 use rustc_span::{self, DUMMY_SP, ErrorGuaranteed, Ident, Span, sym};
@@ -418,7 +419,7 @@ impl<'tcx> HirTyLowerer<'tcx> for FnCtxt<'_, 'tcx> {
         }
     }
 
-    fn record_ty(&self, hir_id: hir::HirId, ty: Ty<'tcx>, span: Span) {
+    fn record_ty(&self, hir_id: HirId, ty: Ty<'tcx>, span: Span) {
         // FIXME: normalization and escaping regions
         let ty = if !ty.has_escaping_bound_vars() {
             // NOTE: These obligations are 100% redundant and are implied by
@@ -438,6 +439,10 @@ impl<'tcx> HirTyLowerer<'tcx> for FnCtxt<'_, 'tcx> {
         self.write_ty(hir_id, ty)
     }
 
+    fn record_res(&self, hir_id: HirId, result: TypeDependentDef) {
+        self.write_resolution(hir_id, result);
+    }
+
     fn infcx(&self) -> Option<&infer::InferCtxt<'tcx>> {
         Some(&self.infcx)
     }
@@ -446,7 +451,7 @@ impl<'tcx> HirTyLowerer<'tcx> for FnCtxt<'_, 'tcx> {
         &self,
         decl: &rustc_hir::FnDecl<'tcx>,
         _generics: Option<&rustc_hir::Generics<'_>>,
-        _hir_id: rustc_hir::HirId,
+        _hir_id: HirId,
         _hir_ty: Option<&hir::Ty<'_>>,
     ) -> (Vec<Ty<'tcx>>, Ty<'tcx>) {
         let input_tys = decl.inputs.iter().map(|a| self.lowerer().lower_ty(a)).collect();
@@ -463,6 +468,12 @@ impl<'tcx> HirTyLowerer<'tcx> for FnCtxt<'_, 'tcx> {
     }
 }
 
+impl HasTypeDependentDefs for FnCtxt<'_, '_> {
+    fn type_dependent_def(&self, id: HirId) -> Option<(DefKind, DefId)> {
+        self.typeck_results.borrow().type_dependent_def(id)
+    }
+}
+
 /// The `ty` representation of a user-provided type. Depending on the use-site
 /// we want to either use the unnormalized or the normalized form of this type.
 ///

compiler/rustc_middle/src/ty/mod.rs

@@ -138,6 +138,7 @@ pub mod print;
 pub mod relate;
 pub mod significant_drop_order;
 pub mod trait_def;
+pub mod typeck_results;
 pub mod util;
 pub mod vtable;
 
@@ -162,7 +163,6 @@ mod region;
 mod structural_impls;
 #[allow(hidden_glob_reexports)]
 mod sty;
-mod typeck_results;
 mod visit;
 
 // Data types

compiler/rustc_middle/src/ty/typeck_results.rs

@@ -32,9 +32,7 @@ pub struct TypeckResults<'tcx> {
     /// The `HirId::owner` all `ItemLocalId`s in this table are relative to.
     pub hir_owner: OwnerId,
 
-    /// Resolved definitions for `<T>::X` associated paths and
-    /// method calls, including those of overloaded operators.
-    type_dependent_defs: ItemLocalMap<Result<(DefKind, DefId), ErrorGuaranteed>>,
+    type_dependent_defs: TypeDependentDefs,
 
     /// Resolved field indices for field accesses in expressions (`S { field }`, `obj.field`)
     /// or patterns (`S { field }`). The index is often useful by itself, but to learn more
@@ -258,32 +256,22 @@ impl<'tcx> TypeckResults<'tcx> {
 
     /// Returns the final resolution of a `QPath` in an `Expr` or `Pat` node.
     pub fn qpath_res(&self, qpath: &hir::QPath<'_>, id: HirId) -> Res {
-        match *qpath {
-            hir::QPath::Resolved(_, path) => path.res,
-            hir::QPath::TypeRelative(..) => self
-                .type_dependent_def(id)
-                .map_or(Res::Err, |(kind, def_id)| Res::Def(kind, def_id)),
-        }
+        HasTypeDependentDefs::qpath_res(self, qpath, id)
     }
 
-    pub fn type_dependent_defs(
-        &self,
-    ) -> LocalTableInContext<'_, Result<(DefKind, DefId), ErrorGuaranteed>> {
+    pub fn type_dependent_defs(&self) -> LocalTableInContext<'_, TypeDependentDef> {
         LocalTableInContext { hir_owner: self.hir_owner, data: &self.type_dependent_defs }
     }
 
     pub fn type_dependent_def(&self, id: HirId) -> Option<(DefKind, DefId)> {
-        validate_hir_id_for_typeck_results(self.hir_owner, id);
-        self.type_dependent_defs.get(&id.local_id).cloned().and_then(|r| r.ok())
+        self.type_dependent_defs().get(id).copied().and_then(|result| result.ok())
     }
 
     pub fn type_dependent_def_id(&self, id: HirId) -> Option<DefId> {
         self.type_dependent_def(id).map(|(_, def_id)| def_id)
     }
 
-    pub fn type_dependent_defs_mut(
-        &mut self,
-    ) -> LocalTableInContextMut<'_, Result<(DefKind, DefId), ErrorGuaranteed>> {
+    pub fn type_dependent_defs_mut(&mut self) -> LocalTableInContextMut<'_, TypeDependentDef> {
         LocalTableInContextMut { hir_owner: self.hir_owner, data: &mut self.type_dependent_defs }
     }
 
@@ -569,6 +557,33 @@ impl<'tcx> TypeckResults<'tcx> {
     }
 }
 
+/// Resolved definitions for `<T>::X` associated paths and
+/// method calls, including those of overloaded operators.
+pub type TypeDependentDefs = ItemLocalMap<TypeDependentDef>;
+
+pub type TypeDependentDef = Result<(DefKind, DefId), ErrorGuaranteed>;
+
+// FIXME(fmease): Yuck!
+pub trait HasTypeDependentDefs {
+    fn type_dependent_def(&self, id: HirId) -> Option<(DefKind, DefId)>;
+
+    /// Returns the final resolution of a `QPath`.
+    fn qpath_res(&self, qpath: &hir::QPath<'_>, id: HirId) -> Res {
+        match qpath {
+            hir::QPath::Resolved(_, path) => path.res,
+            hir::QPath::TypeRelative(..) => self
+                .type_dependent_def(id)
+                .map_or(Res::Err, |(kind, def_id)| Res::Def(kind, def_id)),
+        }
+    }
+}
+
+impl HasTypeDependentDefs for TypeckResults<'_> {
+    fn type_dependent_def(&self, id: HirId) -> Option<(DefKind, DefId)> {
+        self.type_dependent_def(id)
+    }
+}
+
 /// Validate that the given HirId (respectively its `local_id` part) can be
 /// safely used as a key in the maps of a TypeckResults. For that to be
 /// the case, the HirId must have the same `owner` as all the other IDs in
@@ -601,6 +616,10 @@ pub struct LocalTableInContext<'a, V> {
 }
 
 impl<'a, V> LocalTableInContext<'a, V> {
+    pub fn new(hir_owner: OwnerId, data: &'a ItemLocalMap<V>) -> Self {
+        Self { hir_owner, data }
+    }
+
     pub fn contains_key(&self, id: HirId) -> bool {
         validate_hir_id_for_typeck_results(self.hir_owner, id);
         self.data.contains_key(&id.local_id)
@@ -639,6 +658,10 @@ pub struct LocalTableInContextMut<'a, V> {
 }
 
 impl<'a, V> LocalTableInContextMut<'a, V> {
+    pub fn new(hir_owner: OwnerId, data: &'a mut ItemLocalMap<V>) -> Self {
+        Self { hir_owner, data }
+    }
+
     pub fn get_mut(&mut self, id: HirId) -> Option<&mut V> {
         validate_hir_id_for_typeck_results(self.hir_owner, id);
         self.data.get_mut(&id.local_id)

tests/ui/associated-types/non-consecutive-shorthand-projections.eager.stderr

@@ -0,0 +1,15 @@
+error[E0223]: ambiguous associated type
+  --> $DIR/non-consecutive-shorthand-projections.rs:14:12
+   |
+LL |     let _: Identity<T::Project>::Project;
+   |            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+   |
+help: if there were a trait named `Example` with associated type `Project` implemented for `<T as Trait>::Project`, you could use the fully-qualified path
+   |
+LL -     let _: Identity<T::Project>::Project;
+LL +     let _: <<T as Trait>::Project as Example>::Project;
+   |
+
+error: aborting due to 1 previous error
+
+For more information about this error, try `rustc --explain E0223`.

tests/ui/associated-types/non-consecutive-shorthand-projections.lazy.stderr

@@ -1,5 +1,5 @@
 error[E0223]: ambiguous associated type
-  --> $DIR/non-consecutive-shorthand-projections.rs:16:12
+  --> $DIR/non-consecutive-shorthand-projections.rs:14:12
    |
 LL |     let _: Identity<T::Project>::Project;
    |            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

tests/ui/associated-types/non-consecutive-shorthand-projections.rs

@@ -2,7 +2,6 @@
 // Of course, under #22519 (arbitrary shorthand projections), this should obviously typeck.
 // For reference, `T::Alias` typeck'ing does *not* imply `Identity<T>::Alias` typeck'ing.
 //@ revisions: eager lazy
-//@[eager] check-pass
 #![cfg_attr(lazy, feature(lazy_type_alias), allow(incomplete_features))]
 
 type Identity<T> = T;
@@ -12,8 +11,7 @@ trait Trait {
 }
 
 fn scope<T: Trait>() {
-    // FIXME(fmease): Reject this under [eager], too!
-    let _: Identity<T::Project>::Project; //[lazy]~ ERROR ambiguous associated type
+    let _: Identity<T::Project>::Project; //~ ERROR ambiguous associated type
 }
 
 fn main() {}