From 6fda78301c4c8d5878538239d646f6008c4f0137 Mon Sep 17 00:00:00 2001
From: Craig Roberts <craig.roberts@crdevel.com>
Date: Wed, 29 Aug 2018 18:56:26 -0400
Subject: [PATCH] async/await

---
 .../Asterisk.NET/Manager/ManagerConnection.cs | 56 +++++++++++++++++--
 .../ResponseHandlers/TaskResponseHandler.cs   | 30 ++++++++++
 2 files changed, 81 insertions(+), 5 deletions(-)
 create mode 100644 Asterisk.2013/Asterisk.NET/Manager/ResponseHandlers/TaskResponseHandler.cs

diff --git a/Asterisk.2013/Asterisk.NET/Manager/ManagerConnection.cs b/Asterisk.2013/Asterisk.NET/Manager/ManagerConnection.cs
index d1fe667..3e7bdba 100644
--- a/Asterisk.2013/Asterisk.NET/Manager/ManagerConnection.cs
+++ b/Asterisk.2013/Asterisk.NET/Manager/ManagerConnection.cs
@@ -9,6 +9,7 @@ using System.Text;
 using System.Collections.Generic;
 using System.Reflection;
 using AsterNET.IO;
+using System.Threading.Tasks;
 
 namespace AsterNET.Manager
 {
@@ -1314,13 +1315,13 @@ namespace AsterNET.Manager
         /// <param name="action">action to send</param>
         /// <param name="timeout">timeout in milliseconds</param>
         /// <returns></returns>
-        public Response.ManagerResponse SendAction(ManagerAction action, int timeOut)
+        public Response.ManagerResponse SendAction(ManagerAction action, int timeout)
         {
             AutoResetEvent autoEvent = new AutoResetEvent(false);
             ResponseHandler handler = new ResponseHandler(action, autoEvent);
 
             int hash = SendAction(action, handler);
-            bool result = autoEvent.WaitOne(timeOut <= 0 ? -1 : timeOut, true);
+            bool result = autoEvent.WaitOne(timeout <= 0 ? -1 : timeout, true);
 
             RemoveResponseHandler(handler);
 
@@ -1331,7 +1332,13 @@ namespace AsterNET.Manager
         #endregion
 
         #region SendAction(action, responseHandler)
-        public int SendAction(ManagerAction action, ResponseHandler responseHandler)
+        /// <summary>
+        /// Send action ans with timeout (milliseconds)
+        /// </summary>
+        /// <param name="action">action to send</param>
+        /// <param name="responseHandler">Response Handler</param>
+        /// <returns></returns>
+        public int SendAction(ManagerAction action, IResponseHandler responseHandler)
         {
             if (action == null)
                 throw new ArgumentException("Unable to send action: action is null.");
@@ -1354,6 +1361,42 @@ namespace AsterNET.Manager
         }
         #endregion
 
+
+
+        #region SendActionAsync(action)
+        /// <summary>
+        /// Asynchronously send Action async with default timeout.
+        /// </summary>
+        /// <param name="action">action to send</param>
+        public Task<ManagerResponse> SendActionAsync(ManagerAction action)
+        {
+          return SendActionAsync(action, null);
+        }
+        #endregion
+
+        #region SendActionAsync(action, timeout)
+        /// <summary>
+        /// Asynchronously send Action async.
+        /// </summary>
+        /// <param name="action">action to send</param>
+        /// <param name="cancellationToken">cancellation Token</param>
+        public Task<ManagerResponse> SendActionAsync(ManagerAction action, CancellationTokenSource cancellationToken)
+        {
+          var handler = new TaskResponseHandler(action);
+          var source = handler.TaskCompletionSource;
+
+          SendAction(action, handler);
+
+          if (cancellationToken != null)
+            cancellationToken.Token.Register(() => { source.TrySetCanceled(); });
+
+          return source.Task.ContinueWith(x =>
+          {
+            RemoveResponseHandler(handler);
+            return x.Result;
+          });
+        }
+        #endregion
         #region SendEventGeneratingAction(action)
         public ResponseEvents SendEventGeneratingAction(ManagerActionEvent action)
         {
@@ -1420,7 +1463,11 @@ namespace AsterNET.Manager
                 responseEventHandlers[handler.Hash] = handler;
         }
 
-        internal void RemoveResponseHandler(IResponseHandler handler)
+        /// <summary>
+        /// Delete an instance of a class <see cref="IResponseHandler"/> from handlers list.
+        /// </summary>
+        /// <param name="handler">Class instance <see cref="IResponseHandler"/>.</param>
+		public void RemoveResponseHandler(IResponseHandler handler)
         {
             int hash = handler.Hash;
             if (hash != 0)
@@ -1437,7 +1484,6 @@ namespace AsterNET.Manager
                     if (responseEventHandlers.ContainsKey(hash))
                         responseEventHandlers.Remove(hash);
         }
-
         private IResponseHandler GetRemoveResponseHandler(int hash)
         {
             IResponseHandler handler = null;
diff --git a/Asterisk.2013/Asterisk.NET/Manager/ResponseHandlers/TaskResponseHandler.cs b/Asterisk.2013/Asterisk.NET/Manager/ResponseHandlers/TaskResponseHandler.cs
new file mode 100644
index 0000000..d96ccbf
--- /dev/null
+++ b/Asterisk.2013/Asterisk.NET/Manager/ResponseHandlers/TaskResponseHandler.cs
@@ -0,0 +1,30 @@
+using System.Threading.Tasks;
+using AsterNET.Manager.Action;
+
+namespace AsterNET.Manager.Response
+{
+    /// <summary>
+    ///     
+    /// </summary>
+    public class TaskResponseHandler : IResponseHandler
+    {
+        public TaskResponseHandler(ManagerAction action)
+        {
+            TaskCompletionSource = new TaskCompletionSource<ManagerResponse>();
+            Action = action;
+        }
+
+        public TaskCompletionSource<ManagerResponse> TaskCompletionSource { get; }
+
+        public ManagerAction Action { get; }
+
+        public int Hash { get; set; }
+
+        public void HandleResponse(ManagerResponse response)
+        {
+            TaskCompletionSource.TrySetResult(response);
+        }
+
+        public void Free() { }
+    }
+}