bee/parser/formatter.go

package beeParser

import (
	"encoding/json"
	"io/ioutil"
	"strings"

	"github.com/talos-systems/talos/pkg/machinery/config/encoder"
)

type JsonFormatter struct {
}

func (f *JsonFormatter) FieldFormatFunc(field *StructField) ([]byte, error) {
	annotation := NewAnnotation(field.Doc+field.Comment, field.Name, field.Type)
	res := map[string]interface{}{}
	if field.NestedType != nil {
		res[annotation.Key] = field.NestedType
	} else {
		res[annotation.Key] = annotation.Default
	}
	return json.Marshal(res)
}

func (f *JsonFormatter) StructFormatFunc(node *StructNode) ([]byte, error) {
	return json.Marshal(node.Fields)
}

func (f *JsonFormatter) Marshal(node *StructNode) ([]byte, error) {
	return json.MarshalIndent(node, "", "	")
}

type YamlFormatter struct {
}

var result encoder.Doc

type Result map[string]interface{}

func (c Result) Doc() *encoder.Doc {
	return &result
}

func (f *YamlFormatter) FieldFormatFunc(field *StructField) ([]byte, error) {
	annotation := NewAnnotation(field.Doc+field.Comment, field.Name, field.Type)
	res := Result{}
	// add head comment for this field
	res.Doc().Comments[encoder.HeadComment] = annotation.Description
	if field.NestedType != nil {
		// nestedType format result as this field value
		b, err := field.NestedType.FormatFunc(field.NestedType)
		if err != nil {
			return nil, err
		}
		res[annotation.Key] = string(b)
	} else {
		res[annotation.Key] = annotation.Default
	}

	encoder := encoder.NewEncoder(&res, []encoder.Option{
		encoder.WithComments(encoder.CommentsAll),
	}...)
	encodeByte, err := encoder.Encode()
	if err != nil {
		return nil, err
	}
	// when field.NestedType != nil, the key and nested value strings are encoded with "|"
	// remove "|" by string replace
	encodeByte = []byte(strings.Replace(string(encodeByte), annotation.Key+": |", annotation.Key+":", 1))
	return encodeByte, nil
}

func (f *YamlFormatter) StructFormatFunc(node *StructNode) ([]byte, error) {
	res := make([]byte, 0)
	for _, f := range node.Fields {
		b, _ := f.FormatFunc(f)
		res = append(res, b...)
	}
	return res, nil
}

func (f *YamlFormatter) Marshal(node *StructNode) ([]byte, error) {
	res, err := node.FormatFunc(node)
	if err != nil {
		return nil, err
	}
	ioutil.WriteFile(node.Name+".yaml", res, 0667)
	return res, nil
}