From 7809a8cbfbaf41fcc81de980c903e11f08bd1b12 Mon Sep 17 00:00:00 2001 From: 魏曹先生 <1992414357@qq.com> Date: Sun, 29 Mar 2026 01:18:25 +0800 Subject: Generate struct names from function names in chain and renderer macros --- mingling_macros/src/renderer.rs | 27 ++++++--------------------- 1 file changed, 6 insertions(+), 21 deletions(-) (limited to 'mingling_macros/src/renderer.rs') diff --git a/mingling_macros/src/renderer.rs b/mingling_macros/src/renderer.rs index 14c26df..54a2526 100644 --- a/mingling_macros/src/renderer.rs +++ b/mingling_macros/src/renderer.rs @@ -5,23 +5,8 @@ use proc_macro::TokenStream; use quote::quote; -use syn::parse::{Parse, ParseStream}; use syn::spanned::Spanned; -use syn::{ - FnArg, Ident, ItemFn, Pat, PatType, ReturnType, Signature, Type, TypePath, parse_macro_input, -}; - -/// Parses the renderer attribute arguments -struct RendererAttribute { - struct_name: Ident, -} - -impl Parse for RendererAttribute { - fn parse(input: ParseStream) -> syn::Result { - let struct_name = input.parse()?; - Ok(RendererAttribute { struct_name }) - } -} +use syn::{FnArg, ItemFn, Pat, PatType, ReturnType, Signature, Type, TypePath, parse_macro_input}; /// Extracts the previous type and parameter name from function arguments fn extract_previous_info(sig: &Signature) -> syn::Result<(Pat, TypePath)> { @@ -74,11 +59,7 @@ fn extract_return_type(sig: &Signature) -> syn::Result<()> { } } -pub fn renderer_attr(attr: TokenStream, item: TokenStream) -> TokenStream { - // Parse the attribute arguments - let renderer_attr = parse_macro_input!(attr as RendererAttribute); - let struct_name = renderer_attr.struct_name; - +pub fn renderer_attr(item: TokenStream) -> TokenStream { // Parse the function item let input_fn = parse_macro_input!(item as ItemFn); @@ -115,6 +96,10 @@ pub fn renderer_attr(attr: TokenStream, item: TokenStream) -> TokenStream { // Get function name let fn_name = &input_fn.sig.ident; + // Generate struct name from function name using pascal_case + let pascal_case_name = just_fmt::pascal_case!(fn_name.to_string()); + let struct_name = syn::Ident::new(&pascal_case_name, fn_name.span()); + // Register the renderer in the global list let renderer_entry = quote! { #struct_name => #previous_type, -- cgit