mirror of https://github.com/mikefarah/yq.git
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"container/list"
|
|
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func isTruthy(c *CandidateNode) (bool, error) {
|
|
node := UnwrapDoc(c.Node)
|
|
value := true
|
|
|
|
if node.Tag == "!!null" {
|
|
return false, nil
|
|
}
|
|
if node.Kind == yaml.ScalarNode && node.Tag == "!!bool" {
|
|
errDecoding := node.Decode(&value)
|
|
if errDecoding != nil {
|
|
return false, errDecoding
|
|
}
|
|
|
|
}
|
|
return value, nil
|
|
}
|
|
|
|
type boolOp func(bool, bool) bool
|
|
|
|
func performBoolOp(op boolOp) func(d *dataTreeNavigator, lhs *CandidateNode, rhs *CandidateNode) (*CandidateNode, error) {
|
|
return func(d *dataTreeNavigator, lhs *CandidateNode, rhs *CandidateNode) (*CandidateNode, error) {
|
|
lhs.Node = UnwrapDoc(lhs.Node)
|
|
rhs.Node = UnwrapDoc(rhs.Node)
|
|
|
|
lhsTrue, errDecoding := isTruthy(lhs)
|
|
if errDecoding != nil {
|
|
return nil, errDecoding
|
|
}
|
|
|
|
rhsTrue, errDecoding := isTruthy(rhs)
|
|
if errDecoding != nil {
|
|
return nil, errDecoding
|
|
}
|
|
|
|
return createBooleanCandidate(lhs, op(lhsTrue, rhsTrue)), nil
|
|
}
|
|
}
|
|
|
|
func OrOperator(d *dataTreeNavigator, matchingNodes *list.List, pathNode *PathTreeNode) (*list.List, error) {
|
|
log.Debugf("-- orOp")
|
|
return crossFunction(d, matchingNodes, pathNode, performBoolOp(
|
|
func(b1 bool, b2 bool) bool {
|
|
return b1 || b2
|
|
}))
|
|
}
|
|
|
|
func AndOperator(d *dataTreeNavigator, matchingNodes *list.List, pathNode *PathTreeNode) (*list.List, error) {
|
|
log.Debugf("-- AndOp")
|
|
return crossFunction(d, matchingNodes, pathNode, performBoolOp(
|
|
func(b1 bool, b2 bool) bool {
|
|
return b1 && b2
|
|
}))
|
|
}
|
|
|
|
func NotOperator(d *dataTreeNavigator, matchMap *list.List, pathNode *PathTreeNode) (*list.List, error) {
|
|
log.Debugf("-- notOperation")
|
|
var results = list.New()
|
|
|
|
for el := matchMap.Front(); el != nil; el = el.Next() {
|
|
candidate := el.Value.(*CandidateNode)
|
|
log.Debug("notOperation checking %v", candidate)
|
|
truthy, errDecoding := isTruthy(candidate)
|
|
if errDecoding != nil {
|
|
return nil, errDecoding
|
|
}
|
|
result := createBooleanCandidate(candidate, !truthy)
|
|
results.PushBack(result)
|
|
}
|
|
return results, nil
|
|
}
|