18 lines
285 B
Go
18 lines
285 B
Go
package utils
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func TryFlagFromEnv(cmd *cobra.Command, flag string, envvar string, required bool) {
|
|
if value, ok := os.LookupEnv(envvar); ok {
|
|
_ = cmd.Flags().Set(flag, value)
|
|
return
|
|
}
|
|
if required {
|
|
cmd.MarkFlagRequired(flag)
|
|
}
|
|
}
|