[go: up one dir, main page]

schemars_derive 0.6.1

Macros for #[derive(JsonSchema)], for use with schemars
Documentation 
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

use crate::attr;
use proc_macro2::TokenStream;
use syn::{Attribute, ExprPath};

#[derive(Debug, Clone, Default)]
pub struct SchemaMetadata {
    pub title: Option<String>,
    pub description: Option<String>,
    pub read_only: bool,
    pub write_only: bool,
    pub default: Option<TokenStream>,
    pub skip_default_if: Option<ExprPath>,
}

pub fn set_metadata_on_schema_from_docs(
    schema_expr: TokenStream,
    attrs: &[Attribute],
) -> TokenStream {
    let metadata = get_metadata_from_docs(attrs);
    set_metadata_on_schema(schema_expr, &metadata)
}

pub fn get_metadata_from_docs(attrs: &[Attribute]) -> SchemaMetadata {
    let (title, description) = attr::get_title_and_desc_from_doc(attrs);
    SchemaMetadata {
        title,
        description,
        ..Default::default()
    }
}

pub fn set_metadata_on_schema(schema_expr: TokenStream, metadata: &SchemaMetadata) -> TokenStream {
    let mut setters = Vec::<TokenStream>::new();

    if let Some(title) = &metadata.title {
        setters.push(quote! {
            metadata.title = Some(#title.to_owned());
        });
    }
    if let Some(description) = &metadata.description {
        setters.push(quote! {
            metadata.description = Some(#description.to_owned());
        });
    }

    if metadata.read_only {
        setters.push(quote! {
            metadata.read_only = true;
        });
    }
    if metadata.write_only {
        setters.push(quote! {
            metadata.write_only = true;
        });
    }

    match (&metadata.default, &metadata.skip_default_if) {
        (Some(default), Some(skip_if)) => setters.push(quote! {
            {
                let default = #default;
                if !#skip_if(&default) {
                    metadata.default = schemars::_serde_json::value::to_value(default).ok();
                }
            }
        }),
        (Some(default), None) => setters.push(quote! {
            metadata.default = schemars::_serde_json::value::to_value(#default).ok();
        }),
        _ => {}
    }

    if setters.is_empty() {
        return schema_expr;
    }

    quote! {
        {
            let schema = #schema_expr.into();
            let mut schema_obj = gen.make_extensible(schema);
            let mut metadata = schema_obj.metadata();
            #(#setters)*
            schemars::schema::Schema::Object(schema_obj)
        }
    }
}