1use crate::component::{component_impl, Component, ComponentName};
2use crate::hook::{hook_impl, HookFn};
3use syn::parse_macro_input;
4
5mod body;
6mod component;
7mod hook;
8
9#[proc_macro_attribute]
10pub fn component(
11 attr: proc_macro::TokenStream,
12 item: proc_macro::TokenStream,
13) -> proc_macro::TokenStream {
14 let item = parse_macro_input!(item as Component);
15 let attr = parse_macro_input!(attr as ComponentName);
16
17 component_impl(attr, item)
18 .unwrap_or_else(|err| err.to_compile_error())
19 .into()
20}
21
22#[proc_macro_error::proc_macro_error]
23#[proc_macro_attribute]
24pub fn hook(
25 attr: proc_macro::TokenStream,
26 item: proc_macro::TokenStream,
27) -> proc_macro::TokenStream {
28 let item = parse_macro_input!(item as HookFn);
29
30 if let Some(m) = proc_macro2::TokenStream::from(attr).into_iter().next() {
31 return syn::Error::new_spanned(m, "hook attribute does not accept any arguments")
32 .into_compile_error()
33 .into();
34 }
35
36 hook_impl(item)
37 .unwrap_or_else(|err| err.to_compile_error())
38 .into()
39}