databricks-cli/bundle/config/mutator/select_default_environment.go

package mutator

import (
	"context"
	"fmt"
	"strings"

	"github.com/databricks/cli/bundle"
	"golang.org/x/exp/maps"
)

type selectDefaultEnvironment struct{}

// SelectDefaultEnvironment merges the default environment into the root configuration.
func SelectDefaultEnvironment() bundle.Mutator {
	return &selectDefaultEnvironment{}
}

func (m *selectDefaultEnvironment) Name() string {
	return "SelectDefaultEnvironment"
}

func (m *selectDefaultEnvironment) Apply(ctx context.Context, b *bundle.Bundle) error {
	if len(b.Config.Environments) == 0 {
		return fmt.Errorf("no environments defined")
	}

	// One environment means there's only one default.
	names := maps.Keys(b.Config.Environments)
	if len(names) == 1 {
		return SelectEnvironment(names[0]).Apply(ctx, b)
	}

	// Multiple environments means we look for the `default` flag.
	var defaults []string
	for name, env := range b.Config.Environments {
		if env != nil && env.Default {
			defaults = append(defaults, name)
		}
	}

	// It is invalid to have multiple environments with the `default` flag set.
	if len(defaults) > 1 {
		return fmt.Errorf("multiple environments are marked as default (%s)", strings.Join(defaults, ", "))
	}

	// If no environment has the `default` flag set, ask the user to specify one.
	if len(defaults) == 0 {
		return fmt.Errorf("please specify environment")
	}

	// One default remaining.
	return SelectEnvironment(defaults[0]).Apply(ctx, b)
}