mirror of https://github.com/mikefarah/yq.git
111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"container/list"
|
|
"fmt"
|
|
"time"
|
|
|
|
logging "gopkg.in/op/go-logging.v1"
|
|
)
|
|
|
|
type Context struct {
|
|
MatchingNodes *list.List
|
|
Variables map[string]*list.List
|
|
DontAutoCreate bool
|
|
datetimeLayout string
|
|
}
|
|
|
|
func (n *Context) SingleReadonlyChildContext(candidate *CandidateNode) Context {
|
|
list := list.New()
|
|
list.PushBack(candidate)
|
|
newContext := n.ChildContext(list)
|
|
newContext.DontAutoCreate = true
|
|
return newContext
|
|
}
|
|
|
|
func (n *Context) SingleChildContext(candidate *CandidateNode) Context {
|
|
list := list.New()
|
|
list.PushBack(candidate)
|
|
return n.ChildContext(list)
|
|
}
|
|
|
|
func (n *Context) SetDateTimeLayout(newDateTimeLayout string) {
|
|
n.datetimeLayout = newDateTimeLayout
|
|
}
|
|
|
|
func (n *Context) GetDateTimeLayout() string {
|
|
if n.datetimeLayout != "" {
|
|
return n.datetimeLayout
|
|
}
|
|
return time.RFC3339
|
|
}
|
|
|
|
func (n *Context) GetVariable(name string) *list.List {
|
|
if n.Variables == nil {
|
|
return nil
|
|
}
|
|
return n.Variables[name]
|
|
}
|
|
|
|
func (n *Context) SetVariable(name string, value *list.List) {
|
|
if n.Variables == nil {
|
|
n.Variables = make(map[string]*list.List)
|
|
}
|
|
n.Variables[name] = value
|
|
}
|
|
|
|
func (n *Context) ChildContext(results *list.List) Context {
|
|
clone := Context{DontAutoCreate: n.DontAutoCreate, datetimeLayout: n.datetimeLayout}
|
|
clone.Variables = make(map[string]*list.List)
|
|
for variableKey, originalValueList := range n.Variables {
|
|
|
|
variableCopyList := list.New()
|
|
for el := originalValueList.Front(); el != nil; el = el.Next() {
|
|
// note that we dont make a copy of the candidate node
|
|
// this is so the 'ref' operator can work correctly.
|
|
clonedNode := el.Value.(*CandidateNode)
|
|
variableCopyList.PushBack(clonedNode)
|
|
}
|
|
|
|
clone.Variables[variableKey] = variableCopyList
|
|
}
|
|
|
|
clone.MatchingNodes = results
|
|
return clone
|
|
}
|
|
|
|
func (n *Context) ToString() string {
|
|
if !log.IsEnabledFor(logging.DEBUG) {
|
|
return ""
|
|
}
|
|
result := fmt.Sprintf("Context\nDontAutoCreate: %v\n", n.DontAutoCreate)
|
|
return result + NodesToString(n.MatchingNodes)
|
|
}
|
|
|
|
func (n *Context) DeepClone() Context {
|
|
|
|
clonedContent := list.New()
|
|
for el := n.MatchingNodes.Front(); el != nil; el = el.Next() {
|
|
clonedNode := el.Value.(*CandidateNode).Copy()
|
|
clonedContent.PushBack(clonedNode)
|
|
}
|
|
|
|
return n.ChildContext(clonedContent)
|
|
}
|
|
|
|
func (n *Context) Clone() Context {
|
|
return n.ChildContext(n.MatchingNodes)
|
|
}
|
|
|
|
func (n *Context) ReadOnlyClone() Context {
|
|
clone := n.Clone()
|
|
clone.DontAutoCreate = true
|
|
return clone
|
|
}
|
|
|
|
func (n *Context) WritableClone() Context {
|
|
clone := n.Clone()
|
|
clone.DontAutoCreate = false
|
|
return clone
|
|
}
|