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

1use quote::quote;23pub fn impl_solidity_option<'a>(4	name: &proc_macro2::Ident,5	enum_options: impl Iterator<Item = &'a syn::Ident>,6) -> proc_macro2::TokenStream {7	let enum_options = enum_options.map(|opt| {8		let s = name.to_string() + "." + opt.to_string().as_str();9		let as_string = proc_macro2::Literal::string(s.as_str());10		quote!(#name::#opt => #as_string,)11	});12	quote!(13		#[cfg(feature = "stubgen")]14		impl ::evm_coder::solidity::SolidityEnum for #name {15			fn solidity_option(&self) -> &str {16				match self {17					#(#enum_options)*18				}19			}20		}21	)22}2324pub fn impl_enum_from_u8<'a>(25	name: &proc_macro2::Ident,26	enum_options: impl Iterator<Item = &'a syn::Ident>,27) -> proc_macro2::TokenStream {28	let error_str = format!("Value not convertible into enum \"{name}\"");29	let error_str = proc_macro2::Literal::string(&error_str);30	let enum_options = enum_options.enumerate().map(|(i, opt)| {31		let n = proc_macro2::Literal::u8_suffixed(i as u8);32		quote! {#n => Ok(#name::#opt),}33	});3435	quote!(36		impl TryFrom<u8> for #name {37			type Error = &'static str;3839			fn try_from(value: u8) -> ::core::result::Result<Self, Self::Error> {40				const err: &'static str = #error_str;41				match value {42					#(#enum_options)*43					_ => Err(err)44				}45			}46		}47	)48}4950pub fn impl_enum_abi_type(name: &syn::Ident, option_count: usize) -> proc_macro2::TokenStream {51	quote! {52		impl ::evm_coder::abi::AbiType for #name {53			const SIGNATURE: ::evm_coder::custom_signature::SignatureUnit = <u8 as ::evm_coder::abi::AbiType>::SIGNATURE;54			const FIELDS_COUNT: usize = #option_count;5556			fn is_dynamic() -> bool {57				<u8 as ::evm_coder::abi::AbiType>::is_dynamic()58			}59			fn size() -> usize {60				<u8 as ::evm_coder::abi::AbiType>::size()61			}62		}63	}64}6566pub fn impl_enum_abi_read(name: &syn::Ident) -> proc_macro2::TokenStream {67	quote!(68		impl ::evm_coder::abi::AbiRead for #name {69			fn abi_read(reader: &mut ::evm_coder::abi::AbiReader) -> ::evm_coder::execution::Result<Self> {70				Ok(71					<u8 as ::evm_coder::abi::AbiRead>::abi_read(reader)?72						.try_into()?73				)74			}75		}76	)77}7879pub fn impl_enum_abi_write(name: &syn::Ident) -> proc_macro2::TokenStream {80	quote!(81		impl ::evm_coder::abi::AbiWrite for #name {82			fn abi_write(&self, writer: &mut ::evm_coder::abi::AbiWriter) {83				::evm_coder::abi::AbiWrite::abi_write(&(*self as u8), writer);84			}85		}86	)87}8889pub fn impl_enum_solidity_type_name(name: &syn::Ident) -> proc_macro2::TokenStream {90	quote!(91		#[cfg(feature = "stubgen")]92		impl ::evm_coder::solidity::SolidityTypeName for #name {93			fn solidity_name(94				writer: &mut impl ::core::fmt::Write,95				tc: &::evm_coder::solidity::TypeCollector,96			) -> ::core::fmt::Result {97				write!(writer, "{}", tc.collect_struct::<Self>())98			}99100			fn is_simple() -> bool {101				true102			}103104			fn solidity_default(105				writer: &mut impl ::core::fmt::Write,106				tc: &::evm_coder::solidity::TypeCollector,107			) -> ::core::fmt::Result {108				write!(writer, "{}", <#name as ::evm_coder::solidity::SolidityEnum>::solidity_option(&<#name>::default()))109			}110		}111	)112}113114pub fn impl_enum_solidity_struct_collect<'a>(115	name: &syn::Ident,116	enum_options: impl Iterator<Item = &'a syn::Ident>,117	option_count: usize,118	enum_options_docs: impl Iterator<Item = syn::Result<Vec<proc_macro2::TokenStream>>>,119	docs: &[proc_macro2::TokenStream],120) -> proc_macro2::TokenStream {121	let string_name = name.to_string();122	let enum_options = enum_options123		.zip(enum_options_docs)124		.enumerate()125		.map(|(i, (opt, doc))| {126			let opt = proc_macro2::Literal::string(opt.to_string().as_str());127			let doc = doc.expect("Doc parsing error");128			let comma = if i != option_count - 1 { "," } else { "" };129			quote! {130				#(#doc)*131				writeln!(str, "\t{}{}", #opt, #comma).expect("Enum format option");132			}133		});134135	quote!(136		#[cfg(feature = "stubgen")]137		impl ::evm_coder::solidity::StructCollect for #name {138			fn name() -> String {139				#string_name.into()140			}141142			fn declaration() -> String {143				use std::fmt::Write;144145				let mut str = String::new();146				#(#docs)*147				writeln!(str, "enum {} {{", <Self as ::evm_coder::solidity::StructCollect>::name()).unwrap();148				#(#enum_options)*149				writeln!(str, "}}").unwrap();150				str151			}152		}153	)154}155156pub fn check_and_count_options(de: &syn::DataEnum) -> syn::Result<usize> {157	let mut count = 0;158	for v in de.variants.iter() {159		if !v.fields.is_empty() {160			return Err(syn::Error::new(161				v.ident.span(),162				"Enumeration parameters should not have fields",163			));164		} else if v.discriminant.is_some() {165			return Err(syn::Error::new(166				v.ident.span(),167				"Enumeration options should not have an explicit specified value",168			));169		} else {170			count += 1;171		}172	}173174	Ok(count)175}176177pub fn check_repr_u8(name: &syn::Ident, attrs: &[syn::Attribute]) -> syn::Result<()> {178	let mut has_repr = false;179	for attr in attrs.iter() {180		if attr.path.is_ident("repr") {181			has_repr = true;182			let meta = attr.parse_meta()?;183			check_meta_u8(&meta)?;184		}185	}186187	if !has_repr {188		return Err(syn::Error::new(name.span(), "Enum is not \"repr(u8)\""));189	}190191	Ok(())192}193194fn check_meta_u8(meta: &syn::Meta) -> Result<(), syn::Error> {195	if let syn::Meta::List(p) = meta {196		for nm in p.nested.iter() {197			if let syn::NestedMeta::Meta(syn::Meta::Path(p)) = nm {198				if !p.is_ident("u8") {199					return Err(syn::Error::new(200						p.segments201							.first()202							.expect("repr segments are empty")203							.ident204							.span(),205						"Enum is not \"repr(u8)\"",206					));207				}208			}209		}210	}211	Ok(())212}