mirror of https://github.com/mikefarah/yq.git
56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Encoder interface {
|
|
Encode(node *yaml.Node) error
|
|
}
|
|
|
|
type yamlEncoder struct {
|
|
encoder *yaml.Encoder
|
|
}
|
|
|
|
func NewYamlEncoder(destination io.Writer, indent int) Encoder {
|
|
var encoder = yaml.NewEncoder(destination)
|
|
if indent < 0 {
|
|
indent = 0
|
|
}
|
|
encoder.SetIndent(indent)
|
|
return &yamlEncoder{encoder}
|
|
}
|
|
|
|
func (ye *yamlEncoder) Encode(node *yaml.Node) error {
|
|
return ye.encoder.Encode(node)
|
|
}
|
|
|
|
type jsonEncoder struct {
|
|
encoder *json.Encoder
|
|
}
|
|
|
|
func NewJsonEncoder(destination io.Writer, prettyPrint bool, indent int) Encoder {
|
|
var encoder = json.NewEncoder(destination)
|
|
var indentString = ""
|
|
|
|
for index := 0; index < indent; index++ {
|
|
indentString = indentString + " "
|
|
}
|
|
if prettyPrint {
|
|
encoder.SetIndent("", indentString)
|
|
}
|
|
return &jsonEncoder{encoder}
|
|
}
|
|
|
|
func (je *jsonEncoder) Encode(node *yaml.Node) error {
|
|
var dataBucket interface{}
|
|
errorDecoding := node.Decode(&dataBucket)
|
|
if errorDecoding != nil {
|
|
return errorDecoding
|
|
}
|
|
return je.encoder.Encode(dataBucket)
|
|
}
|