1

Assets/BITKit/Core/Utility/ValidHandle.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using Cysharp.Threading.Tasks;
 
 namespace BITKit
 {
@@ -22,6 +23,41 @@ namespace BITKit
     [CustomType(typeof(IValidHandle))]
     public sealed class ValidHandle: IValidHandle
     {
+        public class MyHandle:IDisposable
+        {
+            private readonly ValidHandle _validHandle;
+            private bool _isDisable = false;
+            public MyHandle(ValidHandle validHandle,bool isDisable = false)
+            {
+                _validHandle = validHandle;
+
+                _isDisable = isDisable;
+                if (isDisable)
+                {
+                    _validHandle.AddDisableElements(this);
+                }
+                else
+                {
+                    _validHandle.AddElement(this);
+                }
+          
+            }
+            public void Dispose()
+            {
+                if (_isDisable)
+                {
+                    _validHandle.RemoveDisableElements(this);
+                }
+                else
+                {
+                    _validHandle.RemoveElement(this);
+                }
+            }
+        }
+
+        public MyHandle GetHandle() => new MyHandle(this);
+        public MyHandle GetDisableHandle()=> new MyHandle(this,true);
+        
         public override string ToString()
         {
             return $"Allow:{enableHandle}\nElements:{string.Join("\n",objs)}\nDisableElements:{string.Join("\n",disableObjs)}";
@@ -51,6 +87,7 @@ namespace BITKit
         public readonly List<object> disableObjs = new List<object>();
         private bool tempEnable;
         private Action<bool> EventOnEnableChanged;
+        private readonly Queue<UniTaskCompletionSource> _completionSources = new();
         
         
         public void AddElement(object obj)
@@ -73,7 +110,14 @@ namespace BITKit
             {
                 enableHandle = tempEnable;
                 if (EventOnEnableChanged is not null)
+                {
                     EventOnEnableChanged.Invoke(enableHandle);
+                }
+                if (tempEnable) return;
+                if (_completionSources.TryDequeue(out var cs))
+                {
+                    cs.TrySetResult();
+                }
             }
         }
         public void RemoveElement(object obj)
@@ -164,7 +208,16 @@ namespace BITKit
                 EventOnEnableChanged -= action;
             }
         }
-
+        public UniTask.Awaiter GetAwaiter()
+        {
+            if (Allow is false)
+            {
+                return UniTask.CompletedTask.GetAwaiter();
+            }
+            var cs = new UniTaskCompletionSource();
+            _completionSources.Enqueue(cs);
+            return cs.Task.GetAwaiter();
+        }
         public void Clear()
         {
             objs.Clear();