From 07534a23113bf377d7524a4c80b87761837c4b57 Mon Sep 17 00:00:00 2001
From: Georgiana Dolocan <georgiana.dolocan@gmail.com>
Date: Wed, 31 Jul 2024 13:41:48 +0300
Subject: [PATCH 1/2] Update command so that it's per hub not per cluster

---
 deployer/commands/exec/infra_components.py | 41 +++++++++-------------
 1 file changed, 17 insertions(+), 24 deletions(-)

diff --git a/deployer/commands/exec/infra_components.py b/deployer/commands/exec/infra_components.py
index 18089e3ba..372a79825 100644
--- a/deployer/commands/exec/infra_components.py
+++ b/deployer/commands/exec/infra_components.py
@@ -21,6 +21,7 @@
 @exec_app.command()
 def root_homes(
     cluster_name: str = typer.Argument(..., help="Name of cluster to operate on"),
+    hub_name: str = typer.Argument(..., help="Name of hub to operate on"),
 ):
     """
     Pop an interactive shell with the entire nfs file system of the given cluster mounted on /root-homes
@@ -31,37 +32,29 @@ def root_homes(
 
     with cluster.auth():
         hubs = cluster.hubs
-        # Get a hub so we can extract the nfs ip and base share name from its config
-        hub = next((hub for hub in hubs if hub.spec["name"] == "staging"), None)
-        if hub:
-            namespace = "staging"
-        else:
-            hub = hubs[0]
-            namespace = hub.spec["name"]
+        hub = next((hub for hub in hubs if hub.spec["name"] == hub_name), None)
+        if not hub:
+            print_colour("Hub does not exist in {cluster_name} cluster}")
+            return
 
+    server_ip = base_share_name = ""
     for values_file in hub.spec["helm_chart_values_files"]:
         if "secret" not in os.path.basename(values_file):
             values_file = config_file_path.parent.joinpath(values_file)
             config = yaml.load(values_file)
 
             if config.get("basehub", {}):
-                server_ip = (
-                    config["basehub"].get("nfs", {}).get("pv", {}).get("serverIP", "")
-                )
-                base_share_name = (
-                    config["basehub"]
-                    .get("nfs", {})
-                    .get("pv", {})
-                    .get("baseShareName", "")
-                )
-            else:
-                server_ip = config.get("nfs", {}).get("pv", {}).get("serverIP", "")
-                base_share_name = (
-                    config.get("nfs", {}).get("pv", {}).get("baseShareName", "")
-                )
-            if server_ip and base_share_name:
-                break
+                config = config["basehub"]
 
+            server_ip = config.get("nfs", {}).get("pv", {}).get("serverIP", server_ip)
+            base_share_name = (
+                config.get("nfs", {})
+                .get("pv", {})
+                .get("baseShareName", base_share_name)
+            )
+
+    print(server_ip)
+    print(base_share_name)
     pod_name = f"{cluster_name}-root-home-shell"
     pod = {
         "apiVersion": "v1",
@@ -97,7 +90,7 @@ def root_homes(
     cmd = [
         "kubectl",
         "-n",
-        namespace,
+        hub_name,
         "run",
         "--rm",  # Remove pod when we're done
         "-it",  # Give us a shell!

From 72f9d6f46e21ec5f2ecec13526582a74dd162ad6 Mon Sep 17 00:00:00 2001
From: Georgiana Dolocan <georgiana.dolocan@gmail.com>
Date: Wed, 31 Jul 2024 13:42:41 +0300
Subject: [PATCH 2/2] Rm debug prints

---
 deployer/commands/exec/infra_components.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/deployer/commands/exec/infra_components.py b/deployer/commands/exec/infra_components.py
index 372a79825..2771d0dc7 100644
--- a/deployer/commands/exec/infra_components.py
+++ b/deployer/commands/exec/infra_components.py
@@ -53,8 +53,6 @@ def root_homes(
                 .get("baseShareName", base_share_name)
             )
 
-    print(server_ip)
-    print(base_share_name)
     pod_name = f"{cluster_name}-root-home-shell"
     pod = {
         "apiVersion": "v1",