1 | use crate::{ |
2 | signature_parser::SignatureParser, utils::*, DynamicDeserialize, DynamicType, Signature, |
3 | }; |
4 | use serde::{ |
5 | de::{Deserialize, DeserializeSeed, Deserializer, Error, Visitor}, |
6 | Serialize, Serializer, |
7 | }; |
8 | use std::marker::PhantomData; |
9 | |
10 | /// A helper type to serialize or deserialize a tuple whose elements implement [DynamicType] but |
11 | /// not [Type]. |
12 | /// |
13 | /// This is required because tuples already have an implementation of [DynamicType] via the blanket |
14 | /// implementation of [DynamicType] where `T: Type`, but that results in a bound of [Type] on each |
15 | /// element, which is stronger than needed for serializing. |
16 | /// |
17 | /// [Type]: trait.Type.html |
18 | #[derive (Debug, Copy, Clone)] |
19 | pub struct DynamicTuple<T>(pub T); |
20 | |
21 | impl DynamicType for DynamicTuple<()> { |
22 | fn dynamic_signature(&self) -> Signature<'_> { |
23 | Signature::from_static_str_unchecked(signature:"" ) |
24 | } |
25 | } |
26 | |
27 | impl<T: Serialize> Serialize for DynamicTuple<T> { |
28 | fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> { |
29 | self.0.serialize(serializer) |
30 | } |
31 | } |
32 | |
33 | impl<'de> Deserialize<'de> for DynamicTuple<()> { |
34 | fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> { |
35 | <()>::deserialize(deserializer).map(op:DynamicTuple) |
36 | } |
37 | } |
38 | |
39 | /// A helper type for [DynamicTuple]'s [DynamicDeserialize] implementation. |
40 | #[derive (Debug, Clone, PartialEq, Eq)] |
41 | pub struct TupleSeed<'a, T, S> { |
42 | sig: Signature<'a>, |
43 | seeds: S, |
44 | marker: PhantomData<T>, |
45 | } |
46 | |
47 | impl<'a, T, S> DynamicType for TupleSeed<'a, T, S> { |
48 | fn dynamic_signature(&self) -> Signature<'_> { |
49 | self.sig.clone() |
50 | } |
51 | } |
52 | |
53 | struct TupleVisitor<T, S> { |
54 | seeds: S, |
55 | marker: PhantomData<T>, |
56 | } |
57 | |
58 | macro_rules! tuple_impls { |
59 | ($($len:expr => ($($n:tt $name:ident)+))+) => { |
60 | $( |
61 | impl<$($name),+> DynamicType for DynamicTuple<($($name,)+)> |
62 | where |
63 | $($name: DynamicType,)+ |
64 | { |
65 | fn dynamic_signature(&self) -> Signature<'_> { |
66 | let mut sig = String::with_capacity(255); |
67 | sig.push(STRUCT_SIG_START_CHAR); |
68 | $( |
69 | sig.push_str(DynamicType::dynamic_signature(&self.0.$n).as_str()); |
70 | )+ |
71 | sig.push(STRUCT_SIG_END_CHAR); |
72 | |
73 | Signature::from_string_unchecked(sig) |
74 | } |
75 | } |
76 | |
77 | impl<'de, $($name),+> DeserializeSeed<'de> for TupleSeed<'de, ($($name,)+), ($(<$name as DynamicDeserialize<'de>>::Deserializer,)+)> |
78 | where |
79 | $($name: DynamicDeserialize<'de>,)+ |
80 | { |
81 | type Value = DynamicTuple<($($name,)+)>; |
82 | |
83 | fn deserialize<D: Deserializer<'de>>(self, deserializer: D) -> Result<Self::Value, D::Error> { |
84 | deserializer.deserialize_tuple($len, TupleVisitor { seeds: self.seeds, marker: self.marker }) |
85 | } |
86 | } |
87 | |
88 | impl<'de, $($name),+> Visitor<'de> for TupleVisitor<($($name,)+), ($(<$name as DynamicDeserialize<'de>>::Deserializer,)+)> |
89 | where |
90 | $($name: DynamicDeserialize<'de>,)+ |
91 | { |
92 | type Value = DynamicTuple<($($name,)+)>; |
93 | |
94 | fn expecting(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { |
95 | formatter.write_str("a tuple" ) |
96 | } |
97 | |
98 | fn visit_seq<V>(self, mut visitor: V) -> Result<DynamicTuple<($($name,)+)>, V::Error> |
99 | where |
100 | V: serde::de::SeqAccess<'de>, |
101 | { |
102 | Ok(DynamicTuple(($({ |
103 | match visitor.next_element_seed(self.seeds.$n) { |
104 | Ok(Some(elt)) => elt, |
105 | Ok(None) => return Err(V::Error::invalid_length($len, &"" )), |
106 | Err(e) => return Err(e), |
107 | } |
108 | },)+))) |
109 | } |
110 | } |
111 | |
112 | impl<'de, $($name),+> DynamicDeserialize<'de> for DynamicTuple<($($name,)+)> |
113 | where |
114 | $($name: DynamicDeserialize<'de>,)+ |
115 | { |
116 | type Deserializer = TupleSeed<'de, ($($name,)+), ($(<$name as DynamicDeserialize<'de>>::Deserializer,)+)>; |
117 | |
118 | fn deserializer_for_signature<S>(signature: S) -> zvariant::Result<Self::Deserializer> |
119 | where S: TryInto<Signature<'de>>, S::Error: Into<zvariant::Error> |
120 | { |
121 | let sig = signature.try_into().map_err(Into::into)?; |
122 | if !sig.starts_with(zvariant::STRUCT_SIG_START_CHAR) { |
123 | return Err(zvariant::Error::IncorrectType); |
124 | } |
125 | if !sig.ends_with(zvariant::STRUCT_SIG_END_CHAR) { |
126 | return Err(zvariant::Error::IncorrectType); |
127 | } |
128 | |
129 | let end = sig.len() - 1; |
130 | let mut sig_parser = SignatureParser::new(sig.slice(1..end)); |
131 | |
132 | let seeds = ($({ |
133 | let elt_sig = sig_parser.parse_next_signature()?; |
134 | $name::deserializer_for_signature(elt_sig)? |
135 | },)+); |
136 | |
137 | Ok(TupleSeed { sig, seeds, marker: PhantomData }) |
138 | } |
139 | } |
140 | )+ |
141 | } |
142 | } |
143 | |
144 | tuple_impls! { |
145 | 1 => (0 T0) |
146 | 2 => (0 T0 1 T1) |
147 | 3 => (0 T0 1 T1 2 T2) |
148 | 4 => (0 T0 1 T1 2 T2 3 T3) |
149 | 5 => (0 T0 1 T1 2 T2 3 T3 4 T4) |
150 | 6 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5) |
151 | 7 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6) |
152 | 8 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7) |
153 | 9 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8) |
154 | 10 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9) |
155 | 11 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9 10 T10) |
156 | 12 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9 10 T10 11 T11) |
157 | 13 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9 10 T10 11 T11 12 T12) |
158 | 14 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9 10 T10 11 T11 12 T12 13 T13) |
159 | 15 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9 10 T10 11 T11 12 T12 13 T13 14 T14) |
160 | 16 => (0 T0 1 T1 2 T2 3 T3 4 T4 5 T5 6 T6 7 T7 8 T8 9 T9 10 T10 11 T11 12 T12 13 T13 14 T14 15 T15) |
161 | } |
162 | |