如何为带有#[component]注解的结构体字段提供#[prop]属性？

咱们一步步来实现你想要的#[component]和#[prop]过程宏，核心思路是：#[prop]作为标记性占位宏，只负责标记哪些字段是组件的外部属性；而#[component]宏则负责解析结构体、识别带#[prop]的字段，最终生成对应的Props结构体并修改原组件结构体的逻辑。

1. 准备依赖 #

首先在你的Cargo.toml里添加过程宏必备的依赖：

[lib]
proc-macro = true

[dependencies]
syn = { version = "2.0", features = ["full", "extra-traits"] }
quote = "1.0"
proc-macro2 = "1.0"

• syn：负责解析Rust代码的Token流，把它转换成AST（抽象语法树）方便操作
• quote：负责把AST转换回Token流，生成最终的代码
• proc-macro2：是标准库proc_macro的替代，提供更灵活的Token处理

2. 实现#[prop]占位宏 #

这个宏很简单，它不需要做任何修改，只是作为标记让#[component]宏识别。直接返回输入的TokenStream即可：

use proc_macro::TokenStream;

#[proc_macro_attribute]
pub fn prop(_attribute: TokenStream, input: TokenStream) -> TokenStream {
    // 啥也不用改，直接返回原Token流，只起到标记作用
    input
}

3. 实现#[component]核心宏 #

这部分是重点，我们需要：

• 解析输入的结构体定义
• 遍历结构体字段，筛选出带有#[prop]属性的字段
• 生成一个对应的{结构体名}Props结构体，包含所有带#[prop]的字段
• （可选）给原组件结构体添加关联方法，比如from_props或者渲染逻辑

以下是完整实现代码：

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, DeriveInput, Fields, ItemStruct};

#[proc_macro_attribute]
pub fn component(_attribute: TokenStream, input: TokenStream) -> TokenStream {
    // 解析输入的结构体
    let input_struct = parse_macro_input!(input as ItemStruct);
    let struct_ident = input_struct.ident;
    let props_struct_ident = quote::format_ident!("{}Props", struct_ident);

    // 提取带有#[prop]属性的字段
    let prop_fields = match &input_struct.fields {
        Fields::Named(fields) => fields.named.iter().filter(|field| {
            // 检查字段是否带有#[prop]属性
            field.attrs.iter().any(|attr| {
                attr.path().is_ident("prop")
            })
        }).collect::<Vec<_>>(),
        _ => panic!("#[component] only supports structs with named fields"),
    };

    // 生成Props结构体的字段定义
    let props_fields = prop_fields.iter().map(|field| {
        let ident = &field.ident;
        let ty = &field.ty;
        quote! {
            pub #ident: #ty,
        }
    });

    // 生成Props结构体
    let props_struct = quote! {
        #[derive(Debug, Clone)]
        pub struct #props_struct_ident {
            #(#props_fields)*
        }
    };

    // 收集非prop字段，用于初始化原结构体
    let non_prop_fields = match &input_struct.fields {
        Fields::Named(fields) => fields.named.iter().filter(|field| {
            !field.attrs.iter().any(|attr| attr.path().is_ident("prop"))
        }).collect::<Vec<_>>(),
        _ => vec![],
    };

    // 生成原结构体的修改版本（保留原结构体，添加从Props创建实例的方法）
    let modified_struct = quote! {
        #input_struct

        impl #struct_ident {
            // 从Props创建组件实例
            pub fn from_props(props: #props_struct_ident) -> Self {
                Self {
                    // 初始化prop字段
                    #(#prop_fields.ident: props.#prop_fields.ident,)*
                    // 初始化非prop字段，这里用Default::default()作为示例
                    #(#non_prop_fields.ident: Default::default(),)*
                }
            }
        }
    };

    // 合并生成的代码并返回
    let output = quote! {
        #props_struct
        #modified_struct
    };

    output.into()
}

4. 使用示例 #

现在你就可以像你期望的那样使用这两个宏了：

use your_crate_name::{component, prop};

#[component]
struct Div {
    #[prop]
    color: String,
    clicked: bool
}

fn main() {
    let props = DivProps {
        color: "red".to_string(),
    };
    let div = Div::from_props(props);
    println!("Div color: {}, clicked: {}", div.color, div.clicked);
}

补充说明 #

• 如果你的组件需要更复杂的逻辑（比如渲染方法、属性验证），可以在impl #struct_ident里添加更多方法
• 对于非#[prop]的字段，示例中用了Default::default()初始化，你可以根据需求修改这部分逻辑，比如让用户手动传入，或者设置固定默认值
• 如果需要支持元组结构体或者单元结构体，可以修改prop_fields的匹配逻辑，不过通常组件结构体都是命名字段

内容的提问来源于stack exchange，提问作者pepsighan