Refactor validate_direct_only_resources and approval log loops

bundle/config/mutator/validate_direct_only_resources.go

@@ -9,51 +9,12 @@ import (
 	"github.com/databricks/cli/libs/diag"
 )
 
-type directOnlyResource struct {
-	resourceType string
-	pluralName   string
-	singularName string
-	getResources func(*bundle.Bundle) map[string]any
-}
-
-// Resources that are only supported in direct deployment mode
-var directOnlyResources = []directOnlyResource{
-	{
-		resourceType: "catalogs",
-		pluralName:   "Catalog",
-		singularName: "catalog",
-		getResources: func(b *bundle.Bundle) map[string]any {
-			result := make(map[string]any)
-			for k, v := range b.Config.Resources.Catalogs {
-				result[k] = v
-			}
-			return result
-		},
-	},
-	{
-		resourceType: "external_locations",
-		pluralName:   "External Location",
-		singularName: "external location",
-		getResources: func(b *bundle.Bundle) map[string]any {
-			result := make(map[string]any)
-			for k, v := range b.Config.Resources.ExternalLocations {
-				result[k] = v
-			}
-			return result
-		},
-	},
-	{
-		resourceType: "vector_search_endpoints",
-		pluralName:   "Vector Search Endpoint",
-		singularName: "vector search endpoint",
-		getResources: func(b *bundle.Bundle) map[string]any {
-			result := make(map[string]any)
-			for k, v := range b.Config.Resources.VectorSearchEndpoints {
-				result[k] = v
-			}
-			return result
-		},
-	},
+// directOnlyResourceTypes lists resources only supported in direct deployment mode.
+// Keys are PluralName values from resources.ResourceDescription.
+var directOnlyResourceTypes = map[string]bool{
+	"catalogs":                true,
+	"external_locations":      true,
+	"vector_search_endpoints": true,
 }
 
 type validateDirectOnlyResources struct {
@@ -76,20 +37,22 @@ func (m *validateDirectOnlyResources) Apply(ctx context.Context, b *bundle.Bundl
 	}
 
 	var diags diag.Diagnostics
-
-	for _, resource := range directOnlyResources {
-		resourceMap := resource.getResources(b)
-		if len(resourceMap) > 0 {
-			diags = diags.Append(diag.Diagnostic{
-				Severity: diag.Error,
-				Summary:  resource.pluralName + " resources are only supported with direct deployment mode",
-				Detail: fmt.Sprintf("%s resources require direct deployment mode. "+
-					"Please set the DATABRICKS_BUNDLE_ENGINE environment variable to 'direct' to use %s resources.\n"+
-					"Learn more at https://docs.databricks.com/dev-tools/bundles/direct",
-					resource.pluralName, resource.singularName),
-				Locations: b.Config.GetLocations("resources." + resource.resourceType),
-			})
+	for _, group := range b.Config.Resources.AllResources() {
+		if !directOnlyResourceTypes[group.Description.PluralName] {
+			continue
+		}
+		if len(group.Resources) == 0 {
+			continue
 		}
+		diags = diags.Append(diag.Diagnostic{
+			Severity: diag.Error,
+			Summary:  group.Description.SingularTitle + " resources are only supported with direct deployment mode",
+			Detail: fmt.Sprintf("%s resources require direct deployment mode. "+
+				"Please set the DATABRICKS_BUNDLE_ENGINE environment variable to 'direct' to use %s resources.\n"+
+				"Learn more at https://docs.databricks.com/dev-tools/bundles/direct",
+				group.Description.SingularTitle, group.Description.SingularName),
+			Locations: b.Config.GetLocations("resources." + group.Description.PluralName),
+		})
 	}
 
 	return diags

bundle/config/mutator/validate_direct_only_resources_test.go

@@ -0,0 +1,110 @@
+package mutator_test
+
+import (
+	"testing"
+
+	"github.com/databricks/cli/bundle"
+	"github.com/databricks/cli/bundle/config"
+	"github.com/databricks/cli/bundle/config/engine"
+	"github.com/databricks/cli/bundle/config/mutator"
+	"github.com/databricks/cli/bundle/config/resources"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestValidateDirectOnlyResourcesDirectEngineReturnsNil(t *testing.T) {
+	b := &bundle.Bundle{
+		Config: config.Root{
+			Resources: config.Resources{
+				Catalogs: map[string]*resources.Catalog{
+					"my_catalog": {},
+				},
+			},
+		},
+	}
+
+	diags := bundle.Apply(t.Context(), b, mutator.ValidateDirectOnlyResources(engine.EngineDirect))
+	assert.Empty(t, diags)
+}
+
+func TestValidateDirectOnlyResourcesTerraformEngineNoDirectOnlyReturnsNil(t *testing.T) {
+	b := &bundle.Bundle{
+		Config: config.Root{
+			Resources: config.Resources{
+				Jobs: map[string]*resources.Job{
+					"my_job": {},
+				},
+			},
+		},
+	}
+
+	diags := bundle.Apply(t.Context(), b, mutator.ValidateDirectOnlyResources(engine.EngineTerraform))
+	assert.Empty(t, diags)
+}
+
+func TestValidateDirectOnlyResourcesTerraformEngineDirectOnlyEmitsError(t *testing.T) {
+	cases := []struct {
+		name            string
+		bundle          *bundle.Bundle
+		expectedSummary string
+		expectedDetail  string
+	}{
+		{
+			name: "catalogs",
+			bundle: &bundle.Bundle{
+				Config: config.Root{
+					Resources: config.Resources{
+						Catalogs: map[string]*resources.Catalog{
+							"my_catalog": {},
+						},
+					},
+				},
+			},
+			expectedSummary: "Catalog resources are only supported with direct deployment mode",
+			expectedDetail: "Catalog resources require direct deployment mode. " +
+				"Please set the DATABRICKS_BUNDLE_ENGINE environment variable to 'direct' to use catalog resources.\n" +
+				"Learn more at https://docs.databricks.com/dev-tools/bundles/direct",
+		},
+		{
+			name: "external_locations",
+			bundle: &bundle.Bundle{
+				Config: config.Root{
+					Resources: config.Resources{
+						ExternalLocations: map[string]*resources.ExternalLocation{
+							"my_location": {},
+						},
+					},
+				},
+			},
+			expectedSummary: "External Location resources are only supported with direct deployment mode",
+			expectedDetail: "External Location resources require direct deployment mode. " +
+				"Please set the DATABRICKS_BUNDLE_ENGINE environment variable to 'direct' to use external_location resources.\n" +
+				"Learn more at https://docs.databricks.com/dev-tools/bundles/direct",
+		},
+		{
+			name: "vector_search_endpoints",
+			bundle: &bundle.Bundle{
+				Config: config.Root{
+					Resources: config.Resources{
+						VectorSearchEndpoints: map[string]*resources.VectorSearchEndpoint{
+							"my_endpoint": {},
+						},
+					},
+				},
+			},
+			expectedSummary: "Vector Search Endpoint resources are only supported with direct deployment mode",
+			expectedDetail: "Vector Search Endpoint resources require direct deployment mode. " +
+				"Please set the DATABRICKS_BUNDLE_ENGINE environment variable to 'direct' to use vector_search_endpoint resources.\n" +
+				"Learn more at https://docs.databricks.com/dev-tools/bundles/direct",
+		},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.name, func(t *testing.T) {
+			diags := bundle.Apply(t.Context(), tc.bundle, mutator.ValidateDirectOnlyResources(engine.EngineTerraform))
+			if assert.Len(t, diags, 1) {
+				assert.Equal(t, tc.expectedSummary, diags[0].Summary)
+				assert.Equal(t, tc.expectedDetail, diags[0].Detail)
+			}
+		})
+	}
+}

bundle/phases/approval.go

@@ -0,0 +1,41 @@
+package phases
+
+import (
+	"context"
+
+	"github.com/databricks/cli/bundle/deployplan"
+	"github.com/databricks/cli/libs/cmdio"
+)
+
+// approvalGroup describes one resource type that needs explicit user consent
+// before a destructive action is applied.
+type approvalGroup struct {
+	group        string // matches config.GetResourceTypeFromKey, e.g. "schemas"
+	message      string // banner shown above the action list
+	skipChildren bool   // skip actions where IsChildResource() is true
+	trailingGap  bool   // print an empty line after the block (destroy only)
+}
+
+// logApprovalGroups filters actions per group and prints non-empty groups.
+// Returns the total number of matched actions across all groups.
+func logApprovalGroups(ctx context.Context, actions []deployplan.Action, groups []approvalGroup, types ...deployplan.ActionType) int {
+	total := 0
+	for _, g := range groups {
+		matched := filterGroup(actions, g.group, types...)
+		if len(matched) == 0 {
+			continue
+		}
+		total += len(matched)
+		cmdio.LogString(ctx, g.message)
+		for _, a := range matched {
+			if g.skipChildren && a.IsChildResource() {
+				continue
+			}
+			cmdio.Log(ctx, a)
+		}
+		if g.trailingGap {
+			cmdio.LogString(ctx, "")
+		}
+	}
+	return total
+}

bundle/phases/deploy.go

@@ -26,6 +26,17 @@ import (
 	"github.com/databricks/cli/libs/sync"
 )
 
+var deployApprovalGroups = []approvalGroup{
+	{group: "schemas", message: deleteOrRecreateSchemaMessage, skipChildren: true},
+	{group: "pipelines", message: deleteOrRecreatePipelineMessage},
+	{group: "volumes", message: deleteOrRecreateVolumeMessage},
+	{group: "dashboards", message: deleteOrRecreateDashboardMessage},
+	{group: "database_instances", message: deleteOrRecreateDatabaseInstanceMessage},
+	{group: "synced_database_tables", message: deleteOrRecreateSyncedDatabaseTableMessage},
+	{group: "postgres_projects", message: deleteOrRecreatePostgresProjectMessage},
+	{group: "postgres_branches", message: deleteOrRecreatePostgresBranchMessage},
+}
+
 func approvalForDeploy(ctx context.Context, b *bundle.Bundle, plan *deployplan.Plan) (bool, error) {
 	actions := plan.GetActions()
 
@@ -34,90 +45,12 @@ func approvalForDeploy(ctx context.Context, b *bundle.Bundle, plan *deployplan.P
 		return false, err
 	}
 
-	types := []deployplan.ActionType{deployplan.Recreate, deployplan.Delete}
-	schemaActions := filterGroup(actions, "schemas", types...)
-	pipelineActions := filterGroup(actions, "pipelines", types...)
-	volumeActions := filterGroup(actions, "volumes", types...)
-	dashboardActions := filterGroup(actions, "dashboards", types...)
-	databaseInstanceActions := filterGroup(actions, "database_instances", types...)
-	syncedDatabaseTableActions := filterGroup(actions, "synced_database_tables", types...)
-	postgresProjectActions := filterGroup(actions, "postgres_projects", types...)
-	postgresBranchActions := filterGroup(actions, "postgres_branches", types...)
-
-	// We don't need to display any prompts in this case.
-	if len(schemaActions) == 0 && len(pipelineActions) == 0 && len(volumeActions) == 0 && len(dashboardActions) == 0 &&
-		len(databaseInstanceActions) == 0 && len(syncedDatabaseTableActions) == 0 &&
-		len(postgresProjectActions) == 0 && len(postgresBranchActions) == 0 {
+	total := logApprovalGroups(ctx, actions, deployApprovalGroups, deployplan.Recreate, deployplan.Delete)
+	if total == 0 {
+		// No destructive actions in any tracked group: skip the prompt.
 		return true, nil
 	}
 
-	// One or more UC schema resources will be deleted or recreated.
-	if len(schemaActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreateSchemaMessage)
-		for _, action := range schemaActions {
-			if action.IsChildResource() {
-				continue
-			}
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more pipelines is being recreated.
-	if len(pipelineActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreatePipelineMessage)
-		for _, action := range pipelineActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more volumes is being recreated.
-	if len(volumeActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreateVolumeMessage)
-		for _, action := range volumeActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more dashboards is being recreated.
-	if len(dashboardActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreateDashboardMessage)
-		for _, action := range dashboardActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more database instances is being deleted or recreated.
-	if len(databaseInstanceActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreateDatabaseInstanceMessage)
-		for _, action := range databaseInstanceActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more synced database tables is being deleted or recreated.
-	if len(syncedDatabaseTableActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreateSyncedDatabaseTableMessage)
-		for _, action := range syncedDatabaseTableActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more Lakebase projects is being deleted or recreated.
-	if len(postgresProjectActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreatePostgresProjectMessage)
-		for _, action := range postgresProjectActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
-	// One or more Lakebase branches is being deleted or recreated.
-	if len(postgresBranchActions) != 0 {
-		cmdio.LogString(ctx, deleteOrRecreatePostgresBranchMessage)
-		for _, action := range postgresBranchActions {
-			cmdio.Log(ctx, action)
-		}
-	}
-
 	if b.AutoApprove {
 		return true, nil
 	}
@@ -127,12 +60,7 @@ func approvalForDeploy(ctx context.Context, b *bundle.Bundle, plan *deployplan.P
 	}
 
 	cmdio.LogString(ctx, "")
-	approved, err := cmdio.AskYesOrNo(ctx, "Would you like to proceed?")
-	if err != nil {
-		return false, err
-	}
-
-	return approved, nil
+	return cmdio.AskYesOrNo(ctx, "Would you like to proceed?")
 }
 
 func deployCore(ctx context.Context, b *bundle.Bundle, plan *deployplan.Plan, targetEngine engine.EngineType) {