mod models;
use models::{FieldAttributeBuilder, TypeAttributeBuilder};
use syn::{Data, DeriveInput, Meta};
use super::TraitHandler;
use crate::{
Trait,
common::{
bound::{BOUND_EXCEPTIONS_COPY, Bound},
quote_mixed,
},
trait_handlers::TraitHandlerContext,
};
pub(crate) fn prerequisites() -> &'static [Trait] {
&[
#[cfg(feature = "Clone")]
Trait::Clone,
]
}
pub(crate) struct CopyHandler;
#[cfg(feature = "Clone")]
pub(crate) fn has_custom_bound(ctx: &TraitHandlerContext) -> syn::Result<bool> {
let Some(meta) = ctx.copy_meta() else {
return Ok(false);
};
let attribute = TypeAttributeBuilder {
enable_flag: true, enable_bound: true
}
.build_from_copy_meta(meta)?;
Ok(matches!(attribute.bound, Bound::Custom(predicates) if !predicates.is_empty()))
}
impl TraitHandler for CopyHandler {
#[inline]
fn trait_meta_handler<'a>(
ast: &'a DeriveInput,
ctx: &mut TraitHandlerContext<'a>,
token_stream: &mut proc_macro2::TokenStream,
traits: &[Trait],
meta: &Meta,
) -> syn::Result<()> {
let generated_impl_attributes =
crate::common::attributes::generated_impl_attributes(&ast.attrs);
let type_attribute = TypeAttributeBuilder {
enable_flag: true, enable_bound: true
}
.build_from_copy_meta(meta)?;
let mut field_types = Vec::new();
match &ast.data {
Data::Struct(data) => {
for field in data.fields.iter() {
let _ = FieldAttributeBuilder.build_from_attributes(&field.attrs, traits)?;
field_types.push(&field.ty);
}
},
Data::Enum(data) => {
for variant in data.variants.iter() {
let _ = TypeAttributeBuilder {
enable_flag: false, enable_bound: false
}
.build_from_attributes(&variant.attrs, traits)?;
for field in variant.fields.iter() {
let _ =
FieldAttributeBuilder.build_from_attributes(&field.attrs, traits)?;
field_types.push(&field.ty);
}
}
},
Data::Union(data) => {
for field in data.fields.named.iter() {
let _ = FieldAttributeBuilder.build_from_attributes(&field.attrs, traits)?;
field_types.push(&field.ty);
}
},
}
let ident = &ast.ident;
let bound_is_auto = matches!(type_attribute.bound, Bound::Auto);
let mut bound =
type_attribute.bound.into_where_predicates_by_generic_parameters_check_types(
&ast.generics.params,
&syn::parse2(quote_mixed!(::core::marker::Copy)).unwrap(),
&field_types,
&ast.ident,
&BOUND_EXCEPTIONS_COPY,
);
if bound_is_auto {
ctx.inherit_from(prerequisites(), &mut bound);
}
ctx.record(Trait::Copy, &bound);
let generics = crate::common::generics::with_predicates(ast.generics.clone(), bound);
let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
token_stream.extend(quote_mixed! {
#generated_impl_attributes
impl #impl_generics ::core::marker::Copy for #ident #ty_generics #where_clause {
}
});
Ok(())
}
}