use crate::{ParamType, TupleParam};
use serde::de::{Error, MapAccess, Visitor};
use serde::{Deserialize, Deserializer};
use std::fmt;
#[derive(Debug, Clone, PartialEq)]
pub struct EventParam {
pub name: String,
pub kind: ParamType,
pub indexed: bool,
}
impl<'a> Deserialize<'a> for EventParam {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'a>,
{
deserializer.deserialize_any(EventParamVisitor)
}
}
struct EventParamVisitor;
impl<'a> Visitor<'a> for EventParamVisitor {
type Value = EventParam;
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
write!(formatter, "a valid event parameter spec")
}
fn visit_map<V>(self, mut map: V) -> Result<Self::Value, V::Error>
where
V: MapAccess<'a>,
{
let mut name = None;
let mut kind = None;
let mut indexed = None;
let mut components = None;
while let Some(ref key) = map.next_key::<String>()? {
match key.as_ref() {
"name" => {
if name.is_some() {
return Err(Error::duplicate_field("name"));
}
name = Some(map.next_value()?);
}
"type" => {
if kind.is_some() {
return Err(Error::duplicate_field("kind"));
}
kind = Some(map.next_value()?);
}
"components" => {
if components.is_some() {
return Err(Error::duplicate_field("components"));
}
let component: Vec<TupleParam> = map.next_value()?;
components = Some(component)
}
"indexed" => {
if indexed.is_some() {
return Err(Error::duplicate_field("indexed"));
}
indexed = Some(map.next_value()?);
}
_ => {}
}
}
let name = name.ok_or_else(|| Error::missing_field("name"))?;
let kind = kind.ok_or_else(|| Error::missing_field("kind")).and_then(|param_type| {
if let ParamType::Tuple(_) = param_type {
let tuple_params = components.ok_or_else(|| Error::missing_field("components"))?;
Ok(ParamType::Tuple(tuple_params.into_iter().map(|param| param.kind).map(Box::new).collect()))
} else {
Ok(param_type)
}
})?;
let indexed = indexed.unwrap_or(false);
Ok(EventParam { name, kind, indexed })
}
}
#[cfg(test)]
mod tests {
use crate::{EventParam, ParamType};
use serde_json;
#[test]
fn event_param_deserialization() {
let s = r#"{
"name": "foo",
"type": "address",
"indexed": true
}"#;
let deserialized: EventParam = serde_json::from_str(s).unwrap();
assert_eq!(deserialized, EventParam { name: "foo".to_owned(), kind: ParamType::Address, indexed: true });
}
#[test]
fn event_param_tuple_deserialization() {
let s = r#"{
"name": "foo",
"type": "tuple",
"indexed": true,
"components": [
{
"name": "amount",
"type": "uint48"
},
{
"name": "things",
"type": "tuple",
"components": [
{
"name": "baseTupleParam",
"type": "address"
}
]
}
]
}"#;
let deserialized: EventParam = serde_json::from_str(s).unwrap();
assert_eq!(
deserialized,
EventParam {
name: "foo".to_owned(),
kind: ParamType::Tuple(vec![
Box::new(ParamType::Uint(48)),
Box::new(ParamType::Tuple(vec![Box::new(ParamType::Address)]))
]),
indexed: true,
}
);
}
}