ruma_common/directory/
room_network_serde.rs

1use std::fmt;
2
3use serde::{
4    de::{Error, MapAccess, Visitor},
5    ser::SerializeStruct,
6    Deserialize, Deserializer, Serialize, Serializer,
7};
8use serde_json::Value as JsonValue;
9
10use super::RoomNetwork;
11
12impl Serialize for RoomNetwork {
13    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
14    where
15        S: Serializer,
16    {
17        let mut state;
18        match self {
19            Self::Matrix => {
20                state = serializer.serialize_struct("RoomNetwork", 0)?;
21            }
22            Self::All => {
23                state = serializer.serialize_struct("RoomNetwork", 1)?;
24                state.serialize_field("include_all_networks", &true)?;
25            }
26            Self::ThirdParty(network) => {
27                state = serializer.serialize_struct("RoomNetwork", 1)?;
28                state.serialize_field("third_party_instance_id", network)?;
29            }
30        }
31        state.end()
32    }
33}
34
35impl<'de> Deserialize<'de> for RoomNetwork {
36    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
37    where
38        D: Deserializer<'de>,
39    {
40        deserializer.deserialize_map(RoomNetworkVisitor)
41    }
42}
43
44struct RoomNetworkVisitor;
45impl<'de> Visitor<'de> for RoomNetworkVisitor {
46    type Value = RoomNetwork;
47
48    fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
49        formatter.write_str("Network selection")
50    }
51
52    fn visit_map<M>(self, mut access: M) -> Result<Self::Value, M::Error>
53    where
54        M: MapAccess<'de>,
55    {
56        let mut include_all_networks = false;
57        let mut third_party_instance_id = None;
58        while let Some((key, value)) = access.next_entry::<String, JsonValue>()? {
59            match key.as_str() {
60                "include_all_networks" => {
61                    include_all_networks = value.as_bool().unwrap_or(false);
62                }
63                "third_party_instance_id" => {
64                    third_party_instance_id = value.as_str().map(|v| v.to_owned());
65                }
66                _ => {}
67            };
68        }
69
70        if include_all_networks {
71            if third_party_instance_id.is_none() {
72                Ok(RoomNetwork::All)
73            } else {
74                Err(M::Error::custom(
75                    "`include_all_networks = true` and `third_party_instance_id` are mutually exclusive.",
76                ))
77            }
78        } else {
79            Ok(match third_party_instance_id {
80                Some(network) => RoomNetwork::ThirdParty(network),
81                None => RoomNetwork::Matrix,
82            })
83        }
84    }
85}