From 73392b83cb0c2e4d9dd27bb93e537610aee98db7 Mon Sep 17 00:00:00 2001
From: Guillaume Boutry <guillaume.boutry@canonical.com>
Date: Fri, 13 Dec 2024 14:51:38 +0100
Subject: [PATCH] Report scale down improvements to local mode

MaaS mode has been improved with a better scale down story. Prechecks,
re-organizing k8s workloads, disabling scheduling on hypervisor...

Signed-off-by: Guillaume Boutry <guillaume.boutry@canonical.com>
---
 .../sunbeam/provider/local/commands.py        | 80 ++++++++++++++-----
 1 file changed, 58 insertions(+), 22 deletions(-)

diff --git a/sunbeam-python/sunbeam/provider/local/commands.py b/sunbeam-python/sunbeam/provider/local/commands.py
index 5335ee8a..1326d42e 100644
--- a/sunbeam-python/sunbeam/provider/local/commands.py
+++ b/sunbeam-python/sunbeam/provider/local/commands.py
@@ -128,12 +128,20 @@
     AddK8SCloudStep,
     AddK8SCredentialStep,
     AddK8SUnitsStep,
+    CheckMysqlK8SDistributionStep,
+    CheckOvnK8SDistributionStep,
+    CheckRabbitmqK8SDistributionStep,
+    CordonK8SUnitStep,
     DeployK8SApplicationStep,
+    DrainK8SUnitStep,
+    MigrateK8SKubeconfigStep,
     RemoveK8SUnitsStep,
     StoreK8SKubeConfigStep,
+    UpdateK8SCloudStep,
 )
 from sunbeam.steps.microceph import (
     AddMicrocephUnitsStep,
+    CheckMicrocephDistributionStep,
     ConfigureMicrocephOSDStep,
     DeployMicrocephApplicationStep,
     RemoveMicrocephUnitsStep,
@@ -962,32 +970,59 @@ def remove(ctx: click.Context, name: str, force: bool, show_hints: bool) -> None
 
     plan = [
         JujuLoginStep(deployment.juju_account),
+        CheckMicrocephDistributionStep(
+            client,
+            name,
+            jhelper,
+            deployment.openstack_machines_model,
+            force=force,
+        ),
+        CheckMysqlK8SDistributionStep(
+            client,
+            name,
+            jhelper,
+            deployment.openstack_machines_model,
+            force=force,
+        ),
+        CheckOvnK8SDistributionStep(
+            client,
+            name,
+            jhelper,
+            deployment.openstack_machines_model,
+            force=force,
+        ),
+        CheckRabbitmqK8SDistributionStep(
+            client,
+            name,
+            jhelper,
+            deployment.openstack_machines_model,
+            force=force,
+        ),
+        MigrateK8SKubeconfigStep(
+            client, name, jhelper, deployment.openstack_machines_model
+        ),
+        UpdateK8SCloudStep(deployment, jhelper),
+        RemoveHypervisorUnitStep(
+            client, name, jhelper, deployment.openstack_machines_model, force
+        ),
+        RemoveMicrocephUnitsStep(
+            client, name, jhelper, deployment.openstack_machines_model
+        ),
+        CordonK8SUnitStep(client, name, jhelper, deployment.openstack_machines_model),
+        DrainK8SUnitStep(client, name, jhelper, deployment.openstack_machines_model),
+        RemoveK8SUnitsStep(client, name, jhelper, deployment.openstack_machines_model),
         RemoveSunbeamMachineUnitsStep(
             client, name, jhelper, deployment.openstack_machines_model
         ),
+        RemoveJujuMachineStep(
+            client, name, jhelper, deployment.openstack_machines_model
+        ),
+        # Cannot remove user as the same user name cannot be resued,
+        # so commenting the RemoveJujuUserStep
+        # RemoveJujuUserStep(name),
+        ClusterRemoveNodeStep(client, name),
     ]
 
-    plan.append(
-        RemoveK8SUnitsStep(client, name, jhelper, deployment.openstack_machines_model)
-    )
-
-    plan.extend(
-        [
-            RemoveMicrocephUnitsStep(
-                client, name, jhelper, deployment.openstack_machines_model
-            ),
-            RemoveHypervisorUnitStep(
-                client, name, jhelper, deployment.openstack_machines_model, force
-            ),
-            RemoveJujuMachineStep(
-                client, name, jhelper, deployment.openstack_machines_model
-            ),
-            # Cannot remove user as the same user name cannot be resued,
-            # so commenting the RemoveJujuUserStep
-            # RemoveJujuUserStep(name),
-            ClusterRemoveNodeStep(client, name),
-        ]
-    )
     run_plan(plan, console, show_hints)
     click.echo(f"Removed node {name} from the cluster")
     # Removing machine does not clean up all deployed juju components. This is
@@ -995,9 +1030,10 @@ def remove(ctx: click.Context, name: str, force: bool, show_hints: bool) -> None
     # Without the workaround mentioned in LP#1851489, it is not possible to
     # reprovision the machine back.
     click.echo(
-        f"Run command 'sudo /sbin/remove-juju-services' on node {name} "
+        f"Run command `sudo /sbin/remove-juju-services` on node {name} "
         "to reuse the machine."
     )
+    click.echo("Run `sunbeam cluster resize` to scale down the cluster")
 
 
 @click.command("deployment")