ai_function/
lib.rs

1extern crate proc_macro;
2
3use proc_macro::TokenStream;
4use quote::{quote, ToTokens};
5use syn::{parse_macro_input, ItemFn};
6
7#[proc_macro_attribute]
8
9pub fn ai_function(_attr: TokenStream, item: TokenStream) -> TokenStream {
10    // parse input fun
11
12    let input_fn: ItemFn = parse_macro_input!(item as ItemFn);
13
14    // create string representation of function
15    let function_str: String = format!("{}", input_fn.to_token_stream());
16
17    // Define a new function with the same signature as the input function
18    let fn_ident: proc_macro2::Ident = input_fn.sig.ident;
19
20    let fn_inputs: syn::punctuated::Punctuated<syn::FnArg, syn::token::Comma> = input_fn.sig.inputs;
21
22    let fn_generics: syn::Generics = input_fn.sig.generics;
23
24    // Generate output function
25
26    let output: proc_macro2::TokenStream = quote! {
27        pub fn #fn_ident #fn_generics(#fn_inputs) -> &'static str {
28            #function_str
29        }
30    };
31    output.into()
32}