37 lines
1 KiB
Go
37 lines
1 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package args
|
|
|
|
import "regexp"
|
|
|
|
var (
|
|
envRe = regexp.MustCompile(`\${[a-zA-Z0-9_\-\.]+}`)
|
|
)
|
|
|
|
// ReplaceEnv takes an arg and replaces all occurrences of environment variables.
|
|
// If the variable is found in the passed map it is replaced, otherwise the
|
|
// original string is returned.
|
|
func ReplaceEnv(arg string, environments ...map[string]string) string {
|
|
return envRe.ReplaceAllStringFunc(arg, func(arg string) string {
|
|
stripped := arg[2 : len(arg)-1]
|
|
for _, env := range environments {
|
|
if value, ok := env[stripped]; ok {
|
|
return value
|
|
}
|
|
}
|
|
|
|
return arg
|
|
})
|
|
}
|
|
|
|
// ReplaceEnvWithPlaceHolder replaces all occurrences of environment variables with the placeholder string.
|
|
func ReplaceEnvWithPlaceHolder(arg string, placeholder string) string {
|
|
return envRe.ReplaceAllString(arg, placeholder)
|
|
}
|
|
|
|
// ContainsEnv takes an arg and returns true if if contains an environment variable reference
|
|
func ContainsEnv(arg string) bool {
|
|
return envRe.MatchString(arg)
|
|
}
|