fix WaitCommand

Author: Dimonik333

Assets/CatCode/Commands/WaitCommand.cs

@@ -37,18 +37,14 @@ private async Awaitable WaitAsync()
             { }
             finally
             {
-                _cts.Dispose();
+                _cts?.Dispose();
                 _cts = null;
             }
         }
 
         private void Cancel()
-        {
-            if (_cts == null)
-                return;
-            _cts.Cancel();
-            _cts.Dispose();
-            _cts = null;
+        {         
+            _cts?.Cancel();
         }
     }
 }

Assets/CommandTester.cs

@@ -0,0 +1,24 @@
+using CatCode.Commands;
+using UnityEngine;
+
+public class CommandTester : MonoBehaviour
+{
+    [SerializeField] private WaitCommand _command;
+
+    private void Awake()
+    {
+        _command = new WaitCommand(5)
+            .AddOnStarted(() => Debug.Log("Started"))
+            .AddOnFinished(() => Debug.Log("Finished"))
+            .AddOnStopped(() => Debug.Log("Stopped"));
+    }
+    private void OnEnable()
+    {
+        _command.Execute();
+    }
+
+    private void OnDisable()
+    {
+        _command.Stop();
+    }
+}