1use proc_macro::TokenStream;
2
3use hooks_derive_core::{
4 darling::FromMeta,
5 proc_macro2,
6 quote::ToTokens,
7 syn::{parse_macro_input, AttributeArgs, ItemFn},
8};
9
10#[proc_macro_attribute]
11pub fn hook(args: TokenStream, input: TokenStream) -> TokenStream {
12 let attr_args = parse_macro_input!(args as AttributeArgs);
13
14 let args = match ::hooks_derive_core::HookArgs::from_list(&attr_args) {
15 Ok(v) => v,
16 Err(e) => {
17 return TokenStream::from(e.write_errors());
18 }
19 };
20
21 let item_fn = parse_macro_input!(input as ItemFn);
22
23 let (item_fn, error) = args.transform_item_fn(item_fn);
24
25 let item_fn = item_fn.into_token_stream();
26
27 if let Some(error) = error {
28 proc_macro2::TokenStream::from_iter([item_fn, error.write_errors()])
29 } else {
30 item_fn
31 }
32 .into()
33}