From 5b29451add93d424bdc3fc81eac5384f2b5ac2a0 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Sat, 3 Jun 2023 20:35:36 +0200
Subject: [PATCH] Add methods to opt out specific activities

---
 .../Extensions/ModuleExtensions.cs            |  5 ++++
 .../Features/WorkflowManagementFeature.cs     | 25 ++++++++++++++++---
 2 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/src/modules/Elsa.Workflows.Management/Extensions/ModuleExtensions.cs b/src/modules/Elsa.Workflows.Management/Extensions/ModuleExtensions.cs
index 033260e5d..ec82714a4 100644
--- a/src/modules/Elsa.Workflows.Management/Extensions/ModuleExtensions.cs
+++ b/src/modules/Elsa.Workflows.Management/Extensions/ModuleExtensions.cs
@@ -69,4 +69,9 @@ public static class ModuleExtensions
     /// Adds the specified activity type to the system.
     /// </summary>
     public static IModule AddActivity<T>(this IModule module) where T : IActivity => module.UseWorkflowManagement(management => management.AddActivity<T>());
+    
+    /// <summary>
+    /// Removes the specified activity type from the system.
+    /// </summary>
+    public static IModule RemoveActivity<T>(this IModule module) where T : IActivity => module.UseWorkflowManagement(management => management.RemoveActivity<T>());
 }
\ No newline at end of file
diff --git a/src/modules/Elsa.Workflows.Management/Features/WorkflowManagementFeature.cs b/src/modules/Elsa.Workflows.Management/Features/WorkflowManagementFeature.cs
index c3d2d1e1d..e825788e4 100644
--- a/src/modules/Elsa.Workflows.Management/Features/WorkflowManagementFeature.cs
+++ b/src/modules/Elsa.Workflows.Management/Features/WorkflowManagementFeature.cs
@@ -68,15 +68,20 @@ public class WorkflowManagementFeature : FeatureBase
         new(typeof(TimeSpan), PrimitivesCategory, "Represents a duration of time."),
         new(typeof(IDictionary<string, string>), LookupsCategory, "A dictionary with string key and values."),
         new(typeof(IDictionary<string, object>), LookupsCategory, "A dictionary with string key and object values."),
-        new (typeof(ExpandoObject), DynamicCategory, "A dictionary that can be typed as dynamic to access members using dot notation.")
+        new(typeof(ExpandoObject), DynamicCategory, "A dictionary that can be typed as dynamic to access members using dot notation.")
     };
 
     /// <summary>
     /// Adds the specified activity type to the system.
     /// </summary>
-    public WorkflowManagementFeature AddActivity<T>() where T : IActivity
+    public WorkflowManagementFeature AddActivity<T>() where T : IActivity => AddActivity(typeof(T));
+
+    /// <summary>
+    /// Adds the specified activity type to the system.
+    /// </summary>
+    public WorkflowManagementFeature AddActivity(Type activityType)
     {
-        ActivityTypes.Add(typeof(T));
+        ActivityTypes.Add(activityType);
         return this;
     }
 
@@ -100,6 +105,20 @@ public class WorkflowManagementFeature : FeatureBase
         return this;
     }
 
+    /// <summary>
+    /// Removes the specified activity type from the system.
+    /// </summary>
+    public WorkflowManagementFeature RemoveActivity<T>() where T : IActivity => RemoveActivity(typeof(T));
+
+    /// <summary>
+    /// Adds the specified activity type to the system.
+    /// </summary>
+    public WorkflowManagementFeature RemoveActivity(Type activityType)
+    {
+        ActivityTypes.Remove(activityType);
+        return this;
+    }
+
     /// <summary>
     /// Adds the specified variable type to the system.
     /// </summary>