a3s-boot-macros 0.1.2

Attribute macros for a3s-boot
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use quote::quote;
use syn::{Attribute, Ident, LitStr, Result, Type};

#[derive(Clone)]
pub(crate) struct UploadedFileExtractor {
    pub(crate) name: LitStr,
}

pub(crate) struct UploadOpenApiField {
    pub(crate) name: LitStr,
    pub(crate) multiple: bool,
    pub(crate) required: bool,
}

pub(crate) fn parse_uploaded_file_extractor(
    attr: &Attribute,
    name: &str,
) -> Result<UploadedFileExtractor> {
    attr.parse_args::<LitStr>()
        .map(|name| UploadedFileExtractor { name })
        .map_err(|_| {
            syn::Error::new_spanned(
                attr,
                format!("#[{name}] requires a multipart field-name string literal"),
            )
        })
}

pub(crate) fn uploaded_file_extractor_tokens(
    ident: Ident,
    ty: Box<Type>,
    spec: UploadedFileExtractor,
    optional: bool,
) -> proc_macro2::TokenStream {
    let name = spec.name;
    if optional {
        quote! {
            let #ident: #ty = __a3s_boot_multipart_form.file(#name).cloned();
        }
    } else {
        quote! {
            let #ident: #ty = __a3s_boot_multipart_form
                .file(#name)
                .cloned()
                .ok_or_else(|| ::a3s_boot::BootError::BadRequest(
                    format!("missing uploaded file: {}", #name)
                ))?;
        }
    }
}

pub(crate) fn uploaded_files_extractor_tokens(
    ident: Ident,
    ty: Box<Type>,
    spec: UploadedFileExtractor,
) -> proc_macro2::TokenStream {
    let name = spec.name;
    quote! {
        let #ident: #ty = __a3s_boot_multipart_form
            .files_by_name(#name)
            .into_iter()
            .cloned()
            .collect();
    }
}

pub(crate) fn multipart_openapi_token(
    fields: Vec<UploadOpenApiField>,
) -> Option<proc_macro2::TokenStream> {
    if fields.is_empty() {
        return None;
    }

    let mut properties = Vec::new();
    let mut required = Vec::new();

    for field in fields {
        let name = field.name;
        if field.multiple {
            properties.push(quote! {
                (
                    #name,
                    ::a3s_boot::OpenApiSchema::array(
                        ::a3s_boot::OpenApiSchema::binary_file()
                    )
                )
            });
        } else {
            properties.push(quote! {
                (#name, ::a3s_boot::OpenApiSchema::binary_file())
            });
        }

        if field.required {
            required.push(quote!(#name));
        }
    }

    let required = if required.is_empty() {
        quote!(::std::iter::empty::<&str>())
    } else {
        quote!([#(#required),*])
    };

    Some(quote! {
        with_request_body_content_type(
            "multipart/form-data",
            ::a3s_boot::OpenApiSchema::object_with_properties(
                [#(#properties),*],
                #required,
            )
        )
    })
}