1 | use crate::{
|
2 | de::key::QNameDeserializer,
|
3 | de::resolver::EntityResolver,
|
4 | de::simple_type::SimpleTypeDeserializer,
|
5 | de::{DeEvent, Deserializer, XmlRead, TEXT_KEY},
|
6 | errors::serialize::DeError,
|
7 | };
|
8 | use serde::de::value::BorrowedStrDeserializer;
|
9 | use serde::de::{self, DeserializeSeed, Deserializer as _, Visitor};
|
10 |
|
11 | /// An enum access
|
12 | pub struct EnumAccess<'de, 'a, R, E>
|
13 | where
|
14 | R: XmlRead<'de>,
|
15 | E: EntityResolver,
|
16 | {
|
17 | de: &'a mut Deserializer<'de, R, E>,
|
18 | }
|
19 |
|
20 | impl<'de, 'a, R, E> EnumAccess<'de, 'a, R, E>
|
21 | where
|
22 | R: XmlRead<'de>,
|
23 | E: EntityResolver,
|
24 | {
|
25 | pub fn new(de: &'a mut Deserializer<'de, R, E>) -> Self {
|
26 | EnumAccess { de }
|
27 | }
|
28 | }
|
29 |
|
30 | impl<'de, 'a, R, E> de::EnumAccess<'de> for EnumAccess<'de, 'a, R, E>
|
31 | where
|
32 | R: XmlRead<'de>,
|
33 | E: EntityResolver,
|
34 | {
|
35 | type Error = DeError;
|
36 | type Variant = VariantAccess<'de, 'a, R, E>;
|
37 |
|
38 | fn variant_seed<V>(self, seed: V) -> Result<(V::Value, VariantAccess<'de, 'a, R, E>), DeError>
|
39 | where
|
40 | V: DeserializeSeed<'de>,
|
41 | {
|
42 | let decoder = self.de.reader.decoder();
|
43 | let (name, is_text) = match self.de.peek()? {
|
44 | DeEvent::Start(e) => (
|
45 | seed.deserialize(QNameDeserializer::from_elem(e.raw_name(), decoder)?)?,
|
46 | false,
|
47 | ),
|
48 | DeEvent::Text(_) => (
|
49 | seed.deserialize(BorrowedStrDeserializer::<DeError>::new(TEXT_KEY))?,
|
50 | true,
|
51 | ),
|
52 | DeEvent::End(e) => return Err(DeError::UnexpectedEnd(e.name().into_inner().to_vec())),
|
53 | DeEvent::Eof => return Err(DeError::UnexpectedEof),
|
54 | };
|
55 | Ok((
|
56 | name,
|
57 | VariantAccess {
|
58 | de: self.de,
|
59 | is_text,
|
60 | },
|
61 | ))
|
62 | }
|
63 | }
|
64 |
|
65 | pub struct VariantAccess<'de, 'a, R, E>
|
66 | where
|
67 | R: XmlRead<'de>,
|
68 | E: EntityResolver,
|
69 | {
|
70 | de: &'a mut Deserializer<'de, R, E>,
|
71 | /// `true` if variant should be deserialized from a textual content
|
72 | /// and `false` if from tag
|
73 | is_text: bool,
|
74 | }
|
75 |
|
76 | impl<'de, 'a, R, E> de::VariantAccess<'de> for VariantAccess<'de, 'a, R, E>
|
77 | where
|
78 | R: XmlRead<'de>,
|
79 | E: EntityResolver,
|
80 | {
|
81 | type Error = DeError;
|
82 |
|
83 | fn unit_variant(self) -> Result<(), DeError> {
|
84 | match self.de.next()? {
|
85 | // Consume subtree
|
86 | DeEvent::Start(e) => self.de.read_to_end(e.name()),
|
87 | // Does not needed to deserialize using SimpleTypeDeserializer, because
|
88 | // it returns `()` when `deserialize_unit()` is requested
|
89 | DeEvent::Text(_) => Ok(()),
|
90 | // SAFETY: the other events are filtered in `variant_seed()`
|
91 | _ => unreachable!("Only `Start` or `Text` events are possible here" ),
|
92 | }
|
93 | }
|
94 |
|
95 | fn newtype_variant_seed<T>(self, seed: T) -> Result<T::Value, DeError>
|
96 | where
|
97 | T: DeserializeSeed<'de>,
|
98 | {
|
99 | if self.is_text {
|
100 | match self.de.next()? {
|
101 | DeEvent::Text(e) => seed.deserialize(SimpleTypeDeserializer::from_text_content(e)),
|
102 | // SAFETY: the other events are filtered in `variant_seed()`
|
103 | _ => unreachable!("Only `Text` events are possible here" ),
|
104 | }
|
105 | } else {
|
106 | seed.deserialize(&mut *self.de)
|
107 | }
|
108 | }
|
109 |
|
110 | fn tuple_variant<V>(self, len: usize, visitor: V) -> Result<V::Value, DeError>
|
111 | where
|
112 | V: Visitor<'de>,
|
113 | {
|
114 | if self.is_text {
|
115 | match self.de.next()? {
|
116 | DeEvent::Text(e) => {
|
117 | SimpleTypeDeserializer::from_text_content(e).deserialize_tuple(len, visitor)
|
118 | }
|
119 | // SAFETY: the other events are filtered in `variant_seed()`
|
120 | _ => unreachable!("Only `Text` events are possible here" ),
|
121 | }
|
122 | } else {
|
123 | self.de.deserialize_tuple(len, visitor)
|
124 | }
|
125 | }
|
126 |
|
127 | fn struct_variant<V>(
|
128 | self,
|
129 | fields: &'static [&'static str],
|
130 | visitor: V,
|
131 | ) -> Result<V::Value, DeError>
|
132 | where
|
133 | V: Visitor<'de>,
|
134 | {
|
135 | if self.is_text {
|
136 | match self.de.next()? {
|
137 | DeEvent::Text(e) => SimpleTypeDeserializer::from_text_content(e)
|
138 | .deserialize_struct("" , fields, visitor),
|
139 | // SAFETY: the other events are filtered in `variant_seed()`
|
140 | _ => unreachable!("Only `Text` events are possible here" ),
|
141 | }
|
142 | } else {
|
143 | self.de.deserialize_struct("" , fields, visitor)
|
144 | }
|
145 | }
|
146 | }
|
147 | |