Fix associated types only working for first generic parameter

Also, removes extraneous `where for`
This commit is contained in:
Osspial 2018-04-12 16:44:32 -04:00
parent def8d6e8af
commit 4415d10c61
2 changed files with 23 additions and 12 deletions

View File

@ -59,24 +59,27 @@ pub fn with_where_predicates_from_fields<F, W>(
) -> syn::Generics ) -> syn::Generics
where where
F: Fn(&attr::Field) -> Option<&[syn::WherePredicate]>, F: Fn(&attr::Field) -> Option<&[syn::WherePredicate]>,
W: for<'a> Fn(&attr::Field) -> bool, W: Fn(&attr::Field) -> bool,
{ {
let predicates = cont.data let predicates = cont.data
.all_fields() .all_fields()
.flat_map(|field| { .flat_map(|field| {
let field_ty = field.ty; let field_ty = field.ty;
let field_bound: Option<syn::WherePredicate> = match (field_ty, gen_bound_where(&field.attrs)) { let matching_generic = |t: &syn::PathSegment, g| match g {
(&syn::Type::Path(ref ty_path), true) => match (ty_path.path.segments.first(), generics.params.first()) { &syn::GenericParam::Type(ref generic_ty)
(Some(syn::punctuated::Pair::Punctuated(ref t, _)), Some(syn::punctuated::Pair::End(&syn::GenericParam::Type(ref generic_ty)))) if generic_ty.ident == t.ident => true,
if generic_ty.ident == t.ident => _ => false
{ };
let predicate: syn::WherePredicate = parse_quote!(#field_ty: #trait_bound);
Some(predicate) let mut field_bound: Option<syn::WherePredicate> = None;
}, if let &syn::Type::Path(ref ty_path) = field_ty {
_ => None field_bound = match (gen_bound_where(&field.attrs), ty_path.path.segments.first()) {
}, (true, Some(syn::punctuated::Pair::Punctuated(ref t, _)))
if generics.params.iter().fold(false, |b, g| b || matching_generic(t, g))
=> Some(parse_quote!(#field_ty: #trait_bound)),
(_, _) => None (_, _) => None
}; };
}
field_bound.into_iter().chain(from_field(&field.attrs).into_iter().flat_map(|predicates| predicates.to_vec())) field_bound.into_iter().chain(from_field(&field.attrs).into_iter().flat_map(|predicates| predicates.to_vec()))
}); });

View File

@ -555,6 +555,14 @@ fn test_gen() {
} }
assert::<AssocDerive<NoSerdeImpl>>(); assert::<AssocDerive<NoSerdeImpl>>();
#[derive(Serialize, Deserialize)]
struct AssocDeriveMulti<S, T: AssocSerde> {
s: S,
assoc: T::Assoc,
}
assert::<AssocDeriveMulti<i32, NoSerdeImpl>>();
} }
////////////////////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////////////