fix: treat enum variants as generic item on their own

This commit is contained in:
Ryo Yoshida 2022-10-04 00:07:34 +09:00
parent 7556f74b16
commit e0a161b2e3
No known key found for this signature in database
GPG Key ID: E25698A930586171
2 changed files with 12 additions and 10 deletions

View File

@ -192,9 +192,7 @@ pub fn subst_for_def(
parent_subst: Option<Substitution>,
) -> TyBuilder<()> {
let generics = generics(db.upcast(), def.into());
// FIXME: this assertion should hold but some adjustment around
// `ValueTyDefId::EnumVariantId` is needed.
// assert!(generics.parent_generics().is_some() == parent_subst.is_some());
assert!(generics.parent_generics().is_some() == parent_subst.is_some());
let params = generics
.iter_self()
.map(|(id, data)| match data {

View File

@ -653,9 +653,13 @@ pub(super) fn substs_from_path(
infer_args: bool,
) -> Substitution {
let last = path.segments().last().expect("path should have at least one segment");
let generic_def = resolved.to_generic_def_id();
let segment = match resolved {
ValueTyDefId::EnumVariantId(_) => {
let (segment, generic_def) = match resolved {
ValueTyDefId::FunctionId(it) => (last, Some(it.into())),
ValueTyDefId::StructId(it) => (last, Some(it.into())),
ValueTyDefId::UnionId(it) => (last, Some(it.into())),
ValueTyDefId::ConstId(it) => (last, Some(it.into())),
ValueTyDefId::StaticId(_) => (last, None),
ValueTyDefId::EnumVariantId(var) => {
// the generic args for an enum variant may be either specified
// on the segment referring to the enum, or on the segment
// referring to the variant. So `Option::<T>::None` and
@ -663,12 +667,12 @@ pub(super) fn substs_from_path(
// preferred). See also `def_ids_for_path_segments` in rustc.
let len = path.segments().len();
let penultimate = len.checked_sub(2).and_then(|idx| path.segments().get(idx));
match penultimate {
let segment = match penultimate {
Some(segment) if segment.args_and_bindings.is_some() => segment,
_ => last,
};
(segment, Some(var.parent.into()))
}
}
_ => last,
};
self.substs_from_path_segment(segment, generic_def, infer_args, None)
}
@ -1660,7 +1664,7 @@ pub(crate) fn to_generic_def_id(self) -> Option<GenericDefId> {
Self::FunctionId(id) => Some(id.into()),
Self::StructId(id) => Some(id.into()),
Self::UnionId(id) => Some(id.into()),
Self::EnumVariantId(var) => Some(var.parent.into()),
Self::EnumVariantId(var) => Some(var.into()),
Self::ConstId(id) => Some(id.into()),
Self::StaticId(_) => None,
}