pass profile to the CLI

Author: fjakobs

experimental/apps-mcp/lib/middlewares/databricks_client.go

@@ -15,7 +15,8 @@ import (
 )
 
 const (
-	DatabricksClientKey = "databricks_client"
+	DatabricksClientKey  = "databricks_client"
+	DatabricksProfileKey = "databricks_profile"
 )
 
 func NewDatabricksClientMiddleware(unauthorizedToolNames []string) mcp.Middleware {
@@ -40,6 +41,18 @@ func NewDatabricksClientMiddleware(unauthorizedToolNames []string) mcp.Middlewar
 	})
 }
 
+func MustGetDatabricksProfile(ctx context.Context) string {
+	sess, err := session.GetSession(ctx)
+	if err != nil {
+		panic(err)
+	}
+	profile, ok := sess.Get(DatabricksProfileKey)
+	if !ok {
+		return ""
+	}
+	return profile.(string)
+}
+
 func MustGetApiClient(ctx context.Context) (*httpclient.ApiClient, error) {
 	w := MustGetDatabricksClient(ctx)
 	clientCfg, err := config.HTTPClientConfigFromConfig(w.Config)

experimental/apps-mcp/lib/providers/clitools/configure_auth.go

@@ -55,6 +55,10 @@ func ConfigureAuth(ctx context.Context, sess *session.Session, host, profile *st
 	// Store client in session data
 	sess.Set(middlewares.DatabricksClientKey, client)
 
+	if profile == nil {
+		sess.Set(middlewares.DatabricksProfileKey, client.Config.Profile)
+	}
+
 	return client, nil
 }
 

experimental/apps-mcp/lib/providers/clitools/invoke_databricks_cli.go

@@ -25,7 +25,14 @@ func InvokeDatabricksCLI(ctx context.Context, command []string, workingDirectory
 	cmd := exec.CommandContext(ctx, cliPath, command...)
 	cmd.Dir = workingDirectory
 	env := os.Environ()
-	env = append(env, "DATABRICKS_HOST="+host)
+
+	profile := middlewares.MustGetDatabricksProfile(ctx)
+	if profile != "" {
+		env = append(env, "DATABRICKS_CONFIG_PROFILE="+profile)
+	}
+	if host != "" {
+		env = append(env, "DATABRICKS_HOST="+host)
+	}
 	cmd.Env = env
 
 	output, err := cmd.CombinedOutput()