sark-gen 0.6.3

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
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 proc_macro2::TokenStream;
use quote::quote;
use syn::{Ident, Result, Type};

use crate::codegen::value::Value;
use crate::model::{HeaderAttrField, QueryAttrField};
use crate::util::TypeExt;

pub(super) struct Hidden<'a> {
    pub(super) name: &'a Ident,
    pub(super) inner_name: &'a Ident,
    pub(super) raw_name: &'a Ident,
    pub(super) headers: &'a [HeaderAttrField],
    pub(super) queries: &'a [QueryAttrField],
    pub(super) ordered_query_state: bool,
}

impl<'a> Hidden<'a> {
    pub(super) fn header_query_exprs(
        headers: &[HeaderAttrField],
        queries: &[QueryAttrField],
    ) -> Result<Vec<TokenStream>> {
        headers
            .iter()
            .map(|f| {
                Value::build_header_query_field(
                    &f.ty,
                    f.default.as_ref(),
                    &f.ident,
                    quote!(req.local_at(range)),
                    "request header/query",
                )
            })
            .chain(queries.iter().map(|f| {
                Value::build_header_query_field(
                    &f.ty,
                    f.default.as_ref(),
                    &f.ident,
                    quote!(req.local_at(range)),
                    "request header/query",
                )
            }))
            .collect()
    }

    pub(super) fn build(self) -> Result<TokenStream> {
        let Self {
            name,
            inner_name,
            raw_name,
            headers,
            queries,
            ordered_query_state,
        } = self;
        let all_ident: Vec<&Ident> = headers
            .iter()
            .map(|f| &f.ident)
            .chain(queries.iter().map(|f| &f.ident))
            .collect();
        let all_ty: Vec<&Type> = headers
            .iter()
            .map(|f| &f.ty)
            .chain(queries.iter().map(|f| &f.ty))
            .collect();
        let all_ty_ref: Vec<Type> = all_ty
            .iter()
            .map(|ty| {
                let mut ty = (*ty).clone();
                ty.rewrite_local_to_ref();
                ty
            })
            .collect();
        let raw_ty: Vec<Type> = all_ty
            .iter()
            .map(|ty| ty.raw_field_ty())
            .collect::<Result<_>>()?;
        let query_state = if ordered_query_state {
            quote! { __query: u8, }
        } else {
            quote! {}
        };
        let typed_field_expr = Self::header_query_exprs(headers, queries)?;
        Ok(quote! {
            #[allow(non_camel_case_types)]
            struct #inner_name<'req> {
                #( #all_ident: #all_ty_ref, )*
                #[doc(hidden)]
                __sark_m: ::core::marker::PhantomData<&'req ()>,
            }

            #[allow(non_camel_case_types, dead_code)]
            type #name = #inner_name<'static>;

            #[derive(Default)]
            struct #raw_name { #( #all_ident: #raw_ty, )* #query_state }

            impl<'req> sark::service::HeaderParams for #inner_name<'req> {}

            impl<'req> #inner_name<'req> {
                #[allow(dead_code, unused_variables)]
                fn from_raw_ref(
                    req: &sark::request::Ref<'req, ()>,
                    headers: #raw_name,
                ) -> sark::error::Result<Self> {
                    let #raw_name { #( #all_ident, )* .. } = headers;
                    Ok(Self {
                        #( #all_ident: #typed_field_expr, )*
                        __sark_m: ::core::marker::PhantomData,
                    })
                }
            }
        })
    }
}