use crate::protocol::*;
use crate::shared::*;
use bevy::prelude::*;
use lightyear::prelude::server::NetConfig;
use lightyear::prelude::*;
use lightyear::server::config::*;
use lightyear::server::plugin::PluginConfig;
use lightyear::server::plugin::ServerPlugin;

pub struct NetworkPlugin {
    pub transport: TransportConfig,
}

impl Plugin for NetworkPlugin {
    fn build(&self, app: &mut App) {
        app.add_plugins(ServerPlugin::new(PluginConfig::new(
            ServerConfig {
                net: [NetConfig::Netcode {
                    config: NetcodeConfig::default()
                        .with_protocol_id(PROTOCOL_ID)
                        .with_key(KEY),
                    io: IoConfig::from_transport(self.transport.clone()),
                }]
                .to_vec(),
                ping: PingConfig::default(),
                ..Default::default()
            },
            protocol(),
        )));
    }
}