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, events::death_listener};
10
11pub mod config;
12pub mod game;
13pub mod login;
14
15pub struct PacketPlugin;
16
17pub fn death_event_on_0_health(
18    query: Query<(Entity, &Health), Changed<Health>>,
19    mut death_events: EventWriter<DeathEvent>,
20) {
21    for (entity, health) in query.iter() {
22        if **health == 0. {
23            death_events.write(DeathEvent {
24                entity,
25                packet: None,
26            });
27        }
28    }
29}
30
31impl Plugin for PacketPlugin {
32    fn build(&self, app: &mut App) {
33        app.add_observer(game::handle_outgoing_packets_observer)
34            .add_observer(config::handle_outgoing_packets_observer)
35            .add_observer(login::handle_outgoing_packets_observer)
36            .add_systems(
37                Update,
38                (
39                    (
40                        config::handle_outgoing_packets,
41                        game::handle_outgoing_packets,
42                        login::handle_outgoing_packets,
43                    )
44                        .chain(),
45                    death_event_on_0_health.before(death_listener),
46                ),
47            )
48            .add_event::<game::ReceiveGamePacketEvent>()
49            .add_event::<config::ReceiveConfigPacketEvent>()
50            .add_event::<login::ReceiveLoginPacketEvent>()
51            //
52            .add_event::<game::SendPacketEvent>()
53            .add_event::<config::SendConfigPacketEvent>()
54            .add_event::<login::SendLoginPacketEvent>()
55            //
56            .add_event::<game::AddPlayerEvent>()
57            .add_event::<game::RemovePlayerEvent>()
58            .add_event::<game::UpdatePlayerEvent>()
59            .add_event::<ChatReceivedEvent>()
60            .add_event::<game::DeathEvent>()
61            .add_event::<game::KeepAliveEvent>()
62            .add_event::<game::ResourcePackEvent>()
63            .add_event::<game::InstanceLoadedEvent>()
64            .add_event::<login::ReceiveCustomQueryEvent>();
65    }
66}
67
68#[macro_export]
69macro_rules! declare_packet_handlers {
70    (
71        $packetenum:ident,
72        $packetvar:expr,
73        $handler:ident,
74        [$($packet:path),+ $(,)?]
75    ) => {
76        paste::paste! {
77           match $packetvar {
78                $(
79                    $packetenum::[< $packet:camel >](p) => $handler.$packet(p),
80                )+
81            }
82        }
83    };
84}
85
86pub(crate) fn as_system<T>(ecs: &mut World, f: impl FnOnce(T::Item<'_, '_>))
87where
88    T: SystemParam + 'static,
89{
90    let mut system_state = SystemState::<T>::new(ecs);
91    let values = system_state.get_mut(ecs);
92    f(values);
93    system_state.apply(ecs);
94}
azalea_client/plugins/packet/mod.rs

azalea_client/plugins/packet/
mod.rs
