Mirrors/leptos
2
0
Fork 0
mirror of https://github.com/leptos-rs/leptos synced 2024-09-19 22:11:54 +00:00
Code Issues Projects Releases Packages Wiki Activity

feat(#2946): add #[store(skip)] for #[derive(Store)] (#2975)

Browse source
This commit is contained in:
Kajetan Welc 2024-09-17 03:57:57 +02:00 committed by GitHub
parent 2bdacf636e
commit e69c7f4ae0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 25 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
reactive_stores_macro/src/lib.rs
View file

@ -80,6 +80,7 @@ impl Parse for Model {
#[derive(Clone)] #[derive(Clone)]
enum SubfieldMode { enum SubfieldMode {
Keyed(ExprClosure, Type), Keyed(ExprClosure, Type),
Skip,
} }
impl Parse for SubfieldMode { impl Parse for SubfieldMode {
@ -91,6 +92,8 @@ impl Parse for SubfieldMode {
let _eq: Token!(=) = input.parse()?; let _eq: Token!(=) = input.parse()?;
let ident: ExprClosure = input.parse()?; let ident: ExprClosure = input.parse()?;
Ok(SubfieldMode::Keyed(ident, ty)) Ok(SubfieldMode::Keyed(ident, ty))
} else if mode == "skip" {
Ok(SubfieldMode::Skip)
} else { } else {
Err(input.error("expected `key = <ident>: <Type>`")) Err(input.error("expected `key = <ident>: <Type>`"))
} }
@ -280,26 +283,29 @@ fn field_to_tokens(
if let Some(modes) = modes { if let Some(modes) = modes {
if modes.len() == 1 { if modes.len() == 1 {
let mode = &modes[0]; let mode = &modes[0];
// Can replace with a match if additional modes added match mode {
let SubfieldMode::Keyed(keyed_by, key_ty) = mode; SubfieldMode::Keyed(keyed_by, key_ty) => {
let signature = quote! { let signature = quote! {
fn #ident(self) -> #library_path::KeyedSubfield<#any_store_field, #name #generics, #key_ty, #ty> fn #ident(self) -> #library_path::KeyedSubfield<#any_store_field, #name #generics, #key_ty, #ty>
}; };
return if include_body { return if include_body {
quote! { quote! {
#signature { #signature {
#library_path::KeyedSubfield::new( #library_path::KeyedSubfield::new(
self, self,
#idx.into(), #idx.into(),
#keyed_by, #keyed_by,
|prev| &prev.#locator, |prev| &prev.#locator,
|prev| &mut prev.#locator, |prev| &mut prev.#locator,
) )
} }
}
} else {
quote! { #signature; }
};
} }
} else { SubfieldMode::Skip => return quote! {},
quote! { #signature; } }
};
} else { } else {
abort!( abort!(
orig_ident orig_ident