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::{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: MessageWriter<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(Update, death_event_on_0_health.before(death_listener))
37 .add_message::<game::ReceiveGamePacketEvent>()
38 .add_message::<config::ReceiveConfigPacketEvent>()
39 .add_message::<login::ReceiveLoginPacketEvent>()
40 .add_message::<game::AddPlayerEvent>()
42 .add_message::<game::RemovePlayerEvent>()
43 .add_message::<game::UpdatePlayerEvent>()
44 .add_message::<ChatReceivedEvent>()
45 .add_message::<game::DeathEvent>()
46 .add_message::<game::KeepAliveEvent>()
47 .add_message::<game::ResourcePackEvent>()
48 .add_message::<game::InstanceLoadedEvent>()
49 .add_message::<login::ReceiveCustomQueryEvent>();
50 }
51}
52
53#[doc(hidden)]
54macro_rules! __declare_packet_handlers {
55 (
56 $packetenum:ident,
57 $packetvar:expr,
58 $handler:ident,
59 [$($packet:path),+ $(,)?]
60 ) => {
61 pastey::paste! {
62 match $packetvar {
63 $(
64 $packetenum::[< $packet:camel >](p) => $handler.$packet(p),
65 )+
66 }
67 }
68 };
69}
70
71pub(crate) use __declare_packet_handlers as declare_packet_handlers;
72
73pub(crate) fn as_system<T>(ecs: &mut World, f: impl FnOnce(T::Item<'_, '_>))
74where
75 T: SystemParam + 'static,
76{
77 let mut system_state = SystemState::<T>::new(ecs);
78 let values = system_state.get_mut(ecs);
79 f(values);
80 system_state.apply(ecs);
81}