mirror of https://github.com/astral-sh/ruff
38 lines
1.1 KiB
Rust
38 lines
1.1 KiB
Rust
use quote::{quote, quote_spanned};
|
|
use syn::spanned::Spanned;
|
|
use syn::{Data, DataStruct, DeriveInput};
|
|
|
|
pub(crate) fn derive_impl(input: DeriveInput) -> syn::Result<proc_macro2::TokenStream> {
|
|
let DeriveInput { ident, data, .. } = input;
|
|
|
|
match data {
|
|
Data::Struct(DataStruct { fields, .. }) => {
|
|
let output: Vec<_> = fields
|
|
.members()
|
|
.map(|member| {
|
|
|
|
quote_spanned!(
|
|
member.span() => ty_combine::Combine::combine_with(&mut self.#member, other.#member)
|
|
)
|
|
})
|
|
.collect();
|
|
|
|
Ok(quote! {
|
|
#[automatically_derived]
|
|
impl ty_combine::Combine for #ident {
|
|
#[allow(deprecated)]
|
|
fn combine_with(&mut self, other: Self) {
|
|
#(
|
|
#output
|
|
);*
|
|
}
|
|
}
|
|
})
|
|
}
|
|
_ => Err(syn::Error::new(
|
|
ident.span(),
|
|
"Can only derive Combine from structs.",
|
|
)),
|
|
}
|
|
}
|