Skip to main content

dce_macro/
lib.rs

1mod funcs;
2mod router;
3
4use proc_macro::TokenStream;
5use quote::quote;
6use syn::{parse_macro_input, ItemFn};
7use router::Api;
8use funcs::DceError;
9
10#[proc_macro_attribute]
11pub fn api(args: TokenStream, input: TokenStream) -> TokenStream {
12    let api = parse_macro_input!(args as Api);
13    let input = parse_macro_input!(input as ItemFn);
14    api.processing(input).into()
15}
16
17
18#[proc_macro]
19pub fn pub_msg(args: TokenStream) -> TokenStream {
20    let err = parse_macro_input!(args as DceError);
21    let call = err.gen_func(true);
22    quote!(#call).into()
23}
24
25#[proc_macro]
26pub fn priv_msg(args: TokenStream) -> TokenStream {
27    let err = parse_macro_input!(args as DceError);
28    let call = err.gen_func(false);
29    quote!(#call).into()
30}
31
32#[proc_macro]
33pub fn err_pub_msg(args: TokenStream) -> TokenStream {
34    let err = parse_macro_input!(args as DceError);
35    let call = err.gen_func(true);
36    quote!(Err(#call)).into()
37}
38
39#[proc_macro]
40pub fn err_priv_msg(args: TokenStream) -> TokenStream {
41    let err = parse_macro_input!(args as DceError);
42    let call = err.gen_func(false);
43    quote!(Err(#call)).into()
44}