1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use proc_macro::TokenStream;

use hooks_derive_core::{
    darling::FromMeta,
    proc_macro2,
    quote::ToTokens,
    syn::{parse_macro_input, AttributeArgs, ItemFn},
};

#[proc_macro_attribute]
pub fn hook(args: TokenStream, input: TokenStream) -> TokenStream {
    let attr_args = parse_macro_input!(args as AttributeArgs);

    let args = match ::hooks_derive_core::HookArgs::from_list(&attr_args) {
        Ok(v) => v,
        Err(e) => {
            return TokenStream::from(e.write_errors());
        }
    };

    let item_fn = parse_macro_input!(input as ItemFn);

    let (item_fn, error) = args.transform_item_fn(item_fn);

    let item_fn = item_fn.into_token_stream();

    if let Some(error) = error {
        proc_macro2::TokenStream::from_iter([item_fn, error.write_errors()])
    } else {
        item_fn
    }
    .into()
}