fire_stream_api_codegen/
lib.rs1mod args;
2mod api;
3mod message;
4mod action;
5mod util;
6
7use args::ApiArgs;
8
9use proc_macro::TokenStream;
10use syn::{parse_macro_input, ItemFn, DeriveInput};
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
32#[proc_macro_derive(IntoMessage, attributes(message))]
37pub fn derive_into_message(input: TokenStream) -> TokenStream {
38 let input = parse_macro_input!(input as DeriveInput);
39
40 message::into_expand(input)
41 .unwrap_or_else(|e| e.to_compile_error())
42 .into()
43}
44
45#[proc_macro_derive(FromMessage, attributes(message))]
46pub fn derive_from_message(input: TokenStream) -> TokenStream {
47 let input = parse_macro_input!(input as DeriveInput);
48
49 message::from_expand(input)
50 .unwrap_or_else(|e| e.to_compile_error())
51 .into()
52}
53
54#[proc_macro_derive(Action)]
55pub fn derive_action(input: TokenStream) -> TokenStream {
56 let input = parse_macro_input!(input as DeriveInput);
57
58 action::expand(input)
59 .unwrap_or_else(|e| e.to_compile_error())
60 .into()
61}