fire_stream_api_codegen/
lib.rs

1mod 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/*
13#[api(Request)]
14async fn request(req: Request, session: &Session) -> Result<Response> {
15	todo!()
16}
17*/
18#[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/*
33#[derive(IntoMessage)]
34#[message(json)]
35*/
36#[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}