tea-codec-macros 0.3.0-dev.7

The TEA SDK
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

use proc_macro2::Ident;
use syn::{
	parse::{Parse, ParseStream},
	spanned::Spanned,
	Attribute, DeriveInput, Generics, Result, Type, TypePath,
};

pub struct Input {
	pub ident: Ident,
	pub generics: Generics,
	pub resp: Option<Type>,
}

const ATTR_RESPONSE_IDENT: &str = "response";

impl Parse for Input {
	fn parse(input: ParseStream) -> Result<Self> {
		let body = DeriveInput::parse(input)?;
		let mut resp = None;
		for attr in &body.attrs {
			if is_single_ident(ATTR_RESPONSE_IDENT)(attr) {
				if resp.is_some() {
					return Err(syn::Error::new(
						attr.span(),
						"Multiple response type specified.",
					));
				}
				resp = Some(attr.parse_args()?);
			}
		}
		if resp.is_none() {
			let ident = body.ident.to_string();
			if let Some(ident) = ident.strip_suffix("Request") {
				resp = Some(Type::Path(TypePath {
					qself: None,
					path: Ident::new((ident.to_string() + "Response").as_str(), body.ident.span())
						.into(),
				}))
			}
		}
		Ok(Self {
			resp,
			ident: body.ident,
			generics: body.generics,
		})
	}
}

const fn is_single_ident(cmp: &str) -> impl Fn(&Attribute) -> bool + '_ {
	move |attr| {
		if let Some(ident) = attr.path.get_ident() {
			ident == cmp
		} else {
			false
		}
	}
}