Add a derive macro for Combine (#4325)

## Summary

Saves us some boilerplate when adding settings in the future.
This commit is contained in:
Charlie Marsh 2024-06-13 21:53:27 -07:00 committed by GitHub
parent 83067c1802
commit 74c05683bb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 71 additions and 123 deletions

View file

@ -0,0 +1,40 @@
use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, DeriveInput};
#[proc_macro_derive(CombineOptions)]
pub fn derive_combine(input: TokenStream) -> TokenStream {
let input = parse_macro_input!(input as DeriveInput);
impl_combine(&input)
}
fn impl_combine(ast: &DeriveInput) -> TokenStream {
let name = &ast.ident;
let fields = if let syn::Data::Struct(syn::DataStruct {
fields: syn::Fields::Named(ref fields),
..
}) = ast.data
{
&fields.named
} else {
unimplemented!();
};
let combines = fields.iter().map(|f| {
let name = &f.ident;
quote! {
#name: self.#name.combine(other.#name)
}
});
let gen = quote! {
impl crate::Combine for #name {
fn combine(self, other: #name) -> #name {
#name {
#(#combines),*
}
}
}
};
gen.into()
}