mirror of https://github.com/databricks/cli.git
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
|
package mutator
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/databricks/cli/bundle"
|
||
|
|
"golang.org/x/exp/maps"
|
||
|
|
)
|
||
|
|
|
||
|
|
type selectDefaultTarget struct{}
|
||
|
|
|
||
|
|
// SelectDefaultTarget merges the default target into the root configuration.
|
||
|
|
func SelectDefaultTarget() bundle.Mutator {
|
||
|
|
return &selectDefaultTarget{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *selectDefaultTarget) Name() string {
|
||
|
|
return "SelectDefaultTarget"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *selectDefaultTarget) Apply(ctx context.Context, b *bundle.Bundle) error {
|
||
|
|
if len(b.Config.Targets) == 0 {
|
||
|
|
return fmt.Errorf("no targets defined")
|
||
|
|
}
|
||
|
|
|
||
|
|
// One target means there's only one default.
|
||
|
|
names := maps.Keys(b.Config.Targets)
|
||
|
|
if len(names) == 1 {
|
||
|
|
return SelectTarget(names[0]).Apply(ctx, b)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Multiple targets means we look for the `default` flag.
|
||
|
|
var defaults []string
|
||
|
|
for name, env := range b.Config.Targets {
|
||
|
|
if env != nil && env.Default {
|
||
|
|
defaults = append(defaults, name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// It is invalid to have multiple targets with the `default` flag set.
|
||
|
|
if len(defaults) > 1 {
|
||
|
|
return fmt.Errorf("multiple targets are marked as default (%s)", strings.Join(defaults, ", "))
|
||
|
|
}
|
||
|
|
|
||
|
|
// If no target has the `default` flag set, ask the user to specify one.
|
||
|
|
if len(defaults) == 0 {
|
||
|
|
return fmt.Errorf("please specify target")
|
||
|
|
}
|
||
|
|
|
||
|
|
// One default remaining.
|
||
|
|
return SelectTarget(defaults[0]).Apply(ctx, b)
|
||
|
|
}
|