mirror of
https://github.com/mikefarah/yq.git
synced 2024-11-12 13:48:06 +00:00
cf8cfbd865
* Refactor ordered_map into separate files Separate json and xml, from the regular yaml. Makes it possible to compile, without those... * Refactor encoder and decoder creation Use more consistent parameters vs globals Return errors instead of calling panic() * Allow build without json and xml support
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
//go:build !yq_nojson
|
|
|
|
package yqlib
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/goccy/go-json"
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type jsonDecoder struct {
|
|
decoder json.Decoder
|
|
}
|
|
|
|
func NewJSONDecoder() Decoder {
|
|
return &jsonDecoder{}
|
|
}
|
|
|
|
func (dec *jsonDecoder) Init(reader io.Reader) error {
|
|
dec.decoder = *json.NewDecoder(reader)
|
|
return nil
|
|
}
|
|
|
|
func (dec *jsonDecoder) Decode() (*CandidateNode, error) {
|
|
|
|
var dataBucket orderedMap
|
|
log.Debug("going to decode")
|
|
err := dec.decoder.Decode(&dataBucket)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
node, err := dec.convertToYamlNode(&dataBucket)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &CandidateNode{
|
|
Node: &yaml.Node{
|
|
Kind: yaml.DocumentNode,
|
|
Content: []*yaml.Node{node},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func (dec *jsonDecoder) convertToYamlNode(data *orderedMap) (*yaml.Node, error) {
|
|
if data.kv == nil {
|
|
switch rawData := data.altVal.(type) {
|
|
case nil:
|
|
return createScalarNode(nil, "null"), nil
|
|
case float64, float32:
|
|
// json decoder returns ints as float.
|
|
return parseSnippet(fmt.Sprintf("%v", rawData))
|
|
case int, int64, int32, string, bool:
|
|
return createScalarNode(rawData, fmt.Sprintf("%v", rawData)), nil
|
|
case []*orderedMap:
|
|
return dec.parseArray(rawData)
|
|
default:
|
|
return nil, fmt.Errorf("unrecognised type :( %v", rawData)
|
|
}
|
|
}
|
|
|
|
var yamlMap = &yaml.Node{Kind: yaml.MappingNode, Tag: "!!map"}
|
|
for _, keyValuePair := range data.kv {
|
|
yamlValue, err := dec.convertToYamlNode(&keyValuePair.V)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
yamlMap.Content = append(yamlMap.Content, createScalarNode(keyValuePair.K, keyValuePair.K), yamlValue)
|
|
}
|
|
return yamlMap, nil
|
|
|
|
}
|
|
|
|
func (dec *jsonDecoder) parseArray(dataArray []*orderedMap) (*yaml.Node, error) {
|
|
|
|
var yamlMap = &yaml.Node{Kind: yaml.SequenceNode, Tag: "!!seq"}
|
|
|
|
for _, value := range dataArray {
|
|
yamlValue, err := dec.convertToYamlNode(value)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
yamlMap.Content = append(yamlMap.Content, yamlValue)
|
|
}
|
|
return yamlMap, nil
|
|
}
|