Add NotifyCompletion handlers

Assets/BetterExtensions/Runtime/Extensions/AssetBundleRequestAwaiterExtensions.cs

@@ -0,0 +1,19 @@
+using System;
+using Better.Extensions.Runtime.Helpers;
+using UnityEngine;
+
+namespace Better.Extensions.Runtime
+{
+    public static class AssetBundleRequestAwaiterExtensions
+    {
+        public static AssetBundleRequestAwaiter GetAwaiter(this AssetBundleRequest self)
+        {
+            if (self == null)
+            {
+                throw new ArgumentNullException(nameof(self));
+            }
+
+            return new AssetBundleRequestAwaiter(self);
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Extensions/AsyncOperationAwaiterExtensions.cs

@@ -0,0 +1,19 @@
+using System;
+using Better.Extensions.Runtime.Helpers;
+using UnityEngine;
+
+namespace Better.Extensions.Runtime
+{
+    public static class AsyncOperationAwaiterExtensions
+    {
+        public static AsyncOperationAwaiter GetAwaiter(this AsyncOperation self)
+        {
+            if (self == null)
+            {
+                throw new ArgumentNullException(nameof(self));
+            }
+
+            return new AsyncOperationAwaiter(self);
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Extensions/ResourceRequestAwaiterExtensions.cs

@@ -0,0 +1,19 @@
+using System;
+using Better.Extensions.Runtime.Helpers;
+using UnityEngine;
+
+namespace Better.Extensions.Runtime
+{
+    public static class ResourceRequestAwaiterExtensions
+    {
+        public static ResourceRequestAwaiter GetAwaiter(this ResourceRequest self)
+        {
+            if (self == null)
+            {
+                throw new ArgumentNullException(nameof(self));
+            }
+
+            return new ResourceRequestAwaiter(self);
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Extensions/UnityWebRequestAsyncOperationExtensions.cs

@@ -0,0 +1,19 @@
+using System;
+using Better.Extensions.Runtime.Helpers;
+using UnityEngine.Networking;
+
+namespace Better.Extensions.Runtime
+{
+    public static class UnityWebRequestAsyncOperationExtensions
+    {
+        public static UnityWebRequestAwaiter GetAwaiter(this UnityWebRequestAsyncOperation self)
+        {
+            if (self == null)
+            {
+                throw new ArgumentNullException(nameof(self));
+            }
+
+            return new UnityWebRequestAwaiter(self);
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Helpers/NotifyCompletions/AssetBundleRequestAwaiter.cs

@@ -0,0 +1,30 @@
+using System;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
+using UnityEngine;
+using Object = UnityEngine.Object;
+
+namespace Better.Extensions.Runtime.Helpers
+{
+    [DebuggerNonUserCode]
+    public readonly struct AssetBundleRequestAwaiter : INotifyCompletion
+    {
+        private readonly AssetBundleRequest _asyncOperation;
+        public bool IsCompleted => _asyncOperation.isDone;
+
+        public AssetBundleRequestAwaiter(AssetBundleRequest asyncOperation)
+        {
+            _asyncOperation = asyncOperation;
+        }
+
+        public void OnCompleted(Action continuation)
+        {
+            _asyncOperation.completed += _ => continuation();
+        }
+
+        public Object GetResult()
+        {
+            return _asyncOperation.asset;
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Helpers/NotifyCompletions/AsyncOperationAwaiter.cs

@@ -0,0 +1,29 @@
+using System;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
+using UnityEngine;
+
+namespace Better.Extensions.Runtime.Helpers
+{
+    [DebuggerNonUserCode]
+    public readonly struct AsyncOperationAwaiter : INotifyCompletion
+    {
+        private readonly AsyncOperation _asyncOperation;
+        public bool IsCompleted => _asyncOperation.isDone;
+
+        public AsyncOperationAwaiter(AsyncOperation asyncOperation)
+        {
+            _asyncOperation = asyncOperation;
+        }
+
+        public void OnCompleted(Action continuation)
+        {
+            _asyncOperation.completed += _ => continuation();
+        }
+
+        public AsyncOperation GetResult()
+        {
+            return _asyncOperation;
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Helpers/NotifyCompletions/ResourceRequestAwaiter.cs

@@ -0,0 +1,29 @@
+using System;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
+using UnityEngine;
+
+namespace Better.Extensions.Runtime.Helpers
+{
+    [DebuggerNonUserCode]
+    public readonly struct ResourceRequestAwaiter : INotifyCompletion
+    {
+        private readonly ResourceRequest _asyncOperation;
+        public bool IsCompleted => _asyncOperation.isDone;
+
+        public ResourceRequestAwaiter(ResourceRequest asyncOperation)
+        {
+            _asyncOperation = asyncOperation;
+        }
+
+        public void OnCompleted(Action continuation)
+        {
+            _asyncOperation.completed += _ => continuation();
+        }
+
+        public ResourceRequest GetResult()
+        {
+            return _asyncOperation;
+        }
+    }
+}

Assets/BetterExtensions/Runtime/Helpers/NotifyCompletions/UnityWebRequestAwaiter.cs

@@ -0,0 +1,30 @@
+using System;
+using System.Diagnostics;
+using System.Runtime.CompilerServices;
+using UnityEngine.Networking;
+
+namespace Better.Extensions.Runtime.Helpers
+{
+    [DebuggerNonUserCode]
+    public readonly struct UnityWebRequestAwaiter : INotifyCompletion
+    {
+        private readonly UnityWebRequestAsyncOperation _asyncOperation;
+
+        public bool IsCompleted => _asyncOperation.isDone;
+
+        public UnityWebRequestAwaiter(UnityWebRequestAsyncOperation asyncOperation)
+        {
+            _asyncOperation = asyncOperation;
+        }
+
+        public void OnCompleted(Action continuation)
+        {
+            _asyncOperation.completed += _ => continuation();
+        }
+
+        public UnityWebRequest GetResult()
+        {
+            return _asyncOperation.webRequest;
+        }
+    }
+}