mirrors/leptos
1
0
Fork 0
mirror of https://github.com/leptos-rs/leptos.git synced 2025-12-27 09:54:41 -05:00
Code Issues Packages Projects Releases 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
1 changed files with 25 additions and 19 deletions
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