mirror of https://github.com/mikefarah/yq.git
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
logging "gopkg.in/op/go-logging.v1"
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type UpdateCommand struct {
|
|
Command string
|
|
Path string
|
|
Value *yaml.Node
|
|
}
|
|
|
|
type YqLib interface {
|
|
DebugNode(node *yaml.Node)
|
|
Get(rootNode *yaml.Node, path string) ([]MatchingNode, error)
|
|
Update(rootNode *yaml.Node, updateCommand UpdateCommand) error
|
|
New(path string) yaml.Node
|
|
}
|
|
|
|
type lib struct {
|
|
navigator DataNavigator
|
|
parser PathParser
|
|
log *logging.Logger
|
|
}
|
|
|
|
func NewYqLib(l *logging.Logger) YqLib {
|
|
return &lib{
|
|
parser: NewPathParser(),
|
|
log: l,
|
|
}
|
|
}
|
|
|
|
func (l *lib) DebugNode(node *yaml.Node) {
|
|
navigator := NewDataNavigator(l.log, false)
|
|
navigator.DebugNode(node)
|
|
}
|
|
|
|
func (l *lib) Get(rootNode *yaml.Node, path string) ([]MatchingNode, error) {
|
|
var paths = l.parser.ParsePath(path)
|
|
navigator := NewDataNavigator(l.log, true)
|
|
return navigator.Get(rootNode, paths)
|
|
}
|
|
|
|
func (l *lib) New(path string) yaml.Node {
|
|
var paths = l.parser.ParsePath(path)
|
|
navigator := NewDataNavigator(l.log, false)
|
|
newNode := yaml.Node{Kind: navigator.GuessKind(paths, 0)}
|
|
return newNode
|
|
}
|
|
|
|
func (l *lib) Update(rootNode *yaml.Node, updateCommand UpdateCommand) error {
|
|
navigator := NewDataNavigator(l.log, false)
|
|
l.log.Debugf("%v to %v", updateCommand.Command, updateCommand.Path)
|
|
switch updateCommand.Command {
|
|
case "update":
|
|
var paths = l.parser.ParsePath(updateCommand.Path)
|
|
return navigator.Update(rootNode, paths, updateCommand.Value)
|
|
case "delete":
|
|
var paths = l.parser.ParsePath(updateCommand.Path)
|
|
return navigator.Delete(rootNode, paths)
|
|
default:
|
|
return fmt.Errorf("Unknown command %v", updateCommand.Command)
|
|
}
|
|
|
|
}
|