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

use log::info;
use specs::{ReadStorage, System, WriteExpect, WriteStorage};

use crate::{
    common::ClientFilter,
    server::models::{Entity, Message, MessageType},
    world::{
        comps::{etype::ETypeComp, flags::EntityFlag, id::IDComp, metadata::MetadataComp},
        messages::MessageQueue,
    },
};

pub struct BroadcastEntitiesSystem;

impl<'a> System<'a> for BroadcastEntitiesSystem {
    type SystemData = (
        WriteExpect<'a, MessageQueue>,
        ReadStorage<'a, EntityFlag>,
        ReadStorage<'a, IDComp>,
        ReadStorage<'a, ETypeComp>,
        WriteStorage<'a, MetadataComp>,
    );

    fn run(&mut self, data: Self::SystemData) {
        use specs::Join;

        let (mut queue, flag, ids, etypes, mut metadatas) = data;

        let mut entities = vec![];
        for (id, etype, metadata, _) in (&ids, &etypes, &mut metadatas, &flag).join() {
            entities.push(Entity {
                id: id.0.to_owned(),
                r#type: etype.0.to_owned(),
                metadata: Some(metadata.to_json_string()),
            });

            metadata.reset();
        }

        if entities.is_empty() {
            return;
        }

        queue.push((
            Message::new(&MessageType::Entity)
                .entities(&entities)
                .build(),
            ClientFilter::All,
        ));
    }
}