22 lines
447 B
Go
22 lines
447 B
Go
package cli
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// FlagOrEnv returns the flag value if set, otherwise the environment variable value,
|
|
// otherwise the fallback.
|
|
func FlagOrEnv(cmd *cobra.Command, flagName, envKey, fallback string) string {
|
|
if cmd.Flags().Changed(flagName) {
|
|
val, _ := cmd.Flags().GetString(flagName)
|
|
return val
|
|
}
|
|
if v := strings.TrimSpace(os.Getenv(envKey)); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|