53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package view
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// Env represent K9s and K8s available environment variables.
|
|
type Env map[string]string
|
|
|
|
// EnvRX match $XXX custom arg.
|
|
var envRX = regexp.MustCompile(`\$(\!?[\w|\d|\-|]+)`)
|
|
|
|
// Substitute replaces env variable keys from in a string with their corresponding values.
|
|
func (e Env) Substitute(arg string) (string, error) {
|
|
kk := envRX.FindAllString(arg, -1)
|
|
if len(kk) == 0 {
|
|
return arg, nil
|
|
}
|
|
|
|
// To prevent the substitution starts with the shorter environment variable,
|
|
// sort with the length of the found environment variables.
|
|
sort.Slice(kk, func(i, j int) bool {
|
|
return len(kk[i]) > len(kk[j])
|
|
})
|
|
|
|
for _, k := range kk {
|
|
key, inverse := k[1:], false
|
|
if key[0] == '!' {
|
|
key, inverse = key[1:], true
|
|
}
|
|
v, ok := e[strings.ToUpper(key)]
|
|
if !ok {
|
|
log.Warn().Msgf("no k9s environment matching key %q:%q", k, key)
|
|
continue
|
|
}
|
|
if b, err := strconv.ParseBool(v); err == nil {
|
|
if inverse {
|
|
b = !b
|
|
}
|
|
v = fmt.Sprintf("%t", b)
|
|
}
|
|
arg = strings.Replace(arg, k, v, -1)
|
|
}
|
|
|
|
return arg, nil
|
|
}
|