Only start abort if cluster provision state is not succeeded.

Azure · Oct 6, 2023 · 3082b0f · 3082b0f
1 parent c3d30e6
commit 3082b0f
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/src/commands/aksAbortLastOperation/aksAbortLastOperation.ts b/src/commands/aksAbortLastOperation/aksAbortLastOperation.ts
@@ -1,7 +1,7 @@
 import * as vscode from 'vscode';
 import * as k8s from 'vscode-kubernetes-tools-api';
 import { IActionContext } from "@microsoft/vscode-azext-utils";
-import { abortLastOperationInCluster, getAksClusterTreeItem } from '../utils/clusters';
+import { abortLastOperationInCluster, determineProvisioningState, getAksClusterTreeItem } from '../utils/clusters';
 import { failed, succeeded } from '../utils/errorable';
 import { longRunning } from '../utils/host';
 
@@ -18,7 +18,17 @@ export default async function aksAbortLastOperation(
   }
 
   const clusterName = cluster.result.name;
+
+  const clusterProvisioingState = await determineProvisioningState(cluster.result, clusterName);
+  if (failed(clusterProvisioingState)) {
+    vscode.window.showErrorMessage(clusterProvisioingState.error);
+    return;
+  }
 
+  if (clusterProvisioingState.succeeded && clusterProvisioingState.result === "Succeeded") {
+    vscode.window.showInformationMessage(`Cluster provisioning state is ${clusterProvisioingState} and there is no operation to abort.`);
+    return;
+  }
   const answer = await vscode.window.showInformationMessage(`Do you want to abort last operation in cluster ${clusterName}?`, "Yes", "No");
 
   if (answer === "Yes") {

diff --git a/src/commands/utils/clusters.ts b/src/commands/utils/clusters.ts
@@ -338,6 +338,21 @@ export async function determineClusterState(
     }
 }
 
+export async function determineProvisioningState(
+    target: AksClusterTreeItem,
+    clusterName: string
+): Promise<Errorable<string>> {
+    try {
+        const containerClient = getContainerClient(target);
+        const clusterInfo = (await containerClient.managedClusters.get(target.resourceGroupName, clusterName));
+
+        return { succeeded: true, result:  clusterInfo.provisioningState ?? "" };
+
+    } catch (ex) {
+        return { succeeded: false, error: `Error invoking ${clusterName} managed cluster: ${ex}` };
+    }
+}
+
 export async function startCluster(
     target: AksClusterTreeItem,
     clusterName: string,