use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, ItemFn};
#[proc_macro_attribute]
pub fn server_component(_attr: TokenStream, item: TokenStream) -> TokenStream {
let input = parse_macro_input!(item as ItemFn);
let fn_name = &input.sig.ident;
let fn_name_str = fn_name.to_string();
let vis = &input.vis;
let block = &input.block;
let output = &input.sig.output;
let expanded = quote! {
#vis fn #fn_name() #output {
next_rs_rsc::global_registry().register_server(module_path!(), #fn_name_str);
(|| #block)()
}
};
expanded.into()
}
#[proc_macro_attribute]
pub fn client_component(_attr: TokenStream, item: TokenStream) -> TokenStream {
let input = parse_macro_input!(item as ItemFn);
let fn_name = &input.sig.ident;
let fn_name_str = fn_name.to_string();
let vis = &input.vis;
let block = &input.block;
let output = &input.sig.output;
let expanded = quote! {
#vis fn #fn_name() #output {
next_rs_rsc::global_registry().register_client(module_path!(), #fn_name_str);
(|| #block)()
}
};
expanded.into()
}
#[proc_macro_attribute]
pub fn server_action(_attr: TokenStream, item: TokenStream) -> TokenStream {
let input = parse_macro_input!(item as ItemFn);
let fn_name = &input.sig.ident;
let fn_name_str = fn_name.to_string();
let vis = &input.vis;
let sig = &input.sig;
let block = &input.block;
let expanded = quote! {
#vis #sig {
next_rs_rsc::global_registry().register_server(module_path!(), #fn_name_str);
#block
}
};
expanded.into()
}