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<Owned: de::Deserialize<'de>>,
47{
48    #[inline]
49    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
50    where
51        D: de::Deserializer<'de>,
52    {
53        T::Owned::deserialize(deserializer).map(Cow::Owned)
54    }
55}
56
57struct StringVisitor;
58struct StringInPlaceVisitor<'a>(&'a mut String);
59
60impl de::Visitor<'_> for StringVisitor {
61    type Value = String;
62
63    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
64        formatter.write_str("a string")
65    }
66
67    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
68    where
69        E: Error,
70    {
71        v.try_to_owned().map_err(E::custom)
72    }
73
74    fn visit_string<E>(self, v: rust_alloc::string::String) -> Result<Self::Value, E>
75    where
76        E: Error,
77    {
78        String::try_from(v).map_err(E::custom)
79    }
80
81    fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
82    where
83        E: Error,
84    {
85        match core::str::from_utf8(v) {
86            Ok(s) => s.try_to_owned().map_err(E::custom),
87            Err(_) => Err(Error::invalid_value(Unexpected::Bytes(v), &self)),
88        }
89    }
90
91    fn visit_byte_buf<E>(self, v: rust_alloc::vec::Vec<u8>) -> Result<Self::Value, E>
92    where
93        E: Error,
94    {
95        let v = Vec::try_from(v).map_err(E::custom)?;
96
97        match String::from_utf8(v) {
98            Ok(s) => Ok(s),
99            Err(e) => Err(Error::invalid_value(
100                Unexpected::Bytes(&e.into_bytes()),
101                &self,
102            )),
103        }
104    }
105}
106
107impl de::Visitor<'_> for StringInPlaceVisitor<'_> {
108    type Value = ();
109
110    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
111        formatter.write_str("a string")
112    }
113
114    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
115    where
116        E: Error,
117    {
118        self.0.clear();
119        self.0.try_push_str(v).map_err(E::custom)?;
120        Ok(())
121    }
122
123    fn visit_string<E>(self, v: rust_alloc::string::String) -> Result<Self::Value, E>
124    where
125        E: Error,
126    {
127        *self.0 = String::try_from(v.as_str()).map_err(E::custom)?;
128        Ok(())
129    }
130
131    fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
132    where
133        E: Error,
134    {
135        match core::str::from_utf8(v) {
136            Ok(s) => {
137                self.0.clear();
138                self.0.try_push_str(s).map_err(E::custom)?;
139                Ok(())
140            }
141            Err(_) => Err(Error::invalid_value(Unexpected::Bytes(v), &self)),
142        }
143    }
144
145    fn visit_byte_buf<E>(self, v: rust_alloc::vec::Vec<u8>) -> Result<Self::Value, E>
146    where
147        E: Error,
148    {
149        let v = Vec::try_from(v).map_err(E::custom)?;
150
151        match String::from_utf8(v) {
152            Ok(s) => {
153                *self.0 = s;
154                Ok(())
155            }
156            Err(e) => Err(Error::invalid_value(
157                Unexpected::Bytes(&e.into_bytes()),
158                &self,
159            )),
160        }
161    }
162}