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

difftreelog

refactor change panic to syn::Error

Trubnikov Sergey2022-11-10parent: #4c95237.patch.diff
in: master

3 files changed

modifiedcrates/evm-coder/procedural/src/abi_derive.rsdiffbeforeafterboth
--- a/crates/evm-coder/procedural/src/abi_derive.rs
+++ b/crates/evm-coder/procedural/src/abi_derive.rs
@@ -1,15 +1,15 @@
 use quote::quote;
 
-pub(crate) fn impl_abi_macro(ast: &syn::DeriveInput) -> proc_macro2::TokenStream {
+pub(crate) fn impl_abi_macro(ast: &syn::DeriveInput) -> syn::Result<proc_macro2::TokenStream> {
 	// dbg!(ast);
 	let name = &ast.ident;
 	let can_be_plcaed_in_vec = impl_can_be_placed_in_vec(name);
-	let abi_type = impl_abi_type(ast);
+	let abi_type = impl_abi_type(ast)?;
 	// println!("{}", abi_type);
-	quote! {
+	Ok(quote! {
 		#can_be_plcaed_in_vec
 		#abi_type
-	}
+	})
 }
 
 fn impl_can_be_placed_in_vec(ident: &syn::Ident) -> proc_macro2::TokenStream {
@@ -22,23 +22,23 @@
 	&field.ty
 }
 
-fn impl_abi_type(ast: &syn::DeriveInput) -> proc_macro2::TokenStream {
+fn impl_abi_type(ast: &syn::DeriveInput) -> syn::Result<proc_macro2::TokenStream> {
 	let name = &ast.ident;
 	let (fields, params_count) = match &ast.data {
 		syn::Data::Struct(ds) => match ds.fields {
-			syn::Fields::Named(ref fields) => (
+			syn::Fields::Named(ref fields) => Ok((
 				fields.named.iter().map(map_field_to_type),
 				fields.named.len(),
-			),
-			syn::Fields::Unnamed(ref fields) => (
+			)),
+			syn::Fields::Unnamed(ref fields) => Ok((
 				fields.unnamed.iter().map(map_field_to_type),
 				fields.unnamed.len(),
-			),
-			syn::Fields::Unit => unimplemented!("Unit structs not supported"),
+			)),
+			syn::Fields::Unit => Err(syn::Error::new(name.span(), "Unit structs not supported")),
 		},
-		syn::Data::Enum(_) => unimplemented!("Enums not supported"),
-		syn::Data::Union(_) => unimplemented!("Unions not supported"),
-	};
+		syn::Data::Enum(_) => Err(syn::Error::new(name.span(), "Enums not supported")),
+		syn::Data::Union(_) => Err(syn::Error::new(name.span(), "Unions not supported")),
+	}?;
 
 	let mut params_signature = {
 		let fields = fields.clone();
@@ -46,13 +46,16 @@
 			#(nameof(<#fields as ::evm_coder::abi::AbiType>::SIGNATURE) fixed(","))*
 		)
 	};
-	if params_count > 0 {
-		params_signature.extend(quote!(shift_left(1)))
+
+	if params_count == 0 {
+		return Err(syn::Error::new(name.span(), "Empty structs not supported"));
 	};
 
+	params_signature.extend(quote!(shift_left(1)));
+
 	let fields_for_dynamic = fields.clone();
 
-	quote! {
+	Ok(quote! {
 		impl ::evm_coder::abi::AbiType for #name {
 			const SIGNATURE: ::evm_coder::custom_signature::SignatureUnit = ::evm_coder::make_signature!(
 				new fixed("(")
@@ -69,5 +72,5 @@
 				0 #(+ <#fields as ::evm_coder::abi::AbiType>::size())*
 			}
 		}
-	}
+	})
 }
modifiedcrates/evm-coder/procedural/src/lib.rsdiffbeforeafterboth
247#[proc_macro_derive(AbiCoder)]247#[proc_macro_derive(AbiCoder)]
248pub fn abi_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {248pub fn abi_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
249 let ast = syn::parse(input).unwrap();249 let ast = syn::parse(input).unwrap();
250 let ts = abi_derive::impl_abi_macro(&ast);250 let ts = match abi_derive::impl_abi_macro(&ast) {
251 Ok(e) => e,
252 Err(e) => e.to_compile_error(),
253 };
251 // println!("{}", &ts);254 // println!("{}", &ts);
252 ts.into()255 ts.into()
253}256}
modifiedcrates/evm-coder/tests/abi_derive_generation.rsdiffbeforeafterboth
--- a/crates/evm-coder/tests/abi_derive_generation.rs
+++ b/crates/evm-coder/tests/abi_derive_generation.rs
@@ -4,8 +4,9 @@
 	abi::{AbiType},
 };
 
-#[derive(AbiCoder)]
-struct TypeStructUnit {}
+// TODO: move to build_failed tests
+// #[derive(AbiCoder)]
+// struct TypeStructUnit {}
 
 #[derive(AbiCoder)]
 struct TypeStruct1SimpleParam {
@@ -70,10 +71,6 @@
 #[test]
 fn impl_abi_type_signature() {
 	assert_eq!(
-		<TypeStructUnit as AbiType>::SIGNATURE.as_str().unwrap(),
-		"()"
-	);
-	assert_eq!(
 		<TypeStruct1SimpleParam as AbiType>::SIGNATURE
 			.as_str()
 			.unwrap(),
@@ -137,7 +134,6 @@
 
 #[test]
 fn impl_abi_type_is_dynamic() {
-	assert_eq!(<TypeStructUnit as AbiType>::is_dynamic(), false);
 	assert_eq!(<TypeStruct1SimpleParam as AbiType>::is_dynamic(), false);
 	assert_eq!(<TypeStruct1DynamicParam as AbiType>::is_dynamic(), true);
 	assert_eq!(<TypeStruct2SimpleParam as AbiType>::is_dynamic(), false);
@@ -168,7 +164,6 @@
 #[test]
 fn impl_abi_type_size() {
 	const ABI_ALIGNMENT: usize = 32;
-	assert_eq!(<TypeStructUnit as AbiType>::size(), 0);
 	assert_eq!(<TypeStruct1SimpleParam as AbiType>::size(), ABI_ALIGNMENT);
 	assert_eq!(<TypeStruct1DynamicParam as AbiType>::size(), ABI_ALIGNMENT);
 	assert_eq!(