zephyrus-macros 0.6.0

use proc_macro2::TokenStream as TokenStream2;
use syn::{parse2, spanned::Spanned, Error, ItemFn, Result};
use crate::util;

/// The implementation of before macro, this macro takes the given input, which must be another
/// function and prepares it to be an before hook, wrapping it in a struct and providing a pointer
/// to the actual function
pub fn before(input: TokenStream2) -> Result<TokenStream2> {
    let fun = parse2::<ItemFn>(input)?;
    let ItemFn {
        attrs,
        vis,
        mut sig,
        block,
    } = fun;

    if sig.inputs.len() > 2 {
        // This hook is expected to have a `&SlashContext` and a `&str` parameter.
        return Err(Error::new(
            sig.inputs.span(),
            "Function parameter must only be &SlashContext and &str",
        ));
    }

    // The name of the original macro
    let ident = sig.ident.clone();
    // The name the function will have after this macro's execution
    let fn_ident = quote::format_ident!("_{}", &ident);
    sig.ident = fn_ident.clone();
    /*
        Check the return of the function, returning if it does not match, this function is required
        to return a `bool` indicating if the recognised command should be executed or not
    */
    util::check_return_type(&sig.output, quote::quote!(bool))?;

    let ty = util::get_context_type(&sig, true)?;
    // Get the hook macro so we can fit the function into a normal fn pointer
    let hook = util::get_hook_macro();
    let path = quote::quote!(::zephyrus::hook::BeforeHook);

    Ok(quote::quote! {
        pub fn #ident() -> #path<#ty> {
            #path(#fn_ident)
        }

        #[#hook]
        #(#attrs)*
        #vis #sig #block
    })
}