Cielonos/Assets/Scripts/MainGame/AttackArea/Submodules/TimeSubmodule.cs

using System;
using Lean.Pool;
using UnityEngine;

namespace Cielonos.MainGame
{
    public partial class TimeSubmodule : AttackAreaSubmoduleBase
    {
        public float lifeTime;
        public float delayTime;
        public float enablingTime;
        public float remainingEnableTime;
        public float remainingLifeTime;
        public Action enableAction;
        public Action timeOutAction;

        public TimeSubmodule(AttackAreaBase attackArea, float lifeTime, Action timeOutAction = null) : base(attackArea)
        {
            this.isEnabling = true;
            this.lifeTime = lifeTime;
            this.remainingLifeTime = lifeTime;
            this.enablingTime = 0;
            
            if (attackArea is NormalArea)
            {
                this.remainingEnableTime = lifeTime * 0.25f;
            }
            else if (attackArea is Projectile)
            {
                this.remainingEnableTime = lifeTime;
            }
            else
            {
                this.remainingEnableTime = lifeTime;
            }
            
            this.timeOutAction = timeOutAction ?? (() =>
            {
                if (attackArea is Projectile projectile)
                {
                    projectile.Explode(projectile.transform.position);
                }
                else
                {
                    LeanPool.Despawn(attackArea.topParent);
                }
            });
        }

        public TimeSubmodule(AttackAreaBase attackArea, float lifeTime, float delayTime, float enableTime, Action enableAction, Action timeOutAction) : base(attackArea)
        {
            this.isEnabling = true;
            this.lifeTime = lifeTime;
            this.delayTime = delayTime;
            this.attackArea.isEnabling = delayTime <= 0;
            this.timeOutAction = timeOutAction ?? (() =>
            {
                if (attackArea is Projectile projectile)
                {
                    projectile.Explode(projectile.transform.position);
                }
                else
                {
                    LeanPool.Despawn(attackArea.topParent);
                }
            });
            if (this.attackArea.isEnabling)
            {
                enableAction?.Invoke();
            }
            this.remainingLifeTime = lifeTime;
            this.enablingTime = 0;
            this.remainingEnableTime = enableTime;
            this.enableAction = enableAction;
        }
    }
    
    public partial class TimeSubmodule
    {
        public void Update()
        {
            if (!isEnabling) return;

            if (delayTime > 0)
            {
                delayTime -= attackArea.creator.selfTimeSm.DeltaTime;
                return;
            }

            if (!attackArea.isEnabling)
            {
                attackArea.isEnabling = true;
                enableAction?.Invoke();
            }

            enablingTime += attackArea.creator.selfTimeSm.DeltaTime;
            remainingLifeTime -= attackArea.creator.selfTimeSm.DeltaTime;
            remainingEnableTime -= attackArea.creator.selfTimeSm.DeltaTime;

            if (remainingLifeTime <= 0)
            {
                this.isEnabling = false;
                timeOutAction?.Invoke();
            }

            if (remainingEnableTime <= 0)
            {
                attackArea.isEnabling = false;
            }
        }
        
        public void ModifyLifeTime(float modifyValue)
        {
            lifeTime += modifyValue;
            remainingLifeTime += modifyValue;
        }
    }
}