Reconnect websocket on window focus (#366)

reflex-dev · Jan 28, 2023 · 0056d94 · 0056d94
1 parent 5427dd0
commit 0056d94
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pynecone/compiler/templates.py b/pynecone/compiler/templates.py
@@ -164,8 +164,14 @@ def format_state(
         "  if(!isReady) {{",
         "    return;",
         "  }}",
-        f"  if (!{SOCKET}.current) {{{{",
-        f"    connect({SOCKET}, {{state}}, {{set_state}}, {RESULT}, {SET_RESULT}, {ROUTER}, {EVENT_ENDPOINT})",
+        "  const reconnectSocket = () => {{",
+        f"    {SOCKET}.current.reconnect()",
+        "  }}",
+        f"  if (typeof {SOCKET}.current !== 'undefined') {{{{",
+        f"    if (!{SOCKET}.current) {{{{",
+        f"      window.addEventListener('focus', reconnectSocket)",
+        f"      connect({SOCKET}, {{state}}, {{set_state}}, {RESULT}, {SET_RESULT}, {ROUTER}, {EVENT_ENDPOINT})",
+        "    }}",
         "  }}",
         "  const update = async () => {{",
         f"    if ({RESULT}.{STATE} != null) {{{{",