yq/pkg/yqlib/encoder.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)
}