From 0a590bab229d7e9a729861bcdac2a970682b5ac4 Mon Sep 17 00:00:00 2001
From: griffi-gh <prasol258@gmail.com>
Date: Tue, 14 Mar 2023 01:36:34 +0100
Subject: [PATCH] handle disconnect

---
 kubi/src/networking.rs | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/kubi/src/networking.rs b/kubi/src/networking.rs
index 6abfbb3..9e1e608 100644
--- a/kubi/src/networking.rs
+++ b/kubi/src/networking.rs
@@ -134,6 +134,19 @@ fn check_server_hello_response(
   }
 }
 
+fn handle_disconnect(
+  network_events: View<NetworkEvent>,
+  mut join_state: UniqueViewMut<ClientJoinState>
+) {
+  for event in network_events.iter() {
+    if matches!(event.0, ClientEvent::Disconnect) {
+      log::warn!("Disconnected from server");
+      *join_state = ClientJoinState::Disconnected;
+      return;
+    }
+  }
+}
+
 pub fn update_networking() -> Workload {
   (
     connect_client.run_if_missing_unique::<UdpClient>(),
@@ -143,8 +156,9 @@ pub fn update_networking() -> Workload {
       say_hello,
     ).into_sequential_workload().run_if(if_just_connected),
     (
-      check_server_hello_response
-    ).run_if(is_join_state::<{ClientJoinState::Connected as u8}>),
+      check_server_hello_response,
+      handle_disconnect,
+    ).into_sequential_workload().run_if(is_join_state::<{ClientJoinState::Connected as u8}>),
     (
       recv_block_place_events
     ).run_if(is_join_state::<{ClientJoinState::Joined as u8}>).run_if(is_ingame_or_loading),