git.delta.rocks / unique-network / refs/commits / 4c9523749b30

difftreelog

feat add derive macro AbiCoder for unnamed struct to implement AbiType

Trubnikov Sergey2022-11-10parent: #7cb2015.patch.diff
in: master

3 files changed

modifiedcrates/evm-coder/procedural/src/abi_derive.rsdiffbeforeafterboth
5 let name = &ast.ident;5 let name = &ast.ident;
6 let can_be_plcaed_in_vec = impl_can_be_placed_in_vec(name);6 let can_be_plcaed_in_vec = impl_can_be_placed_in_vec(name);
7 let abi_type = impl_abi_type(ast);7 let abi_type = impl_abi_type(ast);
8 println!("{}", abi_type);8 // println!("{}", abi_type);
9 quote! {9 quote! {
10 #can_be_plcaed_in_vec10 #can_be_plcaed_in_vec
11 #abi_type11 #abi_type
18 }18 }
19}19}
20
21fn map_field_to_type<'a>(field: &'a syn::Field) -> &'a syn::Type {
22 &field.ty
23}
2024
21fn impl_abi_type(ast: &syn::DeriveInput) -> proc_macro2::TokenStream {25fn impl_abi_type(ast: &syn::DeriveInput) -> proc_macro2::TokenStream {
22 let name = &ast.ident;26 let name = &ast.ident;
23 let (fields, params_count) = match &ast.data {27 let (fields, params_count) = match &ast.data {
24 syn::Data::Struct(ds) => match ds.fields {28 syn::Data::Struct(ds) => match ds.fields {
25 syn::Fields::Named(ref fields) => (29 syn::Fields::Named(ref fields) => (
26 fields.named.iter().map(|field| &field.ty),30 fields.named.iter().map(map_field_to_type),
27 fields.named.len(),31 fields.named.len(),
28 ),32 ),
29 syn::Fields::Unnamed(_) => todo!(),33 syn::Fields::Unnamed(ref fields) => (
34 fields.unnamed.iter().map(map_field_to_type),
35 fields.unnamed.len(),
36 ),
30 syn::Fields::Unit => unimplemented!("Unit structs not supported"),37 syn::Fields::Unit => unimplemented!("Unit structs not supported"),
31 },38 },
32 syn::Data::Enum(_) => unimplemented!("Enums not supported"),39 syn::Data::Enum(_) => unimplemented!("Enums not supported"),
modifiedcrates/evm-coder/procedural/src/lib.rsdiffbeforeafterboth
--- a/crates/evm-coder/procedural/src/lib.rs
+++ b/crates/evm-coder/procedural/src/lib.rs
@@ -248,6 +248,6 @@
 pub fn abi_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
 	let ast = syn::parse(input).unwrap();
 	let ts = abi_derive::impl_abi_macro(&ast);
-	println!("{}", &ts);
+	// println!("{}", &ts);
 	ts.into()
 }
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
@@ -65,6 +65,9 @@
 }
 
 #[test]
+fn empty() {}
+
+#[test]
 fn impl_abi_type_signature() {
 	assert_eq!(
 		<TypeStructUnit as AbiType>::SIGNATURE.as_str().unwrap(),
@@ -201,3 +204,209 @@
 		ABI_ALIGNMENT * 5
 	);
 }
+
+#[derive(AbiCoder)]
+struct TupleStruct1SimpleParam(u8);
+
+#[derive(AbiCoder)]
+struct TupleStruct1DynamicParam(String);
+
+#[derive(AbiCoder)]
+struct TupleStruct2SimpleParam(u8, u32);
+
+#[derive(AbiCoder)]
+struct TupleStruct2DynamicParam(String, bytes);
+
+#[derive(AbiCoder)]
+struct TupleStruct2MixedParam(u8, bytes);
+
+#[derive(AbiCoder)]
+struct TupleStruct1DerivedSimpleParam(TupleStruct1SimpleParam);
+
+#[derive(AbiCoder)]
+struct TupleStruct2DerivedSimpleParam(TupleStruct1SimpleParam, TupleStruct2SimpleParam);
+
+#[derive(AbiCoder)]
+struct TupleStruct1DerivedDynamicParam(TupleStruct1DynamicParam);
+
+#[derive(AbiCoder)]
+struct TupleStruct2DerivedDynamicParam(TupleStruct1DynamicParam, TupleStruct2DynamicParam);
+
+#[derive(AbiCoder)]
+struct TupleStruct3DerivedMixedParam(
+	TupleStruct1SimpleParam,
+	TupleStruct2DynamicParam,
+	TupleStruct2MixedParam,
+);
+
+#[test]
+fn impl_abi_type_signature_same_for_structs() {
+	assert_eq!(
+		<TypeStruct1SimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct1SimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap()
+	);
+	assert_eq!(
+		<TypeStruct1DynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct1DynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap()
+	);
+	assert_eq!(
+		<TypeStruct2SimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct2SimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap()
+	);
+	assert_eq!(
+		<TypeStruct2DynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct2DynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap()
+	);
+	assert_eq!(
+		<TypeStruct2MixedParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct2MixedParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+	);
+	assert_eq!(
+		<TypeStruct1DerivedSimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct1DerivedSimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+	);
+	assert_eq!(
+		<TypeStruct2DerivedSimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct2DerivedSimpleParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+	);
+	assert_eq!(
+		<TypeStruct1DerivedDynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct1DerivedDynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+	);
+	assert_eq!(
+		<TypeStruct2DerivedDynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct2DerivedDynamicParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+	);
+	assert_eq!(
+		<TypeStruct3DerivedMixedParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+		<TupleStruct3DerivedMixedParam as AbiType>::SIGNATURE
+			.as_str()
+			.unwrap(),
+	);
+}
+
+#[test]
+fn impl_abi_type_is_dynamic_same_for_structs() {
+	assert_eq!(
+		<TypeStruct1SimpleParam as AbiType>::is_dynamic(),
+		<TupleStruct1SimpleParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct1DynamicParam as AbiType>::is_dynamic(),
+		<TupleStruct1DynamicParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct2SimpleParam as AbiType>::is_dynamic(),
+		<TupleStruct2SimpleParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct2DynamicParam as AbiType>::is_dynamic(),
+		<TupleStruct2DynamicParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct2MixedParam as AbiType>::is_dynamic(),
+		<TupleStruct2MixedParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct1DerivedSimpleParam as AbiType>::is_dynamic(),
+		<TupleStruct1DerivedSimpleParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct2DerivedSimpleParam as AbiType>::is_dynamic(),
+		<TupleStruct2DerivedSimpleParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct1DerivedDynamicParam as AbiType>::is_dynamic(),
+		<TupleStruct1DerivedDynamicParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct2DerivedDynamicParam as AbiType>::is_dynamic(),
+		<TupleStruct2DerivedDynamicParam as AbiType>::is_dynamic()
+	);
+	assert_eq!(
+		<TypeStruct3DerivedMixedParam as AbiType>::is_dynamic(),
+		<TupleStruct3DerivedMixedParam as AbiType>::is_dynamic()
+	);
+}
+
+#[test]
+fn impl_abi_type_size_same_for_structs() {
+	assert_eq!(
+		<TypeStruct1SimpleParam as AbiType>::size(),
+		<TupleStruct1SimpleParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct1DynamicParam as AbiType>::size(),
+		<TupleStruct1DynamicParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct2SimpleParam as AbiType>::size(),
+		<TupleStruct2SimpleParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct2DynamicParam as AbiType>::size(),
+		<TupleStruct2DynamicParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct2MixedParam as AbiType>::size(),
+		<TupleStruct2MixedParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct1DerivedSimpleParam as AbiType>::size(),
+		<TupleStruct1DerivedSimpleParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct2DerivedSimpleParam as AbiType>::size(),
+		<TupleStruct2DerivedSimpleParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct1DerivedDynamicParam as AbiType>::size(),
+		<TupleStruct1DerivedDynamicParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct2DerivedDynamicParam as AbiType>::size(),
+		<TupleStruct2DerivedDynamicParam as AbiType>::size()
+	);
+	assert_eq!(
+		<TypeStruct3DerivedMixedParam as AbiType>::size(),
+		<TupleStruct3DerivedMixedParam as AbiType>::size()
+	);
+}