waiter_codegen 1.6.6

Macro for Waiter DI
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

use proc_macro::TokenStream;

use proc_macro2::TokenStream as TokenStream2;
use quote::ToTokens;
use syn::{Attribute, Error, Path};
use syn::{ExprAssign, LitStr};
use syn::parse::Parser;
use syn::parse_macro_input::parse;
use syn::punctuated::Punctuated;
use syn::token::Comma;

pub(crate) struct ProvidesAttr {
    pub profiles: Vec<Path>,
}

pub(crate) fn parse_provides_attr(attr: TokenStream) -> Result<ProvidesAttr, Error> {
    let profiles_syn = <Punctuated<Path, Comma>>::parse_terminated.parse(attr)?;

    let profiles: Vec<Path> = profiles_syn
        .iter().cloned()
        .collect();

    Ok(ProvidesAttr { profiles })
}


#[derive(Clone)]
pub(crate) struct PropAttr {
    pub(crate) name: Option<String>,
    pub(crate) default_value: Option<TokenStream2>,
}

pub(crate) fn parse_prop_attr(attr: &Attribute) -> Result<PropAttr, Error> {
    if attr.tokens.is_empty() {
        Ok(PropAttr { name: None, default_value: None })
    } else {
        attr.parse_args::<ExprAssign>()
            .and_then(|with_default| {
                let name = parse::<LitStr>(with_default.left.to_token_stream().into())?;

                Ok(PropAttr {
                    name: Some(name.value()),
                    default_value: Some(with_default.right.to_token_stream()),
                })
            })
            .or_else(|_| {
                Ok(PropAttr {
                    name: Some(attr.parse_args::<LitStr>()?.value()),
                    default_value: None,
                })
            })
    }
}