separate state from rsc

macro/src/lib.rs

@@ -2,8 +2,8 @@ extern crate proc_macro;
 use proc_macro::TokenStream;
 use quote::quote;
 use syn::{
-    Attribute, Block, Error, Fields, FieldsNamed, GenericParam, Generics, Ident, ItemStruct,
-    ItemTrait, Meta, Signature, Token, Visibility,
+    Attribute, Block, Error, GenericParam, Generics, Ident, ItemStruct, ItemTrait, Signature,
+    Token, Type, Visibility,
     parse::{Parse, ParseStream, Result},
     parse_macro_input, parse_quote,
     spanned::Spanned,
@@ -89,104 +89,65 @@ pub fn widget_trait(input: TokenStream) -> TokenStream {
     quote! {
         #trai
 
-        impl #generics #name<State, WL, Tag> for WL {
+        impl #generics #name<Rsc, WL, Tag> for WL {
             #(#impls)*
         }
     }
     .into()
 }
 
-#[proc_macro_derive(UiState, attributes(rsc))]
-pub fn derive_ui_state(input: TokenStream) -> TokenStream {
+#[proc_macro_derive(DefaultUiState, attributes(default_ui_state))]
+pub fn derive_default_ui_state(input: TokenStream) -> TokenStream {
     let mut output = proc_macro2::TokenStream::new();
 
     let state: ItemStruct = parse_macro_input!(input);
-    let sname = state.ident;
-    let rscname = Ident::new(&(sname.to_string() + "Rsc"), sname.span());
-    let mut rsc_fields = Vec::new();
 
-    for field in state.fields {
-        let Some(attr) = field.attrs.iter().find(|a| a.path().is_ident("rsc")) else {
+    let mut found_attr = false;
+    let mut state_field = None;
+    for field in &state.fields {
+        if !found_attr
+            && let Type::Path(path) = &field.ty
+            && path.path.is_ident("DefaultUiState")
+        {
+            state_field = Some(field);
+        }
+        let Some(attr) = field
+            .attrs
+            .iter()
+            .find(|a| a.path().is_ident("default_ui_state"))
+        else {
             continue;
         };
-        let Meta::List(list) = &attr.meta else {
-            output.extend(Error::new(attr.span(), "invalid attr syntax").into_compile_error());
+        if found_attr {
+            output.extend(
+                Error::new(
+                    attr.span(),
+                    "cannot have more than one default_ui_state attribute",
+                )
+                .into_compile_error(),
+            );
             continue;
-        };
-        let tname: Ident = match list.parse_args::<Ident>() {
-            Ok(ident) => ident,
-            Err(err) => {
-                output.extend(err.to_compile_error());
-                continue;
-            }
-        };
-        let fty = &field.ty;
-        let fname = &field.ident.unwrap();
-        rsc_fields.extend(quote! {#fname: #fty,});
-        output.extend(quote! {
-            impl #tname for #sname {
-                fn get(&self) -> &#fty {
-                    &self.#fname
-                }
-                fn get_mut(&mut self) -> &mut #fty {
-                    &mut self.#fname
-                }
-            }
-            impl #tname for #rscname {
-                fn get(&self) -> &#fty {
-                    &self.#fname
-                }
-                fn get_mut(&mut self) -> &mut #fty {
-                    &mut self.#fname
-                }
-            }
-        });
+        }
+        found_attr = true;
+        state_field = Some(field);
     }
-    let vis = state.vis;
+    let Some(field) = state_field else {
+        output.extend(
+            Error::new(state.ident.span(), "no DefaultUiState field found").into_compile_error(),
+        );
+        return output.into();
+    };
+    let sname = &state.ident;
+    let fname = field.ident.as_ref().unwrap();
     output.extend(quote! {
-        #vis struct #rscname {
-            #(#rsc_fields)*
-        }
-
-        impl HasState for #sname {
-            type State = #sname;
-        }
-
-        impl HasState for #rscname {
-            type State = #sname;
-        }
-
-        impl StateLike<#sname> for #sname {
-            fn as_state(&mut self) -> &mut Self {
-                self
+        impl iris::default::HasDefaultUiState for #sname {
+            fn default_state(&self) -> &iris::default::DefaultUiState {
+                &self.#fname
             }
-        }
-
-        impl StateLike<#rscname> for #rscname {
-            fn as_state(&mut self) -> &mut Self {
-                self
+            fn default_state_mut(&mut self) -> &mut iris::default::DefaultUiState {
+                &mut self.#fname
             }
         }
     });
     output.into()
 }
-
-#[proc_macro_attribute]
-pub fn default_ui_state(_attr: TokenStream, input: TokenStream) -> TokenStream {
-    let mut state: ItemStruct = parse_macro_input!(input);
-    let Fields::Named(fields) = &mut state.fields else {
-        panic!("must be on named fields struct");
-    };
-    let name = &state.ident;
-    state.attrs.push(parse_quote! {#[derive(UiState)]});
-    let new: FieldsNamed = parse_quote! {{
-        #[rsc(HasUi)]
-        pub ui: Ui,
-        #[rsc(HasDefaultUiState)]
-        pub ui_state: DefaultUiState,
-        #[rsc(HasEvents)]
-        pub events: iris::prelude::EventManager<#name>,
-    }};
-    fields.named.extend(new.named);
-    quote! {#state}.into()
-}