1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

// TODO: delete this module in favor of valence_chat.

use bevy_app::prelude::*;
use bevy_ecs::prelude::*;
use valence_protocol::encode::WritePacket;
use valence_protocol::packets::play::{ChatMessageC2s, GameMessageS2c};
use valence_protocol::text::IntoText;

use crate::event_loop::{EventLoopPreUpdate, PacketEvent};

pub struct MessagePlugin;

impl Plugin for MessagePlugin {
    fn build(&self, app: &mut App) {
        app.add_event::<ChatMessageEvent>()
            .add_systems(EventLoopPreUpdate, handle_chat_message);
    }
}

pub trait SendMessage {
    /// Sends a system message visible in the chat.
    fn send_chat_message<'a>(&mut self, msg: impl IntoText<'a>);
    /// Displays a message in the player's action bar (text above the hotbar).
    fn send_action_bar_message<'a>(&mut self, msg: impl IntoText<'a>);
}

impl<T: WritePacket> SendMessage for T {
    fn send_chat_message<'a>(&mut self, msg: impl IntoText<'a>) {
        self.write_packet(&GameMessageS2c {
            chat: msg.into_cow_text(),
            overlay: false,
        });
    }

    fn send_action_bar_message<'a>(&mut self, msg: impl IntoText<'a>) {
        self.write_packet(&GameMessageS2c {
            chat: msg.into_cow_text(),
            overlay: true,
        });
    }
}

#[derive(Event, Clone, Debug)]
pub struct ChatMessageEvent {
    pub client: Entity,
    pub message: Box<str>,
    pub timestamp: u64,
}

pub fn handle_chat_message(
    mut packets: EventReader<PacketEvent>,
    mut events: EventWriter<ChatMessageEvent>,
) {
    for packet in packets.read() {
        if let Some(pkt) = packet.decode::<ChatMessageC2s>() {
            events.send(ChatMessageEvent {
                client: packet.client,
                message: pkt.message.0.into(),
                timestamp: pkt.timestamp,
            });
        }
    }
}