mirrors/salsa
1
0
Fork 0
mirror of https://github.com/salsa-rs/salsa.git synced 2025-01-15 01:39:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

refactor error reporting, detect too few argments

Browse source
This commit is contained in:
Niko Matsakis 2022-08-05 01:24:07 -04:00
parent 89f801276c
commit 1f1950c145
1 changed files with 17 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
components/salsa-entity-macros/src/tracked_fn.rs
View file

@ -10,18 +10,29 @@ pub(crate) fn tracked(
item_fn: syn::ItemFn, item_fn: syn::ItemFn,
) -> proc_macro::TokenStream { ) -> proc_macro::TokenStream {
let args = syn::parse_macro_input!(args as Args); let args = syn::parse_macro_input!(args as Args);
match tracked_fn(args, item_fn) {
Ok(p) => p.into(),
Err(e) => return e.into_compile_error().into(),
}
}
fn tracked_fn(args: Args, item_fn: syn::ItemFn) -> syn::Result<TokenStream> {
if item_fn.sig.inputs.len() <= 1 {
return Err(syn::Error::new(
item_fn.sig.ident.span(),
"tracked functions must have at least a database and salsa struct argument",
));
}
let struct_item = configuration_struct(&item_fn); let struct_item = configuration_struct(&item_fn);
let configuration = fn_configuration(&args, &item_fn); let configuration = fn_configuration(&args, &item_fn);
let struct_item_ident = &struct_item.ident; let struct_item_ident = &struct_item.ident;
let struct_ty: syn::Type = parse_quote!(#struct_item_ident); let struct_ty: syn::Type = parse_quote!(#struct_item_ident);
let configuration_impl = configuration.to_impl(&struct_ty); let configuration_impl = configuration.to_impl(&struct_ty);
let ingredients_for_impl = ingredients_for_impl(&args, &struct_ty); let ingredients_for_impl = ingredients_for_impl(&args, &item_fn, &struct_ty);
let (getter, setter) = match wrapper_fns(&args, &item_fn, &struct_ty) { let (getter, setter) = wrapper_fns(&args, &item_fn, &struct_ty)?;
Ok(p) => p,
Err(e) => return e.into_compile_error().into(),
};
proc_macro::TokenStream::from(quote! { Ok(quote! {
#struct_item #struct_item
#configuration_impl #configuration_impl
#ingredients_for_impl #ingredients_for_impl