veloren/common/src/comp/states/wield.rs

use crate::comp::{ActionState::*, EcsStateData, IdleState, StateHandle, StateUpdate};
use std::time::Duration;

#[derive(Clone, Copy, Debug, PartialEq, Serialize, Deserialize, Eq, Hash)]
pub struct WieldState {
    /// How long before a new action can be performed
    /// after equipping
    pub equip_delay: Duration,
}

impl StateHandle for WieldState {
    fn handle(&self, ecs_data: &EcsStateData) -> StateUpdate {
        let mut update = StateUpdate {
            character: *ecs_data.character,
            pos: *ecs_data.pos,
            vel: *ecs_data.vel,
            ori: *ecs_data.ori,
        };

        // Only act once equip_delay has expired
        if self.equip_delay == Duration::default() {
            // Toggle Weapons
            if ecs_data.inputs.toggle_wield.is_just_pressed()
                && ecs_data.character.action_state.is_equip_finished()
            {
                update.character.action_state = Idle(IdleState);
                return update;
            }

            // Try weapon actions
            if ecs_data.inputs.primary.is_pressed() {
                // TODO: PrimaryStart
            } else if ecs_data.inputs.secondary.is_pressed() {
                // TODO: SecondaryStart
            }
        } else {
            // Equip delay hasn't expired yet
            // Update wield delay
            update.character.action_state = Wield(WieldState {
                equip_delay: self
                    .equip_delay
                    .checked_sub(Duration::from_secs_f32(ecs_data.dt.0))
                    .unwrap_or_default(),
            });
        }

        return update;
    }
}