fix: Preserve profile name in Databricks CLI auth provider (#1877)

## Changes

When the databricks-cli auth type is used with a named profile (e.g.
from ~/.databrickscfg), the profile name was not being threaded through
to DatabricksCliAuthProvider. As a result `auth login` always used
--host, causing issues with new databricks cli when same host exists
under multiple profiles.

Notes:
The process.env["DATABRICKS_CONFIG_PROFILE"] mutation is a workaround and should be revisited once the
upstream JS SDK exposes a proper API for this.

## Tests

Manually

Author: misha-db

packages/databricks-vscode/src/configuration/auth/AuthProvider.ts

@@ -78,6 +78,15 @@ export abstract class AuthProvider {
             return true;
         }
 
+        // TODO: Temporary workaround until the JS SDK supports passing a profile
+        // directly to auth types. Once supported, remove this env mutation.
+        const profile = this.toJSON()["profile"];
+        if (profile !== undefined) {
+            process.env["DATABRICKS_CONFIG_PROFILE"] = profile;
+        } else {
+            delete process.env["DATABRICKS_CONFIG_PROFILE"];
+        }
+
         const checkFn = async (token?: CancellationToken) => {
             this.checked = await this._check(token);
         };
@@ -165,7 +174,8 @@ export abstract class AuthProvider {
                 return new DatabricksCliAuthProvider(
                     host,
                     json.databricksPath ?? cli.cliPath,
-                    cli
+                    cli,
+                    json.profile
                 );
 
             case "profile":
@@ -198,7 +208,8 @@ export abstract class AuthProvider {
                 return new DatabricksCliAuthProvider(
                     host,
                     config.databricksCliPath ?? cli.cliPath,
-                    cli
+                    cli,
+                    config.profile
                 );
 
             default:
@@ -308,7 +319,8 @@ export class DatabricksCliAuthProvider extends AuthProvider {
     constructor(
         host: URL,
         readonly cliPath: string,
-        cli: CliWrapper
+        cli: CliWrapper,
+        readonly profile?: string
     ) {
         super(host, "databricks-cli", cli);
     }
@@ -322,6 +334,7 @@ export class DatabricksCliAuthProvider extends AuthProvider {
             host: this.host.toString(),
             authType: this.authType,
             databricksPath: this.cliPath,
+            ...(this.profile ? {profile: this.profile} : {}),
         };
     }
 
@@ -330,14 +343,19 @@ export class DatabricksCliAuthProvider extends AuthProvider {
             host: this.host.toString(),
             authType: "databricks-cli",
             databricksCliPath: this.cliPath,
+            ...(this.profile ? {profile: this.profile} : {}),
         });
     }
 
     toEnv(): Record<string, string> {
-        return {
+        const env: Record<string, string> = {
             DATABRICKS_HOST: this.host.toString(),
             DATABRICKS_AUTH_TYPE: "databricks-cli",
         };
+        if (this.profile) {
+            env["DATABRICKS_CONFIG_PROFILE"] = this.profile;
+        }
+        return env;
     }
 
     toIni() {

packages/databricks-vscode/src/configuration/auth/DatabricksCliCheck.ts

@@ -84,6 +84,7 @@ export class DatabricksCliCheck implements Disposable {
                 host: this.authProvider.host.toString(),
                 authType: "databricks-cli",
                 databricksCliPath: this.authProvider.cliPath,
+                profile: this.authProvider.profile,
             },
             {
                 product: "databricks-vscode",
@@ -105,9 +106,16 @@ export class DatabricksCliCheck implements Disposable {
     private async login(cancellationToken?: CancellationToken): Promise<void> {
         try {
             const host = this.authProvider.host.toString().replace(/\/+$/, "");
+            const profile = this.authProvider.profile;
+            const args = ["auth", "login"];
+            if (profile) {
+                args.push("--profile", profile);
+            } else {
+                args.push("--host", host);
+            }
             await execFile(
                 this.authProvider.cliPath,
-                ["auth", "login", "--host", host],
+                args,
                 {},
                 cancellationToken
             );