vampire-like/Assets/GameMain/Scripts/Runtime/Entity/EntityLogic/Weapon/WeaponLightning/WeaponLightning.AttackState.cs

namespace SepCore.Entity.Weapon
{
    public partial class WeaponLightning
    {
        private class AttackState : WeaponStateBase
        {
            private WeaponLightning _weapon;

            public override WeaponStateType State => WeaponStateType.Attack;
            public override void OnInit(WeaponBase weapon) => _weapon = weapon as WeaponLightning;

            public override void OnEnter()
            {
                _weapon.ResetAttackCooldown(_weapon._weaponData.Cooldown);
                _weapon.Attack();
            }

            public override void OnUpdate(float elapseSeconds, float realElapseSeconds)
            {
                if (!_weapon._isAttacking)
                {
                    _weapon.TransitionTo(WeaponStateType.Check_InRange);
                }
            }

            public override void OnLeave()
            {
            }
        }
    }
}