yq/pkg/yqlib/operator_env.go

50 lines
1.1 KiB
Go
Raw Normal View History

2021-01-08 10:07:46 +00:00
package yqlib
import (
2021-01-11 03:38:53 +00:00
"fmt"
2021-01-08 10:07:46 +00:00
"os"
2021-01-09 01:06:19 +00:00
"strings"
2021-01-08 10:07:46 +00:00
yaml "gopkg.in/yaml.v3"
)
type envOpPreferences struct {
2021-01-09 01:06:19 +00:00
StringValue bool
}
2021-01-08 10:07:46 +00:00
func envOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
2021-01-12 23:18:53 +00:00
envName := expressionNode.Operation.CandidateNode.Node.Value
2021-01-08 10:07:46 +00:00
log.Debug("EnvOperator, env name:", envName)
rawValue := os.Getenv(envName)
preferences := expressionNode.Operation.Preferences.(envOpPreferences)
2021-01-09 01:06:19 +00:00
var node *yaml.Node
if preferences.StringValue {
node = &yaml.Node{
Kind: yaml.ScalarNode,
Tag: "!!str",
Value: rawValue,
}
2021-01-11 03:38:53 +00:00
} else if rawValue == "" {
return Context{}, fmt.Errorf("Value for env variable '%v' not provided in env()", envName)
2021-01-09 01:06:19 +00:00
} else {
var dataBucket yaml.Node
decoder := yaml.NewDecoder(strings.NewReader(rawValue))
errorReading := decoder.Decode(&dataBucket)
if errorReading != nil {
return Context{}, errorReading
2021-01-09 01:06:19 +00:00
}
//first node is a doc
2021-01-12 23:00:51 +00:00
node = unwrapDoc(&dataBucket)
2021-01-09 01:06:19 +00:00
}
log.Debug("ENV tag", node.Tag)
log.Debug("ENV value", node.Value)
log.Debug("ENV Kind", node.Kind)
target := &CandidateNode{Node: node}
2021-01-08 10:07:46 +00:00
return context.SingleChildContext(target), nil
2021-01-08 10:07:46 +00:00
}