mirror of https://github.com/mikefarah/yq.git
74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func DeleteNavigationStrategy(pathElementToDelete string) NavigationStrategy {
|
|
parser := NewPathParser()
|
|
return &NavigationStrategyImpl{
|
|
visitedNodes: []*NodeContext{},
|
|
pathParser: parser,
|
|
followAlias: func(nodeContext NodeContext) bool {
|
|
return false
|
|
},
|
|
autoCreateMap: func(nodeContext NodeContext) bool {
|
|
return true
|
|
},
|
|
visit: func(nodeContext NodeContext) error {
|
|
node := nodeContext.Node
|
|
log.Debug("need to find and delete %v in here", pathElementToDelete)
|
|
DebugNode(node)
|
|
if node.Kind == yaml.SequenceNode {
|
|
newContent := deleteFromArray(parser, node.Content, nodeContext.PathStack, pathElementToDelete)
|
|
node.Content = newContent
|
|
} else if node.Kind == yaml.MappingNode {
|
|
node.Content = deleteFromMap(parser, node.Content, nodeContext.PathStack, pathElementToDelete)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
}
|
|
func deleteFromMap(pathParser PathParser, contents []*yaml.Node, pathStack []interface{}, pathElementToDelete string) []*yaml.Node {
|
|
newContents := make([]*yaml.Node, 0)
|
|
for index := 0; index < len(contents); index = index + 2 {
|
|
keyNode := contents[index]
|
|
valueNode := contents[index+1]
|
|
if !pathParser.MatchesNextPathElement(NewNodeContext(keyNode, pathElementToDelete, []string{}, pathStack), keyNode.Value) {
|
|
log.Debug("adding node %v", keyNode.Value)
|
|
newContents = append(newContents, keyNode, valueNode)
|
|
} else {
|
|
log.Debug("skipping node %v", keyNode.Value)
|
|
}
|
|
}
|
|
return newContents
|
|
}
|
|
|
|
func deleteFromArray(pathParser PathParser, content []*yaml.Node, pathStack []interface{}, pathElementToDelete string) []*yaml.Node {
|
|
|
|
var indexToDelete, err = strconv.ParseInt(pathElementToDelete, 10, 64) // nolint
|
|
if err == nil {
|
|
return deleteIndexInArray(content, indexToDelete)
|
|
}
|
|
log.Debug("%v is not a numeric index, finding matching patterns", pathElementToDelete)
|
|
var newArray = make([]*yaml.Node, 0)
|
|
|
|
for _, childValue := range content {
|
|
if !pathParser.MatchesNextPathElement(NewNodeContext(childValue, pathElementToDelete, []string{}, pathStack), childValue.Value) {
|
|
newArray = append(newArray, childValue)
|
|
}
|
|
}
|
|
return newArray
|
|
}
|
|
|
|
func deleteIndexInArray(content []*yaml.Node, index int64) []*yaml.Node {
|
|
log.Debug("deleting index %v in array", index)
|
|
if index >= int64(len(content)) {
|
|
log.Debug("index %v is greater than content length %v", index, len(content))
|
|
return content
|
|
}
|
|
return append(content[:index], content[index+1:]...)
|
|
}
|