Fix pretty hir for anon consts in diagnostics

compiler-errors · compiler-errors · commit 619e04417841 · 2024-04-15T18:48:12.000-04:00
diff --git a/Cargo.lock b/Cargo.lock
@@ -4278,6 +4278,7 @@ dependencies = [
  "rustc_fluent_macro",
  "rustc_graphviz",
  "rustc_hir",
+ "rustc_hir_pretty",
  "rustc_index",
  "rustc_macros",
  "rustc_query_system",
diff --git a/compiler/rustc_driver_impl/src/pretty.rs b/compiler/rustc_driver_impl/src/pretty.rs
@@ -24,20 +24,6 @@ struct AstNoAnn;
 
 impl pprust_ast::PpAnn for AstNoAnn {}
 
-struct HirNoAnn<'tcx> {
-    tcx: TyCtxt<'tcx>,
-}
-
-impl<'tcx> pprust_hir::PpAnn for HirNoAnn<'tcx> {
-    fn nested(&self, state: &mut pprust_hir::State<'_>, nested: pprust_hir::Nested) {
-        pprust_hir::PpAnn::nested(
-            &(&self.tcx.hir() as &dyn hir::intravisit::Map<'_>),
-            state,
-            nested,
-        )
-    }
-}
-
 struct AstIdentifiedAnn;
 
 impl pprust_ast::PpAnn for AstIdentifiedAnn {
@@ -300,10 +286,7 @@ pub fn print<'tcx>(sess: &Session, ppm: PpMode, ex: PrintExtra<'tcx>) {
                 )
             };
             match s {
-                PpHirMode::Normal => {
-                    let annotation = HirNoAnn { tcx };
-                    f(&annotation)
-                }
+                PpHirMode::Normal => f(&tcx),
                 PpHirMode::Identified => {
                     let annotation = HirIdentifiedAnn { tcx };
                     f(&annotation)
diff --git a/compiler/rustc_hir_analysis/src/check/errs.rs b/compiler/rustc_hir_analysis/src/check/errs.rs
@@ -12,7 +12,7 @@ pub fn maybe_expr_static_mut(tcx: TyCtxt<'_>, expr: hir::Expr<'_>) {
     let hir_id = expr.hir_id;
     if let hir::ExprKind::AddrOf(borrow_kind, m, expr) = expr.kind
         && matches!(borrow_kind, hir::BorrowKind::Ref)
-        && let Some(var) = is_path_static_mut(*expr)
+        && let Some(var) = path_if_static_mut(tcx, expr)
     {
         handle_static_mut_ref(tcx, span, var, span.edition().at_least_rust_2024(), m, hir_id);
     }
@@ -24,7 +24,7 @@ pub fn maybe_stmt_static_mut(tcx: TyCtxt<'_>, stmt: hir::Stmt<'_>) {
         && let hir::PatKind::Binding(ba, _, _, _) = loc.pat.kind
         && let hir::ByRef::Yes(rmutbl) = ba.0
         && let Some(init) = loc.init
-        && let Some(var) = is_path_static_mut(*init)
+        && let Some(var) = path_if_static_mut(tcx, init)
     {
         handle_static_mut_ref(
             tcx,
@@ -37,13 +37,13 @@ pub fn maybe_stmt_static_mut(tcx: TyCtxt<'_>, stmt: hir::Stmt<'_>) {
     }
 }
 
-fn is_path_static_mut(expr: hir::Expr<'_>) -> Option<String> {
+fn path_if_static_mut(tcx: TyCtxt<'_>, expr: &hir::Expr<'_>) -> Option<String> {
     if let hir::ExprKind::Path(qpath) = expr.kind
         && let hir::QPath::Resolved(_, path) = qpath
         && let hir::def::Res::Def(def_kind, _) = path.res
         && let hir::def::DefKind::Static { mutability: Mutability::Mut, nested: false } = def_kind
     {
-        return Some(qpath_to_string(&qpath));
+        return Some(qpath_to_string(&tcx, &qpath));
     }
     None
 }
diff --git a/compiler/rustc_hir_pretty/src/lib.rs b/compiler/rustc_hir_pretty/src/lib.rs
@@ -49,10 +49,6 @@ pub trait PpAnn {
     fn post(&self, _state: &mut State<'_>, _node: AnnNode<'_>) {}
 }
 
-pub struct NoAnn;
-
-impl PpAnn for NoAnn {}
-
 impl PpAnn for &dyn rustc_hir::intravisit::Map<'_> {
     fn nested(&self, state: &mut State<'_>, nested: Nested) {
         match nested {
@@ -190,16 +186,16 @@ where
     printer.s.eof()
 }
 
-pub fn ty_to_string(ty: &hir::Ty<'_>) -> String {
-    to_string(&NoAnn, |s| s.print_type(ty))
+pub fn ty_to_string(ann: &dyn PpAnn, ty: &hir::Ty<'_>) -> String {
+    to_string(ann, |s| s.print_type(ty))
 }
 
-pub fn qpath_to_string(segment: &hir::QPath<'_>) -> String {
-    to_string(&NoAnn, |s| s.print_qpath(segment, false))
+pub fn qpath_to_string(ann: &dyn PpAnn, segment: &hir::QPath<'_>) -> String {
+    to_string(ann, |s| s.print_qpath(segment, false))
 }
 
-pub fn pat_to_string(pat: &hir::Pat<'_>) -> String {
-    to_string(&NoAnn, |s| s.print_pat(pat))
+pub fn pat_to_string(ann: &dyn PpAnn, pat: &hir::Pat<'_>) -> String {
+    to_string(ann, |s| s.print_pat(pat))
 }
 
 impl<'a> State<'a> {
diff --git a/compiler/rustc_hir_typeck/src/_match.rs b/compiler/rustc_hir_typeck/src/_match.rs
@@ -395,7 +395,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                     return self.get_fn_decl(hir_id).map(|(_, fn_decl, _)| {
                         let (ty, span) = match fn_decl.output {
                             hir::FnRetTy::DefaultReturn(span) => ("()".to_string(), span),
-                            hir::FnRetTy::Return(ty) => (ty_to_string(ty), ty.span),
+                            hir::FnRetTy::Return(ty) => (ty_to_string(&self.tcx, ty), ty.span),
                         };
                         (span, format!("expected `{ty}` because of this return type"))
                     });
diff --git a/compiler/rustc_hir_typeck/src/callee.rs b/compiler/rustc_hir_typeck/src/callee.rs
@@ -724,7 +724,8 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                 def::CtorOf::Variant => "enum variant",
             };
             let removal_span = callee_expr.span.shrink_to_hi().to(call_expr.span.shrink_to_hi());
-            unit_variant = Some((removal_span, descr, rustc_hir_pretty::qpath_to_string(qpath)));
+            unit_variant =
+                Some((removal_span, descr, rustc_hir_pretty::qpath_to_string(&self.tcx, qpath)));
         }
 
         let callee_ty = self.resolve_vars_if_possible(callee_ty);
diff --git a/compiler/rustc_hir_typeck/src/lib.rs b/compiler/rustc_hir_typeck/src/lib.rs
@@ -374,7 +374,7 @@ fn report_unexpected_variant_res(
         Res::Def(DefKind::Variant, _) => "struct variant",
         _ => res.descr(),
     };
-    let path_str = rustc_hir_pretty::qpath_to_string(qpath);
+    let path_str = rustc_hir_pretty::qpath_to_string(&tcx, qpath);
     let err = tcx
         .dcx()
         .struct_span_err(span, format!("expected {expected}, found {res_descr} `{path_str}`"))
diff --git a/compiler/rustc_hir_typeck/src/pat.rs b/compiler/rustc_hir_typeck/src/pat.rs
@@ -1561,7 +1561,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
         {
             let has_shorthand_field_name = field_patterns.iter().any(|field| field.is_shorthand);
             if has_shorthand_field_name {
-                let path = rustc_hir_pretty::qpath_to_string(qpath);
+                let path = rustc_hir_pretty::qpath_to_string(&self.tcx, qpath);
                 let mut err = struct_span_code_err!(
                     self.dcx(),
                     pat.span,
@@ -1743,7 +1743,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                 return None;
             }
 
-            let path = rustc_hir_pretty::qpath_to_string(qpath);
+            let path = rustc_hir_pretty::qpath_to_string(&self.tcx, qpath);
             let mut err = struct_span_code_err!(
                 self.dcx(),
                 pat.span,
@@ -1793,7 +1793,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                             f
                         }
                     }
-                    Err(_) => rustc_hir_pretty::pat_to_string(field.pat),
+                    Err(_) => rustc_hir_pretty::pat_to_string(&self.tcx, field.pat),
                 }
             })
             .collect::<Vec<String>>()
diff --git a/compiler/rustc_middle/Cargo.toml b/compiler/rustc_middle/Cargo.toml
@@ -25,6 +25,7 @@ rustc_feature = { path = "../rustc_feature" }
 rustc_fluent_macro = { path = "../rustc_fluent_macro" }
 rustc_graphviz = { path = "../rustc_graphviz" }
 rustc_hir = { path = "../rustc_hir" }
+rustc_hir_pretty = { path = "../rustc_hir_pretty" }
 rustc_index = { path = "../rustc_index" }
 rustc_macros = { path = "../rustc_macros" }
 rustc_query_system = { path = "../rustc_query_system" }
diff --git a/compiler/rustc_middle/src/hir/map/mod.rs b/compiler/rustc_middle/src/hir/map/mod.rs
@@ -13,6 +13,7 @@ use rustc_hir::def_id::{DefId, LocalDefId, LocalModDefId, LOCAL_CRATE};
 use rustc_hir::definitions::{DefKey, DefPath, DefPathHash};
 use rustc_hir::intravisit::Visitor;
 use rustc_hir::*;
+use rustc_hir_pretty as pprust_hir;
 use rustc_middle::hir::nested_filter;
 use rustc_span::def_id::StableCrateId;
 use rustc_span::symbol::{kw, sym, Ident, Symbol};
@@ -999,6 +1000,12 @@ impl<'hir> intravisit::Map<'hir> for Map<'hir> {
     }
 }
 
+impl<'tcx> pprust_hir::PpAnn for TyCtxt<'tcx> {
+    fn nested(&self, state: &mut pprust_hir::State<'_>, nested: pprust_hir::Nested) {
+        pprust_hir::PpAnn::nested(&(&self.hir() as &dyn intravisit::Map<'_>), state, nested)
+    }
+}
+
 pub(super) fn crate_hash(tcx: TyCtxt<'_>, _: LocalCrate) -> Svh {
     let krate = tcx.hir_crate(());
     let hir_body_hash = krate.opt_hir_hash.expect("HIR hash missing while computing crate hash");
diff --git a/tests/ui/consts/value-suggestion-ice-123906.stderr b/tests/ui/consts/value-suggestion-ice-123906.stderr
@@ -2,7 +2,7 @@ error[E0308]: mismatched types
   --> $DIR/value-suggestion-ice-123906.rs:3:9
    |
 LL | fn as_chunks<const N: usize>() -> [u8; N] {
-   |                                   ------- expected `[u8; ]` because of this return type
+   |                                   ------- expected `[u8; N]` because of this return type
 LL |     loop {
    |     ---- this loop is expected to be of type `[u8; N]`
 LL |         break;
diff --git a/tests/ui/sized/ensure-overriding-bindings-in-pattern-with-ty-err-doesnt-ice.rs b/tests/ui/sized/ensure-overriding-bindings-in-pattern-with-ty-err-doesnt-ice.rs
@@ -3,5 +3,5 @@ fn main() {
 //~^ ERROR expected a pattern, found an expression
 //~| ERROR cannot find type `T` in this scope
 //~| ERROR const and type arguments are not allowed on builtin type `str`
-//~| ERROR expected unit struct, unit variant or constant, found associated function `str<, T>::as_bytes`
+//~| ERROR expected unit struct, unit variant or constant, found associated function `str<
 }
diff --git a/tests/ui/sized/ensure-overriding-bindings-in-pattern-with-ty-err-doesnt-ice.stderr b/tests/ui/sized/ensure-overriding-bindings-in-pattern-with-ty-err-doesnt-ice.stderr
@@ -24,7 +24,9 @@ LL -     let str::<{fn str() { let str::T>>::as_bytes; }}, T>::as_bytes;
 LL +     let str::as_bytes;
    |
 
-error[E0533]: expected unit struct, unit variant or constant, found associated function `str<, T>::as_bytes`
+error[E0533]: expected unit struct, unit variant or constant, found associated function `str<{
+                      fn str() { let (/*ERROR*/); }
+                  }, T>::as_bytes`
   --> $DIR/ensure-overriding-bindings-in-pattern-with-ty-err-doesnt-ice.rs:2:9
    |
 LL |     let str::<{fn str() { let str::T>>::as_bytes; }}, T>::as_bytes;

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ pub fn maybe_expr_static_mut(tcx: TyCtxt<'_>, expr: hir::Expr<'_>) {`
`12`	`12`	`let hir_id = expr.hir_id;`
`13`	`13`	`if let hir::ExprKind::AddrOf(borrow_kind, m, expr) = expr.kind`
`14`	`14`	`&& matches!(borrow_kind, hir::BorrowKind::Ref)`
`15`		`- && let Some(var) = is_path_static_mut(*expr)`
	`15`	`+ && let Some(var) = path_if_static_mut(tcx, expr)`
`16`	`16`	`{`
`17`	`17`	`handle_static_mut_ref(tcx, span, var, span.edition().at_least_rust_2024(), m, hir_id);`
`18`	`18`	`}`
`@@ -24,7 +24,7 @@ pub fn maybe_stmt_static_mut(tcx: TyCtxt<'_>, stmt: hir::Stmt<'_>) {`
`24`	`24`	`&& let hir::PatKind::Binding(ba, _, _, _) = loc.pat.kind`
`25`	`25`	`&& let hir::ByRef::Yes(rmutbl) = ba.0`
`26`	`26`	`&& let Some(init) = loc.init`
`27`		`- && let Some(var) = is_path_static_mut(*init)`
	`27`	`+ && let Some(var) = path_if_static_mut(tcx, init)`
`28`	`28`	`{`
`29`	`29`	`handle_static_mut_ref(`
`30`	`30`	`tcx,`
`@@ -37,13 +37,13 @@ pub fn maybe_stmt_static_mut(tcx: TyCtxt<'_>, stmt: hir::Stmt<'_>) {`
`37`	`37`	`}`
`38`	`38`	`}`
`39`	`39`
`40`		`-fn is_path_static_mut(expr: hir::Expr<'_>) -> Option<String> {`
	`40`	`+fn path_if_static_mut(tcx: TyCtxt<'_>, expr: &hir::Expr<'_>) -> Option<String> {`
`41`	`41`	`if let hir::ExprKind::Path(qpath) = expr.kind`
`42`	`42`	`&& let hir::QPath::Resolved(_, path) = qpath`
`43`	`43`	`&& let hir::def::Res::Def(def_kind, _) = path.res`
`44`	`44`	`&& let hir::def::DefKind::Static { mutability: Mutability::Mut, nested: false } = def_kind`
`45`	`45`	`{`
`46`		`- return Some(qpath_to_string(&qpath));`
	`46`	`+ return Some(qpath_to_string(&tcx, &qpath));`
`47`	`47`	`}`
`48`	`48`	`None`
`49`	`49`	`}`
Original file line number	Diff line number	Diff line change
`@@ -49,10 +49,6 @@ pub trait PpAnn {`
`49`	`49`	`fn post(&self, _state: &mut State<'_>, _node: AnnNode<'_>) {}`
`50`	`50`	`}`
`51`	`51`
`52`		`-pub struct NoAnn;`
`53`		`-`
`54`		`-impl PpAnn for NoAnn {}`
`55`		`-`
`56`	`52`	`impl PpAnn for &dyn rustc_hir::intravisit::Map<'_> {`
`57`	`53`	`fn nested(&self, state: &mut State<'_>, nested: Nested) {`
`58`	`54`	`match nested {`
`@@ -190,16 +186,16 @@ where`
`190`	`186`	`printer.s.eof()`
`191`	`187`	`}`
`192`	`188`
`193`		`-pub fn ty_to_string(ty: &hir::Ty<'_>) -> String {`
`194`		`- to_string(&NoAnn, \|s\| s.print_type(ty))`
	`189`	`+pub fn ty_to_string(ann: &dyn PpAnn, ty: &hir::Ty<'_>) -> String {`
	`190`	`+ to_string(ann, \|s\| s.print_type(ty))`
`195`	`191`	`}`
`196`	`192`
`197`		`-pub fn qpath_to_string(segment: &hir::QPath<'_>) -> String {`
`198`		`- to_string(&NoAnn, \|s\| s.print_qpath(segment, false))`
	`193`	`+pub fn qpath_to_string(ann: &dyn PpAnn, segment: &hir::QPath<'_>) -> String {`
	`194`	`+ to_string(ann, \|s\| s.print_qpath(segment, false))`
`199`	`195`	`}`
`200`	`196`
`201`		`-pub fn pat_to_string(pat: &hir::Pat<'_>) -> String {`
`202`		`- to_string(&NoAnn, \|s\| s.print_pat(pat))`
	`197`	`+pub fn pat_to_string(ann: &dyn PpAnn, pat: &hir::Pat<'_>) -> String {`
	`198`	`+ to_string(ann, \|s\| s.print_pat(pat))`
`203`	`199`	`}`
`204`	`200`
`205`	`201`	`impl<'a> State<'a> {`
Original file line number	Diff line number	Diff line change
`@@ -724,7 +724,8 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {`
`724`	`724`	`def::CtorOf::Variant => "enum variant",`
`725`	`725`	`};`
`726`	`726`	`let removal_span = callee_expr.span.shrink_to_hi().to(call_expr.span.shrink_to_hi());`
`727`		`- unit_variant = Some((removal_span, descr, rustc_hir_pretty::qpath_to_string(qpath)));`
	`727`	`+ unit_variant =`
	`728`	`+ Some((removal_span, descr, rustc_hir_pretty::qpath_to_string(&self.tcx, qpath)));`
`728`	`729`	`}`
`729`	`730`
`730`	`731`	`let callee_ty = self.resolve_vars_if_possible(callee_ty);`
Original file line number	Diff line number	Diff line change
`@@ -1561,7 +1561,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {`
`1561`	`1561`	`{`
`1562`	`1562`	`let has_shorthand_field_name = field_patterns.iter().any(\|field\| field.is_shorthand);`
`1563`	`1563`	`if has_shorthand_field_name {`
`1564`		`- let path = rustc_hir_pretty::qpath_to_string(qpath);`
	`1564`	`+ let path = rustc_hir_pretty::qpath_to_string(&self.tcx, qpath);`
`1565`	`1565`	`let mut err = struct_span_code_err!(`
`1566`	`1566`	`self.dcx(),`
`1567`	`1567`	`pat.span,`
`@@ -1743,7 +1743,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {`
`1743`	`1743`	`return None;`
`1744`	`1744`	`}`
`1745`	`1745`
`1746`		`- let path = rustc_hir_pretty::qpath_to_string(qpath);`
	`1746`	`+ let path = rustc_hir_pretty::qpath_to_string(&self.tcx, qpath);`
`1747`	`1747`	`let mut err = struct_span_code_err!(`
`1748`	`1748`	`self.dcx(),`
`1749`	`1749`	`pat.span,`
`@@ -1793,7 +1793,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {`
`1793`	`1793`	`f`
`1794`	`1794`	`}`
`1795`	`1795`	`}`
`1796`		`- Err(_) => rustc_hir_pretty::pat_to_string(field.pat),`
	`1796`	`+ Err(_) => rustc_hir_pretty::pat_to_string(&self.tcx, field.pat),`
`1797`	`1797`	`}`
`1798`	`1798`	`})`
`1799`	`1799`	`.collect::<Vec<String>>()`