Ron/bdc script (#4221)

* WIP adding scripting support.

* Adding deploy command along with additional env vars needed.

* Adding script generation that sets envars, kube context, and mssqlctl

* Adding test email for docker email envar until we update UI.

* Adding cluster platform detection and disabling generate script after first click.

* Fix spacing and adding comment.

extensions/big-data-cluster/src/kubectl/kubectl.ts

@@ -14,6 +14,8 @@ import { getToolPath } from '../config/config';
 export interface Kubectl {
     checkPresent(errorMessageMode: CheckPresentMessageMode): Promise<boolean>;
     asJson<T>(command: string): Promise<Errorable<T>>;
+    invokeAsync(command: string, stdin?: string): Promise<ShellResult | undefined>;
+    getContext(): Context;
 }
 
 interface Context {
@@ -30,7 +32,7 @@ class KubectlImpl implements Kubectl {
         this.context = { host : host, fs : fs, shell : shell, installDependenciesCallback : installDependenciesCallback, binFound : kubectlFound, binPath : 'kubectl' };
     }
 
-    private readonly context: Context;
+    readonly context: Context;
 
     checkPresent(errorMessageMode: CheckPresentMessageMode): Promise<boolean> {
         return checkPresent(this.context, errorMessageMode);
@@ -38,6 +40,13 @@ class KubectlImpl implements Kubectl {
     asJson<T>(command: string): Promise<Errorable<T>> {
         return asJson(this.context, command);
     }
+    invokeAsync(command: string, stdin?: string): Promise<ShellResult | undefined> {
+        return invokeAsync(this.context, command, stdin);
+    }
+
+    getContext(): Context {
+        return this.context;
+    }
 }
 
 export function create(host: Host, fs: FS, shell: Shell, installDependenciesCallback: () => void): Kubectl {
@@ -108,7 +117,7 @@ async function checkPossibleIncompatibility(context: Context): Promise<void> {
 }
 
 
-function baseKubectlPath(context: Context): string {
+export function baseKubectlPath(context: Context): string {
     let bin = getToolPath(context.host, context.shell, 'kubectl');
     if (!bin) {
         bin = 'kubectl';

extensions/big-data-cluster/src/kubectl/kubectlUtils.ts

@@ -5,7 +5,7 @@
 
 import * as vscode from "vscode";
 import { Kubectl } from "./kubectl";
-import { failed } from "../interfaces";
+import { failed, ClusterType } from "../interfaces";
 
 export interface KubectlContext {
     readonly clusterName: string;
@@ -83,4 +83,50 @@ export async function getContexts(kubectl: Kubectl): Promise<KubectlContext[]> {
             active: c.name === currentContext
         };
     });
+}
+
+export async function setContext(kubectl: Kubectl, targetContext: string): Promise<void> {
+    const shellResult = await kubectl.invokeAsync(`config use-context ${targetContext}`);
+    if (!shellResult || shellResult.code != 0) {
+        // TODO: Update error handling for now.
+        vscode.window.showErrorMessage(`Failed to set '${targetContext}' as current cluster: ${shellResult ? shellResult.stderr : "Unable to run kubectl"}`);
+    }
+}
+
+export async function inferCurrentClusterType(kubectl: Kubectl): Promise<ClusterType> {
+    let latestContextName = "";
+
+    const ctxsr = await kubectl.invokeAsync('config current-context');
+    if (ctxsr && ctxsr.code === 0) {
+        latestContextName = ctxsr.stdout.trim();
+    } else {
+        return ClusterType.Other;
+    }
+
+    const cisr = await kubectl.invokeAsync('cluster-info');
+    if (!cisr || cisr.code !== 0) {
+        return ClusterType.Unknown;
+    }
+    const masterInfos = cisr.stdout.split('\n')
+                                   .filter((s) => s.indexOf('master is running at') >= 0);
+
+    if (masterInfos.length === 0) {
+        return ClusterType.Other;
+    }
+
+    const masterInfo = masterInfos[0];
+    if (masterInfo.indexOf('azmk8s.io') >= 0 || masterInfo.indexOf('azure.com') >= 0) {
+        return ClusterType.AKS;
+    }
+
+    if (latestContextName) {
+        const gcsr = await kubectl.invokeAsync(`config get-contexts ${latestContextName}`);
+        if (gcsr && gcsr.code === 0) {
+            if (gcsr.stdout.indexOf('minikube') >= 0) {
+                return ClusterType.Minikube;
+            }
+        }
+    }
+
+    return ClusterType.Other;
 }