sark-gen 0.26.0

Sark proc-macro generators
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

use crate::{
    app::{serve, spec},
    runtime::paths::Paths,
};

pub(super) fn define<M>(
    paths: &Paths,
    vis: &syn::Visibility,
    entries: &[serve::RouteSpec<'_, M>],
) -> Vec<proc_macro2::TokenStream> {
    use quote::quote;

    let ns_core = &paths.core;
    let ns_http = &paths.http;

    let heads = entries.iter().map(|item| {
        let slot = &item.head;
        let entry = item.entry;
        let route = &entry.route;
        quote! {
            #[doc(hidden)]
            #vis struct #slot {
                head: #ns_core::request::raw::Parsed<
                    #ns_core::request::Head<
                        #ns_http::field::metadata::section::storage::Bytes,
                        <<#route as #ns_core::route::Spec>::Request
                            as #ns_core::route::Schema>::Frame,
                    >,
                    <#route as #ns_core::route::Spec>::Request,
                >,
                content_length: ::core::option::Option<::core::primitive::usize>,
            }
        }
    });
    let requests = entries.iter().map(|item| {
        let slot = &item.request;
        let entry = item.entry;
        let route = &entry.route;
        let length = match item.kind {
            spec::Kind::Duplex => {
                quote! { content_length: ::core::option::Option<::core::primitive::usize>, }
            }
            spec::Kind::Sync | spec::Kind::Fiber | spec::Kind::Stream => {
                proc_macro2::TokenStream::new()
            }
        };
        quote! {
            #[doc(hidden)]
            #vis struct #slot {
                head: #ns_core::request::raw::Parsed<
                    #ns_core::request::Head<
                        #ns_http::field::metadata::section::storage::Bytes,
                        <<#route as #ns_core::route::Spec>::Request
                            as #ns_core::route::Schema>::Frame,
                    >,
                    <#route as #ns_core::route::Spec>::Request,
                >,
                #length
            }
        }
    });
    heads.chain(requests).collect()
}

pub(super) fn implement<M>(
    paths: &Paths,
    prepared: &syn::Ident,
    entries: &[serve::RouteSpec<'_, M>],
) -> Vec<proc_macro2::TokenStream> {
    use quote::quote;

    let ns_core = &paths.core;

    entries
        .iter()
        .map(|item| {
            let slot = &item.request;
            let variant = &item.prepared;
            quote! {
                impl #ns_core::decode::state::Slot<#prepared> for #slot {
                    fn prepare(self) -> #prepared { #prepared::#variant(self) }
                }
            }
        })
        .collect()
}