From 284f294c6d898ab67cd65f7fab6f5c891787ebad Mon Sep 17 00:00:00 2001
From: Stephen Seo <seo.disparate@gmail.com>
Date: Fri, 8 Apr 2022 11:55:31 +0900
Subject: [PATCH] Fix not "disconnecting" on refresh

---
 front_end/src/yew_components.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/front_end/src/yew_components.rs b/front_end/src/yew_components.rs
index 08d7c20..0b38736 100644
--- a/front_end/src/yew_components.rs
+++ b/front_end/src/yew_components.rs
@@ -191,6 +191,9 @@ impl Component for MainMenu {
                                     window
                                         .add_event_listener_with_callback("pagehide", &resolve)
                                         .expect("Should be able to set \"pagehide\" callback");
+                                    window
+                                        .add_event_listener_with_callback("beforeunload", &resolve)
+                                        .expect("Should be able to set \"beforeunload\" callback");
                                 });
                             let js_fut = JsFuture::from(promise);
                             js_fut.await.ok();