using AIMachineTools;
using System.Collections;
using UnityEngine;

public class ChaseState : State
{
    private static ChaseState _instance; //only declared once

    private ChaseState() //set instance to itself
    {
        if (_instance != null)
        {
            return;
        }

        _instance = this;
    }

    public static ChaseState instance //creates the instance of the first state if it does not exist
    {
        get
        {
            if (_instance == null)
            {
                new ChaseState();
            }

            return _instance;
        }
    }

    public override IEnumerator InState(AIMachine owner)
    {
        owner.RemoveMarker();
        float timeBeforeAttack = 2.0f;
        float timeSinceEnteredState = 0.0f;

        while (owner.botMachine.currentState == ChaseState.instance)
        {
            if (!owner.disabled)
            {
                if (timeSinceEnteredState < timeBeforeAttack)
                    timeSinceEnteredState += Time.deltaTime;

                if (owner.enemyObject == null)
                {
                    owner.canSeeEnemy = false;
                    owner.enemyHealth = null;
                    if (owner.botMachine.baseState == SearchState.instance)
                        owner.botMachine.ChangeState(SearchState.instance);
                    else if (owner.botMachine.baseState == CaptureAreaState.instance)
                        owner.botMachine.ChangeState(CaptureAreaState.instance);
                    else if (owner.botMachine.baseState == PartnerState.instance)
                        owner.botMachine.ChangeState(PartnerState.instance);
                    else if (owner.botMachine.baseState == MoveToState.instance)
                        owner.botMachine.ChangeState(MoveToState.instance);
                    yield break;
                }

                if (owner.enemyObject != null)
                {
                    if (Vector3.Angle(owner.transform.forward, owner.enemyObject.transform.position - owner.transform.position) <= 15)
                    {
                        MoveTurret(owner, owner.enemyObject.transform.position);

                        owner.thisAgent.SetDestination(owner.enemyObject.transform.position);

                        if (Vector3.Distance(owner.transform.position, owner.enemyObject.transform.position) <= owner.attackDistance && owner.canSeeEnemy && timeSinceEnteredState >= timeBeforeAttack)
                        {
                            owner.botMachine.ChangeState(AttackState.instance);
                            yield break;
                        }
                    }
                    else
                        RotateBody(owner, owner.enemyObject.transform.position);

                    if (Vector3.Distance(owner.transform.position, owner.enemyObject.transform.position) <= owner.attackDistance && owner.canSeeEnemy && timeSinceEnteredState >= timeBeforeAttack)
                    {
                        owner.botMachine.ChangeState(AttackState.instance);
                        yield break;
                    }
                }
            }
            else
            {
                owner.thisAgent.SetDestination(owner.transform.position);
            }

            yield return null;
        }
    }

    private void RotateBody(AIMachine _owner, Vector3 navPoint) //called to have the bot rotate to face the waypoint before navigating to it
    {
        //declare variables that will be used to determine how to move the bot
        Quaternion targetRot = Quaternion.LookRotation(navPoint - _owner.transform.position);

        //rotate the bot over time to face the enemy
        _owner.transform.rotation = Quaternion.RotateTowards(_owner.transform.rotation, targetRot, Time.deltaTime * _owner.bodyRotateSpeed);

        //set the bots y euler angle so that it only moves on that transform
        _owner.transform.localEulerAngles = new Vector3(0, _owner.transform.localEulerAngles.y, 0);
    }

    private void MoveTurret(AIMachine _owner, Vector3 hitPos) //called to move the bots turret so that it faces whatever Vector3 is sent
    {
        //declare variables that will be used to determine how to move the turret
        Quaternion hitRot = Quaternion.LookRotation(hitPos - _owner.transform.position);
        Quaternion turretRot = Quaternion.LookRotation(hitPos - _owner.turretPos.transform.position);

        if (Vector3.Angle(_owner.turretPos.transform.forward, hitPos - _owner.turretPos.transform.position) > 5) //rotate the turret until it is facing within 5 degrees of the target
        {
            //rotate the turret over time to the target
            _owner.turretPos.transform.rotation = Quaternion.RotateTowards(_owner.turretPos.transform.rotation, turretRot, Time.deltaTime * _owner.turretRotateSpeed);

            //set the turrets y euler angle so that it only moves on that transform
            _owner.turretPos.transform.localEulerAngles = new Vector3(0, _owner.turretPos.transform.localEulerAngles.y, 0);
        }

        //rotate the barrel over time to the target
        _owner.barrelPos.transform.rotation = Quaternion.RotateTowards(_owner.barrelPos.transform.rotation, turretRot, Time.deltaTime * _owner.barrelMoveSpeed);

        //set the barrels y euler angle so that it only moves on that transform
        _owner.barrelPos.transform.localEulerAngles = new Vector3(0, _owner.barrelPos.transform.localEulerAngles.y, 0);
    }
}