Fix pretty hir for anon consts in diagnostics

This commit is contained in:
Michael Goulet 2024-04-14 10:26:59 -04:00
parent 3493a56529
commit 619e044178
13 changed files with 32 additions and 41 deletions

View file

@ -4278,6 +4278,7 @@ dependencies = [
"rustc_fluent_macro", "rustc_fluent_macro",
"rustc_graphviz", "rustc_graphviz",
"rustc_hir", "rustc_hir",
"rustc_hir_pretty",
"rustc_index", "rustc_index",
"rustc_macros", "rustc_macros",
"rustc_query_system", "rustc_query_system",

View file

@ -24,20 +24,6 @@ struct AstNoAnn;
impl pprust_ast::PpAnn for 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; struct AstIdentifiedAnn;
impl pprust_ast::PpAnn for AstIdentifiedAnn { impl pprust_ast::PpAnn for AstIdentifiedAnn {
@ -300,10 +286,7 @@ pub fn print<'tcx>(sess: &Session, ppm: PpMode, ex: PrintExtra<'tcx>) {
) )
}; };
match s { match s {
PpHirMode::Normal => { PpHirMode::Normal => f(&tcx),
let annotation = HirNoAnn { tcx };
f(&annotation)
}
PpHirMode::Identified => { PpHirMode::Identified => {
let annotation = HirIdentifiedAnn { tcx }; let annotation = HirIdentifiedAnn { tcx };
f(&annotation) f(&annotation)

View file

@ -12,7 +12,7 @@ pub fn maybe_expr_static_mut(tcx: TyCtxt<'_>, expr: hir::Expr<'_>) {
let hir_id = expr.hir_id; let hir_id = expr.hir_id;
if let hir::ExprKind::AddrOf(borrow_kind, m, expr) = expr.kind if let hir::ExprKind::AddrOf(borrow_kind, m, expr) = expr.kind
&& matches!(borrow_kind, hir::BorrowKind::Ref) && 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); 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::PatKind::Binding(ba, _, _, _) = loc.pat.kind
&& let hir::ByRef::Yes(rmutbl) = ba.0 && let hir::ByRef::Yes(rmutbl) = ba.0
&& let Some(init) = loc.init && 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( handle_static_mut_ref(
tcx, 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 if let hir::ExprKind::Path(qpath) = expr.kind
&& let hir::QPath::Resolved(_, path) = qpath && let hir::QPath::Resolved(_, path) = qpath
&& let hir::def::Res::Def(def_kind, _) = path.res && let hir::def::Res::Def(def_kind, _) = path.res
&& let hir::def::DefKind::Static { mutability: Mutability::Mut, nested: false } = def_kind && 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 None
} }

View file

@ -49,10 +49,6 @@ pub trait PpAnn {
fn post(&self, _state: &mut State<'_>, _node: AnnNode<'_>) {} fn post(&self, _state: &mut State<'_>, _node: AnnNode<'_>) {}
} }
pub struct NoAnn;
impl PpAnn for NoAnn {}
impl PpAnn for &dyn rustc_hir::intravisit::Map<'_> { impl PpAnn for &dyn rustc_hir::intravisit::Map<'_> {
fn nested(&self, state: &mut State<'_>, nested: Nested) { fn nested(&self, state: &mut State<'_>, nested: Nested) {
match nested { match nested {
@ -190,16 +186,16 @@ where
printer.s.eof() printer.s.eof()
} }
pub fn ty_to_string(ty: &hir::Ty<'_>) -> String { pub fn ty_to_string(ann: &dyn PpAnn, ty: &hir::Ty<'_>) -> String {
to_string(&NoAnn, |s| s.print_type(ty)) to_string(ann, |s| s.print_type(ty))
} }
pub fn qpath_to_string(segment: &hir::QPath<'_>) -> String { pub fn qpath_to_string(ann: &dyn PpAnn, segment: &hir::QPath<'_>) -> String {
to_string(&NoAnn, |s| s.print_qpath(segment, false)) to_string(ann, |s| s.print_qpath(segment, false))
} }
pub fn pat_to_string(pat: &hir::Pat<'_>) -> String { pub fn pat_to_string(ann: &dyn PpAnn, pat: &hir::Pat<'_>) -> String {
to_string(&NoAnn, |s| s.print_pat(pat)) to_string(ann, |s| s.print_pat(pat))
} }
impl<'a> State<'a> { impl<'a> State<'a> {

View file

@ -395,7 +395,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
return self.get_fn_decl(hir_id).map(|(_, fn_decl, _)| { return self.get_fn_decl(hir_id).map(|(_, fn_decl, _)| {
let (ty, span) = match fn_decl.output { let (ty, span) = match fn_decl.output {
hir::FnRetTy::DefaultReturn(span) => ("()".to_string(), span), 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")) (span, format!("expected `{ty}` because of this return type"))
}); });

View file

@ -724,7 +724,8 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
def::CtorOf::Variant => "enum variant", def::CtorOf::Variant => "enum variant",
}; };
let removal_span = callee_expr.span.shrink_to_hi().to(call_expr.span.shrink_to_hi()); 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); let callee_ty = self.resolve_vars_if_possible(callee_ty);

View file

@ -374,7 +374,7 @@ fn report_unexpected_variant_res(
Res::Def(DefKind::Variant, _) => "struct variant", Res::Def(DefKind::Variant, _) => "struct variant",
_ => res.descr(), _ => 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 let err = tcx
.dcx() .dcx()
.struct_span_err(span, format!("expected {expected}, found {res_descr} `{path_str}`")) .struct_span_err(span, format!("expected {expected}, found {res_descr} `{path_str}`"))

View file

@ -1561,7 +1561,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
{ {
let has_shorthand_field_name = field_patterns.iter().any(|field| field.is_shorthand); let has_shorthand_field_name = field_patterns.iter().any(|field| field.is_shorthand);
if has_shorthand_field_name { 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!( let mut err = struct_span_code_err!(
self.dcx(), self.dcx(),
pat.span, pat.span,
@ -1743,7 +1743,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
return None; 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!( let mut err = struct_span_code_err!(
self.dcx(), self.dcx(),
pat.span, pat.span,
@ -1793,7 +1793,7 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
f f
} }
} }
Err(_) => rustc_hir_pretty::pat_to_string(field.pat), Err(_) => rustc_hir_pretty::pat_to_string(&self.tcx, field.pat),
} }
}) })
.collect::<Vec<String>>() .collect::<Vec<String>>()

View file

@ -25,6 +25,7 @@ rustc_feature = { path = "../rustc_feature" }
rustc_fluent_macro = { path = "../rustc_fluent_macro" } rustc_fluent_macro = { path = "../rustc_fluent_macro" }
rustc_graphviz = { path = "../rustc_graphviz" } rustc_graphviz = { path = "../rustc_graphviz" }
rustc_hir = { path = "../rustc_hir" } rustc_hir = { path = "../rustc_hir" }
rustc_hir_pretty = { path = "../rustc_hir_pretty" }
rustc_index = { path = "../rustc_index" } rustc_index = { path = "../rustc_index" }
rustc_macros = { path = "../rustc_macros" } rustc_macros = { path = "../rustc_macros" }
rustc_query_system = { path = "../rustc_query_system" } rustc_query_system = { path = "../rustc_query_system" }

View file

@ -13,6 +13,7 @@ use rustc_hir::def_id::{DefId, LocalDefId, LocalModDefId, LOCAL_CRATE};
use rustc_hir::definitions::{DefKey, DefPath, DefPathHash}; use rustc_hir::definitions::{DefKey, DefPath, DefPathHash};
use rustc_hir::intravisit::Visitor; use rustc_hir::intravisit::Visitor;
use rustc_hir::*; use rustc_hir::*;
use rustc_hir_pretty as pprust_hir;
use rustc_middle::hir::nested_filter; use rustc_middle::hir::nested_filter;
use rustc_span::def_id::StableCrateId; use rustc_span::def_id::StableCrateId;
use rustc_span::symbol::{kw, sym, Ident, Symbol}; 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 { pub(super) fn crate_hash(tcx: TyCtxt<'_>, _: LocalCrate) -> Svh {
let krate = tcx.hir_crate(()); let krate = tcx.hir_crate(());
let hir_body_hash = krate.opt_hir_hash.expect("HIR hash missing while computing crate hash"); let hir_body_hash = krate.opt_hir_hash.expect("HIR hash missing while computing crate hash");

View file

@ -2,7 +2,7 @@ error[E0308]: mismatched types
--> $DIR/value-suggestion-ice-123906.rs:3:9 --> $DIR/value-suggestion-ice-123906.rs:3:9
| |
LL | fn as_chunks<const N: usize>() -> [u8; N] { 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 { LL | loop {
| ---- this loop is expected to be of type `[u8; N]` | ---- this loop is expected to be of type `[u8; N]`
LL | break; LL | break;

View file

@ -3,5 +3,5 @@ fn main() {
//~^ ERROR expected a pattern, found an expression //~^ ERROR expected a pattern, found an expression
//~| ERROR cannot find type `T` in this scope //~| ERROR cannot find type `T` in this scope
//~| ERROR const and type arguments are not allowed on builtin type `str` //~| 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<
} }

View file

@ -24,7 +24,9 @@ LL - let str::<{fn str() { let str::T>>::as_bytes; }}, T>::as_bytes;
LL + let str::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 --> $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; LL | let str::<{fn str() { let str::T>>::as_bytes; }}, T>::as_bytes;