use crate::{types::Selector, utils::id};
pub use ethabi::Contract as Abi;
pub use ethabi::*;
mod tokens;
pub use tokens::{Detokenize, InvalidOutputType, Tokenizable, TokenizableItem, Tokenize};
pub mod struct_def;
pub use struct_def::SolStruct;
mod error;
pub use error::ParseError;
mod human_readable;
pub use human_readable::{parse as parse_abi, parse_str as parse_abi_str, AbiParser};
pub trait FunctionExt {
fn abi_signature(&self) -> String;
fn selector(&self) -> Selector;
}
impl FunctionExt for Function {
fn abi_signature(&self) -> String {
let mut full_signature = self.signature();
if let Some(colon) = full_signature.find(':') {
full_signature.truncate(colon);
}
full_signature
}
fn selector(&self) -> Selector {
id(self.abi_signature())
}
}
pub trait EventExt {
fn abi_signature(&self) -> String;
}
impl EventExt for Event {
fn abi_signature(&self) -> String {
format!(
"{}({}){}",
self.name,
self.inputs
.iter()
.map(|input| input.kind.to_string())
.collect::<Vec<_>>()
.join(","),
if self.anonymous { " anonymous" } else { "" },
)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn format_function_signature() {
for (f, expected) in &[
(
r#"{"name":"foo","inputs":[],"outputs":[], "stateMutability": "nonpayable"}"#,
"foo()",
),
(
r#"{"name":"bar","inputs":[{"name":"a","type":"uint256"},{"name":"b","type":"bool"}],"outputs":[], "stateMutability": "nonpayable"}"#,
"bar(uint256,bool)",
),
(
r#"{"name":"baz","inputs":[{"name":"a","type":"uint256"}],"outputs":[{"name":"b","type":"bool"}], "stateMutability": "nonpayable"}"#,
"baz(uint256)",
),
(
r#"{"name":"bax","inputs":[],"outputs":[{"name":"a","type":"uint256"},{"name":"b","type":"bool"}], "stateMutability": "nonpayable"}"#,
"bax()",
),
] {
let function: Function = serde_json::from_str(f).expect("invalid function JSON");
let signature = function.abi_signature();
assert_eq!(signature, *expected);
}
}
#[test]
fn format_event_signature() {
for (e, expected) in &[
(r#"{"name":"foo","inputs":[],"anonymous":false}"#, "foo()"),
(
r#"{"name":"bar","inputs":[{"name":"a","type":"uint256"},{"name":"b","type":"bool"}],"anonymous":false}"#,
"bar(uint256,bool)",
),
(
r#"{"name":"baz","inputs":[{"name":"a","type":"uint256"}],"anonymous":true}"#,
"baz(uint256) anonymous",
),
(
r#"{"name":"bax","inputs":[],"anonymous":true}"#,
"bax() anonymous",
),
] {
let event: Event = serde_json::from_str(e).expect("invalid event JSON");
let signature = event.abi_signature();
assert_eq!(signature, *expected);
}
}
}