siege-song/Data/Assets/Malbers Animations/Common/Scripts/Animal Controller/AI Brain/Tasks/MessageTask.cs

using MalbersAnimations.Utilities;
using UnityEngine;


namespace MalbersAnimations.Controller.AI
{
    [CreateAssetMenu(menuName = "Malbers Animations/Pluggable AI/Tasks/Message Task", fileName = "new Message Task")]
    public class MessageTask : MTask
    {
        public override string DisplayName => "General/Send Message";

        [Space, Tooltip("Apply the Task to the Animal(Self) or the Target(Target)")]
        public Affected affect = Affected.Self;
        [Tooltip("When you want to send the Message")]
        public ExecuteTask when = ExecuteTask.OnStart;
        public bool UseSendMessage = false;
        public bool SendToChildren = false;
        [Tooltip("The message will be send to the Root of the Hierarchy")]
        public bool SendToRoot = true;
        public MesssageItem[] messages;                                     //Store messages to send it when Enter the animation State


        public override void StartTask(MAnimalBrain brain, int index)
        {
            if (when == ExecuteTask.OnStart)
            {
                Execute_Task(brain);
                brain.TaskDone(index); //Set Done to this task
            }
        }

        public override void UpdateTask(MAnimalBrain brain, int index)
        {
            if (when == ExecuteTask.OnUpdate)
            {
                Execute_Task(brain);
            }
        }

        public override void ExitAIState(MAnimalBrain brain, int index)
        {
            if (when == ExecuteTask.OnExit)
            {
                Execute_Task(brain);
            }
            brain.TaskDone(index); //Set Done to this task
        }

        private void Execute_Task(MAnimalBrain brain)
        {
            if (affect == Affected.Self)
            {
                SendMessage(SendToRoot ?  brain.transform.root : brain.transform);
            }
            else
            {
                if (brain.Target != null) SendMessage(SendToRoot ? brain.Target.root : brain.Target);
            }
        }


        public virtual void SendMessage(Transform t)
        {
            IAnimatorListener[] listeners;

            if (SendToChildren)
                listeners = t.GetComponentsInChildren<IAnimatorListener>();
            else
                listeners = t.GetComponents<IAnimatorListener>();

            foreach (var msg in messages)
            {
                if (UseSendMessage)
                    msg.DeliverMessage(t, SendToChildren);
                else
                    foreach (var animListener in listeners)
                        msg.DeliverAnimListener(animListener);
            }
        }


        void Reset()
        { Description = "Send messages to the Root game Object of the Target or the Animal using the Brain"; }
    }
}