From 02a3df468c39a405620ee0751cfbfd39aaf19981 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Wed, 13 Jan 2021 19:57:49 +0100
Subject: [PATCH] Add IfTrue overload

---
 .../Activities/ControlFlow/IfElse/IfElseBuilderExtensions.cs  | 4 ++++
 .../Activities/ControlFlow/IfElse/IfElseExtensions.cs         | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseBuilderExtensions.cs b/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseBuilderExtensions.cs
index 6bc49ca11..cf557e8c0 100644
--- a/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseBuilderExtensions.cs
+++ b/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseBuilderExtensions.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Threading.Tasks;
 using Elsa.Builders;
 using Elsa.Services.Models;
 
@@ -18,6 +19,9 @@ namespace Elsa.Activities.ControlFlow
         public static IActivityBuilder IfElse(this IBuilder builder, Func<bool> condition, Action<IActivityBuilder>? activity = default) => builder.IfElse(x => x.WithCondition(condition), activity);
         public static IActivityBuilder IfElse(this IBuilder builder, bool condition, Action<IActivityBuilder>? activity = default) => builder.IfElse(x => x.WithCondition(condition), activity);
 
+        public static IActivityBuilder IfTrue(this IBuilder builder, Func<ActivityExecutionContext, ValueTask<bool>> condition, Action<IOutcomeBuilder> whenTrue) =>
+            builder.IfElse(x => x.WithCondition(condition), ifElse => whenTrue(ifElse.When(OutcomeNames.True)));
+        
         public static IActivityBuilder IfTrue(this IBuilder builder, Func<ActivityExecutionContext, bool> condition, Action<IOutcomeBuilder> whenTrue) =>
             builder.IfElse(x => x.WithCondition(condition), ifElse => whenTrue(ifElse.When(OutcomeNames.True)));
         
diff --git a/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseExtensions.cs b/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseExtensions.cs
index 3b801d82d..4e408d9b1 100644
--- a/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseExtensions.cs
+++ b/src/core/Elsa.Core/Activities/ControlFlow/IfElse/IfElseExtensions.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Threading.Tasks;
 using Elsa.Builders;
 using Elsa.Services.Models;
 
@@ -7,6 +8,7 @@ namespace Elsa.Activities.ControlFlow
 {
     public static class IfElseExtensions
     {
+        public static ISetupActivity<IfElse> WithCondition(this ISetupActivity<IfElse> activity, Func<ActivityExecutionContext, ValueTask<bool>> value) => activity.Set(x => x.Condition, value);
         public static ISetupActivity<IfElse> WithCondition(this ISetupActivity<IfElse> activity, Func<ActivityExecutionContext, bool> value) => activity.Set(x => x.Condition, value);
         public static ISetupActivity<IfElse> WithCondition(this ISetupActivity<IfElse> activity, Func<bool> value) => activity.Set(x => x.Condition, value);
         public static ISetupActivity<IfElse> WithCondition(this ISetupActivity<IfElse> activity, bool value) => activity.Set(x => x.Condition, value);