azalea_client/plugins/packet/
mod.rs1use azalea_entity::metadata::Health;
2use bevy_app::{App, Plugin, Update};
3use bevy_ecs::{
4 prelude::*,
5 system::{SystemParam, SystemState},
6};
7
8use self::game::DeathEvent;
9use crate::client_chat::ChatReceivedEvent;
10
11pub mod config;
12pub mod game;
13pub mod login;
14pub mod relative_updates;
15
16pub struct PacketPlugin;
17
18pub fn death_event_on_0_health(
19 query: Query<(Entity, &Health), Changed<Health>>,
20 mut death_events: MessageWriter<DeathEvent>,
21) {
22 for (entity, health) in query.iter() {
23 if **health == 0. {
24 death_events.write(DeathEvent {
25 entity,
26 packet: None,
27 });
28 }
29 }
30}
31
32impl Plugin for PacketPlugin {
33 fn build(&self, app: &mut App) {
34 app.add_systems(
35 Update,
36 relative_updates::debug_detect_updates_received_on_local_entities,
37 )
38 .add_observer(game::handle_outgoing_packets_observer)
39 .add_observer(config::handle_outgoing_packets_observer)
40 .add_observer(login::handle_outgoing_packets_observer)
41 .add_systems(Update, death_event_on_0_health)
42 .add_message::<game::ReceiveGamePacketEvent>()
43 .add_message::<config::ReceiveConfigPacketEvent>()
44 .add_message::<login::ReceiveLoginPacketEvent>()
45 .add_message::<game::AddPlayerEvent>()
47 .add_message::<game::RemovePlayerEvent>()
48 .add_message::<game::UpdatePlayerEvent>()
49 .add_message::<ChatReceivedEvent>()
50 .add_message::<game::DeathEvent>()
51 .add_message::<game::ResourcePackEvent>()
52 .add_message::<game::WorldLoadedEvent>()
53 .add_message::<login::ReceiveCustomQueryEvent>();
54 }
55}
56
57pub(crate) use azalea_protocol::azalea_protocol_macros::declare_packet_handlers;
58
59#[derive(Resource)]
60struct CachedSystemState<T: SystemParam + 'static>(SystemState<T>);
61
62pub(crate) fn as_system<T>(ecs: &mut World, f: impl FnOnce(T::Item<'_, '_>))
63where
64 T: SystemParam + 'static,
65{
66 let mut system_state = match ecs.remove_resource::<CachedSystemState<T>>() {
69 Some(s) => s.0,
70 None => SystemState::<T>::new(ecs),
71 };
72 let values = system_state.get_mut(ecs);
73 f(values);
74 system_state.apply(ecs);
75 ecs.insert_resource(CachedSystemState(system_state));
76}