rune_alloc/string/
serde.rs

1use core::fmt;
2
3use serde::de::{self, Error, Unexpected};
4use serde::ser;
5
6use crate::borrow::{Cow, TryToOwned};
7use crate::{Box, String, Vec};
8
9impl ser::Serialize for String {
10    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
11    where
12        S: ser::Serializer,
13    {
14        serializer.serialize_str(self.as_str())
15    }
16}
17
18impl<'de> de::Deserialize<'de> for String {
19    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
20    where
21        D: de::Deserializer<'de>,
22    {
23        deserializer.deserialize_string(StringVisitor)
24    }
25
26    fn deserialize_in_place<D>(deserializer: D, place: &mut Self) -> Result<(), D::Error>
27    where
28        D: de::Deserializer<'de>,
29    {
30        deserializer.deserialize_string(StringInPlaceVisitor(place))
31    }
32}
33
34impl<'de> de::Deserialize<'de> for Box<str> {
35    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
36    where
37        D: de::Deserializer<'de>,
38    {
39        let string = String::deserialize(deserializer)?;
40        string.try_into_boxed_str().map_err(D::Error::custom)
41    }
42}
43
44impl<'de, T: ?Sized> de::Deserialize<'de> for Cow<'_, T>
45where
46    T: TryToOwned,
47    T::Owned: de::Deserialize<'de>,
48{
49    #[inline]
50    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
51    where
52        D: de::Deserializer<'de>,
53    {
54        T::Owned::deserialize(deserializer).map(Cow::Owned)
55    }
56}
57
58struct StringVisitor;
59struct StringInPlaceVisitor<'a>(&'a mut String);
60
61impl de::Visitor<'_> for StringVisitor {
62    type Value = String;
63
64    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
65        formatter.write_str("a string")
66    }
67
68    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
69    where
70        E: Error,
71    {
72        v.try_to_owned().map_err(E::custom)
73    }
74
75    fn visit_string<E>(self, v: ::rust_alloc::string::String) -> Result<Self::Value, E>
76    where
77        E: Error,
78    {
79        String::try_from(v).map_err(E::custom)
80    }
81
82    fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
83    where
84        E: Error,
85    {
86        match core::str::from_utf8(v) {
87            Ok(s) => s.try_to_owned().map_err(E::custom),
88            Err(_) => Err(Error::invalid_value(Unexpected::Bytes(v), &self)),
89        }
90    }
91
92    fn visit_byte_buf<E>(self, v: ::rust_alloc::vec::Vec<u8>) -> Result<Self::Value, E>
93    where
94        E: Error,
95    {
96        let v = Vec::try_from(v).map_err(E::custom)?;
97
98        match String::from_utf8(v) {
99            Ok(s) => Ok(s),
100            Err(e) => Err(Error::invalid_value(
101                Unexpected::Bytes(&e.into_bytes()),
102                &self,
103            )),
104        }
105    }
106}
107
108impl de::Visitor<'_> for StringInPlaceVisitor<'_> {
109    type Value = ();
110
111    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
112        formatter.write_str("a string")
113    }
114
115    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
116    where
117        E: Error,
118    {
119        self.0.clear();
120        self.0.try_push_str(v).map_err(E::custom)?;
121        Ok(())
122    }
123
124    fn visit_string<E>(self, v: ::rust_alloc::string::String) -> Result<Self::Value, E>
125    where
126        E: Error,
127    {
128        *self.0 = String::try_from(v.as_str()).map_err(E::custom)?;
129        Ok(())
130    }
131
132    fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
133    where
134        E: Error,
135    {
136        match core::str::from_utf8(v) {
137            Ok(s) => {
138                self.0.clear();
139                self.0.try_push_str(s).map_err(E::custom)?;
140                Ok(())
141            }
142            Err(_) => Err(Error::invalid_value(Unexpected::Bytes(v), &self)),
143        }
144    }
145
146    fn visit_byte_buf<E>(self, v: ::rust_alloc::vec::Vec<u8>) -> Result<Self::Value, E>
147    where
148        E: Error,
149    {
150        let v = Vec::try_from(v).map_err(E::custom)?;
151
152        match String::from_utf8(v) {
153            Ok(s) => {
154                *self.0 = s;
155                Ok(())
156            }
157            Err(e) => Err(Error::invalid_value(
158                Unexpected::Bytes(&e.into_bytes()),
159                &self,
160            )),
161        }
162    }
163}