using System.Collections; // For Coroutines.

public class MageEnemy : EnemyBase
{
    public float attackRange = 5f;          // Optimal distance for ranged attack.
    public GameObject projectilePrefab;     // Projectile to spawn.
    public Transform projectileSpawnPoint;  // Where projectiles emerge.
    public float attackDamage = 15f;        // Damage per projectile.

    private bool isAttacking = false;       // True when in attack animation.

    /// <summary>
    /// Overrides base Update for ranged attack logic.
    /// </summary>
    protected override void Update()
    {
        if (currentHealth <= 0) { Die(); return; } // Handles death early.

        float distanceToPlayer = Vector3.Distance(transform.position, player.position);

        // If in range and not attacking, initiate attack.
        if (distanceToPlayer <= attackRange && !isAttacking)
        {
            if (Time.time >= nextAttackTime)
            {
                Attack(); // Calls the overridden Attack method.
                nextAttackTime = Time.time + attackRate;
            }
            StopMoving(); // Stops movement while attacking.
        }
        else if (!isAttacking)
        {
            base.Update(); // Continues base movement (chasing).
        }
    }

    /// <summary>
    /// Implements specific ranged attack logic.
    /// </summary>
    protected override void Attack()
    {
        Debug.Log("Mage attacks!");
        // Spawns projectile, setting its direction and damage.
        GameObject projectile = Instantiate(projectilePrefab, projectileSpawnPoint.position, Quaternion.LookRotation(player.position - transform.position));
        projectile.GetComponent<ProjectileMoverEnemy>().damage = attackDamage;

        // Triggers attack animation and manages cooldown via Coroutine.
        if (animator != null)
        {
            animator.SetBool("IsAttacking", true);
            isAttacking = true;
            StartCoroutine(ResetAttackAnimation(animator.GetCurrentAnimatorStateInfo(0).length));
        }
    }

    /// <summary>
    /// Coroutine to reset attack animation state.
    /// </summary>
    private IEnumerator ResetAttackAnimation(float waitTime)
    {
        yield return new WaitForSeconds(waitTime); // Waits for animation to complete.
        if (animator != null) animator.SetBool("IsAttacking", false);
        isAttacking = false; // Resets attacking flag.
        // Resume movement if player is out of range.
        if (Vector3.Distance(transform.position, player.position) > attackRange) animator.SetBool("IsRunning", true);
    }

    /// <summary>
    /// Stops mage movement and sets idle animation.
    /// </summary>
    private void StopMoving()
    {
        agent.isStopped = true;
        animator.SetBool("IsRunning", false);
        animator.SetBool("IsIdle", true);
    }

    /// <summary>
    /// Overrides base Die for mage-specific death.
    /// </summary>
    protected override void Die()
    {
        if (animator != null) animator.SetBool("IsDead", true);
        enabled = false; agent.isStopped = true; // Disable components.
        foreach (Renderer renderer in GetComponentsInChildren<Renderer>()) renderer.enabled = false;
        StartCoroutine(RemoveAfterAnimation(animator.GetCurrentAnimatorStateInfo(0).length));
    }
}