rustpatcher_macros/lib.rs
1use proc_macro::TokenStream;
2use quote::quote;
3use syn::{parse_macro_input, ItemFn};
4
5#[proc_macro_attribute]
6pub fn main(_args: TokenStream, input: TokenStream) -> TokenStream {
7 let input_fn = parse_macro_input!(input as ItemFn);
8 let expanded = quote! {
9 // Create a static initializer that runs before main
10 #[ctor::ctor]
11 fn __init_version() {
12 rustpatcher::version_embed::__set_version(env!("CARGO_PKG_VERSION"));
13 }
14
15 #input_fn
16 };
17
18 TokenStream::from(expanded)
19}