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

use std::{marker::PhantomData, net::SocketAddr};

use serde::{de::DeserializeOwned, Serialize};

use amethyst_core::{bundle::SystemBundle, ecs::World, shred::DispatcherBuilder};
use amethyst_error::{Error, ResultExt};

use crate::{server::ServerConfig, NetSocketSystem};

/// A convenience bundle to create the infrastructure needed to send and receive network messages.
#[allow(missing_debug_implementations)] // TODO: Revisit for laminar
pub struct NetworkBundle<T> {
    /// the configuration used for the networking crate.
    config: ServerConfig,
    _data: PhantomData<T>,
}

impl<T> NetworkBundle<T> {
    /// Creates a new NetworkBundle.
    ///
    /// `receive_addr`: this is the address on which we will receive incoming packets.
    /// `send_addr`: this is the address from which we will send outgoing packets.
    pub fn new(udp_socket_addr: SocketAddr) -> Self {
        let config = ServerConfig {
            udp_socket_addr,
            ..Default::default()
        };

        NetworkBundle {
            config,
            _data: PhantomData,
        }
    }

    /// Construct a new `NetworkBundle` with the specified configuration.
    pub fn from_config(config: ServerConfig) -> NetworkBundle<T> {
        NetworkBundle {
            config,
            _data: PhantomData,
        }
    }
}

impl<'a, 'b, T> SystemBundle<'a, 'b> for NetworkBundle<T>
where
    T: Send + Sync + PartialEq + Serialize + Clone + DeserializeOwned + 'static,
{
    /// Build the networking bundle by adding the networking system to the application.
    fn build(
        self,
        _world: &mut World,
        builder: &mut DispatcherBuilder<'_, '_>,
    ) -> Result<(), Error> {
        let socket_system = NetSocketSystem::<T>::new(self.config)
            .with_context(|_| Error::from_string("Failed to open network system."))?;
        builder.add(socket_system, "net_socket", &[]);

        Ok(())
    }
}