| use crate::ImplTraitPosition; |
| |
| use super::errors::{GenericTypeWithParentheses, UseAngleBrackets}; |
| use super::ResolverAstLoweringExt; |
| use super::{GenericArgsCtor, LifetimeRes, ParenthesizedGenericArgs}; |
| use super::{ImplTraitContext, LoweringContext, ParamMode}; |
| |
| use rustc_ast::{self as ast, *}; |
| use rustc_hir as hir; |
| use rustc_hir::def::{DefKind, PartialRes, Res}; |
| use rustc_hir::GenericArg; |
| use rustc_span::symbol::{kw, sym, Ident}; |
| use rustc_span::{BytePos, Span, DUMMY_SP}; |
| |
| use smallvec::{smallvec, SmallVec}; |
| |
| impl<'a, 'hir> LoweringContext<'a, 'hir> { |
| #[instrument(level = "trace", skip(self))] |
| pub(crate) fn lower_qpath( |
| &mut self, |
| id: NodeId, |
| qself: &Option<ptr::P<QSelf>>, |
| p: &Path, |
| param_mode: ParamMode, |
| itctx: &ImplTraitContext, |
| ) -> hir::QPath<'hir> { |
| let qself_position = qself.as_ref().map(|q| q.position); |
| let qself = qself.as_ref().map(|q| self.lower_ty(&q.ty, itctx)); |
| |
| let partial_res = |
| self.resolver.get_partial_res(id).unwrap_or_else(|| PartialRes::new(Res::Err)); |
| let base_res = partial_res.base_res(); |
| let unresolved_segments = partial_res.unresolved_segments(); |
| |
| let path_span_lo = p.span.shrink_to_lo(); |
| let proj_start = p.segments.len() - unresolved_segments; |
| let path = self.arena.alloc(hir::Path { |
| res: self.lower_res(base_res), |
| segments: self.arena.alloc_from_iter(p.segments[..proj_start].iter().enumerate().map( |
| |(i, segment)| { |
| let param_mode = match (qself_position, param_mode) { |
| (Some(j), ParamMode::Optional) if i < j => { |
| // This segment is part of the trait path in a |
| // qualified path - one of `a`, `b` or `Trait` |
| // in `<X as a::b::Trait>::T::U::method`. |
| ParamMode::Explicit |
| } |
| _ => param_mode, |
| }; |
| |
| let parenthesized_generic_args = match base_res { |
| // `a::b::Trait(Args)` |
| Res::Def(DefKind::Trait, _) if i + 1 == proj_start => { |
| ParenthesizedGenericArgs::ParenSugar |
| } |
| // `a::b::Trait(Args)::TraitItem` |
| Res::Def(DefKind::AssocFn, _) |
| | Res::Def(DefKind::AssocConst, _) |
| | Res::Def(DefKind::AssocTy, _) |
| if i + 2 == proj_start => |
| { |
| ParenthesizedGenericArgs::ParenSugar |
| } |
| // Avoid duplicated errors. |
| Res::Err => ParenthesizedGenericArgs::ParenSugar, |
| // An error |
| _ => ParenthesizedGenericArgs::Err, |
| }; |
| |
| self.lower_path_segment( |
| p.span, |
| segment, |
| param_mode, |
| parenthesized_generic_args, |
| itctx, |
| ) |
| }, |
| )), |
| span: self.lower_span( |
| p.segments[..proj_start] |
| .last() |
| .map_or(path_span_lo, |segment| path_span_lo.to(segment.span())), |
| ), |
| }); |
| |
| // Simple case, either no projections, or only fully-qualified. |
| // E.g., `std::mem::size_of` or `<I as Iterator>::Item`. |
| if unresolved_segments == 0 { |
| return hir::QPath::Resolved(qself, path); |
| } |
| |
| // Create the innermost type that we're projecting from. |
| let mut ty = if path.segments.is_empty() { |
| // If the base path is empty that means there exists a |
| // syntactical `Self`, e.g., `&i32` in `<&i32>::clone`. |
| qself.expect("missing QSelf for <T>::...") |
| } else { |
| // Otherwise, the base path is an implicit `Self` type path, |
| // e.g., `Vec` in `Vec::new` or `<I as Iterator>::Item` in |
| // `<I as Iterator>::Item::default`. |
| let new_id = self.next_id(); |
| self.arena.alloc(self.ty_path(new_id, path.span, hir::QPath::Resolved(qself, path))) |
| }; |
| |
| // Anything after the base path are associated "extensions", |
| // out of which all but the last one are associated types, |
| // e.g., for `std::vec::Vec::<T>::IntoIter::Item::clone`: |
| // * base path is `std::vec::Vec<T>` |
| // * "extensions" are `IntoIter`, `Item` and `clone` |
| // * type nodes are: |
| // 1. `std::vec::Vec<T>` (created above) |
| // 2. `<std::vec::Vec<T>>::IntoIter` |
| // 3. `<<std::vec::Vec<T>>::IntoIter>::Item` |
| // * final path is `<<<std::vec::Vec<T>>::IntoIter>::Item>::clone` |
| for (i, segment) in p.segments.iter().enumerate().skip(proj_start) { |
| let hir_segment = self.arena.alloc(self.lower_path_segment( |
| p.span, |
| segment, |
| param_mode, |
| ParenthesizedGenericArgs::Err, |
| itctx, |
| )); |
| let qpath = hir::QPath::TypeRelative(ty, hir_segment); |
| |
| // It's finished, return the extension of the right node type. |
| if i == p.segments.len() - 1 { |
| return qpath; |
| } |
| |
| // Wrap the associated extension in another type node. |
| let new_id = self.next_id(); |
| ty = self.arena.alloc(self.ty_path(new_id, path_span_lo.to(segment.span()), qpath)); |
| } |
| |
| // We should've returned in the for loop above. |
| |
| self.diagnostic().span_bug( |
| p.span, |
| format!( |
| "lower_qpath: no final extension segment in {}..{}", |
| proj_start, |
| p.segments.len() |
| ), |
| ); |
| } |
| |
| pub(crate) fn lower_use_path( |
| &mut self, |
| res: SmallVec<[Res; 3]>, |
| p: &Path, |
| param_mode: ParamMode, |
| ) -> &'hir hir::UsePath<'hir> { |
| self.arena.alloc(hir::UsePath { |
| res, |
| segments: self.arena.alloc_from_iter(p.segments.iter().map(|segment| { |
| self.lower_path_segment( |
| p.span, |
| segment, |
| param_mode, |
| ParenthesizedGenericArgs::Err, |
| &ImplTraitContext::Disallowed(ImplTraitPosition::Path), |
| ) |
| })), |
| span: self.lower_span(p.span), |
| }) |
| } |
| |
| pub(crate) fn lower_path_segment( |
| &mut self, |
| path_span: Span, |
| segment: &PathSegment, |
| param_mode: ParamMode, |
| parenthesized_generic_args: ParenthesizedGenericArgs, |
| itctx: &ImplTraitContext, |
| ) -> hir::PathSegment<'hir> { |
| debug!("path_span: {:?}, lower_path_segment(segment: {:?})", path_span, segment,); |
| let (mut generic_args, infer_args) = if let Some(generic_args) = segment.args.as_deref() { |
| match generic_args { |
| GenericArgs::AngleBracketed(data) => { |
| self.lower_angle_bracketed_parameter_data(data, param_mode, itctx) |
| } |
| GenericArgs::Parenthesized(data) => match parenthesized_generic_args { |
| ParenthesizedGenericArgs::ParenSugar => { |
| self.lower_parenthesized_parameter_data(data, itctx) |
| } |
| ParenthesizedGenericArgs::Err => { |
| // Suggest replacing parentheses with angle brackets `Trait(params...)` to `Trait<params...>` |
| let sub = if !data.inputs.is_empty() { |
| // Start of the span to the 1st character of 1st argument |
| let open_param = data.inputs_span.shrink_to_lo().to(data |
| .inputs |
| .first() |
| .unwrap() |
| .span |
| .shrink_to_lo()); |
| // Last character position of last argument to the end of the span |
| let close_param = data |
| .inputs |
| .last() |
| .unwrap() |
| .span |
| .shrink_to_hi() |
| .to(data.inputs_span.shrink_to_hi()); |
| |
| Some(UseAngleBrackets { open_param, close_param }) |
| } else { |
| None |
| }; |
| self.tcx.sess.emit_err(GenericTypeWithParentheses { span: data.span, sub }); |
| ( |
| self.lower_angle_bracketed_parameter_data( |
| &data.as_angle_bracketed_args(), |
| param_mode, |
| itctx, |
| ) |
| .0, |
| false, |
| ) |
| } |
| }, |
| } |
| } else { |
| ( |
| GenericArgsCtor { |
| args: Default::default(), |
| bindings: &[], |
| parenthesized: hir::GenericArgsParentheses::No, |
| span: path_span.shrink_to_hi(), |
| }, |
| param_mode == ParamMode::Optional, |
| ) |
| }; |
| |
| let has_lifetimes = |
| generic_args.args.iter().any(|arg| matches!(arg, GenericArg::Lifetime(_))); |
| |
| // FIXME(return_type_notation): Is this correct? I think so. |
| if generic_args.parenthesized != hir::GenericArgsParentheses::ParenSugar && !has_lifetimes { |
| self.maybe_insert_elided_lifetimes_in_path( |
| path_span, |
| segment.id, |
| segment.ident.span, |
| &mut generic_args, |
| ); |
| } |
| |
| let res = self.expect_full_res(segment.id); |
| let hir_id = self.lower_node_id(segment.id); |
| debug!( |
| "lower_path_segment: ident={:?} original-id={:?} new-id={:?}", |
| segment.ident, segment.id, hir_id, |
| ); |
| |
| hir::PathSegment { |
| ident: self.lower_ident(segment.ident), |
| hir_id, |
| res: self.lower_res(res), |
| infer_args, |
| args: if generic_args.is_empty() && generic_args.span.is_empty() { |
| None |
| } else { |
| Some(generic_args.into_generic_args(self)) |
| }, |
| } |
| } |
| |
| fn maybe_insert_elided_lifetimes_in_path( |
| &mut self, |
| path_span: Span, |
| segment_id: NodeId, |
| segment_ident_span: Span, |
| generic_args: &mut GenericArgsCtor<'hir>, |
| ) { |
| let (start, end) = match self.resolver.get_lifetime_res(segment_id) { |
| Some(LifetimeRes::ElidedAnchor { start, end }) => (start, end), |
| None => return, |
| Some(_) => panic!(), |
| }; |
| let expected_lifetimes = end.as_usize() - start.as_usize(); |
| debug!(expected_lifetimes); |
| |
| // Note: these spans are used for diagnostics when they can't be inferred. |
| // See rustc_resolve::late::lifetimes::LifetimeContext::add_missing_lifetime_specifiers_label |
| let elided_lifetime_span = if generic_args.span.is_empty() { |
| // If there are no brackets, use the identifier span. |
| // HACK: we use find_ancestor_inside to properly suggest elided spans in paths |
| // originating from macros, since the segment's span might be from a macro arg. |
| segment_ident_span.find_ancestor_inside(path_span).unwrap_or(path_span) |
| } else if generic_args.is_empty() { |
| // If there are brackets, but not generic arguments, then use the opening bracket |
| generic_args.span.with_hi(generic_args.span.lo() + BytePos(1)) |
| } else { |
| // Else use an empty span right after the opening bracket. |
| generic_args.span.with_lo(generic_args.span.lo() + BytePos(1)).shrink_to_lo() |
| }; |
| |
| generic_args.args.insert_many( |
| 0, |
| (start.as_u32()..end.as_u32()).map(|i| { |
| let id = NodeId::from_u32(i); |
| let l = self.lower_lifetime(&Lifetime { |
| id, |
| ident: Ident::new(kw::Empty, elided_lifetime_span), |
| }); |
| GenericArg::Lifetime(l) |
| }), |
| ); |
| } |
| |
| pub(crate) fn lower_angle_bracketed_parameter_data( |
| &mut self, |
| data: &AngleBracketedArgs, |
| param_mode: ParamMode, |
| itctx: &ImplTraitContext, |
| ) -> (GenericArgsCtor<'hir>, bool) { |
| let has_non_lt_args = data.args.iter().any(|arg| match arg { |
| AngleBracketedArg::Arg(ast::GenericArg::Lifetime(_)) |
| | AngleBracketedArg::Constraint(_) => false, |
| AngleBracketedArg::Arg(ast::GenericArg::Type(_) | ast::GenericArg::Const(_)) => true, |
| }); |
| let args = data |
| .args |
| .iter() |
| .filter_map(|arg| match arg { |
| AngleBracketedArg::Arg(arg) => Some(self.lower_generic_arg(arg, itctx)), |
| AngleBracketedArg::Constraint(_) => None, |
| }) |
| .collect(); |
| let bindings = self.arena.alloc_from_iter(data.args.iter().filter_map(|arg| match arg { |
| AngleBracketedArg::Constraint(c) => Some(self.lower_assoc_ty_constraint(c, itctx)), |
| AngleBracketedArg::Arg(_) => None, |
| })); |
| let ctor = GenericArgsCtor { |
| args, |
| bindings, |
| parenthesized: hir::GenericArgsParentheses::No, |
| span: data.span, |
| }; |
| (ctor, !has_non_lt_args && param_mode == ParamMode::Optional) |
| } |
| |
| fn lower_parenthesized_parameter_data( |
| &mut self, |
| data: &ParenthesizedArgs, |
| itctx: &ImplTraitContext, |
| ) -> (GenericArgsCtor<'hir>, bool) { |
| // Switch to `PassThrough` mode for anonymous lifetimes; this |
| // means that we permit things like `&Ref<T>`, where `Ref` has |
| // a hidden lifetime parameter. This is needed for backwards |
| // compatibility, even in contexts like an impl header where |
| // we generally don't permit such things (see #51008). |
| let ParenthesizedArgs { span, inputs, inputs_span, output } = data; |
| let inputs = self.arena.alloc_from_iter(inputs.iter().map(|ty| { |
| self.lower_ty_direct(ty, &ImplTraitContext::Disallowed(ImplTraitPosition::FnTraitParam)) |
| })); |
| let output_ty = match output { |
| // Only allow `impl Trait` in return position. i.e.: |
| // ```rust |
| // fn f(_: impl Fn() -> impl Debug) -> impl Fn() -> impl Debug |
| // // disallowed --^^^^^^^^^^ allowed --^^^^^^^^^^ |
| // ``` |
| FnRetTy::Ty(ty) if matches!(itctx, ImplTraitContext::ReturnPositionOpaqueTy { .. }) => { |
| if self.tcx.features().impl_trait_in_fn_trait_return { |
| self.lower_ty(&ty, itctx) |
| } else { |
| self.lower_ty( |
| &ty, |
| &ImplTraitContext::FeatureGated( |
| ImplTraitPosition::FnTraitReturn, |
| sym::impl_trait_in_fn_trait_return, |
| ), |
| ) |
| } |
| } |
| FnRetTy::Ty(ty) => { |
| self.lower_ty(&ty, &ImplTraitContext::Disallowed(ImplTraitPosition::FnTraitReturn)) |
| } |
| FnRetTy::Default(_) => self.arena.alloc(self.ty_tup(*span, &[])), |
| }; |
| let args = smallvec![GenericArg::Type(self.arena.alloc(self.ty_tup(*inputs_span, inputs)))]; |
| let binding = self.output_ty_binding(output_ty.span, output_ty); |
| ( |
| GenericArgsCtor { |
| args, |
| bindings: arena_vec![self; binding], |
| parenthesized: hir::GenericArgsParentheses::ParenSugar, |
| span: data.inputs_span, |
| }, |
| false, |
| ) |
| } |
| |
| /// An associated type binding `Output = $ty`. |
| pub(crate) fn output_ty_binding( |
| &mut self, |
| span: Span, |
| ty: &'hir hir::Ty<'hir>, |
| ) -> hir::TypeBinding<'hir> { |
| let ident = Ident::with_dummy_span(hir::FN_OUTPUT_NAME); |
| let kind = hir::TypeBindingKind::Equality { term: ty.into() }; |
| let args = arena_vec![self;]; |
| let bindings = arena_vec![self;]; |
| let gen_args = self.arena.alloc(hir::GenericArgs { |
| args, |
| bindings, |
| parenthesized: hir::GenericArgsParentheses::No, |
| span_ext: DUMMY_SP, |
| }); |
| hir::TypeBinding { |
| hir_id: self.next_id(), |
| gen_args, |
| span: self.lower_span(span), |
| ident, |
| kind, |
| } |
| } |
| } |
