git.delta.rocks / unique-network / refs/commits / 3d5704a1dece

difftreelog

source

crates/evm-coder/procedural/src/abi_derive.rs2.0 KiBsourcehistory
1use quote::quote;23pub(crate) fn impl_abi_macro(ast: &syn::DeriveInput) -> syn::Result<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	Ok(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 map_field_to_type<'a>(field: &'a syn::Field) -> &'a syn::Type {22	&field.ty23}2425fn impl_abi_type(ast: &syn::DeriveInput) -> syn::Result<proc_macro2::TokenStream> {26	let name = &ast.ident;27	let (fields, params_count) = match &ast.data {28		syn::Data::Struct(ds) => match ds.fields {29			syn::Fields::Named(ref fields) => Ok((30				fields.named.iter().map(map_field_to_type),31				fields.named.len(),32			)),33			syn::Fields::Unnamed(ref fields) => Ok((34				fields.unnamed.iter().map(map_field_to_type),35				fields.unnamed.len(),36			)),37			syn::Fields::Unit => Err(syn::Error::new(name.span(), "Unit structs not supported")),38		},39		syn::Data::Enum(_) => Err(syn::Error::new(name.span(), "Enums not supported")),40		syn::Data::Union(_) => Err(syn::Error::new(name.span(), "Unions not supported")),41	}?;4243	let mut params_signature = {44		let fields = fields.clone();45		quote!(46			#(nameof(<#fields as ::evm_coder::abi::AbiType>::SIGNATURE) fixed(","))*47		)48	};4950	if params_count == 0 {51		return Err(syn::Error::new(name.span(), "Empty structs not supported"));52	};5354	params_signature.extend(quote!(shift_left(1)));5556	let fields_for_dynamic = fields.clone();5758	Ok(quote! {59		impl ::evm_coder::abi::AbiType for #name {60			const SIGNATURE: ::evm_coder::custom_signature::SignatureUnit = ::evm_coder::make_signature!(61				new fixed("(")62				#params_signature63				fixed(")")64			);65			fn is_dynamic() -> bool {66				false67				#(68					|| <#fields_for_dynamic as ::evm_coder::abi::AbiType>::is_dynamic()69				)*70			}71			fn size() -> usize {72				0 #(+ <#fields as ::evm_coder::abi::AbiType>::size())*73			}74		}75	})76}