package yqlib import ( "container/list" yaml "gopkg.in/yaml.v3" ) // A yaml expression evaluator that runs the expression once against all files/nodes in memory. type Evaluator interface { EvaluateFiles(expression string, filenames []string, printer Printer) error // EvaluateNodes takes an expression and one or more yaml nodes, returning a list of matching candidate nodes EvaluateNodes(expression string, nodes ...*yaml.Node) (*list.List, error) // EvaluateCandidateNodes takes an expression and list of candidate nodes, returning a list of matching candidate nodes EvaluateCandidateNodes(expression string, inputCandidateNodes *list.List) (*list.List, error) } type allAtOnceEvaluator struct { treeNavigator DataTreeNavigator treeCreator ExpressionParser } func NewAllAtOnceEvaluator() Evaluator { return &allAtOnceEvaluator{treeNavigator: NewDataTreeNavigator(), treeCreator: NewExpressionParser()} } func (e *allAtOnceEvaluator) EvaluateNodes(expression string, nodes ...*yaml.Node) (*list.List, error) { inputCandidates := list.New() for _, node := range nodes { inputCandidates.PushBack(&CandidateNode{Node: node}) } return e.EvaluateCandidateNodes(expression, inputCandidates) } func (e *allAtOnceEvaluator) EvaluateCandidateNodes(expression string, inputCandidates *list.List) (*list.List, error) { node, err := e.treeCreator.ParseExpression(expression) if err != nil { return nil, err } return e.treeNavigator.GetMatchingNodes(inputCandidates, node) } func (e *allAtOnceEvaluator) EvaluateFiles(expression string, filenames []string, printer Printer) error { fileIndex := 0 var allDocuments *list.List = list.New() for _, filename := range filenames { reader, err := readStream(filename) if err != nil { return err } fileDocuments, err := readDocuments(reader, filename, fileIndex) if err != nil { return err } allDocuments.PushBackList(fileDocuments) fileIndex = fileIndex + 1 } matches, err := e.EvaluateCandidateNodes(expression, allDocuments) if err != nil { return err } return printer.PrintResults(matches) }