yq/pkg/yqlib/operator_file.go

39 lines
1.0 KiB
Go
Raw Normal View History

2020-11-20 04:50:15 +00:00
package yqlib
import (
"container/list"
"fmt"
yaml "gopkg.in/yaml.v3"
)
func getFilenameOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
2020-11-20 04:50:15 +00:00
log.Debugf("GetFilename")
var results = list.New()
for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
2020-11-20 04:50:15 +00:00
candidate := el.Value.(*CandidateNode)
node := &yaml.Node{Kind: yaml.ScalarNode, Value: candidate.Filename, Tag: "!!str"}
result := candidate.CreateChild(nil, node)
results.PushBack(result)
2020-11-20 04:50:15 +00:00
}
return context.ChildContext(results), nil
2020-11-20 04:50:15 +00:00
}
func getFileIndexOperator(d *dataTreeNavigator, context Context, expressionNode *ExpressionNode) (Context, error) {
2020-11-20 04:50:15 +00:00
log.Debugf("GetFileIndex")
var results = list.New()
for el := context.MatchingNodes.Front(); el != nil; el = el.Next() {
2020-11-20 04:50:15 +00:00
candidate := el.Value.(*CandidateNode)
node := &yaml.Node{Kind: yaml.ScalarNode, Value: fmt.Sprintf("%v", candidate.FileIndex), Tag: "!!int"}
result := candidate.CreateChild(nil, node)
results.PushBack(result)
2020-11-20 04:50:15 +00:00
}
return context.ChildContext(results), nil
2020-11-20 04:50:15 +00:00
}