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

1mod derive_enum;2mod derive_struct;34use quote::quote;5use derive_struct::*;6use derive_enum::*;78pub(crate) fn impl_abi_macro(ast: &syn::DeriveInput) -> syn::Result<proc_macro2::TokenStream> {9	let name = &ast.ident;10	match &ast.data {11		syn::Data::Struct(ds) => expand_struct(ds, ast),12		syn::Data::Enum(de) => expand_enum(de, ast),13		syn::Data::Union(_) => Err(syn::Error::new(name.span(), "Unions not supported")),14	}15}1617fn expand_struct(18	ds: &syn::DataStruct,19	ast: &syn::DeriveInput,20) -> syn::Result<proc_macro2::TokenStream> {21	let name = &ast.ident;22	let docs = extract_docs(&ast.attrs, false)?;23	let (is_named_fields, field_names, field_types, field_docs, params_count) = match ds.fields {24		syn::Fields::Named(ref fields) => Ok((25			true,26			fields.named.iter().enumerate().map(map_field_to_name),27			fields.named.iter().map(map_field_to_type),28			fields.named.iter().map(map_field_to_doc),29			fields.named.len(),30		)),31		syn::Fields::Unnamed(ref fields) => Ok((32			false,33			fields.unnamed.iter().enumerate().map(map_field_to_name),34			fields.unnamed.iter().map(map_field_to_type),35			fields.unnamed.iter().map(map_field_to_doc),36			fields.unnamed.len(),37		)),38		syn::Fields::Unit => Err(syn::Error::new(name.span(), "Unit structs not supported")),39	}?;4041	if params_count == 0 {42		return Err(syn::Error::new(name.span(), "Empty structs not supported"));43	};4445	let tuple_type = tuple_type(field_types.clone());46	let tuple_ref_type = tuple_ref_type(field_types.clone());47	let tuple_data = tuple_data_as_ref(is_named_fields, field_names.clone());48	let tuple_names = tuple_names(is_named_fields, field_names.clone());49	let struct_from_tuple = struct_from_tuple(name, is_named_fields, field_names.clone());5051	let can_be_plcaed_in_vec = impl_can_be_placed_in_vec(name);52	let abi_type = impl_struct_abi_type(name, tuple_type.clone(), params_count);53	let abi_read = impl_struct_abi_read(name, tuple_type, tuple_names, struct_from_tuple);54	let abi_write = impl_struct_abi_write(name, is_named_fields, tuple_ref_type, tuple_data);55	let solidity_type = impl_struct_solidity_type(name, field_types.clone(), params_count);56	let solidity_type_name =57		impl_struct_solidity_type_name(name, field_types.clone(), params_count);58	let solidity_struct_collect =59		impl_struct_solidity_struct_collect(name, field_names, field_types, field_docs, &docs)?;6061	Ok(quote! {62		#can_be_plcaed_in_vec63		#abi_type64		#abi_read65		#abi_write66		#solidity_type67		#solidity_type_name68		#solidity_struct_collect69	})70}7172fn expand_enum(73	de: &syn::DataEnum,74	ast: &syn::DeriveInput,75) -> syn::Result<proc_macro2::TokenStream> {76	let name = &ast.ident;77	check_repr_u8(name, &ast.attrs)?;78	let docs = extract_docs(&ast.attrs, false)?;79	let option_count = check_and_count_options(de)?;80	let enum_options = de.variants.iter().map(|v| &v.ident);81	let enum_options_docs = de.variants.iter().map(|v| extract_docs(&v.attrs, true));8283	let from = impl_enum_from_u8(name, enum_options.clone());84	let solidity_option = impl_solidity_option(name, enum_options.clone());85	let can_be_plcaed_in_vec = impl_can_be_placed_in_vec(name);86	let abi_type = impl_enum_abi_type(name, option_count);87	let abi_read = impl_enum_abi_read(name);88	let abi_write = impl_enum_abi_write(name);89	let solidity_type_name = impl_enum_solidity_type_name(name);90	let solidity_struct_collect = impl_enum_solidity_struct_collect(91		name,92		enum_options,93		option_count,94		enum_options_docs,95		&docs,96	);9798	Ok(quote! {99		#from100		#solidity_option101		#can_be_plcaed_in_vec102		#abi_type103		#abi_read104		#abi_write105		#solidity_type_name106		#solidity_struct_collect107	})108}109110fn extract_docs(111	attrs: &[syn::Attribute],112	is_field_doc: bool,113) -> syn::Result<Vec<proc_macro2::TokenStream>> {114	attrs115		.iter()116		.filter_map(|attr| {117			if let Some(ps) = attr.path.segments.first() {118				if ps.ident == "doc" {119					let meta = match attr.parse_meta() {120						Ok(meta) => meta,121						Err(e) => return Some(Err(e)),122					};123					match meta {124						syn::Meta::NameValue(mnv) => match &mnv.lit {125							syn::Lit::Str(ls) => return Some(Ok(ls.value())),126							_ => unreachable!(),127						},128						_ => unreachable!(),129					}130				}131			}132			None133		})134		.enumerate()135		.map(|(i, doc)| {136			let doc = doc?;137			let doc = doc.trim();138			let dev = if i == 0 { " @dev" } else { "" };139			let tab = if is_field_doc { "\t" } else { "" };140			Ok(quote! {141				writeln!(str, "{}///{} {}", #tab, #dev, #doc).unwrap();142			})143		})144		.collect()145}