1 files changed, 104 insertions, 0 deletions
diff --git a/client/go/prog/spec_env.go b/client/go/prog/spec_env.go
new file mode 100644
index 00000000000..498931fe92c
--- /dev/null
+++ b/client/go/prog/spec_env.go
@@ -0,0 +1,104 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// Author: arnej
+
+package prog
+
+import (
+	"os"
+	"strings"
+
+	"github.com/vespa-engine/vespa/client/go/trace"
+)
+
+func (p *Spec) Setenv(k, v string) {
+	p.Env[k] = v
+}
+
+func (p *Spec) Getenv(k string) string {
+	if v, ok := p.Env[k]; ok {
+		return v
+	}
+	return os.Getenv(k)
+}
+
+func (spec *Spec) effectiveEnv() []string {
+	env := make(map[string]string)
+	for _, entry := range os.Environ() {
+		addInMap := func(kv string) bool {
+			for idx, elem := range kv {
+				if elem == '=' {
+					k := kv[:idx]
+					env[k] = kv
+					return true
+				}
+			}
+			return false
+		}
+		if !addInMap(entry) {
+			env[entry] = ""
+		}
+	}
+	for k, v := range spec.Env {
+		trace.Trace("add to environment:", k, "=", v)
+		env[k] = k + "=" + v
+	}
+	envv := make([]string, 0, len(env))
+	for _, v := range env {
+		envv = append(envv, v)
+	}
+	return envv
+}
+
+func (spec *Spec) considerFallback(varName, varValue string) {
+	if spec.Getenv(varName) == "" && varValue != "" {
+		spec.Setenv(varName, varValue)
+	}
+}
+
+func (spec *Spec) considerEnvFallback(targetVar, fallbackVar string) {
+	spec.considerFallback(targetVar, spec.Getenv(fallbackVar))
+}
+
+func (p *Spec) matchesListEnv(envVarName string) bool {
+	return p.matchesListString(p.Getenv(envVarName))
+}
+
+func (p *Spec) matchesListString(env string) bool {
+	if env == "all" {
+		trace.Debug(p.Program, "always matching in:", env)
+		return true
+	}
+	parts := strings.Fields(env)
+	for _, part := range parts {
+		if p.BaseName == part {
+			trace.Debug(p.Program, "has basename matching in:", env)
+			return true
+		}
+		trace.Debug("checking matching:", p.BaseName, "!=", part)
+	}
+	return false
+}
+
+func (p *Spec) valueFromListEnv(envVarName string) string {
+	return p.valueFromListString(p.Getenv(envVarName))
+}
+
+func (p *Spec) valueFromListString(env string) string {
+	parts := strings.Fields(env)
+	for _, part := range parts {
+		idx := strings.Index(part, "=")
+		if idx <= 0 {
+			trace.Trace("expected key=value, but got:", part)
+			continue
+		}
+		partName := part[:idx]
+		idx++
+		partValue := part[idx:]
+		if p.BaseName == partName || partName == "all" {
+			trace.Debug(p.Program, "has basename matching in:", env)
+			return partValue
+		}
+		trace.Debug("checking matching:", p.BaseName, "!=", part)
+	}
+	return ""
+}