apigate-macros 0.2.5

Procedural macros for apigate — #[service], #[hook], #[map], and route attributes
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

use syn::parse::{Parse, ParseStream};
use syn::punctuated::Punctuated;
use syn::{Error, Ident, LitStr, Result, Token};

use crate::parse::{parse_assigned, set_once};

pub(crate) struct ServiceArgs {
    pub name: Option<LitStr>,
    pub prefix: Option<LitStr>,
    pub policy: Option<LitStr>,
}

enum ServiceArg {
    Name(LitStr),
    Prefix(LitStr),
    Policy(LitStr),
}

impl Parse for ServiceArg {
    fn parse(input: ParseStream) -> Result<Self> {
        let key: Ident = input.parse()?;
        let value: LitStr = parse_assigned(input)?;

        match key.to_string().as_str() {
            "name" => Ok(Self::Name(value)),
            "prefix" => Ok(Self::Prefix(value)),
            "policy" => Ok(Self::Policy(value)),
            _ => Err(Error::new(key.span(), format!("unknown argument `{key}`"))),
        }
    }
}

impl Parse for ServiceArgs {
    fn parse(input: ParseStream) -> Result<Self> {
        let args = input.call(Punctuated::<ServiceArg, Token![,]>::parse_terminated)?;

        let mut name = None;
        let mut prefix = None;
        let mut policy = None;

        for arg in args {
            match arg {
                ServiceArg::Name(v) => set_once(&mut name, v.clone(), v.span(), "name")?,
                ServiceArg::Prefix(v) => set_once(&mut prefix, v.clone(), v.span(), "prefix")?,
                ServiceArg::Policy(v) => set_once(&mut policy, v.clone(), v.span(), "policy")?,
            }
        }

        Ok(Self {
            name,
            prefix,
            policy,
        })
    }
}