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

pub(crate) fn generate(
    mut writer: impl std::io::Write,
    type_name: &str,
    or: Or,
    json_schema_props_type_name: &str,
    map_namespace: &impl crate::MapNamespace,
) -> Result<(), crate::Error> {
    let local = crate::map_namespace_local_to_string(map_namespace)?;

    let or_variant_name = match or {
        Or::Array => "Schemas",
        Or::Bool => "Bool",
        Or::StringArray => "Strings",
    };

    let or_variant_type: std::borrow::Cow<'_, str> = match or {
        Or::Array => format!("Vec<{json_schema_props_type_name}>").into(),
        Or::Bool => "bool".into(),
        Or::StringArray => "Vec<String>".into(),
    };

    let mut or_visit = String::new();
    match or {
        Or::Array => {
            use std::fmt::Write;

            writeln!(or_visit, "            fn visit_seq<A>(self, seq: A) -> Result<Self::Value, A::Error> where A: {local}serde::de::SeqAccess<'de> {{")?;
            writeln!(or_visit,
                "                Ok({type_name}::Schemas({local}serde::de::Deserialize::deserialize({local}serde::de::value::SeqAccessDeserializer::new(seq))?))")?;
            writeln!(or_visit, "            }}")?;
        },

        Or::Bool => {
            use std::fmt::Write;

            writeln!(or_visit, "            fn visit_bool<E>(self, v: bool) -> Result<Self::Value, E> where E: {local}serde::de::Error {{")?;
            writeln!(or_visit, "                Ok({type_name}::Bool(v))")?;
            writeln!(or_visit, "            }}")?;
        },

        Or::StringArray => {
            use std::fmt::Write;

            writeln!(or_visit, "            fn visit_seq<A>(self, seq: A) -> Result<Self::Value, A::Error> where A: {local}serde::de::SeqAccess<'de> {{")?;
            writeln!(or_visit,
                "                Ok({type_name}::Strings({local}serde::de::Deserialize::deserialize({local}serde::de::value::SeqAccessDeserializer::new(seq))?))")?;
            writeln!(or_visit, "            }}")?;
        },
    }

    writeln!(
        writer,
        include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/templates/json_schema_props_or.rs")),
        local = local,
        type_name = type_name,
        json_schema_props_type_name = json_schema_props_type_name,
        or_variant_name = or_variant_name,
        or_variant_type = or_variant_type,
        or_visit = or_visit,
    )?;

    Ok(())
}

#[derive(Clone, Copy)]
pub(crate) enum Or {
    Array,
    Bool,
    StringArray,
}