mirror of
https://github.com/mikefarah/yq.git
synced 2024-12-19 20:19:04 +00:00
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/goccy/go-yaml/ast"
|
|
)
|
|
|
|
func (o *CandidateNode) goccyDecodeIntoChild(childNode ast.Node, anchorMap map[string]*CandidateNode) (*CandidateNode, error) {
|
|
newChild := o.CreateChild()
|
|
|
|
err := newChild.UnmarshalGoccyYAML(childNode, anchorMap)
|
|
return newChild, err
|
|
}
|
|
|
|
func (o *CandidateNode) UnmarshalGoccyYAML(node ast.Node, anchorMap map[string]*CandidateNode) error {
|
|
log.Debugf("UnmarshalYAML %v", node)
|
|
log.Debugf("UnmarshalYAML %v", node.Type().String())
|
|
|
|
o.Value = node.String()
|
|
switch node.Type() {
|
|
case ast.IntegerType:
|
|
o.Kind = ScalarNode
|
|
o.Tag = "!!int"
|
|
case ast.FloatType:
|
|
o.Kind = ScalarNode
|
|
o.Tag = "!!float"
|
|
case ast.StringType:
|
|
o.Kind = ScalarNode
|
|
o.Tag = "!!str"
|
|
case ast.TagType:
|
|
o.UnmarshalGoccyYAML(node.(*ast.TagNode).Value, anchorMap)
|
|
o.Tag = node.(*ast.TagNode).Start.Value
|
|
case ast.MappingValueType, ast.MappingType:
|
|
log.Debugf("UnmarshalYAML - a mapping node")
|
|
o.Kind = MappingNode
|
|
o.Tag = "!!map"
|
|
|
|
if node.Type() == ast.MappingType {
|
|
o.Style = FlowStyle
|
|
}
|
|
|
|
astMapIter := node.(ast.MapNode).MapRange()
|
|
for astMapIter.Next() {
|
|
log.Debug("UnmarshalYAML map entry %v", astMapIter.Key().String())
|
|
keyNode, err := o.goccyDecodeIntoChild(astMapIter.Key(), anchorMap)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
keyNode.IsMapKey = true
|
|
log.Debug("UnmarshalYAML map value %v", astMapIter.Value().String())
|
|
valueNode, err := o.goccyDecodeIntoChild(astMapIter.Value(), anchorMap)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
o.Content = append(o.Content, keyNode, valueNode)
|
|
}
|
|
case ast.SequenceType:
|
|
log.Debugf("UnmarshalYAML - a sequence node")
|
|
o.Kind = SequenceNode
|
|
o.Tag = "!!seq"
|
|
sequenceNode := node.(*ast.SequenceNode)
|
|
if sequenceNode.IsFlowStyle {
|
|
o.Style = FlowStyle
|
|
}
|
|
astSeq := sequenceNode.Values
|
|
o.Content = make([]*CandidateNode, len(astSeq))
|
|
for i := 0; i < len(astSeq); i++ {
|
|
keyNode := o.CreateChild()
|
|
keyNode.IsMapKey = true
|
|
keyNode.Tag = "!!int"
|
|
keyNode.Kind = ScalarNode
|
|
keyNode.Value = fmt.Sprintf("%v", i)
|
|
|
|
valueNode, err := o.goccyDecodeIntoChild(astSeq[i], anchorMap)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
valueNode.Key = keyNode
|
|
o.Content[i] = valueNode
|
|
}
|
|
|
|
default:
|
|
log.Debugf("UnmarshalYAML - node idea of the type!!")
|
|
}
|
|
log.Debugf("KIND: %v", o.Kind)
|
|
return nil
|
|
}
|