darklua_core/utils/
serde_string_or_struct.rs

1use std::{fmt, marker::PhantomData, str::FromStr};
2
3use serde::{
4    de::{value::MapAccessDeserializer, MapAccess, Visitor},
5    Deserialize, Deserializer,
6};
7
8pub(crate) fn string_or_struct<'de, T, D>(deserializer: D) -> Result<T, D::Error>
9where
10    T: Deserialize<'de> + FromStr<Err = String>,
11    D: Deserializer<'de>,
12{
13    struct StringOrStruct<T>(PhantomData<T>);
14
15    impl<'de, T> Visitor<'de> for StringOrStruct<T>
16    where
17        T: Deserialize<'de> + FromStr<Err = String>,
18    {
19        type Value = T;
20
21        fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
22            formatter.write_str("string or object")
23        }
24
25        fn visit_str<E>(self, value: &str) -> Result<T, E>
26        where
27            E: serde::de::Error,
28        {
29            T::from_str(value).map_err(E::custom)
30        }
31
32        fn visit_map<M>(self, map: M) -> Result<T, M::Error>
33        where
34            M: MapAccess<'de>,
35        {
36            Deserialize::deserialize(MapAccessDeserializer::new(map))
37        }
38    }
39
40    deserializer.deserialize_any(StringOrStruct(PhantomData))
41}