yq/pkg/yqlib/operator_has.go

55 lines
1.5 KiB
Go
Raw Normal View History

2020-11-24 00:38:39 +00:00
package yqlib
import (
"container/list"
"strconv"
yaml "gopkg.in/yaml.v3"
)
func hasOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
2020-11-24 00:38:39 +00:00
log.Debugf("-- hasOperation")
var results = list.New()
rhs, err := d.GetMatchingNodes(context, expressionNode.Rhs)
wanted := rhs.MatchingNodes.Front().Value.(*CandidateNode).Node
2020-11-24 00:38:39 +00:00
wantedKey := wanted.Value
if err != nil {
return Context{}, err
2020-11-24 00:38:39 +00:00
}
for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
2020-11-24 00:38:39 +00:00
candidate := el.Value.(*CandidateNode)
// grab the first value
2021-01-12 23:00:51 +00:00
candidateNode := unwrapDoc(candidate.Node)
2021-01-11 22:30:24 +00:00
var contents = candidateNode.Content
switch candidateNode.Kind {
2020-11-24 00:38:39 +00:00
case yaml.MappingNode:
candidateHasKey := false
for index := 0; index < len(contents) && !candidateHasKey; index = index + 2 {
key := contents[index]
if key.Value == wantedKey {
candidateHasKey = true
}
}
results.PushBack(createBooleanCandidate(candidate, candidateHasKey))
case yaml.SequenceNode:
candidateHasKey := false
if wanted.Tag == "!!int" {
var number, errParsingInt = strconv.ParseInt(wantedKey, 10, 64) // nolint
if errParsingInt != nil {
return Context{}, errParsingInt
2020-11-24 00:38:39 +00:00
}
candidateHasKey = int64(len(contents)) > number
}
results.PushBack(createBooleanCandidate(candidate, candidateHasKey))
default:
results.PushBack(createBooleanCandidate(candidate, false))
}
}
return context.ChildContext(results), nil
2020-11-24 00:38:39 +00:00
}