naia-server 0.15.0

use std::{collections::HashMap, hash::Hash, net::SocketAddr};

use naia_shared::ProtocolKindType;

use super::mut_channel::{MutChannel, MutReceiver, MutReceiverBuilder, MutSender};

pub struct GlobalDiffHandler<E: Copy + Eq + Hash, K: ProtocolKindType> {
    mut_receiver_builders: HashMap<(E, K), MutReceiverBuilder>,
}

impl<E: Copy + Eq + Hash, K: ProtocolKindType> Default for GlobalDiffHandler<E, K> {
    fn default() -> Self {
        Self {
            mut_receiver_builders: HashMap::default(),
        }
    }
}

impl<E: Copy + Eq + Hash, K: ProtocolKindType> GlobalDiffHandler<E, K> {
    // For Server
    pub fn register_component(
        &mut self,
        entity: &E,
        component_kind: &K,
        diff_mask_length: u8,
    ) -> MutSender {
        if self
            .mut_receiver_builders
            .contains_key(&(*entity, *component_kind))
        {
            panic!("Component cannot register with Server more than once!");
        }

        let (sender, builder) = MutChannel::new_channel(diff_mask_length);

        self.mut_receiver_builders
            .insert((*entity, *component_kind), builder);

        sender
    }

    pub fn deregister_component(&mut self, entity: &E, component_kind: &K) {
        self.mut_receiver_builders
            .remove(&(*entity, *component_kind));
    }

    pub fn receiver(
        &self,
        addr: &SocketAddr,
        entity: &E,
        component_kind: &K,
    ) -> Option<MutReceiver> {
        if let Some(builder) = self.mut_receiver_builders.get(&(*entity, *component_kind)) {
            return builder.build(addr);
        }
        None
    }
}