DisconnectEvent

1 files changed, 31 insertions, 0 deletions

diff --git a/azalea-client/src/disconnect.rs b/azalea-client/src/disconnect.rs
new file mode 100644
index 00000000..100618c3
--- /dev/null
+++ b/azalea-client/src/disconnect.rs
@@ -0,0 +1,31 @@
+//! Disconnect a client from the server.
+
+use azalea_ecs::{
+    app::{App, CoreStage, Plugin},
+    entity::Entity,
+    event::EventReader,
+    system::Commands,
+};
+
+use crate::client::JoinedClientBundle;
+
+pub struct DisconnectPlugin;
+impl Plugin for DisconnectPlugin {
+    fn build(&self, app: &mut App) {
+        app.add_event::<DisconnectEvent>()
+            .add_system_to_stage(CoreStage::PostUpdate, handle_disconnect);
+    }
+}
+
+/// An event sent when a client is getting disconnected.
+pub struct DisconnectEvent {
+    pub entity: Entity,
+}
+
+/// System that removes the [`JoinedClientBundle`] from the entity when it
+/// receives a [`DisconnectEvent`].
+pub fn handle_disconnect(mut commands: Commands, mut events: EventReader<DisconnectEvent>) {
+    for DisconnectEvent { entity } in events.iter() {
+        commands.entity(*entity).remove::<JoinedClientBundle>();
+    }
+}