ActionTowerDefense/ActionTowerDefense/Assets/Scripts/Skills/Rock/RockRoller.cs

using Base.Common;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using cfg;
using TMPro;

public class RockRoller : Enemy
{
    public int hitTime = 1;
    public int curHitTime = 0;
    public RockAttackTrigger attackTrigger0;
    public AttackTrigger attackEnemyTrigger0;
    public List<TargetType> initialTargetTypes = new List<TargetType>();

    public GameObject stone;
    public string smallStone;
    public float baseRotateSpeed = 90f;
    public float rotateMul = 1f;
    public float rotateDirection = -1f;
    public float dieTime = 0.1f;
    public float deathTimer = 0f;
    public bool isDying;
    public bool isFalled = false;
    public Enemy owner;

    public bool isGather = false;
    public float gatherTime = 1f;
    public float gatherTimer = 0f;

    public bool isWeak = false;
    public float weakTime = 1f;
    public float weakTimer = 0f;
    public float addForce = 1f;

    public bool isBig;
    private string enemyStr;

    public override void Init()
    {
        base.Init();
    }

    private void Awake()
    {
        base.Awake();
        InitializeComponents();
        InitializeAttackMethods();
        InitializeTargetTypes();
        InitializeStoneReference();
    }

    private void InitializeComponents()
    {
        TryGetComponent<Rigidbody>(out rb);
    }

    private void InitializeAttackMethods()
    {
        if (attackTrigger0 != null && attackTrigger0.attackMethod.attackInfo.damage == 0 &&
            attackEnemyTrigger0 != null && attackEnemyTrigger0.attackMethod.attackInfo.damage == 0)
        {
            attackTrigger0.attackMethod = attackController.attackMethod_march[0];
            attackEnemyTrigger0.attackMethod = attackController.attackMethod_summon[0];
        }
    }

    private void InitializeTargetTypes()
    {
        if (attackController != null && attackController.targetTypes != null)
        {
            initialTargetTypes.Clear();
            initialTargetTypes.AddRange(attackController.targetTypes);
        }
    }

    private void InitializeStoneReference()
    {
        if (stone == null)
        {
            stone = gameObject;
        }
        enemyStr = $"Prefab/Enemy/{smallStone}";
    }

    protected override void OnEnable()
    {
        base.OnEnable();
        ResetState();
    }

    private void ResetState()
    {
        hitFeedbackSystem.canBeHitStun = false;
        hitFeedbackSystem.canFreeze = false;
        isDying = false;
        isFalled = false;
        deathTimer = 0f;
        gatherTimer = 0f;
        isGather = false;
        weakTimer = 0f;
        isWeak = false;
        rotateDirection = -1f;
        curHitTime = 0;

        if (attackEnemyTrigger0 != null)
            attackEnemyTrigger0.gameObject.SetActive(false);
        if (attackTrigger0 != null)
            attackTrigger0.gameObject.SetActive(true);

        ResetTargetTypes();
    }

    private void ResetTargetTypes()
    {
        if (attackController != null && attackController.targetTypes != null)
        {
            attackController.targetTypes.Clear();
            attackController.targetTypes.AddRange(initialTargetTypes);
        }
    }

    public void Update()
    {
        UpdateHitFeedback();
        if (owner == null) return;

        HandleOwnerSpecialState();
        Rotate();
    }

    public void Rotate()
    {
        if (stone == null) return;

        float currentSpeed = baseRotateSpeed;
        if (rb != null)
        {
            currentSpeed = rb.velocity.magnitude * rotateMul * baseRotateSpeed;
            currentSpeed = Mathf.Clamp(currentSpeed, 0, baseRotateSpeed * 5);
        }

        stone.transform.Rotate(Vector3.forward, rotateDirection * currentSpeed * Time.deltaTime);
    }


    private void UpdateHitFeedback()
    {
        hitFeedbackSystem.canBeHitStun = false;
        hitFeedbackSystem.canFreeze = false;
    }

    private void HandleOwnerSpecialState()
    {
        if (owner.state == CharacterState.SpecialStatus_BlowUp)
        {
            if (attackTrigger0 != null)
                attackTrigger0.gameObject.SetActive(false);

            UpdateTargetTypesForAttack();
        }
    }

    private void UpdateTargetTypesForAttack()
    {
        if (attackController != null)
        {
            attackController.targetTypes.Clear();
            attackController.targetTypes.Add(TargetType.Enemy);
            attackController.targetTypes.Add(TargetType.EnemyTower);
        }
    }

    public override void FixedUpdate()
    {
        OnState();
        HandleDyingState();
    }

    private void HandleDyingState()
    {
        if (owner != null && owner.state == CharacterState.SpecialStatus_BlowUp &&
            owner.attributeStatus.hitState == 1 && !isDying)
        {
            StartDying();
        }

        if (isDying)
        {
            deathTimer += Time.deltaTime;

            if (deathTimer >= dieTime)
            {
                HandleDeath();
            }
        }
    }

    private void StartDying()
    {
        if (attackEnemyTrigger0 != null)
            attackEnemyTrigger0.gameObject.SetActive(true);

        isDying = true;
        isFalled = true;
        deathTimer = 0f;
    }

    private void HandleDeath()
    {
        if (isBig && isFalled)
        {
            SpawnSmallStones();
        }

        if (owner != null && curHitTime >= hitTime)
        {
            owner.gameObject.SetActive(false);
        }

        isDying = false;
    }

    private void SpawnSmallStones()
    {
        attackEnemyTrigger0.gameObject.SetActive(false);
        ResetTargetTypes();

        SingleEnemyConfig cfgEnemy = GameManager.instance.allCfgData.CfgEnemy.Get("小滚石怪");

        // 生成第一个小石头
        GameObject enemyObj = Util.Instantiate(enemyStr, transform.position - new Vector3(-1, 0, 0), active: true);
        ConfigureSmallStone(enemyObj, cfgEnemy);

        // 生成第二个小石头
        GameObject enemyObj1 = Util.Instantiate(enemyStr, transform.position - new Vector3(1, 0, 0), active: true);
        ConfigureSmallStone(enemyObj1, cfgEnemy);

        isFalled = false;
    }

    private void ConfigureSmallStone(GameObject stoneObj, SingleEnemyConfig cfgEnemy)
    {
        RockRoller rock = stoneObj.GetComponent<RockRoller>();
        if (rock != null)
        {
            rock.attackTrigger0.attackMethod.attackInfo.damage = cfgEnemy.AttackMarch[0];
            rock.attackEnemyTrigger0.attackMethod.attackInfo.damage = cfgEnemy.AttackSummon;
            rock.owner.moveSpeed = Random.Range(cfgEnemy.MinMoveSpeed, cfgEnemy.MaxMoveSpeed);
        }
    }

    public override void ChangeState(CharacterState newState)
    {
        if (state == newState || newState == CharacterState.FramePause)
        {
            return;
        }

        ExitCurrentState();
        EnterNewState(newState);
    }

    private void ExitCurrentState()
    {
        switch (state)
        {
            case CharacterState.Idle:
                rb.velocity = Vector3.zero;
                break;
            case CharacterState.Die:
                isDie = false;
                break;
        }
    }

    private void EnterNewState(CharacterState newState)
    {
        CharacterState oldState = state;
        state = newState;

        switch (newState)
        {
            case CharacterState.Idle:
                EnterIdleState();
                break;
            case CharacterState.Rush:
                EnterRushState();
                break;
            case CharacterState.Rise:
                EnterRiseState();
                break;
            case CharacterState.Fall:
                EnterFallState();
                break;
            case CharacterState.Die:
                EnterDieState();
                break;
            case CharacterState.SpecialStatus_BlowUp:
                EnterBlowUpState();
                break;
        }
    }

    private void EnterIdleState()
    {
        rotateDirection = -1f;
        rb.velocity = Vector3.zero;
        if (attackEnemyTrigger0 != null) attackEnemyTrigger0.gameObject.SetActive(false);
        if (attackTrigger0 != null) attackTrigger0.gameObject.SetActive(false);
    }

    private void EnterRushState()
    {
        rb.velocity = Vector3.zero;
        if (attackEnemyTrigger0 != null) attackEnemyTrigger0.gameObject.SetActive(false);
        if (attackTrigger0 != null) attackTrigger0.gameObject.SetActive(true);
    }

    private void EnterRiseState()
    {
        isGather = true;
        gatherTimer = 0f;
    }

    private void EnterFallState()
    {
        rotateDirection = 1f;
        weakTimer = 0f;
    }

    private void EnterDieState()
    {
        isDie = true;
        dieKeepTime = totalDieKeepTime;
        if (GameManager.instance.gameType != GameType.GameEnd)
        {
            DropSouls();
        }
    }

    private void EnterBlowUpState()
    {
        rotateDirection = 1f;
        curHitTime += 1;
        isDying = false;
        isFalled = true;
        deathTimer = 0f;
        if (owner != null) owner.attributeStatus.hitState = 0;
    }

    public override void OnState()
    {
        switch (state)
        {
            case CharacterState.Idle:
                ChangeState(CharacterState.Rise);
                break;
            case CharacterState.Rise:
                HandleRiseState();
                break;
            case CharacterState.Rush:
                HandleRushState();
                break;
            case CharacterState.Fall:
                HandleFallState();
                break;
            case CharacterState.SpecialStatus_BlowUp:
                attributeStatus.SpecialStateEffect(SpecialState.BlownUp);
                break;
            case CharacterState.Die:
                dieKeepTime -= Time.deltaTime;
                if (dieKeepTime <= 0)
                {
                    if (killer != null && killer.GetComponent<Demonic>())
                    {
                        SoldierType st = killer.GetComponent<Demonic>().soldierType;
                        SoldierEXP.expInstance.AddEXP(st, (int)(exp * LevelSelect.EXPRatio + 0.5f));
                    }
                    if (dieEffect)
                    {
                        PoolManager.Instantiate(dieEffect, transform.position);
                    }
                    GameManager gameManager = GameManager.instance;
                    if (gameManager.gameType == GameType.GameEnd)
                    {
                        GameObject fx = PoolManager.Instantiate(gameManager.dropGoldFX);
                        fx.transform.position = transform.position;
                        GameObject injuryNum = PoolManager.Instantiate(gameManager.dropGoldText);
                        injuryNum.transform.position = new Vector3(
                            transform.position.x + injuryNumPos_march.x + Random.Range(-injuryNumRandom_march.x / 2f, injuryNumRandom_march.x / 2f),
                            transform.position.y + injuryNumPos_march.y + Random.Range(-injuryNumRandom_march.y / 2f, injuryNumRandom_march.y / 2f),
                            transform.position.z);
                        TextMeshProUGUI text = injuryNum.GetComponentInChildren<TextMeshProUGUI>();
                        text.text = $"+{gameManager.enemyGoldDrop}";
                    }
                    gameObject.SetActive(false);
                    break;
                }
                break;
                break;
        }
    }

    private void HandleRiseState()
    {
        if (isGather && gatherTimer < gatherTime)
        {
            gatherTimer += Time.deltaTime;
        }
        else
        {
            Debug.Log("gatherFinsh");
            isGather = false;
            ChangeState(CharacterState.Rush);
        }
    }

    private void HandleRushState()
    {
        Vector3 velocity = rb.velocity;
        if (Mathf.Abs(velocity.x) < moveSpeed)
        {
            velocity.x += addForce;
        }
        rb.velocity = velocity;
    }

    private void HandleFallState()
    {
        if (!foot.TrigGround)
        {
            isWeak = true;
        }

        if (isWeak && weakTimer < weakTime && foot.TrigGround)
        {
            rb.velocity = Vector3.zero;
            weakTimer += Time.deltaTime;
        }
        else if (weakTimer >= weakTime)
        {
            isWeak = false;
            ChangeState(CharacterState.Idle);
        }
    }

    public void OnDisable()
    {
        EnemyCreater.instance.OnEnemyRecycle(this);
        ResetState();
    }
}