aetheris-client-wasm 0.3.10

//! Client-side world state and interpolation buffer.
//!
//! This module implements the `WorldState` trait for the WASM client,
//! providing the foundation for the two-tick interpolation buffer.

use crate::shared_world::SabSlot;
use aetheris_protocol::error::WorldError;
use aetheris_protocol::events::{ComponentUpdate, ReplicationEvent};
use aetheris_protocol::traits::WorldState;
use aetheris_protocol::types::{
    ClientId, ComponentKind, LocalId, NetworkId, ShipClass, ShipStats, Transform,
};
use std::collections::BTreeMap;

/// A simplified client-side world that tracks entity states using `SabSlot`.
#[derive(Debug)]
pub struct ClientWorld {
    /// Map of `NetworkId` to the last known authoritative state.
    pub entities: BTreeMap<NetworkId, SabSlot>,
    /// The latest tick received from the server.
    pub latest_tick: u64,
}

impl Default for ClientWorld {
    fn default() -> Self {
        Self::new()
    }
}

impl ClientWorld {
    /// Creates a new, empty `ClientWorld`.
    #[must_use]
    pub fn new() -> Self {
        Self {
            entities: BTreeMap::new(),
            latest_tick: 0,
        }
    }
}

impl WorldState for ClientWorld {
    fn get_local_id(&self, network_id: NetworkId) -> Option<LocalId> {
        Some(LocalId(network_id.0))
    }

    fn get_network_id(&self, local_id: LocalId) -> Option<NetworkId> {
        Some(NetworkId(local_id.0))
    }

    fn extract_deltas(&mut self) -> Vec<ReplicationEvent> {
        Vec::new()
    }

    fn apply_updates(&mut self, updates: &[(ClientId, ComponentUpdate)]) {
        for (_, update) in updates {
            if update.tick > self.latest_tick {
                self.latest_tick = update.tick;
            }

            // Ensure entity exists for component updates
            self.entities.entry(update.network_id).or_insert_with(|| {
                tracing::debug!(
                    network_id = update.network_id.0,
                    kind = update.component_kind.0,
                    "New entity from server"
                );
                SabSlot {
                    network_id: update.network_id.0,
                    x: 0.0,
                    y: 0.0,
                    z: 0.0,
                    rotation: 0.0,
                    dx: 0.0,
                    dy: 0.0,
                    dz: 0.0,
                    hp: 100,
                    shield: 0,
                    entity_type: 0,
                    flags: 1,
                    cargo_ore: 0,
                    mining_target_id: 0,
                    mining_active: 0,
                }
            });

            self.apply_component_update(update);
        }
    }

    fn simulate(&mut self) {
        let dt = 0.05; // 20Hz
        let drag_base = 0.05;
        for slot in self.entities.values_mut() {
            // In a simple velocity integration without input, we just move by DX/DY.
            slot.x += slot.dx * dt;
            slot.y += slot.dy * dt;

            // Apply Drag (Local approximation)
            slot.dx *= 1.0 - (drag_base * dt);
            slot.dy *= 1.0 - (drag_base * dt);
        }
    }

    fn spawn_networked(&mut self) -> NetworkId {
        NetworkId(0)
    }

    fn spawn_networked_for(&mut self, _client_id: ClientId) -> NetworkId {
        self.spawn_networked()
    }

    fn despawn_networked(&mut self, network_id: NetworkId) -> Result<(), WorldError> {
        self.entities
            .remove(&network_id)
            .map(|_| ())
            .ok_or(WorldError::EntityNotFound(network_id))
    }

    fn stress_test(&mut self, _count: u16, _rotate: bool) {}

    fn spawn_kind(&mut self, _kind: u16, _x: f32, _y: f32, _rot: f32) -> NetworkId {
        NetworkId(1)
    }

    fn clear_world(&mut self) {
        self.entities.clear();
    }
}

impl ClientWorld {
    fn apply_component_update(&mut self, update: &ComponentUpdate) {
        match update.component_kind {
            // ComponentKind(1) == Transform (Spatial data)
            ComponentKind(1) => match rmp_serde::from_slice::<Transform>(&update.payload) {
                Ok(transform) => {
                    if let Some(entry) = self.entities.get_mut(&update.network_id) {
                        entry.x = transform.x;
                        entry.y = transform.y;
                        entry.z = transform.z;
                        entry.rotation = transform.rotation;
                        if transform.entity_type != 0 {
                            entry.entity_type = transform.entity_type;
                        }
                    }
                }
                Err(e) => {
                    tracing::warn!(network_id = update.network_id.0, error = ?e, "Failed to decode Transform");
                }
            },
            // ComponentKind(5) == ShipClass (Drives rendering type)
            ComponentKind(5) => match rmp_serde::from_slice::<ShipClass>(&update.payload) {
                Ok(ship_class) => {
                    if let Some(entry) = self.entities.get_mut(&update.network_id) {
                        entry.entity_type = match ship_class {
                            ShipClass::Interceptor => 1,
                            ShipClass::Dreadnought => 3,
                            ShipClass::Hauler => 4,
                        };
                    }
                }
                Err(e) => {
                    tracing::warn!(network_id = update.network_id.0, error = ?e, "Failed to decode ShipClass");
                }
            },
            // ComponentKind(6) == ShipStats (HP/Shield)
            ComponentKind(6) => match rmp_serde::from_slice::<ShipStats>(&update.payload) {
                Ok(stats) => {
                    if let Some(entry) = self.entities.get_mut(&update.network_id) {
                        entry.hp = stats.hp;
                        entry.shield = stats.shield;
                    }
                }
                Err(e) => {
                    tracing::warn!(network_id = update.network_id.0, error = ?e, "Failed to decode ShipStats");
                }
            },
            // ComponentKind(1024) == MiningBeam
            aetheris_protocol::types::MINING_BEAM_KIND => {
                use aetheris_protocol::types::MiningBeam;
                match rmp_serde::from_slice::<MiningBeam>(&update.payload) {
                    Ok(beam) => {
                        if let Some(entry) = self.entities.get_mut(&update.network_id) {
                            entry.mining_active = u8::from(beam.active);
                            #[allow(clippy::cast_possible_truncation)]
                            {
                                entry.mining_target_id = beam.target.map_or(0, |id| id.0 as u16);
                            }
                        }
                    }
                    Err(e) => {
                        tracing::warn!(network_id = update.network_id.0, error = ?e, "Failed to decode MiningBeam");
                    }
                }
            }
            // ComponentKind(1025) == CargoHold
            aetheris_protocol::types::CARGO_HOLD_KIND => {
                use aetheris_protocol::types::CargoHold;
                match rmp_serde::from_slice::<CargoHold>(&update.payload) {
                    Ok(cargo) => {
                        if let Some(entry) = self.entities.get_mut(&update.network_id) {
                            entry.cargo_ore = cargo.ore_count;
                        }
                    }
                    Err(e) => {
                        tracing::warn!(network_id = update.network_id.0, error = ?e, "Failed to decode CargoHold");
                    }
                }
            }
            // ComponentKind(1026) == Asteroid
            aetheris_protocol::types::ASTEROID_KIND => {
                if let Some(entry) = self.entities.get_mut(&update.network_id) {
                    // Mark as Asteroid type (5) if not already set
                    if entry.entity_type == 0 {
                        entry.entity_type = 5;
                    }
                }
            }
            kind => {
                tracing::debug!(
                    network_id = update.network_id.0,
                    kind = kind.0,
                    "Unhandled component kind"
                );
            }
        }
    }
}