34 lines
1.3 KiB
Rust
34 lines
1.3 KiB
Rust
//! Helper functions for deriving `EthAbiType`
|
|
|
|
use ethers_core::macros::ethers_core_crate;
|
|
use quote::quote;
|
|
use syn::DeriveInput;
|
|
|
|
/// Generates the `AbiEncode` + `AbiDecode` implementation
|
|
pub fn derive_codec_impl(input: &DeriveInput) -> proc_macro2::TokenStream {
|
|
let name = &input.ident;
|
|
let ethers_core = ethers_core_crate();
|
|
|
|
quote! {
|
|
impl #ethers_core::abi::AbiDecode for #name {
|
|
fn decode(bytes: impl AsRef<[u8]>) -> ::std::result::Result<Self, #ethers_core::abi::AbiError> {
|
|
if let #ethers_core::abi::ParamType::Tuple(params) = <Self as #ethers_core::abi::AbiType>::param_type() {
|
|
let tokens = #ethers_core::abi::decode(¶ms, bytes.as_ref())?;
|
|
Ok(<Self as #ethers_core::abi::Tokenizable>::from_token(#ethers_core::abi::Token::Tuple(tokens))?)
|
|
} else {
|
|
Err(
|
|
#ethers_core::abi::InvalidOutputType("Expected tuple".to_string()).into()
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl #ethers_core::abi::AbiEncode for #name {
|
|
fn encode(self) -> ::std::vec::Vec<u8> {
|
|
let tokens = #ethers_core::abi::Tokenize::into_tokens(self);
|
|
#ethers_core::abi::encode(&tokens)
|
|
}
|
|
}
|
|
}
|
|
}
|