mirror of
https://github.com/mikefarah/yq.git
synced 2024-11-14 15:18:06 +00:00
132 lines
3.5 KiB
Go
132 lines
3.5 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/jinzhu/copier"
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type CandidateNode struct {
|
|
Node *yaml.Node // the actual node
|
|
Path []interface{} /// the path we took to get to this node
|
|
Document uint // the document index of this node
|
|
Filename string
|
|
FileIndex int
|
|
// when performing op against all nodes given, this will treat all the nodes as one
|
|
// (e.g. top level cross document merge). This property does not propegate to child nodes.
|
|
EvaluateTogether bool
|
|
}
|
|
|
|
func (n *CandidateNode) GetKey() string {
|
|
return fmt.Sprintf("%v - %v", n.Document, n.Path)
|
|
}
|
|
|
|
func (n *CandidateNode) CreateChild(path interface{}, node *yaml.Node) *CandidateNode {
|
|
return &CandidateNode{
|
|
Node: node,
|
|
Path: n.createChildPath(path),
|
|
Document: n.Document,
|
|
Filename: n.Filename,
|
|
FileIndex: n.FileIndex,
|
|
}
|
|
}
|
|
|
|
func (n *CandidateNode) createChildPath(path interface{}) []interface{} {
|
|
if path == nil {
|
|
newPath := make([]interface{}, len(n.Path))
|
|
copy(newPath, n.Path)
|
|
return newPath
|
|
}
|
|
|
|
//don't use append as they may actually modify the path of the orignal node!
|
|
newPath := make([]interface{}, len(n.Path)+1)
|
|
copy(newPath, n.Path)
|
|
newPath[len(n.Path)] = path
|
|
return newPath
|
|
}
|
|
|
|
func (n *CandidateNode) Copy() (*CandidateNode, error) {
|
|
clone := &CandidateNode{}
|
|
err := copier.Copy(clone, n)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return clone, nil
|
|
}
|
|
|
|
// updates this candidate from the given candidate node
|
|
func (n *CandidateNode) UpdateFrom(other *CandidateNode) {
|
|
|
|
n.UpdateAttributesFrom(other)
|
|
n.Node.Content = other.Node.Content
|
|
n.Node.Value = other.Node.Value
|
|
n.Node.Alias = other.Node.Alias
|
|
n.Node.Anchor = other.Node.Anchor
|
|
}
|
|
|
|
func (n *CandidateNode) UpdateAttributesFrom(other *CandidateNode) {
|
|
if n.Node.Kind != other.Node.Kind {
|
|
// clear out the contents when switching to a different type
|
|
// e.g. map to array
|
|
n.Node.Content = make([]*yaml.Node, 0)
|
|
n.Node.Value = ""
|
|
}
|
|
n.Node.Kind = other.Node.Kind
|
|
n.Node.Tag = other.Node.Tag
|
|
|
|
// merge will pickup the style of the new thing
|
|
// when autocreating nodes
|
|
if n.Node.Style == 0 {
|
|
n.Node.Style = other.Node.Style
|
|
}
|
|
n.Node.FootComment = n.Node.FootComment + other.Node.FootComment
|
|
n.Node.HeadComment = n.Node.HeadComment + other.Node.HeadComment
|
|
n.Node.LineComment = n.Node.LineComment + other.Node.LineComment
|
|
}
|
|
|
|
func (n *CandidateNode) PathStackToString() string {
|
|
return mergePathStackToString(n.Path)
|
|
}
|
|
|
|
func mergePathStackToString(pathStack []interface{}) string {
|
|
var sb strings.Builder
|
|
for index, path := range pathStack {
|
|
switch path.(type) {
|
|
case int, int64:
|
|
// if arrayMergeStrategy == AppendArrayMergeStrategy {
|
|
// sb.WriteString("[+]")
|
|
// } else {
|
|
sb.WriteString(fmt.Sprintf("[%v]", path))
|
|
// }
|
|
|
|
default:
|
|
s := fmt.Sprintf("%v", path)
|
|
var _, errParsingInt = strconv.ParseInt(s, 10, 64) // nolint
|
|
|
|
hasSpecial := strings.Contains(s, ".") || strings.Contains(s, "[") || strings.Contains(s, "]") || strings.Contains(s, "\"")
|
|
hasDoubleQuotes := strings.Contains(s, "\"")
|
|
wrappingCharacterStart := "\""
|
|
wrappingCharacterEnd := "\""
|
|
if hasDoubleQuotes {
|
|
wrappingCharacterStart = "("
|
|
wrappingCharacterEnd = ")"
|
|
}
|
|
if hasSpecial || errParsingInt == nil {
|
|
sb.WriteString(wrappingCharacterStart)
|
|
}
|
|
sb.WriteString(s)
|
|
if hasSpecial || errParsingInt == nil {
|
|
sb.WriteString(wrappingCharacterEnd)
|
|
}
|
|
}
|
|
|
|
if index < len(pathStack)-1 {
|
|
sb.WriteString(".")
|
|
}
|
|
}
|
|
return sb.String()
|
|
}
|