Connectedk8s extension helm chart cleanup

src/connectedk8s/azext_connectedk8s/_utils.py

@@ -4,6 +4,7 @@
 # --------------------------------------------------------------------------------------------
 
 import os
+import shutil
 import subprocess
 from subprocess import Popen, PIPE
 
@@ -22,6 +23,15 @@
 logger = get_logger(__name__)
 
 # pylint: disable=line-too-long
+# pylint: disable=bare-except
+
+
+def cleanup_helm_chart_directory(path):
+    try:
+        if os.path.isdir(path):
+            shutil.rmtree(path)
+    except:
+        logger.warning("Unable to cleanup the azure-arc helm charts already present on the machine. In case of failure, please cleanup the directory '%s' and try again.", path)
 
 
 def validate_location(cmd, location):
@@ -52,6 +62,7 @@ def get_chart_path(registry_path, kube_config, kube_context):
 
     # Exporting helm chart
     chart_export_path = os.path.join(os.path.expanduser('~'), '.azure', 'AzureArcCharts')
+    cleanup_helm_chart_directory(chart_export_path)
     export_helm_chart(registry_path, chart_export_path, kube_config, kube_context)
     # Helm Install
     helm_chart_path = os.path.join(chart_export_path, 'azure-arc-k8sagents')
@@ -74,7 +85,6 @@ def pull_helm_chart(registry_path, kube_config, kube_context):
 
 
 def export_helm_chart(registry_path, chart_export_path, kube_config, kube_context):
-    chart_export_path = os.path.join(os.path.expanduser('~'), '.azure', 'AzureArcCharts')
     cmd_helm_chart_export = ["helm", "chart", "export", registry_path, "--destination", chart_export_path]
     if kube_config:
         cmd_helm_chart_export.extend(["--kubeconfig", kube_config])

src/connectedk8s/azext_connectedk8s/custom.py

@@ -115,6 +115,11 @@ def create_connectedk8s(cmd, client, resource_group_name, cluster_name, https_pr
     helm_version = check_helm_version(kube_config, kube_context)
     telemetry.add_extension_event('connectedk8s', {'Context.Default.AzureCLI.HelmVersion': helm_version})
 
+    # Check for faulty pre-release helm versions
+    if "3.3.0-rc" in helm_version:
+        telemetry.set_user_fault()
+        raise CLIError("The current helm version is not supported for azure-arc onboarding. Please upgrade the helm version and try again.")
+
     # Validate location
     utils.validate_location(cmd, location)
     resourceClient = _resource_client_factory(cmd.cli_ctx, subscription_id=subscription_id)
@@ -688,6 +693,11 @@ def update_agents(cmd, client, resource_group_name, cluster_name, https_proxy=""
     helm_version = check_helm_version(kube_config, kube_context)
     telemetry.add_extension_event('connectedk8s', {'Context.Default.AzureCLI.HelmVersion': helm_version})
 
+    # Check for faulty pre-release helm versions
+    if "3.3.0-rc" in helm_version:
+        telemetry.set_user_fault()
+        raise CLIError("The current helm version is not supported for azure-arc onboarding. Please upgrade the helm version and try again.")
+
     # Check whether Connected Cluster is present
     if not connected_cluster_exists(client, resource_group_name, cluster_name):
         telemetry.set_user_fault()