mirror of
https://github.com/mikefarah/yq.git
synced 2026-06-27 15:37:47 +00:00
7d8d3ab902
* Initial plan * Replace gopkg.in/op/go-logging.v1 with log/slog Co-authored-by: mikefarah <1151925+mikefarah@users.noreply.github.com> Agent-Logs-Url: https://github.com/mikefarah/yq/sessions/aa9c12f4-21b9-4633-9868-6b56585b247f --------- Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: mikefarah <1151925+mikefarah@users.noreply.github.com>
110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"container/list"
|
|
"fmt"
|
|
"log/slog"
|
|
"time"
|
|
)
|
|
|
|
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(slog.LevelDebug) {
|
|
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
|
|
}
|