Continentis/Assets/Mods/Basic/Characters/CombatBuffs/Knight/EstablishFormation.cs

using Continentis.MainGame;
using Continentis.MainGame.Character;
using SLSFramework.General;
using UnityEngine;

namespace Continentis.Mods.Basic.Buffs
{
    public class EstablishFormation : CharacterCombatBuffBase
    {
        public EstablishFormation(int stack, int count)
        {
            Initialize(BuffType.Positive, BuffDispelLevel.Basic);

            this.contentSubmodule = new ContentSubmodule(this)
                .AddParameterGetter("Stack", () => unitedStackSubmodule.stackAmount.ToString())
                .AddParameterGetter("Count", () => actionCountSubmodule.remainingCount.ToString());
            
            this.iconSubmodule = new IconSubmodule(this);
            
            this.unitedStackSubmodule = new UnitedStackSubmodule(this, stack);
            
            this.actionCountSubmodule = new CountSubmodule(this, count);
            
            this.eventSubmodule = new EventSubmodule(this);
            this.eventSubmodule.onGetAttacked.Add("EstablishFormation", new PrioritizedAction<AttackResult>(atkResult =>
            {
                attachedCharacter.Attack(atkResult.attacker, unitedStackSubmodule.stackAmount, null, false, true);
            }));
        }
        
        public override bool OnBuffApply(out CharacterCombatBuffBase existingBuff) 
        {
            MainGameManager.Instance.basePrefabs.GenerateInfoText(contentSubmodule.displayName, attachedCharacter.characterView);
            
            if (FindExistingSameBuff(out existingBuff))
            {
                existingBuff.unitedStackSubmodule.PickHigherStack(this.unitedStackSubmodule);
                existingBuff.actionCountSubmodule.PickHigherCount(this.actionCountSubmodule);
                return false;
            }
            
            return true;
        }
    }
}