From a770decc7fcc5b3d5950290ffdf35f6069ffde08 Mon Sep 17 00:00:00 2001
From: Sipke Schoorstra <sipkeschoorstra@outlook.com>
Date: Thu, 12 Oct 2023 11:52:05 +0200
Subject: [PATCH] Update Flowchart to adhere to CanStartWorkflow

---
 .../Activities/Flowchart/Activities/Flowchart.cs         | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/modules/Elsa.Workflows.Core/Activities/Flowchart/Activities/Flowchart.cs b/src/modules/Elsa.Workflows.Core/Activities/Flowchart/Activities/Flowchart.cs
index c20938669..6e9c8fb3e 100644
--- a/src/modules/Elsa.Workflows.Core/Activities/Flowchart/Activities/Flowchart.cs
+++ b/src/modules/Elsa.Workflows.Core/Activities/Flowchart/Activities/Flowchart.cs
@@ -92,6 +92,15 @@ public class Flowchart : Container
             logger.LogDebug("A Start activity was found: {StartActivityId}", startActivity.Id);
             return startActivity;
         }
+        
+        // If there's an activity marked as "Can Start Workflow", use that.
+        var canStartWorkflowActivity = Activities.FirstOrDefault(x => x.GetCanStartWorkflow());
+        
+        if(canStartWorkflowActivity != null)
+        {
+            logger.LogDebug("An activity marked as 'Can Start Workflow' was found: {CanStartWorkflowActivityId}", canStartWorkflowActivity.Id);
+            return canStartWorkflowActivity;
+        }
 
         // If there is a single activity that has no inbound connections, use that.
         var root = GetRootActivity();