1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use std::{collections::BTreeMap, fmt, sync::Arc};

use async_graphql_parser::types::Type;
use serde::{self, de::Visitor, Deserializer, Serialize, Serializer};

pub fn serde_type_serializer<S>(value: &Type, serializer: S) -> Result<S::Ok, S::Error>
where
    S: Serializer,
{
    value.to_string().serialize(serializer)
}

pub fn serde_type_deserializer<'de, D>(deserializer: D) -> Result<Type, D::Error>
where
    D: Deserializer<'de>,
{
    struct TypeDeserializer;

    impl<'de> Visitor<'de> for TypeDeserializer {
        type Value = Type;

        fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
            formatter.write_str("GraphQL type")
        }

        fn visit_str<E>(self, s: &str) -> Result<Self::Value, E>
        where
            E: serde::de::Error,
        {
            let ty =
                Type::new(s).ok_or_else(|| serde::de::Error::custom("not a valid GraphQL type"))?;
            Ok(ty)
        }
    }

    deserializer.deserialize_str(TypeDeserializer)
}

pub fn serde_variables_serializer<S>(
    value: &BTreeMap<Arc<str>, Type>,
    serializer: S,
) -> Result<S::Ok, S::Error>
where
    S: Serializer,
{
    let converted: BTreeMap<&str, String> = value
        .iter()
        .map(|(k, v)| (k.as_ref(), v.to_string()))
        .collect();
    converted.serialize(serializer)
}

pub fn serde_variables_deserializer<'de, D>(
    deserializer: D,
) -> Result<BTreeMap<Arc<str>, Type>, D::Error>
where
    D: Deserializer<'de>,
{
    struct TypeDeserializer;

    impl<'de> Visitor<'de> for TypeDeserializer {
        type Value = BTreeMap<Arc<str>, &'de str>;

        fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
            formatter.write_str("map of variable names -> types")
        }

        fn visit_map<A>(self, mut map: A) -> Result<Self::Value, A::Error>
        where
            A: serde::de::MapAccess<'de>,
        {
            let mut result: BTreeMap<Arc<str>, &'de str> = BTreeMap::new();
            while let Some((key, value)) = map.next_entry()? {
                result.insert(key, value);
            }
            Ok(result)
        }
    }

    deserializer.deserialize_map(TypeDeserializer).map(|value| {
        let mut result: BTreeMap<Arc<str>, Type> = Default::default();
        for (k, v) in value {
            let ty = Type::new(v).unwrap();
            result.insert(k, ty);
        }
        result
    })
}