lafere_api_codegen/
lib.rs1mod action;
2mod api;
3mod args;
4mod message;
5mod util;
6
7use args::ApiArgs;
8
9use proc_macro::TokenStream;
10use syn::{DeriveInput, ItemFn, parse_macro_input};
11
12#[proc_macro_attribute]
19pub fn api(attrs: TokenStream, item: TokenStream) -> TokenStream {
20 let args = parse_macro_input!(attrs as ApiArgs);
21 let item = parse_macro_input!(item as ItemFn);
22
23 let stream = api::expand(args, item);
24
25 stream
26 .map(|stream| stream.into())
27 .unwrap_or_else(|e| e.to_compile_error())
28 .into()
29}
30
31#[proc_macro_derive(IntoMessage, attributes(message))]
36pub fn derive_into_message(input: TokenStream) -> TokenStream {
37 let input = parse_macro_input!(input as DeriveInput);
38
39 message::into_expand(input)
40 .unwrap_or_else(|e| e.to_compile_error())
41 .into()
42}
43
44#[proc_macro_derive(FromMessage, attributes(message))]
45pub fn derive_from_message(input: TokenStream) -> TokenStream {
46 let input = parse_macro_input!(input as DeriveInput);
47
48 message::from_expand(input)
49 .unwrap_or_else(|e| e.to_compile_error())
50 .into()
51}
52
53#[proc_macro_derive(Action)]
54pub fn derive_action(input: TokenStream) -> TokenStream {
55 let input = parse_macro_input!(input as DeriveInput);
56
57 action::expand(input)
58 .unwrap_or_else(|e| e.to_compile_error())
59 .into()
60}