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

use crate::*;
use std::collections::{BTreeMap, HashMap};

#[derive(Clone, Debug, Default, PartialEq)]
pub struct ObjectType {
    pub key_type: Box<Schema>,
    pub max_length: Option<usize>,
    pub min_length: Option<usize>,
    pub required: Option<Vec<String>>,
    pub value_type: Box<Schema>,
}

impl ObjectType {
    /// Create an indexed/mapable object schema with the provided key and value types.
    pub fn new(key_type: impl Into<Schema>, value_type: impl Into<Schema>) -> Self {
        ObjectType {
            key_type: Box::new(key_type.into()),
            value_type: Box::new(value_type.into()),
            ..ObjectType::default()
        }
    }
}

impl<K: Schematic, V: Schematic> Schematic for BTreeMap<K, V> {
    fn build_schema(mut schema: SchemaBuilder) -> Schema {
        schema.object(ObjectType::new(schema.infer::<K>(), schema.infer::<V>()))
    }
}

impl<K: Schematic, V: Schematic, S> Schematic for HashMap<K, V, S> {
    fn build_schema(mut schema: SchemaBuilder) -> Schema {
        schema.object(ObjectType::new(schema.infer::<K>(), schema.infer::<V>()))
    }
}