1use proc_macro::TokenStream;
2use syn::parse_macro_input;
3
4mod agent_fn;
5mod oneshot;
6mod reactor;
7
8use agent_fn::{AgentFn, AgentName};
9use oneshot::{oneshot_impl, OneshotFn};
10use reactor::{reactor_impl, ReactorFn};
11
12#[proc_macro_attribute]
13pub fn reactor(attr: TokenStream, item: TokenStream) -> TokenStream {
14 let item = parse_macro_input!(item as AgentFn<ReactorFn>);
15 let attr = parse_macro_input!(attr as AgentName);
16
17 reactor_impl(attr, item)
18 .unwrap_or_else(|err| err.to_compile_error())
19 .into()
20}
21
22#[proc_macro_attribute]
23pub fn oneshot(attr: TokenStream, item: TokenStream) -> TokenStream {
24 let item = parse_macro_input!(item as AgentFn<OneshotFn>);
25 let attr = parse_macro_input!(attr as AgentName);
26
27 oneshot_impl(attr, item)
28 .unwrap_or_else(|err| err.to_compile_error())
29 .into()
30}