mirror of https://github.com/mikefarah/yq.git
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
|
|
yaml "gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Encoder interface {
|
|
Encode(node *yaml.Node) error
|
|
}
|
|
|
|
type yamlEncoder struct {
|
|
destination io.Writer
|
|
indent int
|
|
colorise bool
|
|
firstDoc bool
|
|
}
|
|
|
|
func NewYamlEncoder(destination io.Writer, indent int, colorise bool) Encoder {
|
|
if indent < 0 {
|
|
indent = 0
|
|
}
|
|
return &yamlEncoder{destination, indent, colorise, true}
|
|
}
|
|
|
|
func (ye *yamlEncoder) Encode(node *yaml.Node) error {
|
|
|
|
destination := ye.destination
|
|
tempBuffer := bytes.NewBuffer(nil)
|
|
if ye.colorise {
|
|
destination = tempBuffer
|
|
}
|
|
|
|
var encoder = yaml.NewEncoder(destination)
|
|
|
|
encoder.SetIndent(ye.indent)
|
|
// TODO: work out if the first doc had a separator or not.
|
|
if ye.firstDoc {
|
|
ye.firstDoc = false
|
|
} else if _, err := destination.Write([]byte("---\n")); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := encoder.Encode(node); err != nil {
|
|
return err
|
|
}
|
|
|
|
if ye.colorise {
|
|
return ColorizeAndPrint(tempBuffer.Bytes(), ye.destination)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
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)
|
|
}
|