mirror of https://github.com/databricks/cli.git
32 lines
567 B
Go
32 lines
567 B
Go
|
package bundle
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
const envName = "DATABRICKS_BUNDLE_ENV"
|
||
|
|
|
||
|
|
const defaultEnvironment = "default"
|
||
|
|
|
||
|
|
// getEnvironment returns the name of the environment to operate in.
|
||
|
|
func getEnvironment(cmd *cobra.Command) (value string) {
|
||
|
|
// The command line flag takes precedence.
|
||
|
|
flag := cmd.Flag("environment")
|
||
|
|
if flag != nil {
|
||
|
|
value = flag.Value.String()
|
||
|
|
if value != "" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// If it's not set, use the environment variable.
|
||
|
|
value = os.Getenv(envName)
|
||
|
|
if value != "" {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
return defaultEnvironment
|
||
|
|
}
|