resuma-macros 1.2.0

Proc-macros for the Resuma Rust SSR framework: view!, #[component], #[server], rs2js.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

//! `#[derive(Store)]` — field getters/setters via a local extension trait.

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, Data, DeriveInput, Fields};

pub fn derive(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    let name = &input.ident;
    let trait_name = syn::Ident::new(&format!("{name}Store"), name.span());

    let fields = match &input.data {
        Data::Struct(data) => match &data.fields {
            Fields::Named(fields) => &fields.named,
            _ => {
                return syn::Error::new_spanned(
                    &input,
                    "#[derive(Store)] requires a struct with named fields",
                )
                .to_compile_error()
                .into()
            }
        },
        _ => {
            return syn::Error::new_spanned(&input, "#[derive(Store)] supports structs only")
                .to_compile_error()
                .into()
        }
    };

    let mut getters = Vec::new();
    let mut setters = Vec::new();
    let mut trait_methods = Vec::new();

    for field in fields {
        let ident = field.ident.as_ref().expect("named field");
        let ty = &field.ty;
        let setter = syn::Ident::new(&format!("set_{ident}"), ident.span());

        trait_methods.push(quote! {
            fn #ident(&self) -> #ty;
            fn #setter(&self, value: #ty);
        });
        getters.push(quote! {
            fn #ident(&self) -> #ty {
                ::resuma::Store::get(self).#ident
            }
        });
        setters.push(quote! {
            fn #setter(&self, value: #ty) {
                ::resuma::Store::update(self, |s| s.#ident = value);
            }
        });
    }

    let expanded = quote! {
        pub trait #trait_name {
            #(#trait_methods)*
        }

        impl #trait_name for ::resuma::Store<#name> {
            #(#getters)*
            #(#setters)*
        }
    };

    expanded.into()
}