ic_web3_rs/types/
bytes.rs

1use serde::{
2    de::{Error, Unexpected, Visitor},
3    Deserialize, Deserializer, Serialize, Serializer,
4};
5use std::fmt;
6
7/// Raw bytes wrapper
8#[derive(Clone, Debug, Default, PartialEq, Eq, Hash)]
9pub struct Bytes(pub Vec<u8>);
10
11impl<T: Into<Vec<u8>>> From<T> for Bytes {
12    fn from(data: T) -> Self {
13        Bytes(data.into())
14    }
15}
16
17impl Serialize for Bytes {
18    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
19    where
20        S: Serializer,
21    {
22        let mut serialized = "0x".to_owned();
23        serialized.push_str(&hex::encode(&self.0));
24        serializer.serialize_str(serialized.as_ref())
25    }
26}
27
28impl<'a> Deserialize<'a> for Bytes {
29    fn deserialize<D>(deserializer: D) -> Result<Bytes, D::Error>
30    where
31        D: Deserializer<'a>,
32    {
33        deserializer.deserialize_identifier(BytesVisitor)
34    }
35}
36
37struct BytesVisitor;
38
39impl<'a> Visitor<'a> for BytesVisitor {
40    type Value = Bytes;
41
42    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
43        write!(formatter, "a 0x-prefixed hex-encoded vector of bytes")
44    }
45
46    fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
47    where
48        E: Error,
49    {
50        if value.len() >= 2 && &value[0..2] == "0x" {
51            let bytes = hex::decode(&value[2..]).map_err(|e| Error::custom(format!("Invalid hex: {}", e)))?;
52            Ok(Bytes(bytes))
53        } else {
54            Err(Error::invalid_value(Unexpected::Str(value), &"0x prefix"))
55        }
56    }
57
58    fn visit_string<E>(self, value: String) -> Result<Self::Value, E>
59    where
60        E: Error,
61    {
62        self.visit_str(value.as_ref())
63    }
64}