mje/commands/render/render.go

package render

import (
	"encoding"
	"encoding/json"
	"io"

	"gopkg.in/yaml.v3"
)

func Render(w io.Writer, format string, v any) error {
	if format == "yaml" || format == "yml" {
		return YAML(w, v)
	}

	if format == "json" {
		return JSON(w, v)
	}

	if wt, ok := v.(io.WriterTo); ok {
		_, err := wt.WriteTo(w)

		return err
	}

	if tm, ok := v.(encoding.TextMarshaler); ok {
		b, err := tm.MarshalText()
		if err != nil {
			return err
		}

		_, err = w.Write(b)

		return err
	}

	return JSON(w, v)
}

func YAML(w io.Writer, v interface{}) error {
	enc := yaml.NewEncoder(w)
	enc.SetIndent(2)

	return enc.Encode(v)
}

func JSON(w io.Writer, v interface{}) error {
	enc := json.NewEncoder(w)
	enc.SetIndent("", "  ")

	return enc.Encode(v)
}