zod-derive 0.1.0

Rust integraions with the `zod` typescript library.
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

use proc_macro2::{Span, TokenStream};
use quote::quote_spanned;
use serde_derive_internals::attr::TagType;

/// represents a unit variant of an enum, it has no fields and it is represented in typescript as the
/// stringifyied name
pub struct UnitVariant<'a> {
    pub span: Span,
    pub tag: &'a TagType,
    pub name: String,
}

impl<'a> UnitVariant<'a> {
    pub fn expand_schema(&self) -> TokenStream {
        let name = &self.name;
        match self.tag {
            TagType::External => {
                quote_spanned!(self.span => format!("z.literal(\"{}\")", #name))
            }
            TagType::Internal { tag } => {
                quote_spanned!(self.span=> format!("z.object({{ {}: z.literal(\"{}\") }})", #tag, #name))
            }
            TagType::Adjacent { tag, .. } => {
                quote_spanned!(self.span=> format!("z.object({{ {}: z.literal(\"{}\") }})", #tag, #name))
            }
            TagType::None => {
                quote_spanned!(self.span => String::from("z.null()"))
            }
        }
    }

    /// Example `A`  ->  `"A"`
    pub fn expand_type_defs(&self) -> TokenStream {
        let name = &self.name;
        match self.tag {
            TagType::External => {
                quote_spanned!(self.span => format!("\"{}\"", #name))
            }
            TagType::Internal { tag } => {
                quote_spanned!(self.span => format!("{{ {}: \"{}\" }}", #tag, #name))
            }
            TagType::Adjacent { tag, .. } => {
                quote_spanned!(self.span => format!("{{ {}: \"{}\" }}", #tag, #name))
            }
            TagType::None => {
                quote_spanned!(self.span => String::from("null"))
            }
        }
    }
}