git.delta.rocks / unique-network / refs/commits / 7cb20157d174

difftreelog

source

crates/evm-coder/procedural/src/abi_derive.rs1.7 KiBsourcehistory
1use quote::quote;23pub(crate) fn impl_abi_macro(ast: &syn::DeriveInput) -> proc_macro2::TokenStream {4	// dbg!(ast);5	let name = &ast.ident;6	let can_be_plcaed_in_vec = impl_can_be_placed_in_vec(name);7	let abi_type = impl_abi_type(ast);8	println!("{}", abi_type);9	quote! {10		#can_be_plcaed_in_vec11		#abi_type12	}13}1415fn impl_can_be_placed_in_vec(ident: &syn::Ident) -> proc_macro2::TokenStream {16	quote! {17		impl ::evm_coder::abi::sealed::CanBePlacedInVec for #ident {}18	}19}2021fn impl_abi_type(ast: &syn::DeriveInput) -> proc_macro2::TokenStream {22	let name = &ast.ident;23	let (fields, params_count) = match &ast.data {24		syn::Data::Struct(ds) => match ds.fields {25			syn::Fields::Named(ref fields) => (26				fields.named.iter().map(|field| &field.ty),27				fields.named.len(),28			),29			syn::Fields::Unnamed(_) => todo!(),30			syn::Fields::Unit => unimplemented!("Unit structs not supported"),31		},32		syn::Data::Enum(_) => unimplemented!("Enums not supported"),33		syn::Data::Union(_) => unimplemented!("Unions not supported"),34	};3536	let mut params_signature = {37		let fields = fields.clone();38		quote!(39			#(nameof(<#fields as ::evm_coder::abi::AbiType>::SIGNATURE) fixed(","))*40		)41	};42	if params_count > 0 {43		params_signature.extend(quote!(shift_left(1)))44	};4546	let fields_for_dynamic = fields.clone();4748	quote! {49		impl ::evm_coder::abi::AbiType for #name {50			const SIGNATURE: ::evm_coder::custom_signature::SignatureUnit = ::evm_coder::make_signature!(51				new fixed("(")52				#params_signature53				fixed(")")54			);55			fn is_dynamic() -> bool {56				false57				#(58					|| <#fields_for_dynamic as ::evm_coder::abi::AbiType>::is_dynamic()59				)*60			}61			fn size() -> usize {62				0 #(+ <#fields as ::evm_coder::abi::AbiType>::size())*63			}64		}65	}66}