ProjectZ/Assets/Scripts/Characters/Enemy/States/StunnedState.cs

using System.Collections;
using Koptilnya;
using Koptilnya.StateMachine;
using Mirror;
using UnityEditor.Timeline.Actions;
using UnityEngine;

namespace Characters.Enemy.States
{
    public class StunnedState : State<EnemyState>
    {
        private readonly Enemy _enemy;
        private Task _task;
        
        private static readonly int DamageAnimHash = Animator.StringToHash("damage");

        public StunnedState(Enemy enemy) : base(EnemyState.Stunned)
        {
            _enemy = enemy;
        }

        public override void Enter()
        {
            if (_enemy.isServer)
            {
                _enemy.canAttack = false;
                _enemy.target = null;

                _enemy.SetSpeedMul(0f);
                _enemy.GetComponent<NetworkAnimator>().SetTrigger(DamageAnimHash);
                
                _task = new Task(GetStunned());
            }
            else
            {
                _enemy.hitVFX.Stop();
                _enemy.hitVFX.Play();
            }
        }

        [Server]
        public override void ReEnter()
        {
            _task.Stop();
            _task.Start();
        }

        public override void Exit()
        {
            if (!_enemy.isServer) return;
            
            _task.Stop();
            _enemy.SetSpeedMul(1f);
            
            _enemy.canAttack = true;
        }

        IEnumerator GetStunned()
        {
            yield return new WaitForSeconds(6);
            
            _enemy.state = EnemyState.Patrol;
        }
    }
}