naia-shared 0.24.0

Common functionality shared between naia-server & naia-client crates
Documentation
use std::sync::{Arc, RwLock};

use crate::{
    world::delegation::entity_auth_status::{EntityAuthStatus, HostEntityAuthStatus},
    HostType,
};

// EntityAuthChannel
#[derive(Clone)]
pub(crate) struct EntityAuthChannel {
    data: Arc<RwLock<EntityAuthData>>,
}

impl EntityAuthChannel {
    pub(crate) fn new_channel(host_type: HostType) -> (EntityAuthMutator, EntityAuthAccessor) {
        let channel = Self {
            data: Arc::new(RwLock::new(EntityAuthData::new(host_type))),
        };

        let sender = EntityAuthMutator::new(&channel);
        let receiver = EntityAuthAccessor::new(&channel);

        (sender, receiver)
    }

    fn auth_status(&self) -> HostEntityAuthStatus {
        let data = self
            .data
            .as_ref()
            .read()
            .expect("Lock on AuthStatus is held by current thread.");
        return data.auth_status();
    }

    fn set_auth_status(&self, auth_status: EntityAuthStatus) {
        let mut data = self
            .data
            .as_ref()
            .write()
            .expect("Lock on AuthStatus is held by current thread.");
        data.set_auth_status(auth_status);
    }
}

// EntityAuthData
struct EntityAuthData {
    host_type: HostType,
    status: EntityAuthStatus,
}

impl EntityAuthData {
    fn new(host_type: HostType) -> Self {
        let status = match host_type {
            HostType::Server => EntityAuthStatus::Available,
            HostType::Client => EntityAuthStatus::Requested,
        };
        Self { host_type, status }
    }

    fn auth_status(&self) -> HostEntityAuthStatus {
        HostEntityAuthStatus::new(self.host_type, self.status)
    }

    fn set_auth_status(&mut self, auth_status: EntityAuthStatus) {
        self.status = auth_status;
    }
}

// EntityAuthAccessor
#[derive(Clone)]
pub struct EntityAuthAccessor {
    channel: EntityAuthChannel,
}

impl EntityAuthAccessor {
    fn new(channel: &EntityAuthChannel) -> Self {
        Self {
            channel: channel.clone(),
        }
    }

    pub(crate) fn auth_status(&self) -> HostEntityAuthStatus {
        self.channel.auth_status()
    }
}

// EntityAuthMutator
// no Clone necessary
pub(crate) struct EntityAuthMutator {
    channel: EntityAuthChannel,
}

impl EntityAuthMutator {
    fn new(channel: &EntityAuthChannel) -> Self {
        Self {
            channel: channel.clone(),
        }
    }

    pub(crate) fn set_auth_status(&self, auth_status: EntityAuthStatus) {
        self.channel.set_auth_status(auth_status);
    }
}