Automate pipeline management (#147)

* Extract business logic from UI menus to Actions.

Correcting name spaces to match folder structure

* Fix naming issues

* Enable draining and starting the pipelines from the command line

* Create Action for running initialization steps

Separate the logic of creating and running steps from the EnableScaleUnitFeature menu, making it simpler and more decoupled.

* Move deployer to Utilities folder

Use StepGenerator in Deployer

* Fix import path

* Use GetScaleUnitId in menus

Also change access modifiers for IAction

Author: Antonstockmarr

src/CLI/Actions/CleanUpStorageAccountAction.cs

@@ -0,0 +1,27 @@
+using System;
+using System.Threading.Tasks;
+using ScaleUnitManagement.DatabaseManager;
+
+namespace CLI.Actions
+{
+    internal class CleanUpStorageAccountAction : ContextualAction
+    {
+        public CleanUpStorageAccountAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            try
+            {
+                var storageAccountManager = new StorageAccountManager();
+                await storageAccountManager.CleanStorageAccount();
+                Console.WriteLine("Done");
+            }
+            catch (Exception ex)
+            {
+                Console.WriteLine($"An exception occured while cleaning up storage: \n{ex}");
+            }
+        }
+    }
+}

src/CLI/Actions/ConfigureEnvironmentAction.cs

@@ -0,0 +1,21 @@
+using System.Threading.Tasks;
+using ScaleUnitManagement.Utilities;
+using ScaleUnitManagement.WorkloadSetupOrchestrator;
+
+namespace CLI.Actions
+{
+    internal class ConfigureEnvironmentAction : ContextualAction
+    {
+        public ConfigureEnvironmentAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            if (ScaleUnitContext.GetScaleUnitId() == "@@")
+                await new HubConfigurationManager().Configure();
+            else
+                await new ScaleUnitConfigurationManager().Configure();
+        }
+    }
+}

src/CLI/Actions/DeleteWorkloadsAction.cs

@@ -0,0 +1,21 @@
+using System;
+using System.Threading.Tasks;
+using ScaleUnitManagement.WorkloadSetupOrchestrator;
+
+namespace CLI.Actions
+{
+    internal class DeleteWorkloadsAction : ContextualAction
+    {
+        public DeleteWorkloadsAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            var workloadDeleter = new WorkloadDeleter();
+            await workloadDeleter.DeleteWorkloadsFromScaleUnit();
+
+            Console.WriteLine("Done.");
+        }
+    }
+}

src/CLI/Actions/DisableScaleUnitFeatureAction.cs

@@ -0,0 +1,33 @@
+using System;
+using System.Threading.Tasks;
+using ScaleUnitManagement.ScaleUnitFeatureManager.Common;
+using ScaleUnitManagement.Utilities;
+
+namespace CLI.Actions
+{
+    internal class DisableScaleUnitFeatureAction : ContextualAction
+    {
+        public DisableScaleUnitFeatureAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override Task ExecuteInScaleUnitContext()
+        {
+            try
+            {
+                new StopServices().Run();
+                using (var webConfig = new WebConfig())
+                {
+                    SharedWebConfig.Configure(webConfig, isScaleUnitFeatureEnabled: false);
+                }
+                new RunDBSync().Run(isScaleUnitFeatureEnabled: false);
+                new StartServices().Run();
+            }
+            catch (Exception ex)
+            {
+                Console.Error.WriteLine($"An error occured while trying to disable scale unit feature:\n{ex}");
+            }
+            return Task.CompletedTask;
+        }
+    }
+}

src/CLI/Actions/DrainPipelinesAction.cs

@@ -0,0 +1,18 @@
+using System.Threading.Tasks;
+using ScaleUnitManagement.WorkloadSetupOrchestrator;
+
+namespace CLI.Actions
+{
+    internal class DrainPipelinesAction : ContextualAction
+    {
+        public DrainPipelinesAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            var pipelineManager = new PipelineManager();
+            await pipelineManager.DrainWorkloadDataPipelines();
+        }
+    }
+}

src/CLI/Actions/ImportWorkloadBlobAction.cs

@@ -0,0 +1,30 @@
+using System;
+using System.Threading.Tasks;
+using ScaleUnitManagement.DatabaseManager;
+
+namespace CLI.Actions
+{
+    internal class ImportWorkloadBlobAction : ContextualAction
+    {
+        private readonly string sasToken;
+
+        public ImportWorkloadBlobAction(string scaleUnitId, string sasToken) : base(scaleUnitId)
+        {
+            this.sasToken = sasToken;
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            var storageAccountManager = new StorageAccountManager();
+            try
+            {
+                await storageAccountManager.ImportWorkloadsBlob(sasToken);
+                Console.WriteLine("Done");
+            }
+            catch (Exception ex)
+            {
+                Console.WriteLine($"An exception occured while importing blobs: {ex.Message}");
+            }
+        }
+    }
+}

src/CLI/Actions/InstallWorkloadsAction.cs

@@ -0,0 +1,21 @@
+using System.Threading.Tasks;
+using ScaleUnitManagement.Utilities;
+using ScaleUnitManagement.WorkloadSetupOrchestrator;
+
+namespace CLI.Actions
+{
+    internal class InstallWorkloadsAction : ContextualAction
+    {
+        public InstallWorkloadsAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            if (ScaleUnitContext.GetScaleUnitId() == "@@")
+                await new HubWorkloadInstaller().Install();
+            else
+                await new ScaleUnitWorkloadInstaller().Install();
+        }
+    }
+}

…WorkloadMovementOptions/MoveWorkloads.cs src/CLI/Actions/MoveAllWorkloadsAction.cssrc/CLI/Menus/WorkloadMovementOptions/MoveWorkloads.cs renamed to src/CLI/Actions/MoveAllWorkloadsAction.cs src/CLI/Menus/WorkloadMovementOptions/MoveWorkloads.cs renamed to src/CLI/Actions/MoveAllWorkloadsAction.cs

@@ -1,18 +1,19 @@
+using System;
+using System.Collections.Generic;
 using System.Threading.Tasks;
 using ScaleUnitManagement.Utilities;
 using ScaleUnitManagement.WorkloadSetupOrchestrator;
-using System;
 
-namespace CLI.WorkloadMovementOptions
+namespace CLI.Actions
 {
-    internal class MoveWorkloads
+    internal class MoveAllWorkloadsAction : IAction
     {
-        public async Task MoveAllWorkloads(int input, string selectionHistory)
+        public async Task Execute()
         {
             try
             {
                 Console.WriteLine("Moving all workloads to the hub");
-                System.Collections.Generic.List<ScaleUnitInstance> scaleUnitInstances = Config.ScaleUnitInstances();
+                List<ScaleUnitInstance> scaleUnitInstances = Config.ScaleUnitInstances();
 
                 foreach (ScaleUnitInstance scaleUnit in scaleUnitInstances)
                 {

src/CLI/Actions/PerformEmergencyTransitionToHubAction.cs

@@ -0,0 +1,27 @@
+using System;
+using System.Threading.Tasks;
+using ScaleUnitManagement.WorkloadSetupOrchestrator;
+
+namespace CLI.Actions
+{
+    internal class PerformEmergencyTransitionToHubAction : ContextualAction
+    {
+        public PerformEmergencyTransitionToHubAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            try
+            {
+                var workloadMover = new WorkloadMover();
+                await workloadMover.EmergencyTransitionToHub();
+                Console.WriteLine("Done");
+            }
+            catch (Exception ex)
+            {
+                Console.Error.WriteLine($"An error occured while trying to run emergency transition to hub:\n{ex}");
+            }
+        }
+    }
+}

src/CLI/Actions/StartPipelinesAction.cs

@@ -0,0 +1,18 @@
+using System.Threading.Tasks;
+using ScaleUnitManagement.WorkloadSetupOrchestrator;
+
+namespace CLI.Actions
+{
+    internal class StartPipelinesAction : ContextualAction
+    {
+        public StartPipelinesAction(string scaleUnitId) : base(scaleUnitId)
+        {
+        }
+
+        protected override async Task ExecuteInScaleUnitContext()
+        {
+            var pipelineManager = new PipelineManager();
+            await pipelineManager.StartWorkloadDataPipelines();
+        }
+    }
+}