Feature reconcile cluster using update (#280)

Azure · Oct 26, 2023 · 405cd2d · 405cd2d
1 parent a2c1962
commit 405cd2d
Show file tree

Hide file tree

Showing 4 changed files with 67 additions and 0 deletions.
diff --git a/package.json b/package.json
@@ -232,6 +232,10 @@
                 "command": "aks.aksAbortLastOperationInCluster",
                 "title": "Abort Last Operation"
             },
+            {
+                "command": "aks.aksReconcileCluster",
+                "title": "Reconcile Cluster"
+            },
             {
                 "command": "aks.aksInspektorGadgetShow",
                 "title": "Show Inspektor Gadget"
@@ -369,6 +373,10 @@
                 {
                     "command": "aks.aksAbortLastOperationInCluster",
                     "group": "navigation"
+                }, 
+                {
+                    "command": "aks.aksReconcileCluster",
+                    "group": "navigation"
                 }
             ]
         },

diff --git a/src/commands/aksReconcileCluster/aksReconcileCluster.ts b/src/commands/aksReconcileCluster/aksReconcileCluster.ts
@@ -0,0 +1,39 @@
+import * as vscode from 'vscode';
+import * as k8s from 'vscode-kubernetes-tools-api';
+import { IActionContext } from "@microsoft/vscode-azext-utils";
+import { reconcileUsingUpdateInCluster, getAksClusterTreeItem } from '../utils/clusters';
+import { failed, succeeded } from '../utils/errorable';
+import { longRunning } from '../utils/host';
+
+export default async function aksReconcileCluster(
+  _context: IActionContext,
+  target: any
+): Promise<void> {
+  const cloudExplorer = await k8s.extension.cloudExplorer.v1;
+
+  const cluster = getAksClusterTreeItem(target, cloudExplorer);
+  if (failed(cluster)) {
+    vscode.window.showErrorMessage(cluster.error);
+    return;
+  }
+
+  const clusterName = cluster.result.name;
+
+  const answer = await vscode.window.showInformationMessage(`Do you want to reconcile/update operation on cluster ${clusterName}?`, "Yes", "No");
+
+  if (answer === "Yes") {
+    const result = await longRunning(
+      `Reconciling/update last cluster operation in ${clusterName}.`,
+      async () => {
+        return await reconcileUsingUpdateInCluster(cluster.result, clusterName)
+      });
+
+    if (failed(result)) {
+      vscode.window.showErrorMessage(result.error);
+    }
+
+    if (succeeded(result)) {
+      vscode.window.showInformationMessage(result.result);
+    }
+  }
+}
diff --git a/src/commands/utils/clusters.ts b/src/commands/utils/clusters.ts
@@ -476,6 +476,24 @@ export async function abortLastOperationInCluster(
     }
 }
 
+export async function reconcileUsingUpdateInCluster(
+    target: AksClusterTreeItem,
+    clusterName: string
+): Promise<Errorable<string>> {
+    try {
+        // Pleaset note: here is in the only place this way of reconcile is documented: https://learn.microsoft.com/en-us/cli/azure/aks?view=azure-cli-latest#az-aks-update()-examples
+        const containerClient = getContainerClient(target);
+        const getClusterInfo = await containerClient.managedClusters.get(target.resourceGroupName, clusterName);
+        await containerClient.managedClusters.beginCreateOrUpdateAndWait(target.resourceGroupName, clusterName, {
+            location: getClusterInfo.location,
+        })
+
+        return { succeeded: true, result: "Reconcile/Update cluster succeeded." };
+    } catch (ex) {
+        return { succeeded: false, error: `Error invoking ${clusterName} managed cluster: ${getErrorMessage(ex)}` };
+    }
+}
+
 export async function rotateClusterCert(
     target: AksClusterTreeItem,
     clusterName: string

diff --git a/src/extension.ts b/src/extension.ts
@@ -31,6 +31,7 @@ import aksRotateClusterCert from './commands/aksRotateClusterCert/aksRotateClust
 import { aksInspektorGadgetShow } from './commands/aksInspektorGadget/aksInspektorGadget';
 import aksCreateCluster from './commands/aksCreateCluster/aksCreateCluster';
 import aksAbortLastOperation from './commands/aksAbortLastOperation/aksAbortLastOperation';
+import aksReconcileCluster from './commands/aksReconcileCluster/aksReconcileCluster';
 
 export async function activate(context: vscode.ExtensionContext) {
     const cloudExplorer = await k8s.extension.cloudExplorer.v1;
@@ -73,6 +74,7 @@ export async function activate(context: vscode.ExtensionContext) {
         registerCommandWithTelemetry('aks.aksDeleteCluster', aksDeleteCluster);
         registerCommandWithTelemetry('aks.aksRotateClusterCert', aksRotateClusterCert);
         registerCommandWithTelemetry('aks.aksAbortLastOperationInCluster', aksAbortLastOperation);
+        registerCommandWithTelemetry('aks.aksReconcileCluster', aksReconcileCluster);
         registerCommandWithTelemetry('aks.aksInspektorGadgetShow', aksInspektorGadgetShow);
         registerCommandWithTelemetry('aks.createCluster', aksCreateCluster);