mirror of https://github.com/mikefarah/yq.git
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package treeops
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/elliotchance/orderedmap"
|
|
"gopkg.in/op/go-logging.v1"
|
|
)
|
|
|
|
type dataTreeNavigator struct {
|
|
navigationPrefs NavigationPrefs
|
|
}
|
|
|
|
type NavigationPrefs struct {
|
|
FollowAlias bool
|
|
}
|
|
|
|
type DataTreeNavigator interface {
|
|
GetMatchingNodes(matchingNodes []*CandidateNode, pathNode *PathTreeNode) ([]*CandidateNode, error)
|
|
}
|
|
|
|
func NewDataTreeNavigator(navigationPrefs NavigationPrefs) DataTreeNavigator {
|
|
return &dataTreeNavigator{navigationPrefs}
|
|
}
|
|
|
|
func (d *dataTreeNavigator) GetMatchingNodes(matchingNodes []*CandidateNode, pathNode *PathTreeNode) ([]*CandidateNode, error) {
|
|
var matchingNodeMap = orderedmap.NewOrderedMap()
|
|
|
|
for _, n := range matchingNodes {
|
|
matchingNodeMap.Set(n.GetKey(), n)
|
|
}
|
|
|
|
matchedNodes, err := d.getMatchingNodes(matchingNodeMap, pathNode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
values := make([]*CandidateNode, 0, matchedNodes.Len())
|
|
|
|
for el := matchedNodes.Front(); el != nil; el = el.Next() {
|
|
values = append(values, el.Value.(*CandidateNode))
|
|
}
|
|
return values, nil
|
|
}
|
|
|
|
func (d *dataTreeNavigator) getMatchingNodes(matchingNodes *orderedmap.OrderedMap, pathNode *PathTreeNode) (*orderedmap.OrderedMap, error) {
|
|
if pathNode == nil {
|
|
log.Debugf("getMatchingNodes - nothing to do")
|
|
return matchingNodes, nil
|
|
}
|
|
log.Debugf("Processing Path: %v", pathNode.PathElement.toString())
|
|
if log.IsEnabledFor(logging.DEBUG) {
|
|
for el := matchingNodes.Front(); el != nil; el = el.Next() {
|
|
log.Debug(NodeToString(el.Value.(*CandidateNode)))
|
|
}
|
|
}
|
|
log.Debug(">>")
|
|
handler := pathNode.PathElement.OperationType.Handler
|
|
if handler != nil {
|
|
return handler(d, matchingNodes, pathNode)
|
|
}
|
|
return nil, fmt.Errorf("Unknown operator %v", pathNode.PathElement.OperationType)
|
|
|
|
}
|