mirror of https://github.com/databricks/cli.git
64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
package mutator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config/variable"
|
|
)
|
|
|
|
const bundleVarPrefix = "BUNDLE_VAR_"
|
|
|
|
type setVariables struct{}
|
|
|
|
func SetVariables() bundle.Mutator {
|
|
return &setVariables{}
|
|
}
|
|
|
|
func (m *setVariables) Name() string {
|
|
return "SetVariables"
|
|
}
|
|
|
|
func setVariable(v *variable.Variable, name string) error {
|
|
// case: variable already has value initialized, so skip
|
|
if v.HasValue() {
|
|
return nil
|
|
}
|
|
|
|
// case: read and set variable value from process environment
|
|
envVarName := bundleVarPrefix + name
|
|
if val, ok := os.LookupEnv(envVarName); ok {
|
|
err := v.Set(val)
|
|
if err != nil {
|
|
return fmt.Errorf(`failed to assign value "%s" to variable %s from environment variable %s with error: %w`, val, name, envVarName, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// case: Set the variable to its default value
|
|
if v.HasDefault() {
|
|
err := v.Set(*v.Default)
|
|
if err != nil {
|
|
return fmt.Errorf(`failed to assign default value from config "%s" to variable %s with error: %w`, *v.Default, name, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// We should have had a value to set for the variable at this point.
|
|
// TODO: use cmdio to request values for unassigned variables if current
|
|
// terminal is a tty. Tracked in https://github.com/databricks/cli/issues/379
|
|
return fmt.Errorf(`no value assigned to required variable %s. Assignment can be done through the "--var" flag or by setting the %s environment variable`, name, bundleVarPrefix+name)
|
|
}
|
|
|
|
func (m *setVariables) Apply(ctx context.Context, b *bundle.Bundle) ([]bundle.Mutator, error) {
|
|
for name, variable := range b.Config.Variables {
|
|
err := setVariable(variable, name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|