From f3a74b2aeaee6089c23fb8f43f9c8e565904f1e4 Mon Sep 17 00:00:00 2001
From: Blake LaFleur <blake.k.lafleur@gmail.com>
Date: Fri, 7 Jun 2024 11:05:19 -0500
Subject: [PATCH] Add strongly-typed extension method for SignalWithStart

---
 .../Client/ITemporalClientExtensions.cs       | 45 ++++++++++++++++++-
 1 file changed, 44 insertions(+), 1 deletion(-)

diff --git a/src/Temporalio/Client/ITemporalClientExtensions.cs b/src/Temporalio/Client/ITemporalClientExtensions.cs
index 2beb2d26..b1c6a91e 100644
--- a/src/Temporalio/Client/ITemporalClientExtensions.cs
+++ b/src/Temporalio/Client/ITemporalClientExtensions.cs
@@ -43,6 +43,49 @@ public static async Task<TResult> ExecuteWorkflowAsync<TWorkflow, TResult>(
             return await handle.GetResultAsync(rpcOptions: options.Rpc).ConfigureAwait(false);
         }
 
+        /// <summary>
+        /// Signal and (optionally) start a workflow via lambda invoking the signalWithStart method.
+        /// </summary>
+        /// <typeparam name="TWorkflow">Workflow class type.</typeparam>
+        /// <param name="client">Client to use.</param>
+        /// <param name="workflowRunCall">Invocation of workflow run method with a result.</param>
+        /// <param name="signalCall">Invocation of signal method without a result.</param>
+        /// <param name="options">Start workflow options. ID and TaskQueue are required.</param>
+        /// <returns>Workflow handle for the started workflow.</returns>
+        /// <exception cref="ArgumentException">Invalid run call or options.</exception>
+        /// <exception cref="Exceptions.RpcException">Server-side error.</exception>
+        public static async Task<WorkflowHandle<TWorkflow>> SignalWorkflowWithStartAsync<TWorkflow>(
+            this ITemporalClient client,
+            Expression<Func<TWorkflow, Task>> workflowRunCall,
+            Expression<Func<TWorkflow, Task>> signalCall,
+            WorkflowOptions options)
+        {
+            options.SignalWithStart(signalCall);
+            return await client.StartWorkflowAsync(workflowRunCall, options).ConfigureAwait(false);
+        }
+
+        /// <summary>
+        /// Signal and (optionally) start a workflow via lambda invoking the signalWithStart method.
+        /// </summary>
+        /// <typeparam name="TWorkflow">Workflow class type.</typeparam>
+        /// <typeparam name="TResult">Workflow result type.</typeparam>
+        /// <param name="client">Client to use.</param>
+        /// <param name="workflowRunCall">Invocation of workflow run method with a result.</param>
+        /// <param name="signalCall">Invocation of signal method without a result.</param>
+        /// <param name="options">Start workflow options. ID and TaskQueue are required.</param>
+        /// <returns>Workflow handle for the started workflow.</returns>
+        /// <exception cref="ArgumentException">Invalid run call or options.</exception>
+        /// <exception cref="Exceptions.RpcException">Server-side error.</exception>
+        public static async Task<WorkflowHandle<TWorkflow, TResult>> SignalWorkflowWithStartAsync<TWorkflow, TResult>(
+            this ITemporalClient client,
+            Expression<Func<TWorkflow, Task<TResult>>> workflowRunCall,
+            Expression<Func<TWorkflow, Task>> signalCall,
+            WorkflowOptions options)
+        {
+            options.SignalWithStart(signalCall);
+            return await client.StartWorkflowAsync(workflowRunCall, options).ConfigureAwait(false);
+        }
+
         /// <summary>
         /// Shortcut for
         /// <see cref="ITemporalClient.StartWorkflowAsync{T}(Expression{Func{T, Task}}, WorkflowOptions)" />
@@ -158,4 +201,4 @@ public static async IAsyncEnumerable<WorkflowHistory> ListWorkflowHistoriesAsync
         }
 #endif
     }
-}
\ No newline at end of file
+}