-
Notifications
You must be signed in to change notification settings - Fork 16
/
env.go
79 lines (70 loc) · 1.54 KB
/
env.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package dynaml
import (
"os"
"strconv"
"strings"
"github.com/mandelsoft/spiff/yaml"
)
var environ []string = os.Environ()
func ReloadEnv() {
environ = os.Environ()
}
func getenv(name string) (string, bool) {
name += "="
for _, s := range environ {
if strings.HasPrefix(s, name) {
return s[len(name):], true
}
}
return "", false
}
func func_env(arguments []interface{}, binding Binding) (interface{}, EvaluationInfo, bool) {
info := DefaultInfo()
if len(arguments) < 1 {
return nil, info, false
}
args := make([]string, 0)
for i, arg := range arguments {
switch v := arg.(type) {
case string:
args = append(args, v)
case int64:
args = append(args, strconv.FormatInt(v, 10))
case bool:
args = append(args, strconv.FormatBool(v))
case []yaml.Node:
for _, elem := range v {
switch e := elem.Value().(type) {
case string:
args = append(args, e)
case int64:
args = append(args, strconv.FormatInt(e, 10))
case bool:
args = append(args, strconv.FormatBool(e))
default:
return info.Error("elements of list(arg %d) to join must be simple values", i)
}
}
case nil:
default:
return info.Error("env argument %d must be simple value or list", i)
}
}
if len(args) == 1 {
s, ok := getenv(args[0])
if ok {
return s, info, ok
} else {
return info.Error("environment variable '%s' not set", args[0])
}
} else {
m := make(map[string]yaml.Node)
for _, n := range args {
s, ok := getenv(n)
if ok {
m[n] = NewNode(s, nil)
}
}
return m, info, true
}
}