如何解析自定义 rust proc_macro 属性中的其他属性？

Question

我正在编写一个 proc_macro 属性，该属性将字段添加到结构中，并且还实现了我的特征（以及其他通过添加#[derive(...)] ）到扩展结构

这是我想做的事情的简化版本：

#[foo("some value")]
#[derive(Debug)]
struct A {
    bar: u32,
}

扩展后：

#[derive(Debug, Default, Serialize)]
struct A {
    foo: u64
    bar: u32,
}

impl FooTrait for A {
    ...
}

我如何解析 derive 属性以便添加Debug特性以及我使用foo proc_macro 添加的特性的所有 rest？

Answer 1

使用syn ，检查并更改ItemStruct::attrs ：

use std::collections::HashSet;

use proc_macro2::{Delimiter, TokenTree};
use quote::ToTokens;
use syn::parse::Parser;
use syn::punctuated::Punctuated;

#[proc_macro_attribute]
pub fn foo(
    attr: proc_macro::TokenStream,
    input: proc_macro::TokenStream,
) -> proc_macro::TokenStream {
    let mut input = syn::parse_macro_input!(input as syn::ItemStruct);

    let mut all_derived_traits = HashSet::new();
    for i in 0..input.attrs.len() {
        if !input.attrs[i].path.is_ident("derive") {
            continue;
        }

        let derive = input.attrs.remove(i);
        let mut tokens = derive.tokens.clone().into_iter();
        match [tokens.next(), tokens.next()] {
            [Some(TokenTree::Group(group)), None]
                if group.delimiter() == Delimiter::Parenthesis =>
            {
                match Punctuated::<syn::Path, syn::Token![,]>::parse_terminated
                    .parse2(group.stream())
                {
                    Ok(derived_traits) => all_derived_traits.extend(derived_traits),
                    Err(e) => return e.into_compile_error().into(),
                }
            }
            _ => {
                return syn::Error::new_spanned(derive, "malformed derive")
                    .into_compile_error()
                    .into()
            }
        }
    }
    
    all_derived_traits.extend([
        syn::parse_quote!(Default),
        syn::parse_quote!(Serialize),
    ]);
    
    let all_derived_traits = all_derived_traits.into_iter();
    input.attrs.push(syn::parse_quote! {
        #[derive( #(#all_derived_traits),* )]
    });
    
    input.into_token_stream().into()
}

首先，我们将所有derive()属性收集到哈希集中，因此如果它们已经存在，我们将不会生成派生。 这个逻辑并不完美：例如，我们不会将std::default::Default标识为与Default相同。 但它应该足够了。 然后我们添加我们的特征并重新生成派生。 请注意，多个#[derive()]行是统一的，并且可以对特征进行排序，但这并不重要。