git.delta.rocks / unique-network / refs/commits / 14137a17488f

1use super::extract_docs;2use quote::quote;34pub fn tuple_type<'a>(5	field_types: impl Iterator<Item = &'a syn::Type> + Clone,6) -> proc_macro2::TokenStream {7	let field_types = field_types.map(|ty| quote!(#ty,));8	quote! {(#(#field_types)*)}9}1011pub fn tuple_ref_type<'a>(12	field_types: impl Iterator<Item = &'a syn::Type> + Clone,13) -> proc_macro2::TokenStream {14	let field_types = field_types.map(|ty| quote!(&#ty,));15	quote! {(#(#field_types)*)}16}1718pub fn tuple_data_as_ref(19	is_named_fields: bool,20	field_names: impl Iterator<Item = syn::Ident> + Clone,21) -> proc_macro2::TokenStream {22	let field_names = field_names.enumerate().map(|(i, field)| {23		if is_named_fields {24			quote!(&self.#field,)25		} else {26			let field = proc_macro2::Literal::usize_unsuffixed(i);27			quote!(&self.#field,)28		}29	});30	quote! {(#(#field_names)*)}31}3233pub fn tuple_names(34	is_named_fields: bool,35	field_names: impl Iterator<Item = syn::Ident> + Clone,36) -> proc_macro2::TokenStream {37	let field_names = field_names.enumerate().map(|(i, field)| {38		if is_named_fields {39			quote!(#field,)40		} else {41			let field = proc_macro2::Ident::new(42				format!("field{}", i).as_str(),43				proc_macro2::Span::call_site(),44			);45			quote!(#field,)46		}47	});48	quote! {(#(#field_names)*)}49}5051pub fn struct_from_tuple(52	name: &syn::Ident,53	is_named_fields: bool,54	field_names: impl Iterator<Item = syn::Ident> + Clone,55) -> proc_macro2::TokenStream {56	let field_names = field_names.enumerate().map(|(i, field)| {57		if is_named_fields {58			quote!(#field,)59		} else {60			let field = proc_macro2::Ident::new(61				format!("field{}", i).as_str(),62				proc_macro2::Span::call_site(),63			);64			quote!(#field,)65		}66	});6768	if is_named_fields {69		quote! {#name {#(#field_names)*}}70	} else {71		quote! {#name (#(#field_names)*)}72	}73}7475pub fn map_field_to_name(field: (usize, &syn::Field)) -> syn::Ident {76	match field.1.ident.as_ref() {77		Some(name) => name.clone(),78		None => {79			let mut name = "field".to_string();80			name.push_str(field.0.to_string().as_str());81			syn::Ident::new(name.as_str(), proc_macro2::Span::call_site())82		}83	}84}8586pub fn map_field_to_type(field: &syn::Field) -> &syn::Type {87	&field.ty88}8990pub fn map_field_to_doc(field: &syn::Field) -> syn::Result<Vec<proc_macro2::TokenStream>> {91	extract_docs(&field.attrs, true)92}9394pub fn impl_can_be_placed_in_vec(ident: &syn::Ident) -> proc_macro2::TokenStream {95	quote! {96		impl ::evm_coder::sealed::CanBePlacedInVec for #ident {}97	}98}99100pub fn impl_struct_abi_type(101	name: &syn::Ident,102	tuple_type: proc_macro2::TokenStream,103	fields_count: usize,104) -> proc_macro2::TokenStream {105	quote! {106		impl ::evm_coder::abi::AbiType for #name {107			const SIGNATURE: ::evm_coder::custom_signature::SignatureUnit = <#tuple_type as ::evm_coder::abi::AbiType>::SIGNATURE;108			const FIELDS_COUNT: usize = #fields_count;109			fn is_dynamic() -> bool {110				<#tuple_type as ::evm_coder::abi::AbiType>::is_dynamic()111			}112			fn size() -> usize {113				<#tuple_type as ::evm_coder::abi::AbiType>::size()114			}115		}116	}117}118119pub fn impl_struct_abi_read(120	name: &syn::Ident,121	tuple_type: proc_macro2::TokenStream,122	tuple_names: proc_macro2::TokenStream,123	struct_from_tuple: proc_macro2::TokenStream,124) -> proc_macro2::TokenStream {125	quote!(126		impl ::evm_coder::abi::AbiRead for #name {127			fn abi_read(reader: &mut ::evm_coder::abi::AbiReader) -> ::evm_coder::execution::Result<Self> {128				let #tuple_names = <#tuple_type as ::evm_coder::abi::AbiRead>::abi_read(reader)?;129				Ok(#struct_from_tuple)130			}131		}132	)133}134135pub fn impl_struct_abi_write(136	name: &syn::Ident,137	_is_named_fields: bool,138	tuple_type: proc_macro2::TokenStream,139	tuple_data: proc_macro2::TokenStream,140) -> proc_macro2::TokenStream {141	quote!(142		impl ::evm_coder::abi::AbiWrite for #name {143			fn abi_write(&self, writer: &mut ::evm_coder::abi::AbiWriter) {144				<#tuple_type as ::evm_coder::abi::AbiWrite>::abi_write(&#tuple_data, writer)145			}146		}147	)148}149150pub fn impl_struct_solidity_type<'a>(151	name: &syn::Ident,152	field_types: impl Iterator<Item = &'a syn::Type> + Clone,153	params_count: usize,154) -> proc_macro2::TokenStream {155	let len = proc_macro2::Literal::usize_suffixed(params_count);156	quote! {157		#[cfg(feature = "stubgen")]158		impl ::evm_coder::solidity::SolidityType for #name {159			fn names(tc: &::evm_coder::solidity::TypeCollector) -> Vec<String> {160				let mut collected =161					Vec::with_capacity(<Self as ::evm_coder::solidity::SolidityType>::len());162				#({163					let mut out = String::new();164					<#field_types as ::evm_coder::solidity::SolidityTypeName>::solidity_name(&mut out, tc)165						.expect("no fmt error");166					collected.push(out);167				})*168				collected169			}170171			fn len() -> usize {172				#len173			}174		}175	}176}177178pub fn impl_struct_solidity_type_name<'a>(179	name: &syn::Ident,180	field_types: impl Iterator<Item = &'a syn::Type> + Clone,181	params_count: usize,182) -> proc_macro2::TokenStream {183	let arg_dafaults = field_types.enumerate().map(|(i, ty)| {184		let mut defult_value = quote!(<#ty as ::evm_coder::solidity::SolidityTypeName185			>::solidity_default(writer, tc)?;);186		let last_item = params_count - 1;187		if i != last_item {188			defult_value.extend(quote! {write!(writer, ",")?;})189		}190		defult_value191	});192193	quote! {194		#[cfg(feature = "stubgen")]195		impl ::evm_coder::solidity::SolidityTypeName for #name {196			fn solidity_name(197				writer: &mut impl ::core::fmt::Write,198				tc: &::evm_coder::solidity::TypeCollector,199			) -> ::core::fmt::Result {200				write!(writer, "{}", tc.collect_struct::<Self>())201			}202203			fn is_simple() -> bool {204				false205			}206207			fn solidity_default(208				writer: &mut impl ::core::fmt::Write,209				tc: &::evm_coder::solidity::TypeCollector,210			) -> ::core::fmt::Result {211				write!(writer, "{}(", tc.collect_struct::<Self>())?;212213				#(#arg_dafaults)*214215				write!(writer, ")")216			}217		}218	}219}220221pub fn impl_struct_solidity_struct_collect<'a>(222	name: &syn::Ident,223	field_names: impl Iterator<Item = proc_macro2::Ident> + Clone,224	field_types: impl Iterator<Item = &'a syn::Type> + Clone,225	field_docs: impl Iterator<Item = syn::Result<Vec<proc_macro2::TokenStream>>> + Clone,226	docs: &[proc_macro2::TokenStream],227) -> syn::Result<proc_macro2::TokenStream> {228	let string_name = name.to_string();229	let name_type = field_names230		.into_iter()231		.zip(field_types)232		.zip(field_docs)233		.map(|((name, ty), doc)| {234			let field_docs = doc.expect("Doc parse error");235			let name = format!("{}", name);236			quote!(237				#(#field_docs)*238				write!(str, "\t{} ", <#ty as ::evm_coder::solidity::StructCollect>::name()).unwrap();239				writeln!(str, "{};", #name).unwrap();240			)241		});242243	Ok(quote! {244		#[cfg(feature = "stubgen")]245		impl ::evm_coder::solidity::StructCollect for #name {246			fn name() -> String {247				#string_name.into()248			}249250			fn declaration() -> String {251				use std::fmt::Write;252253				let mut str = String::new();254				#(#docs)*255				writeln!(str, "struct {} {{", Self::name()).unwrap();256				#(#name_type)*257				writeln!(str, "}}").unwrap();258				str259			}260		}261	})262}