🎨 Add hydrate function.

celestia-island · Dec 26, 2023 · aa788b9 · aa788b9
1 parent 1a4a02c
commit aa788b9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/packages/macro-types/src/register.rs b/packages/macro-types/src/register.rs
@@ -46,6 +46,10 @@ where
         root: web_sys::Element,
         states: <Self as DeclType>::AppStates,
     ) -> ::yew::prelude::AppHandle<Self::ClientApp>;
+    fn hydrate_with_root(
+        root: web_sys::Element,
+        states: <Self as DeclType>::AppStates,
+    ) -> ::yew::prelude::AppHandle<Self::ClientApp>;
 }
 
 pub trait DeclType

diff --git a/packages/macro/src/utils/app.rs b/packages/macro/src/utils/app.rs
@@ -170,6 +170,18 @@ pub fn root(input: DeriveApp) -> TokenStream {
                     }
                 ).render()
             }
+
+            fn hydrate_with_root(
+                root: web_sys::Element,
+                states: <#ident as ::hikari_boot::DeclType>::AppStates
+            ) -> ::yew::prelude::AppHandle<<#ident as ::hikari_boot::Application>::ClientApp> {
+                ::yew::Renderer::<<#ident as ::hikari_boot::Application>::ClientApp>::with_root_and_props(
+                    root,
+                    ::hikari_boot::AppContextForClient {
+                        states
+                    }
+                ).hydrate()
+            }
         }
     }
 }