yq/pkg/yqlib/data_tree_navigator.go

66 lines
1.8 KiB
Go
Raw Normal View History

2020-11-03 23:48:43 +00:00
package yqlib
2020-10-08 23:59:03 +00:00
2020-10-09 01:04:19 +00:00
import (
2020-10-10 04:00:39 +00:00
"fmt"
2020-11-22 00:56:28 +00:00
logging "gopkg.in/op/go-logging.v1"
yaml "gopkg.in/yaml.v3"
2020-10-09 01:04:19 +00:00
)
2020-10-08 23:59:03 +00:00
type DataTreeNavigator interface {
// given the context and a expressionNode,
// this will process the against the given expressionNode and return
// a new context of matching candidates
GetMatchingNodes(context Context, expressionNode *ExpressionNode) (Context, error)
DeeplyAssign(context Context, path []interface{}, rhsNode *yaml.Node) error
2020-10-08 23:59:03 +00:00
}
2020-11-22 00:56:28 +00:00
type dataTreeNavigator struct {
}
func NewDataTreeNavigator() DataTreeNavigator {
return &dataTreeNavigator{}
2020-10-08 23:59:03 +00:00
}
func (d *dataTreeNavigator) DeeplyAssign(context Context, path []interface{}, rhsNode *yaml.Node) error {
rhsCandidateNode := &CandidateNode{
Path: path,
Node: rhsNode,
}
assignmentOp := &Operation{OperationType: assignOpType, Preferences: assignPreferences{}}
rhsOp := &Operation{OperationType: valueOpType, CandidateNode: rhsCandidateNode}
assignmentOpNode := &ExpressionNode{
Operation: assignmentOp,
LHS: createTraversalTree(path, traversePreferences{}, false),
RHS: &ExpressionNode{Operation: rhsOp},
}
_, err := d.GetMatchingNodes(context, assignmentOpNode)
return err
}
func (d *dataTreeNavigator) GetMatchingNodes(context Context, expressionNode *ExpressionNode) (Context, error) {
2021-01-12 23:18:53 +00:00
if expressionNode == nil {
2020-10-13 01:51:37 +00:00
log.Debugf("getMatchingNodes - nothing to do")
return context, nil
2020-10-13 01:51:37 +00:00
}
2021-01-12 23:18:53 +00:00
log.Debugf("Processing Op: %v", expressionNode.Operation.toString())
2020-10-17 11:10:47 +00:00
if log.IsEnabledFor(logging.DEBUG) {
for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
2020-10-17 11:10:47 +00:00
log.Debug(NodeToString(el.Value.(*CandidateNode)))
}
}
log.Debug(">>")
2021-01-12 23:18:53 +00:00
handler := expressionNode.Operation.OperationType.Handler
2020-10-20 02:53:26 +00:00
if handler != nil {
return handler(d, context, expressionNode)
2020-10-08 23:59:03 +00:00
}
return Context{}, fmt.Errorf("Unknown operator %v", expressionNode.Operation.OperationType)
2020-10-08 23:59:03 +00:00
}