1
Fork 0

rustdoc: Introduce new DynTrait type for better representation of trait objects

This commit is contained in:
Justus K 2021-06-18 21:47:42 +02:00
parent 1f65f56461
commit 4ea27484c9
No known key found for this signature in database
GPG key ID: 8C62FE98A62FC462
9 changed files with 117 additions and 116 deletions

View file

@ -353,12 +353,7 @@ impl<'a, 'tcx> AutoTraitFinder<'a, 'tcx> {
let (poly_trait, output) = let (poly_trait, output) =
(data.0.as_ref().expect("as_ref failed").clone(), data.1.as_ref().cloned()); (data.0.as_ref().expect("as_ref failed").clone(), data.1.as_ref().cloned());
let new_ty = match poly_trait.trait_ { let new_ty = match poly_trait.trait_ {
Type::ResolvedPath { Type::ResolvedPath { ref path, ref did, ref is_generic } => {
ref path,
ref param_names,
ref did,
ref is_generic,
} => {
let mut new_path = path.clone(); let mut new_path = path.clone();
let last_segment = let last_segment =
new_path.segments.pop().expect("segments were empty"); new_path.segments.pop().expect("segments were empty");
@ -395,7 +390,6 @@ impl<'a, 'tcx> AutoTraitFinder<'a, 'tcx> {
Type::ResolvedPath { Type::ResolvedPath {
path: new_path, path: new_path,
param_names: param_names.clone(),
did: *did, did: *did,
is_generic: *is_generic, is_generic: *is_generic,
} }
@ -570,7 +564,6 @@ impl<'a, 'tcx> AutoTraitFinder<'a, 'tcx> {
match **trait_ { match **trait_ {
Type::ResolvedPath { Type::ResolvedPath {
path: ref trait_path, path: ref trait_path,
ref param_names,
ref did, ref did,
ref is_generic, ref is_generic,
} => { } => {
@ -617,7 +610,6 @@ impl<'a, 'tcx> AutoTraitFinder<'a, 'tcx> {
PolyTrait { PolyTrait {
trait_: Type::ResolvedPath { trait_: Type::ResolvedPath {
path: new_trait_path, path: new_trait_path,
param_names: param_names.clone(),
did: *did, did: *did,
is_generic: *is_generic, is_generic: *is_generic,
}, },

View file

@ -180,7 +180,7 @@ impl Clean<Type> for (ty::TraitRef<'_>, &[TypeBinding]) {
debug!("ty::TraitRef\n subst: {:?}\n", trait_ref.substs); debug!("ty::TraitRef\n subst: {:?}\n", trait_ref.substs);
ResolvedPath { path, param_names: None, did: trait_ref.def_id, is_generic: false } ResolvedPath { path, did: trait_ref.def_id, is_generic: false }
} }
} }
@ -1378,30 +1378,9 @@ impl Clean<Type> for hir::Ty<'_> {
} }
TyKind::Path(_) => clean_qpath(&self, cx), TyKind::Path(_) => clean_qpath(&self, cx),
TyKind::TraitObject(ref bounds, ref lifetime, _) => { TyKind::TraitObject(ref bounds, ref lifetime, _) => {
let cleaned = bounds[0].clean(cx); let bounds = bounds.iter().map(|bound| bound.clean(cx)).collect();
match cleaned.trait_ { let lifetime = if !lifetime.is_elided() { Some(lifetime.clean(cx)) } else { None };
ResolvedPath { path, param_names: None, did, is_generic, .. } => { DynTrait(bounds, lifetime)
let mut bounds: Vec<self::GenericBound> = bounds[1..]
.iter()
.map(|bound| {
self::GenericBound::TraitBound(
bound.clean(cx),
hir::TraitBoundModifier::None,
)
})
.collect();
if !lifetime.is_elided() {
bounds.push(self::GenericBound::Outlives(lifetime.clean(cx)));
}
ResolvedPath {
path,
param_names: Some((bounds, cleaned.generic_params)),
did,
is_generic,
}
}
_ => Infer, // shouldn't happen
}
} }
TyKind::BareFn(ref barefn) => BareFunction(box barefn.clean(cx)), TyKind::BareFn(ref barefn) => BareFunction(box barefn.clean(cx)),
TyKind::Infer | TyKind::Err => Infer, TyKind::Infer | TyKind::Err => Infer,
@ -1484,7 +1463,7 @@ impl<'tcx> Clean<Type> for Ty<'tcx> {
}; };
inline::record_extern_fqn(cx, did, kind); inline::record_extern_fqn(cx, did, kind);
let path = external_path(cx, cx.tcx.item_name(did), None, false, vec![], substs); let path = external_path(cx, cx.tcx.item_name(did), None, false, vec![], substs);
ResolvedPath { path, param_names: None, did, is_generic: false } ResolvedPath { path, did, is_generic: false }
} }
ty::Foreign(did) => { ty::Foreign(did) => {
inline::record_extern_fqn(cx, did, ItemType::ForeignType); inline::record_extern_fqn(cx, did, ItemType::ForeignType);
@ -1496,7 +1475,7 @@ impl<'tcx> Clean<Type> for Ty<'tcx> {
vec![], vec![],
InternalSubsts::empty(), InternalSubsts::empty(),
); );
ResolvedPath { path, param_names: None, did, is_generic: false } ResolvedPath { path, did, is_generic: false }
} }
ty::Dynamic(ref obj, ref reg) => { ty::Dynamic(ref obj, ref reg) => {
// HACK: pick the first `did` as the `did` of the trait object. Someone // HACK: pick the first `did` as the `did` of the trait object. Someone
@ -1514,28 +1493,19 @@ impl<'tcx> Clean<Type> for Ty<'tcx> {
inline::record_extern_fqn(cx, did, ItemType::Trait); inline::record_extern_fqn(cx, did, ItemType::Trait);
let mut param_names = vec![]; let lifetime = reg.clean(cx);
if let Some(b) = reg.clean(cx) { let mut bounds = vec![];
param_names.push(GenericBound::Outlives(b));
}
for did in dids { for did in dids {
let empty = cx.tcx.intern_substs(&[]); let empty = cx.tcx.intern_substs(&[]);
let path = let path =
external_path(cx, cx.tcx.item_name(did), Some(did), false, vec![], empty); external_path(cx, cx.tcx.item_name(did), Some(did), false, vec![], empty);
inline::record_extern_fqn(cx, did, ItemType::Trait); inline::record_extern_fqn(cx, did, ItemType::Trait);
let bound = GenericBound::TraitBound( let bound = PolyTrait {
PolyTrait { trait_: ResolvedPath { path, did, is_generic: false },
trait_: ResolvedPath { generic_params: Vec::new(),
path, };
param_names: None, bounds.push(bound);
did,
is_generic: false,
},
generic_params: Vec::new(),
},
hir::TraitBoundModifier::None,
);
param_names.push(bound);
} }
let mut bindings = vec![]; let mut bindings = vec![];
@ -1548,12 +1518,15 @@ impl<'tcx> Clean<Type> for Ty<'tcx> {
let path = let path =
external_path(cx, cx.tcx.item_name(did), Some(did), false, bindings, substs); external_path(cx, cx.tcx.item_name(did), Some(did), false, bindings, substs);
ResolvedPath { bounds.insert(
path, 0,
param_names: Some((param_names, vec![])), PolyTrait {
did, trait_: ResolvedPath { path, did, is_generic: false },
is_generic: false, generic_params: Vec::new(),
} },
);
DynTrait(bounds, lifetime)
} }
ty::Tuple(ref t) => { ty::Tuple(ref t) => {
Tuple(t.iter().map(|t| t.expect_ty()).collect::<Vec<_>>().clean(cx)) Tuple(t.iter().map(|t| t.expect_ty()).collect::<Vec<_>>().clean(cx))
@ -2257,14 +2230,9 @@ impl From<GenericBound> for SimpleBound {
match bound.clone() { match bound.clone() {
GenericBound::Outlives(l) => SimpleBound::Outlives(l), GenericBound::Outlives(l) => SimpleBound::Outlives(l),
GenericBound::TraitBound(t, mod_) => match t.trait_ { GenericBound::TraitBound(t, mod_) => match t.trait_ {
Type::ResolvedPath { path, param_names, .. } => SimpleBound::TraitBound( Type::ResolvedPath { path, .. } => {
path.segments, SimpleBound::TraitBound(path.segments, Vec::new(), t.generic_params, mod_)
param_names.map_or_else(Vec::new, |(v, _)| { }
v.iter().map(|p| SimpleBound::from(p.clone())).collect()
}),
t.generic_params,
mod_,
),
_ => panic!("Unexpected bound {:?}", bound), _ => panic!("Unexpected bound {:?}", bound),
}, },
} }

View file

@ -1141,7 +1141,7 @@ impl GenericBound {
inline::record_extern_fqn(cx, did, ItemType::Trait); inline::record_extern_fqn(cx, did, ItemType::Trait);
GenericBound::TraitBound( GenericBound::TraitBound(
PolyTrait { PolyTrait {
trait_: ResolvedPath { path, param_names: None, did, is_generic: false }, trait_: ResolvedPath { path, did, is_generic: false },
generic_params: Vec::new(), generic_params: Vec::new(),
}, },
hir::TraitBoundModifier::Maybe, hir::TraitBoundModifier::Maybe,
@ -1407,13 +1407,12 @@ crate enum Type {
/// Structs/enums/traits (most that would be an `hir::TyKind::Path`). /// Structs/enums/traits (most that would be an `hir::TyKind::Path`).
ResolvedPath { ResolvedPath {
path: Path, path: Path,
/// If `param_names` is `Some`, this path is a trait object and the Vecs repsresent
/// `(generic bounds, generic parameters)`
param_names: Option<(Vec<GenericBound>, Vec<GenericParamDef>)>,
did: DefId, did: DefId,
/// `true` if is a `T::Name` path for associated types. /// `true` if is a `T::Name` path for associated types.
is_generic: bool, is_generic: bool,
}, },
/// `dyn for<'a> Trait<'a> + Send + 'static`
DynTrait(Vec<PolyTrait>, Option<Lifetime>),
/// For parameterized types, so the consumer of the JSON don't go /// For parameterized types, so the consumer of the JSON don't go
/// looking for types which don't exist anywhere. /// looking for types which don't exist anywhere.
Generic(Symbol), Generic(Symbol),
@ -1600,6 +1599,7 @@ impl Type {
fn inner_def_id(&self, cache: Option<&Cache>) -> Option<DefId> { fn inner_def_id(&self, cache: Option<&Cache>) -> Option<DefId> {
let t: PrimitiveType = match *self { let t: PrimitiveType = match *self {
ResolvedPath { did, .. } => return Some(did.into()), ResolvedPath { did, .. } => return Some(did.into()),
DynTrait(ref bounds, _) => return bounds[0].trait_.inner_def_id(cache),
Primitive(p) => return cache.and_then(|c| c.primitive_locations.get(&p).cloned()), Primitive(p) => return cache.and_then(|c| c.primitive_locations.get(&p).cloned()),
BorrowedRef { type_: box Generic(..), .. } => PrimitiveType::Reference, BorrowedRef { type_: box Generic(..), .. } => PrimitiveType::Reference,
BorrowedRef { ref type_, .. } => return type_.inner_def_id(cache), BorrowedRef { ref type_, .. } => return type_.inner_def_id(cache),

View file

@ -2,7 +2,7 @@ use crate::clean::auto_trait::AutoTraitFinder;
use crate::clean::blanket_impl::BlanketImplFinder; use crate::clean::blanket_impl::BlanketImplFinder;
use crate::clean::{ use crate::clean::{
inline, Clean, Crate, Generic, GenericArg, GenericArgs, ImportSource, Item, ItemKind, Lifetime, inline, Clean, Crate, Generic, GenericArg, GenericArgs, ImportSource, Item, ItemKind, Lifetime,
Path, PathSegment, Primitive, PrimitiveType, ResolvedPath, Type, TypeBinding, Path, PathSegment, PolyTrait, Primitive, PrimitiveType, ResolvedPath, Type, TypeBinding,
}; };
use crate::core::DocContext; use crate::core::DocContext;
use crate::formats::item_type::ItemType; use crate::formats::item_type::ItemType;
@ -163,8 +163,18 @@ pub(super) fn external_path(
crate fn strip_type(ty: Type) -> Type { crate fn strip_type(ty: Type) -> Type {
match ty { match ty {
Type::ResolvedPath { path, param_names, did, is_generic } => { Type::ResolvedPath { path, did, is_generic } => {
Type::ResolvedPath { path: strip_path(&path), param_names, did, is_generic } Type::ResolvedPath { path: strip_path(&path), did, is_generic }
}
Type::DynTrait(mut bounds, lt) => {
let first = bounds.remove(0);
let stripped_trait = strip_type(first.trait_);
bounds.insert(
0,
PolyTrait { trait_: stripped_trait, generic_params: first.generic_params },
);
Type::DynTrait(bounds, lt)
} }
Type::Tuple(inner_tys) => { Type::Tuple(inner_tys) => {
Type::Tuple(inner_tys.iter().map(|t| strip_type(t.clone())).collect()) Type::Tuple(inner_tys.iter().map(|t| strip_type(t.clone())).collect())
@ -431,7 +441,7 @@ crate fn resolve_type(cx: &mut DocContext<'_>, path: Path, id: hir::HirId) -> Ty
_ => false, _ => false,
}; };
let did = register_res(cx, path.res); let did = register_res(cx, path.res);
ResolvedPath { path, param_names: None, did, is_generic } ResolvedPath { path, did, is_generic }
} }
crate fn get_auto_trait_and_blanket_impls( crate fn get_auto_trait_and_blanket_impls(

View file

@ -402,6 +402,15 @@ impl<'a, 'tcx> DocFolder for CacheBuilder<'a, 'tcx> {
self.cache.parent_stack.push(did); self.cache.parent_stack.push(did);
true true
} }
clean::DynTrait(ref bounds, _)
| clean::BorrowedRef { type_: box clean::DynTrait(ref bounds, _), .. } => {
if let Some(did) = bounds[0].trait_.def_id() {
self.cache.parent_stack.push(did);
true
} else {
false
}
}
ref t => { ref t => {
let prim_did = t let prim_did = t
.primitive_type() .primitive_type()
@ -432,6 +441,12 @@ impl<'a, 'tcx> DocFolder for CacheBuilder<'a, 'tcx> {
| clean::BorrowedRef { type_: box clean::ResolvedPath { did, .. }, .. } => { | clean::BorrowedRef { type_: box clean::ResolvedPath { did, .. }, .. } => {
dids.insert(did); dids.insert(did);
} }
clean::DynTrait(ref bounds, _)
| clean::BorrowedRef { type_: box clean::DynTrait(ref bounds, _), .. } => {
if let Some(did) = bounds[0].trait_.def_id() {
dids.insert(did);
}
}
ref t => { ref t => {
let did = t let did = t
.primitive_type() .primitive_type()

View file

@ -646,18 +646,24 @@ fn primitive_link(
/// Helper to render type parameters /// Helper to render type parameters
fn tybounds<'a, 'tcx: 'a>( fn tybounds<'a, 'tcx: 'a>(
param_names: Option<&'a Vec<clean::GenericBound>>, bounds: &'a Vec<clean::PolyTrait>,
lt: &'a Option<clean::Lifetime>,
cx: &'a Context<'tcx>, cx: &'a Context<'tcx>,
) -> impl fmt::Display + 'a + Captures<'tcx> { ) -> impl fmt::Display + 'a + Captures<'tcx> {
display_fn(move |f| match param_names { display_fn(move |f| {
Some(params) => { for (i, bound) in bounds.iter().enumerate() {
for param in params { if i > 0 {
write!(f, " + ")?; write!(f, " + ")?;
fmt::Display::fmt(&param.print(cx), f)?;
} }
Ok(())
fmt::Display::fmt(&bound.print(cx), f)?;
} }
None => Ok(()),
if let Some(lt) = lt {
write!(f, " + ")?;
fmt::Display::fmt(&lt.print(), f)?;
}
Ok(())
}) })
} }
@ -694,32 +700,13 @@ fn fmt_type<'cx>(
match *t { match *t {
clean::Generic(name) => write!(f, "{}", name), clean::Generic(name) => write!(f, "{}", name),
clean::ResolvedPath { did, ref param_names, ref path, is_generic } => { clean::ResolvedPath { did, ref path, is_generic } => {
let generic_params = param_names.as_ref().map(|(_, x)| x);
let param_names = param_names.as_ref().map(|(x, _)| x);
if let Some(generic_params) = generic_params {
f.write_str("dyn ")?;
if !generic_params.is_empty() {
if f.alternate() {
write!(
f,
"for<{:#}> ",
comma_sep(generic_params.iter().map(|g| g.print(cx)))
)?;
} else {
write!(
f,
"for&lt;{}&gt; ",
comma_sep(generic_params.iter().map(|g| g.print(cx)))
)?;
}
}
}
// Paths like `T::Output` and `Self::Output` should be rendered with all segments. // Paths like `T::Output` and `Self::Output` should be rendered with all segments.
resolved_path(f, did, path, is_generic, use_absolute, cx)?; resolved_path(f, did, path, is_generic, use_absolute, cx)
fmt::Display::fmt(&tybounds(param_names, cx), f) }
clean::DynTrait(ref bounds, ref lt) => {
f.write_str("dyn ")?;
fmt::Display::fmt(&tybounds(bounds, lt, cx), f)
} }
clean::Infer => write!(f, "_"), clean::Infer => write!(f, "_"),
clean::Primitive(prim) => primitive_link(f, prim, &*prim.as_sym().as_str(), cx), clean::Primitive(prim) => primitive_link(f, prim, &*prim.as_sym().as_str(), cx),
@ -854,7 +841,9 @@ fn fmt_type<'cx>(
} }
} }
} }
clean::ResolvedPath { param_names: Some((ref v, _)), .. } if !v.is_empty() => { clean::DynTrait(ref bounds, ref trait_lt)
if bounds.len() > 1 || trait_lt.is_some() =>
{
write!(f, "{}{}{}(", amp, lt, m)?; write!(f, "{}{}{}(", amp, lt, m)?;
fmt_type(&ty, f, use_absolute, cx)?; fmt_type(&ty, f, use_absolute, cx)?;
write!(f, ")") write!(f, ")")
@ -915,7 +904,7 @@ fn fmt_type<'cx>(
// the ugliness comes from inlining across crates where // the ugliness comes from inlining across crates where
// everything comes in as a fully resolved QPath (hard to // everything comes in as a fully resolved QPath (hard to
// look at). // look at).
box clean::ResolvedPath { did, ref param_names, .. } => { box clean::ResolvedPath { did, .. } => {
match href(did.into(), cx) { match href(did.into(), cx) {
Some((ref url, _, ref path)) if !f.alternate() => { Some((ref url, _, ref path)) if !f.alternate() => {
write!( write!(
@ -930,9 +919,6 @@ fn fmt_type<'cx>(
} }
_ => write!(f, "{}", name)?, _ => write!(f, "{}", name)?,
} }
// FIXME: `param_names` are not rendered, and this seems bad?
drop(param_names);
Ok(()) Ok(())
} }
_ => write!(f, "{}", name), _ => write!(f, "{}", name),

View file

@ -233,6 +233,7 @@ fn get_index_type_name(clean_type: &clean::Type, accept_generic: bool) -> Option
}); });
Some(path_segment.name) Some(path_segment.name)
} }
clean::DynTrait(ref bounds, _) => get_index_type_name(&bounds[0].trait_, accept_generic),
clean::Generic(s) if accept_generic => Some(s), clean::Generic(s) if accept_generic => Some(s),
clean::Primitive(ref p) => Some(p.as_sym()), clean::Primitive(ref p) => Some(p.as_sym()),
clean::BorrowedRef { ref type_, .. } => get_index_type_name(type_, accept_generic), clean::BorrowedRef { ref type_, .. } => get_index_type_name(type_, accept_generic),

View file

@ -373,14 +373,35 @@ impl FromWithTcx<clean::Type> for Type {
fn from_tcx(ty: clean::Type, tcx: TyCtxt<'_>) -> Self { fn from_tcx(ty: clean::Type, tcx: TyCtxt<'_>) -> Self {
use clean::Type::*; use clean::Type::*;
match ty { match ty {
ResolvedPath { path, param_names, did, is_generic: _ } => Type::ResolvedPath { ResolvedPath { path, did, is_generic: _ } => Type::ResolvedPath {
name: path.whole_name(), name: path.whole_name(),
id: from_def_id(did.into()), id: from_def_id(did.into()),
args: path.segments.last().map(|args| Box::new(args.clone().args.into_tcx(tcx))), args: path.segments.last().map(|args| Box::new(args.clone().args.into_tcx(tcx))),
param_names: param_names param_names: Vec::new(),
.map(|(v, _)| v.into_iter().map(|x| x.into_tcx(tcx)).collect())
.unwrap_or_default(),
}, },
DynTrait(mut bounds, lt) => {
let (path, id) = match bounds.remove(0).trait_ {
ResolvedPath { path, did, .. } => (path, did),
_ => unreachable!(),
};
Type::ResolvedPath {
name: path.whole_name(),
id: from_def_id(id.into()),
args: path
.segments
.last()
.map(|args| Box::new(args.clone().args.into_tcx(tcx))),
param_names: bounds
.into_iter()
.map(|t| {
clean::GenericBound::TraitBound(t, rustc_hir::TraitBoundModifier::None)
})
.chain(lt.into_iter().map(|lt| clean::GenericBound::Outlives(lt)))
.map(|bound| bound.into_tcx(tcx))
.collect(),
}
}
Generic(s) => Type::Generic(s.to_string()), Generic(s) => Type::Generic(s.to_string()),
Primitive(p) => Type::Primitive(p.as_sym().to_string()), Primitive(p) => Type::Primitive(p.as_sym().to_string()),
BareFunction(f) => Type::FunctionPointer(Box::new((*f).into_tcx(tcx))), BareFunction(f) => Type::FunctionPointer(Box::new((*f).into_tcx(tcx))),

View file

@ -51,3 +51,11 @@ pub trait B<'x> {}
// @has - '//code[@class="in-band"]' "impl<'a> B<'a> for dyn for<'b> Trait<'b>" // @has - '//code[@class="in-band"]' "impl<'a> B<'a> for dyn for<'b> Trait<'b>"
impl<'a> B<'a> for dyn for<'b> Trait<'b> {} impl<'a> B<'a> for dyn for<'b> Trait<'b> {}
// @has foo/struct.Bar.html
// @has - '//span[@id="structfield.bar"]' "bar: &'a (dyn for<'b> Trait<'b> + Unpin)"
// @has - '//span[@id="structfield.baz"]' "baz: &'a (dyn Unpin + for<'b> Trait<'b>)"
pub struct Bar<'a> {
pub bar: &'a (dyn for<'b> Trait<'b> + Unpin),
pub baz: &'a (dyn Unpin + for<'b> Trait<'b>),
}