aya/aya-log-ebpf-macros/src/lib.rs

use proc_macro::TokenStream;
use syn::parse_macro_input;

mod expand;

#[proc_macro]
pub fn log(args: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as expand::LogArgs);
    expand::log(args, None)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro]
pub fn error(args: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as expand::LogArgs);
    expand::error(args)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro]
pub fn warn(args: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as expand::LogArgs);
    expand::warn(args)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro]
pub fn info(args: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as expand::LogArgs);
    expand::info(args)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro]
pub fn debug(args: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as expand::LogArgs);
    expand::debug(args)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro]
pub fn trace(args: TokenStream) -> TokenStream {
    let args = parse_macro_input!(args as expand::LogArgs);
    expand::trace(args)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}