use std::marker::PhantomData;
use hecs::{Component, EntityBuilder, EntityRef};
use nbt::CompoundTag;
pub trait EntityCodec: Send + Sync {
fn encode(&self, src: &EntityRef, dst: &mut CompoundTag) -> Result<(), String>;
fn decode(&self, src: &CompoundTag, dst: &mut EntityBuilder) -> Result<(), String>;
fn default(&self, dst: &mut EntityBuilder);
}
pub trait SingleEntityCodec {
type Comp: Default + Component;
fn encode(&self, src: &Self::Comp, dst: &mut CompoundTag);
fn decode(&self, src: &CompoundTag) -> Self::Comp;
}
impl<C, D> EntityCodec for C
where
C: Send + Sync,
C: SingleEntityCodec<Comp=D>,
D: Default + Component
{
fn encode(&self, src: &EntityRef, dst: &mut CompoundTag) -> Result<(), String> {
if let Some(comp) = src.get::<D>() {
<Self as SingleEntityCodec>::encode(self, &*comp, dst);
}
Ok(())
}
fn decode(&self, src: &CompoundTag, dst: &mut EntityBuilder) -> Result<(), String> {
dst.add(<Self as SingleEntityCodec>::decode(self, src));
Ok(())
}
fn default(&self, dst: &mut EntityBuilder) {
dst.add(D::default());
}
}
pub struct DefaultEntityCodec<T>(PhantomData<*const T>);
unsafe impl<T> Send for DefaultEntityCodec<T> {}
unsafe impl<T> Sync for DefaultEntityCodec<T> {}
impl<T> DefaultEntityCodec<T> {
pub const fn new() -> Self {
Self(PhantomData)
}
}
#[allow(unused_variables)]
impl<T> EntityCodec for DefaultEntityCodec<T>
where
T: Default + Component
{
fn encode(&self, src: &EntityRef, dst: &mut CompoundTag) -> Result<(), String> {
Ok(())
}
fn decode(&self, src: &CompoundTag, dst: &mut EntityBuilder) -> Result<(), String> {
Ok(())
}
fn default(&self, dst: &mut EntityBuilder) {
dst.add(<T as Default>::default());
}
}
pub struct NullEntityCodec;
#[allow(unused_variables)]
impl EntityCodec for NullEntityCodec {
fn encode(&self, src: &EntityRef, dst: &mut CompoundTag) -> Result<(), String> {
Ok(())
}
fn decode(&self, src: &CompoundTag, dst: &mut EntityBuilder) -> Result<(), String> {
Ok(())
}
fn default(&self, dst: &mut EntityBuilder) { }
}