LaboratoryProtection/Assets/PMaker/Scripts/Await/Extension/BaseBehaviourExtension.cs

using Cysharp.Threading.Tasks;

using System.Threading;

namespace PMaker.Await
{
    public static partial class BaseBehaviourExtension
    {
        public static async UniTask WaitUI(this BaseBehaviour behaviour, string name, CancellationToken cancellationToken = default)
        {
            var ui = behaviour.transform.Find(name).GetComponent<AwaitBehaviour>();
            await ui.WaitAsync(cancellationToken);
        }

        public static T GetUI<T>(this BaseBehaviour behaviour, string name)
        {
            return behaviour.transform.Find(name).GetComponent<T>();
        }
    }
}