clientix-codegen 0.2.0

Library for building HTTP clients and requests declaratively with procedural macros - no need to write complex imperative or functional logic.
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

use quote::quote;
use syn::__private::TokenStream2;
use crate::attributes::header::HeaderAttributes;
use crate::method::signature::segment::SegmentArgumentCompiler;

#[derive(Clone, Debug)]
pub struct HeaderCompiler {
    attributes: HeaderAttributes,
}

impl HeaderCompiler {

    pub fn new(name: Option<String>, value: Option<String>, sensitive: bool) -> Self {
        Self { attributes: HeaderAttributes::new(name, value, sensitive) }
    }

    pub fn parse(attrs: TokenStream2, dry_run: bool) -> Self {
        Self { attributes: HeaderAttributes::parse(attrs, dry_run) }
    }

    pub fn compile(&self, segments: &Vec<SegmentArgumentCompiler>) -> TokenStream2 {
        if self.attributes.name().is_none() && self.attributes.value().is_none() {
            return quote! {}
        }

        let name = self.attributes.name().clone().unwrap();
        let value = self.attributes.value().clone().unwrap();
        let sensitive = self.attributes.sensitive();

        let mut stream = TokenStream2::new();
        if !segments.is_empty() {
            stream.extend(quote!(let mut arguments = std::collections::HashMap::new();));

            for segment_argument in segments.iter() {
                let name = segment_argument.name();
                let value = segment_argument.value();
                stream.extend(quote! {
                    if let Some(segment) = #value {
                        arguments.insert(#name.to_string(), segment.to_string());
                    }
                });
            }

            stream.extend(quote!(clientix::prelude::strfmt::strfmt(#value, &arguments).expect("failed to format header").as_str()));

            TokenStream2::from(quote!(let builder = builder.header(#name, {#stream}, #sensitive);))
        } else {
            TokenStream2::from(quote!(let builder = builder.header(#name, #value, #sensitive);))
        }
    }

}