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
100

use crate::pieces::Piece;
use hex::{decode_to_slice, FromHex, FromHexError};
use serde::{de, Deserialize, Deserializer, Serialize, Serializer};

impl FromHex for Piece {
    type Error = FromHexError;

    fn from_hex<T: AsRef<[u8]>>(hex: T) -> Result<Self, Self::Error> {
        let hex = hex.as_ref();
        if hex.len() % 2 != 0 {
            return Err(FromHexError::OddLength);
        }
        if hex.len() != 2 * Piece::SIZE {
            return Err(FromHexError::InvalidStringLength);
        }

        let mut out = Self::default();

        decode_to_slice(hex, out.as_mut_slice())?;

        Ok(out.to_shared())
    }
}

impl Serialize for Piece {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        Serializer::serialize_newtype_struct(serializer, "Piece", {
            struct SerializeWith<'a> {
                values: &'a [u8],
            }
            impl<'a> Serialize for SerializeWith<'a> {
                fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
                where
                    S: Serializer,
                {
                    hex::serde::serialize(self.values, serializer)
                }
            }
            &SerializeWith {
                values: self.as_ref(),
            }
        })
    }
}

impl<'de> Deserialize<'de> for Piece {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        struct Visitor;

        impl<'de> de::Visitor<'de> for Visitor {
            type Value = Piece;

            fn expecting(&self, formatter: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
                formatter.write_str("tuple struct Piece")
            }

            #[inline]
            fn visit_newtype_struct<D>(self, deserializer: D) -> Result<Self::Value, D::Error>
            where
                D: Deserializer<'de>,
            {
                hex::serde::deserialize(deserializer)
            }

            #[inline]
            fn visit_seq<A>(self, mut seq: A) -> Result<Self::Value, A::Error>
            where
                A: de::SeqAccess<'de>,
            {
                struct DeserializeWith {
                    value: Piece,
                }
                impl<'de> Deserialize<'de> for DeserializeWith {
                    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
                    where
                        D: Deserializer<'de>,
                    {
                        Ok(DeserializeWith {
                            value: hex::serde::deserialize(deserializer)?,
                        })
                    }
                }

                de::SeqAccess::next_element::<DeserializeWith>(&mut seq)?
                    .map(|wrap| wrap.value)
                    .ok_or(de::Error::invalid_length(
                        0usize,
                        &"tuple struct Piece with 1 element",
                    ))
            }
        }
        Deserializer::deserialize_newtype_struct(deserializer, "Piece", Visitor)
    }
}