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