From 393a7c6e2ac1018083444ba14e7ec61f79b310a8 Mon Sep 17 00:00:00 2001
From: Masen Furer <m_github@0x26.net>
Date: Tue, 5 Sep 2023 23:27:39 -0700
Subject: [PATCH] Send updated client-side storage values when hydrating
 (#1758)

---
 reflex/.templates/jinja/web/pages/index.js.jinja2 | 2 +-
 reflex/compiler/compiler.py                       | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/reflex/.templates/jinja/web/pages/index.js.jinja2 b/reflex/.templates/jinja/web/pages/index.js.jinja2
index 87465e9e4..29de4cecd 100644
--- a/reflex/.templates/jinja/web/pages/index.js.jinja2
+++ b/reflex/.templates/jinja/web/pages/index.js.jinja2
@@ -25,7 +25,7 @@ export default function Component() {
 
   // Route after the initial page hydration.
   useEffect(() => {
-    const change_complete = () => Event([E('{{state_name}}.{{const.hydrate}}', {})])
+    const change_complete = () => Event(initialEvents.map((e) => ({...e})))
     {{const.router}}.events.on('routeChangeComplete', change_complete)
     return () => {
       {{const.router}}.events.off('routeChangeComplete', change_complete)
diff --git a/reflex/compiler/compiler.py b/reflex/compiler/compiler.py
index a177bdd71..c98ee5192 100644
--- a/reflex/compiler/compiler.py
+++ b/reflex/compiler/compiler.py
@@ -33,6 +33,7 @@ DEFAULT_IMPORTS: imports.ImportDict = {
     },
     "/utils/context.js": {
         ImportVar(tag="EventLoopContext"),
+        ImportVar(tag="initialEvents"),
         ImportVar(tag="StateContext"),
     },
     "": {ImportVar(tag="focus-visible/dist/focus-visible")},