azalea_client/plugins/packet/
mod.rs

1use 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::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        //
46        .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::KeepAliveEvent>()
52        .add_message::<game::ResourcePackEvent>()
53        .add_message::<game::WorldLoadedEvent>()
54        .add_message::<login::ReceiveCustomQueryEvent>();
55    }
56}
57
58#[doc(hidden)]
59macro_rules! __declare_packet_handlers {
60    (
61        $packetenum:ident,
62        $packetvar:expr,
63        $handler:ident,
64        [$($packet:path),+ $(,)?]
65    ) => {
66        pastey::paste! {
67           match $packetvar {
68                $(
69                    $packetenum::[< $packet:camel >](p) => $handler.$packet(p),
70                )+
71            }
72        }
73    };
74}
75
76pub(crate) use __declare_packet_handlers as declare_packet_handlers;
77
78#[derive(Resource)]
79struct CachedSystemState<T: SystemParam + 'static>(SystemState<T>);
80
81pub(crate) fn as_system<T>(ecs: &mut World, f: impl FnOnce(T::Item<'_, '_>))
82where
83    T: SystemParam + 'static,
84{
85    // creating a new SystemState is expensive, so we save them as a Resource in the
86    // ecs
87    let mut system_state = match ecs.remove_resource::<CachedSystemState<T>>() {
88        Some(s) => s.0,
89        None => SystemState::<T>::new(ecs),
90    };
91    let values = system_state.get_mut(ecs);
92    f(values);
93    system_state.apply(ecs);
94    ecs.insert_resource(CachedSystemState(system_state));
95}
azalea_client/plugins/packet/mod.rs

azalea_client/plugins/packet/
mod.rs
