1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Copyright 2018-2019 Cryptape Technologies LLC.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

//! Use [`serde`] for serializing and deserializing.
//!
//! [`serde`]: https://crates.io/crates/serde

use crate::fixed_hash::HashConstructor;
use quote::quote;

impl HashConstructor {
    pub fn with_serde(&self) {
        self.with_serde_defun_pub();
    }

    fn with_serde_defun_pub(&self) {
        let name = &self.ts.name;
        let bytes_size = &self.ts.unit_amount;
        let part = quote!(
            #[cfg(feature = "support_serde")]
            impl serde::Serialize for #name {
                fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
                where
                    S: serde::Serializer,
                {
                    let bytes = self.as_bytes();
                    let mut dst = [0u8; #bytes_size * 2 + 2];
                    dst[0] = b'0';
                    dst[1] = b'x';
                    faster_hex::hex_encode(bytes, &mut dst[2..])
                        .map_err(|e| serde::ser::Error::custom(&format!("{}", e)))?;
                    serializer.serialize_str(unsafe { ::std::str::from_utf8_unchecked(&dst) })
                }
            }

            #[cfg(feature = "support_serde")]
            impl<'de> serde::Deserialize<'de> for #name {
                fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
                where
                    D: serde::Deserializer<'de>,
                {
                    struct Visitor;

                    impl<'b> serde::de::Visitor<'b> for Visitor {
                        type Value = #name;

                        fn expecting(
                            &self,
                            formatter: &mut ::std::fmt::Formatter,
                        ) -> ::std::fmt::Result {
                            write!(
                                formatter,
                                "a 0x-prefixed hex string with {} digits",
                                #bytes_size * 2
                            )
                        }

                        fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
                        where
                            E: serde::de::Error,
                        {
                            if v.len() <= 2 || &v[0..2] != "0x" {
                                return Err(E::custom(format_args!(
                                    "invalid format, expected {}",
                                    &self as &serde::de::Expected
                                )));
                            }

                            if v.len() != #bytes_size * 2 + 2 {
                                return Err(E::invalid_length(v.len() - 2, &self));
                            }

                            #name::from_hex_str(&v[2..]).map_err(|e| {
                                E::custom(format_args!(
                                    "invalid hex bytes: {:?}, expected {}",
                                    e, &self as &serde::de::Expected
                                ))
                            })
                        }

                        fn visit_string<E>(self, v: String) -> Result<Self::Value, E>
                        where
                            E: serde::de::Error,
                        {
                            self.visit_str(&v)
                        }
                    }
                    deserializer.deserialize_str(Visitor)
                }
            }
        );
        self.implt(part);
    }
}