Fix "az" on Windows (#318)

and minor improvements to the configuration wizard

Co-authored-by: Kartik Gupta <88345179+kartikgupta-db@users.noreply.github.com>

Author: fjakobs

packages/databricks-sdk-js/src/auth/fromAzureCli.ts

@@ -29,12 +29,16 @@ export const fromAzureCli = (host?: URL): CredentialProvider => {
     return refreshableCredentialProvider(async () => {
         let stdout = "";
         try {
-            ({stdout} = await execFile("az", [
-                "account",
-                "get-access-token",
-                "--resource",
-                azureDatabricksLoginAppID,
-            ]));
+            ({stdout} = await execFile(
+                "az",
+                [
+                    "account",
+                    "get-access-token",
+                    "--resource",
+                    azureDatabricksLoginAppID,
+                ],
+                {shell: true}
+            ));
         } catch (e: any) {
             if (e.code === "ENOENT") {
                 throw new CredentialsProviderError(

packages/databricks-vscode/src/configuration/AzureCliCheck.ts

@@ -182,7 +182,9 @@ export class AzureCliCheck implements Disposable {
     // check if Azure CLI is installed
     public async hasAzureCli(): Promise<boolean> {
         try {
-            const {stdout} = await execFile(this.azBinPath, ["--version"]);
+            const {stdout} = await execFile(this.azBinPath, ["--version"], {
+                shell: true,
+            });
             if (stdout.indexOf("azure-cli") !== -1) {
                 return true;
             }
@@ -213,10 +215,11 @@ export class AzureCliCheck implements Disposable {
     // check if Azure CLI is logged in
     public async isAzureCliLoggedIn(): Promise<boolean> {
         try {
-            const {stdout, stderr} = await execFile(this.azBinPath, [
-                "account",
-                "list",
-            ]);
+            const {stdout, stderr} = await execFile(
+                this.azBinPath,
+                ["account", "list"],
+                {shell: true}
+            );
             if (stdout === "[]") {
                 return false;
             }

packages/databricks-vscode/src/configuration/ConnectionManager.ts

@@ -238,7 +238,9 @@ export class ConnectionManager {
         let config: ProjectConfig | undefined;
         while (true) {
             config = await configureWorkspaceWizard(
-                config?.authProvider?.host.toString()
+                this.databricksWorkspace?.host?.toString() ||
+                    config?.authProvider?.host.toString() ||
+                    process.env.DATABRICKS_HOST
             );
 
             if (!config) {

packages/databricks-vscode/src/test/e2e/configure.e2e.ts

@@ -73,8 +73,6 @@ describe("Configure Databricks Extension", async function () {
         let input = await new InputBox(workbench.locatorMap).wait();
         await sleep(200);
 
-        await input.setText(host);
-        await sleep(500);
         await input.confirm();
         await sleep(200);
 

packages/databricks-vscode/src/ui/wizard.ts

@@ -163,6 +163,7 @@ export class MultiStepInput {
                 input.totalSteps = totalSteps;
                 input.value = value || "";
                 input.prompt = prompt;
+                input.ignoreFocusOut = true;
                 input.buttons = [
                     ...(this.steps.length > 1 ? [QuickInputButtons.Back] : []),
                     ...(buttons || []),